From 5c5d5112e7b999381f8b20c97e53314e6cedada2 Mon Sep 17 00:00:00 2001
From: zyy <zyy@email.com>
Date: Wed, 09 Jul 2025 18:34:52 +0800
Subject: [PATCH] 股票主页、市场页面接口

---
 src/main/java/com/nq/utils/redis/RedisKeyUtil.java |   96 ++++++++++++++++++++++++++++++++++++++++++-----
 1 files changed, 85 insertions(+), 11 deletions(-)

diff --git a/src/main/java/com/nq/utils/redis/RedisKeyUtil.java b/src/main/java/com/nq/utils/redis/RedisKeyUtil.java
index b59513b..fdc5ad3 100644
--- a/src/main/java/com/nq/utils/redis/RedisKeyUtil.java
+++ b/src/main/java/com/nq/utils/redis/RedisKeyUtil.java
@@ -1,17 +1,26 @@
 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.InputStreamReader;
+import java.io.OutputStream;
 import java.math.BigDecimal;
+import java.net.HttpURLConnection;
+import java.net.URL;
+import java.util.Map;
 
 
 public class RedisKeyUtil {
@@ -28,8 +37,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,35 +60,100 @@
             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());
+            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);
+        }
+    }
+
+    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;
+        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);
+            }
+
+            // 读取响应
+            BufferedReader in = new BufferedReader(new InputStreamReader(connection.getInputStream()));
+            String inputLine;
+            StringBuffer response = new StringBuffer();
+
+            while ((inputLine = in.readLine()) != null) {
+                response.append(inputLine);
+            }
+            in.close();
+            result = response.toString();
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        return result;
+    }
+
 
     public  static  void setCacheCompanies(Stock stock,String companiesInfo){
-        RedisShardedPoolUtils.set(RedisKeyConstant.RK_REAL_TIME_STOCK+":"+stock.getStockType()+":"+stock.getStockCode(),
+        RedisShardedPoolUtils.set(RedisKeyConstant.RK_COMPANY_INFO+":"+stock.getStockType()+":"+stock.getStockCode(),
                 new Gson().toJson(companiesInfo));
     }
 
 
     public static JSONObject getCacheCompanies(Stock stock){
-       String  companiesInfo =  RedisShardedPoolUtils.get(RedisKeyConstant.RK_REAL_TIME_STOCK+":"+stock.getStockType()+":"+stock.getStockCode());
+       String  companiesInfo =  RedisShardedPoolUtils.get(RedisKeyConstant.RK_COMPANY_INFO+":"+stock.getStockType()+":"+stock.getStockCode());
        if(companiesInfo.isEmpty()){
            return  null;
        }
        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);
+    }
 
 }

--
Gitblit v1.9.3