1

zj

8 days ago 7d008bf58373926bbdcf67fa7bcf92510f427fb6

 src/main/java/com/nq/service/impl/StockServiceImpl.java

@@ -40,6 +40,8 @@
import java.text.SimpleDateFormat;
import java.util.*;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.locks.ReentrantLock;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
@@ -432,25 +434,146 @@
    }


        /*股票日线-K线*/

    // 缓存键类，包含三个参数
    private static class CacheKey {
        private final String pid;
        private final String interval;
        private final String stockType;

        public CacheKey(String pid, String interval, String stockType) {
            this.pid = pid;
            this.interval = interval;
            this.stockType = stockType;
        }

        @Override
        public boolean equals(Object o) {
            if (this == o) return true;
            if (o == null || getClass() != o.getClass()) return false;
            CacheKey cacheKey = (CacheKey) o;
            return pid.equals(cacheKey.pid) &&
                    interval.equals(cacheKey.interval) &&
                    stockType.equals(cacheKey.stockType);
        }

        @Override
        public int hashCode() {
            int result = pid.hashCode();
            result = 31 * result + interval.hashCode();
            result = 31 * result + stockType.hashCode();
            return result;
        }

        @Override
        public String toString() {
            return "CacheKey{pid='" + pid + "', interval='" + interval + "', stockType='" + stockType + "'}";
        }
    }

    // 缓存条目类
    private static class CacheEntry {
        Object result;
        long lastExecuteTime;

        CacheEntry(Object result, long lastExecuteTime) {
            this.result = result;
            this.lastExecuteTime = lastExecuteTime;
        }
    }

    // 缓存存储：每个CacheKey对应一个CacheEntry
    private static final Map<CacheKey, CacheEntry> cacheMap = new ConcurrentHashMap<>();

    // 锁对象存储：每个CacheKey有自己的锁
    private static final Map<CacheKey, Object> lockMap = new ConcurrentHashMap<>();

    private static final long MIN_INTERVAL_MS = 3000;

    @Override
    public Object getKData(String pid, String interval, String stockType) {
        EStockType eStockType = EStockType.getEStockTypeByCode(stockType);
        Object object = HttpUtil.get(eStockType.stockUrl + "kline?pid=" + pid + "&interval=" + interval + "&key=" + eStockType.stockKey);
        Gson gson = new Gson();
        List<kData> dataList = gson.fromJson(object.toString(), new TypeToken<List<kData>>(){}.getType());
        CacheKey cacheKey = new CacheKey(pid, interval, stockType);
        long currentTime = System.currentTimeMillis();

        Stock stock = stockMapper.selectOne(new LambdaQueryWrapper<Stock>().eq(Stock::getStockCode, pid).eq(Stock::getStockType, "IN"));
        BigDecimal nowPrice = iPriceServices.getNowPrice(stock.getStockCode());
        Map singleStock = getSingleStock(stock.getStockCode());
        StockVO stockVO = (StockVO)singleStock.get("stock");
        // 修改 List 中的最后一条数据
        kData lastData = dataList.get(dataList.size() - 1);
        lastData.setC(nowPrice.toString());
        lastData.setO(stockVO.getOpen_px());
        lastData.setH(stockVO.getToday_max());
        lastData.setL(stockVO.getToday_min());
        return gson.toJson(dataList);
        // 1. 快速检查缓存（无锁）
        CacheEntry cached = cacheMap.get(cacheKey);
        if (cached != null && (currentTime - cached.lastExecuteTime) < MIN_INTERVAL_MS) {
            return cached.result;
        }

        // 2. 获取该缓存键对应的锁
        Object keyLock = lockMap.computeIfAbsent(cacheKey, k -> new Object());

        // 3. 同步块内再次检查
        synchronized (keyLock) {
            currentTime = System.currentTimeMillis();
            cached = cacheMap.get(cacheKey);

            if (cached != null && (currentTime - cached.lastExecuteTime) < MIN_INTERVAL_MS) {
                return cached.result;
            }

            // 4. 获取新数据
            Object newResult = doGetKData(pid, interval, stockType);

            // 5. 更新缓存
            if (newResult != null) {
                cacheMap.put(cacheKey, new CacheEntry(newResult, System.currentTimeMillis()));
                return newResult;
            } else if (cached != null) {
                // 请求失败，返回旧缓存
                return cached.result;
            }

            return "{\"error\":\"获取K线数据失败\"}";
        }
    }

    private Object doGetKData(String pid, String interval, String stockType) {
        try {
            EStockType eStockType = EStockType.getEStockTypeByCode(stockType);
            String url = eStockType.stockUrl + "kline?pid=" + pid +
                    "&interval=" + interval + "&key=" + eStockType.stockKey;

            Object object = HttpUtil.get(url);
            Gson gson = new Gson();
            List<kData> dataList = gson.fromJson(object.toString(),
                    new TypeToken<List<kData>>(){}.getType());

            // 补充实时数据
            if (dataList != null && !dataList.isEmpty()) {
                enrichWithRealTimeData(pid, dataList);
            }

            return gson.toJson(dataList);

        } catch (Exception e) {
            e.printStackTrace();
            return null; // 返回null表示失败
        }
    }

    private void enrichWithRealTimeData(String pid, List<kData> dataList) {
        try {
            Stock stock = stockMapper.selectOne(new LambdaQueryWrapper<Stock>()
                    .eq(Stock::getStockCode, pid)
                    .eq(Stock::getStockType, "IN"));

            if (stock != null) {
                BigDecimal nowPrice = iPriceServices.getNowPrice(stock.getStockCode());
                Map singleStock = getSingleStock(stock.getStockCode());
                StockVO stockVO = (StockVO)singleStock.get("stock");

                kData lastData = dataList.get(dataList.size() - 1);
                lastData.setC(nowPrice.toString());
                lastData.setO(stockVO.getOpen_px());
                lastData.setH(stockVO.getToday_max());
                lastData.setL(stockVO.getToday_min());
            }
        } catch (Exception e) {
            // 实时数据获取失败不影响K线数据返回
            e.printStackTrace();
        }
    }

    @Override