1

zj

2026-01-28 6ed08e7e6a9ecccbb61d5664301da93bbfff67b6

 src/main/java/com/nq/utils/redis/RedisKeyUtil.java

@@ -1,17 +1,28 @@
package com.nq.utils.redis;

import com.alibaba.fastjson2.JSONObject;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.google.gson.Gson;
import com.nq.enums.EStockType;
import com.nq.pojo.DataStockBean;
import com.nq.pojo.Stock;
import com.nq.pojo.StockRealTimeBean;
import com.nq.utils.PropertiesUtil;
import com.nq.utils.stock.sina.StockApi;
import org.apache.http.util.TextUtils;
import org.slf4j.Logger;
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;


public class RedisKeyUtil {
@@ -28,8 +39,8 @@
    /**
     * 获取股票数据
     * */
    public static DataStockBean getCacheBaseStock(Stock stock){
        String cacheBaseData = RedisShardedPoolUtils.get(RedisKeyConstant.RK_BASE_STOCK+":"+stock.getStockType()+":"+stock.getStockCode());
    public static DataStockBean getCacheBaseStock(String stockType, String stockCode){
        String cacheBaseData = RedisShardedPoolUtils.get(RedisKeyConstant.RK_BASE_STOCK+":"+stockType+":"+stockCode);
        return  new Gson().fromJson(cacheBaseData, DataStockBean.class);
    }

@@ -51,18 +62,103 @@
            stockRealTimeBean = new Gson().fromJson(cacheBaseData, StockRealTimeBean.class);
        }
        if(stockRealTimeBean == null){
            String s = doPost(stock.getStockCode(), stock.getStockType());
            Map<String, Object> stringObjectMap = jsonToMap(s);
            stockRealTimeBean = new StockRealTimeBean();
            stockRealTimeBean.setPcp("0.00");
            stockRealTimeBean.setLast("0.00");
            stockRealTimeBean.setHigh("0.00");
            stockRealTimeBean.setLow("0.00");
            stockRealTimeBean.setBid("0.00");
            stockRealTimeBean.setPc("0.00");
            stockRealTimeBean.setAsk("0.00");
            stockRealTimeBean.setPcp(stringObjectMap.get("ChgPct").toString());
            stockRealTimeBean.setLast(stringObjectMap.get("Last").toString());
            stockRealTimeBean.setHigh(stringObjectMap.get("High").toString());
            stockRealTimeBean.setLow(stringObjectMap.get("Low").toString());
            stockRealTimeBean.setBid(stringObjectMap.get("Id").toString());
            stockRealTimeBean.setPc(stringObjectMap.get("PrevClose").toString());
            if (stringObjectMap.get("Ask") != null) {
                stockRealTimeBean.setAsk(stringObjectMap.get("Ask").toString());
            }
        }
        stockRealTimeBean.setPcp(stockRealTimeBean.getPcp().replace("%",""));
        return  stockRealTimeBean;

    }

    public static Map<String, Object> jsonToMap(String json) {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            Object[] array = objectMapper.readValue(json, Object[].class);
            Gson gson = new Gson();
            String s = gson.toJson(array[0]);
            Map<String, Object> map = objectMapper.readValue(s, Map.class);
            return map;
        } catch (JsonProcessingException e) {
            throw new RuntimeException(e);
        }
    }

    // 首先查看你的RedisShardedPoolUtils有哪些方法
    // 如果只有set和get，可以这样实现：

    public static String doPost(String pid, String stockType) {
        EStockType eStockType = EStockType.getEStockTypeByCode(stockType);
        String apiUrl = eStockType.getStockUrl() + "stock?key=" + eStockType.getStockKey();
        String result = null;

        // 创建Redis限流的key
        String redisKey = "limit:post:" + stockType + ":" + pid;

        try {
            // 尝试从Redis获取，如果存在则直接返回
            String cachedResult = RedisShardedPoolUtils.get(redisKey);
            if (cachedResult != null && !cachedResult.isEmpty() && !"ERROR".equals(cachedResult)) {
                return cachedResult;
            }

            // 实际执行HTTP请求
            result = executeHttpPost(apiUrl, pid);

            // 将结果存入Redis
            if (result != null && !result.isEmpty()) {
                RedisShardedPoolUtils.set(redisKey, result);
                RedisShardedPoolUtils.expire(redisKey, 1); // 设置1秒过期
            }

        } 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();
    }


@@ -80,6 +176,15 @@
       return  JSONObject.parseObject(companiesInfo);
    }

    /**
     * 缓存K线数据源到redis
     * */
    public static void setCaCheStockKData(String stockType, String stockCode, Object kData){
        RedisShardedPoolUtils.set(RedisKeyConstant.RK_STOCK_KDATA+":"+stockType+":"+stockCode, kData.toString());
    }

    public static Object getCaCheStockKData(String stockType, String stockCode){
        return RedisShardedPoolUtils.get(RedisKeyConstant.RK_STOCK_KDATA+":"+stockType+":"+stockCode);
    }

}