From 6ed08e7e6a9ecccbb61d5664301da93bbfff67b6 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Wed, 28 Jan 2026 14:29:32 +0800
Subject: [PATCH] 1
---
src/main/java/com/nq/utils/redis/RedisKeyUtil.java | 81 +++++++++++++++++++++++++++-------------
1 files changed, 55 insertions(+), 26 deletions(-)
diff --git a/src/main/java/com/nq/utils/redis/RedisKeyUtil.java b/src/main/java/com/nq/utils/redis/RedisKeyUtil.java
index 788075d..b1ca87a 100644
--- a/src/main/java/com/nq/utils/redis/RedisKeyUtil.java
+++ b/src/main/java/com/nq/utils/redis/RedisKeyUtil.java
@@ -15,11 +15,13 @@
import org.slf4j.LoggerFactory;
import java.io.BufferedReader;
+import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.math.BigDecimal;
import java.net.HttpURLConnection;
import java.net.URL;
+import java.util.Collections;
import java.util.Map;
@@ -91,47 +93,74 @@
}
}
+ // 首先查看你的RedisShardedPoolUtils有哪些方法
+ // 如果只有set和get，可以这样实现：
+
public static String doPost(String pid, String stockType) {
EStockType eStockType = EStockType.getEStockTypeByCode(stockType);
- // 从配置中获取 API URL，并拼接 key
- //String apiUrl = PropertiesUtil.getProperty("JS_IN_HTTP_URL") + "stock?key=" + PropertiesUtil.getProperty("JS_IN_KEY");
String apiUrl = eStockType.getStockUrl() + "stock?key=" + eStockType.getStockKey();
String result = null;
+
+ // 创建Redis限流的key
+ String redisKey = "limit:post:" + stockType + ":" + pid;
+
try {
- URL url = new URL(apiUrl);
- HttpURLConnection connection = (HttpURLConnection) url.openConnection();
-
- // 设置请求方法为 POST
- connection.setRequestMethod("POST");
- // 设置请求头
- connection.setRequestProperty("Content-Type", "application/x-www-form-urlencoded");
- connection.setDoOutput(true); // 允许向连接输出
-
- // 构建 POST 数据
- String postData = "pid=" + pid;
-
- // 发送 POST 请求
- try (OutputStream os = connection.getOutputStream()) {
- byte[] input = postData.getBytes("utf-8");
- os.write(input, 0, input.length);
+ // 尝试从Redis获取，如果存在则直接返回
+ String cachedResult = RedisShardedPoolUtils.get(redisKey);
+ if (cachedResult != null && !cachedResult.isEmpty() && !"ERROR".equals(cachedResult)) {
+ return cachedResult;
}
- // 读取响应
- BufferedReader in = new BufferedReader(new InputStreamReader(connection.getInputStream()));
- String inputLine;
- StringBuffer response = new StringBuffer();
+ // 实际执行HTTP请求
+ result = executeHttpPost(apiUrl, pid);
- while ((inputLine = in.readLine()) != null) {
- response.append(inputLine);
+ // 将结果存入Redis
+ if (result != null && !result.isEmpty()) {
+ RedisShardedPoolUtils.set(redisKey, result);
+ RedisShardedPoolUtils.expire(redisKey, 1); // 设置1秒过期
}
- in.close();
- result = response.toString();
+
} catch (Exception e) {
e.printStackTrace();
+ // 异常处理
+ try {
+ RedisShardedPoolUtils.set(redisKey, "ERROR");
+ RedisShardedPoolUtils.expire(redisKey, 1);
+ } catch (Exception ex) {
+ ex.printStackTrace();
+ }
}
return result;
}
+ // HTTP请求执行方法
+ private static String executeHttpPost(String apiUrl, String pid) throws IOException {
+ URL url = new URL(apiUrl);
+ HttpURLConnection connection = (HttpURLConnection) url.openConnection();
+
+ connection.setRequestMethod("POST");
+ connection.setRequestProperty("Content-Type", "application/x-www-form-urlencoded");
+ connection.setDoOutput(true);
+
+ String postData = "pid=" + pid;
+
+ try (OutputStream os = connection.getOutputStream()) {
+ byte[] input = postData.getBytes("utf-8");
+ os.write(input, 0, input.length);
+ }
+
+ BufferedReader in = new BufferedReader(new InputStreamReader(connection.getInputStream()));
+ String inputLine;
+ StringBuilder response = new StringBuilder();
+
+ while ((inputLine = in.readLine()) != null) {
+ response.append(inputLine);
+ }
+ in.close();
+
+ return response.toString();
+ }
+
public static void setCacheCompanies(Stock stock,String companiesInfo){
RedisShardedPoolUtils.set(RedisKeyConstant.RK_COMPANY_INFO+":"+stock.getStockType()+":"+stock.getStockCode(),
--
Gitblit v1.9.3