From 658e843113e09d1bf319a29e40a17a2d11ffabab Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Tue, 06 Aug 2024 00:05:49 +0800
Subject: [PATCH] 1

---
 kucoinClient/src/main/java/org/example/kucoinclient/wsClient/KucoinClient.java |   44 +++++++++++++++++++++++++++++---------------
 1 files changed, 29 insertions(+), 15 deletions(-)

diff --git a/kucoinClient/src/main/java/org/example/kucoinclient/wsClient/KucoinClient.java b/kucoinClient/src/main/java/org/example/kucoinclient/wsClient/KucoinClient.java
index 18827f7..aca5da9 100644
--- a/kucoinClient/src/main/java/org/example/kucoinclient/wsClient/KucoinClient.java
+++ b/kucoinClient/src/main/java/org/example/kucoinclient/wsClient/KucoinClient.java
@@ -5,7 +5,6 @@
 import com.google.common.reflect.TypeToken;
 import com.google.gson.Gson;
 import com.google.gson.JsonSyntaxException;
-import com.sun.istack.internal.NotNull;
 import lombok.extern.slf4j.Slf4j;
 import org.example.kucoinclient.pojo.Currency;
 import org.example.kucoinclient.util.RedisUtil;
@@ -13,6 +12,7 @@
 
 import javax.websocket.*;
 import java.io.UnsupportedEncodingException;
+import java.math.BigDecimal;
 import java.net.URI;
 import java.net.URLEncoder;
 import java.nio.ByteBuffer;
@@ -144,19 +144,34 @@
             ObjectMapper mapper = new ObjectMapper(); // 创建 ObjectMapper 实例
 
             // 空值检查,避免存储 null 值到 Redis
-            if (resultMap.get("bids") != null) {
-                hashMap.put("bids", resultMap.get("bids")); // 存储 bids
-            }
-            if (resultMap.get("asks") != null) {
-                hashMap.put("asks", resultMap.get("asks")); // 存储 asks
-            }
+            if (resultMap.get("bids") != null && resultMap.get("asks") != null) {
+                Object bidsObj = resultMap.get("bids");
+                Object asksObj = resultMap.get("asks");
 
-            String symbol = extractSymbolFromTopic(topic); // 从 topic 提取符号
-            String key = PREFIX + symbol; // 创建 Redis 缓存键
-            try {
-                RedisUtil.set(key, mapper.writeValueAsString(hashMap)); // 存储到 Redis
-            } catch (JsonProcessingException e) {
-                log.error("将数据存入 Redis 时出错: {}", e.getMessage()); // 输出数据存储错误日志
+                if(bidsObj instanceof List && !((List<?>) bidsObj).isEmpty() && asksObj instanceof List && !((List<?>) asksObj).isEmpty()){
+                    if (bidsObj instanceof List && !((List<?>) bidsObj).isEmpty()) {
+                        List<String> bidsList = (List<String>) bidsObj;
+                        HashMap<String, Object> pvMap = new HashMap<>(); // 创建新的 HashMap 保存 bids 和 asks
+                        pvMap.put("p", new BigDecimal(String.valueOf(bidsList.get(0))).toPlainString());
+                        pvMap.put("v", new BigDecimal(String.valueOf(bidsList.get(1))).toPlainString());
+                        hashMap.put("bids",pvMap);
+                    }
+
+                    if (asksObj instanceof List && !((List<?>) asksObj).isEmpty()) {
+                        List<String> asksList = (List<String>) asksObj;
+                        HashMap<String, Object> pvMap = new HashMap<>(); // 创建新的 HashMap 保存 bids 和 asks
+                        pvMap.put("p", new BigDecimal(String.valueOf(asksList.get(0))).toPlainString());
+                        pvMap.put("v", new BigDecimal(String.valueOf(asksList.get(1))).toPlainString());
+                        hashMap.put("asks",pvMap);
+                    }
+                    String symbol = extractSymbolFromTopic(topic); // 从 topic 提取符号
+                    String key = PREFIX + symbol; // 创建 Redis 缓存键
+                    try {
+                        RedisUtil.set(key, mapper.writeValueAsString(hashMap)); // 存储到 Redis
+                    } catch (JsonProcessingException e) {
+                        log.error("将数据存入 Redis 时出错: {}", e.getMessage()); // 输出数据存储错误日志
+                    }
+                }
             }
         } else {
             log.error("topic--->存入redis失败"); // 输出处理失败日志
@@ -232,7 +247,7 @@
         Map<String, Object> jsonMap = new HashMap<>();
         jsonMap.put("id", id); // 会话 ID
         jsonMap.put("type", "subscribe"); // 订阅类型
-        jsonMap.put("topic", "/spotMarket/level2Depth50:" + symbol); // 订阅的主题
+        jsonMap.put("topic", "/spotMarket/level1:" + symbol); // 订阅的主题
         jsonMap.put("privateChannel", false); // 是否私有通道
         jsonMap.put("response", true); // 是否返回响应
 
@@ -240,7 +255,6 @@
         return mapper.writeValueAsString(jsonMap); // 返回 JSON 字符串
     }
 
-    @NotNull
     private String getSymbol() {
         List<String> symbolList = subscriptions.stream()
                 .map(currency -> currency.getSymbol().replaceAll("USDT", "-USDT")) // 替换符号

--
Gitblit v1.9.3