From 8ed60795a7c278f9699616f72eb05ce49800ba6f Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Wed, 07 Aug 2024 01:20:50 +0800
Subject: [PATCH] 1

---
 kucoinClient/src/main/java/org/example/kucoinclient/wsClient/KucoinClient.java |  130 +++++++++++++++++++++++--------------------
 1 files changed, 69 insertions(+), 61 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 05487b5..e5042fa 100644
--- a/kucoinClient/src/main/java/org/example/kucoinclient/wsClient/KucoinClient.java
+++ b/kucoinClient/src/main/java/org/example/kucoinclient/wsClient/KucoinClient.java
@@ -6,9 +6,16 @@
 import com.google.gson.Gson;
 import com.google.gson.JsonSyntaxException;
 import lombok.extern.slf4j.Slf4j;
+import org.example.kucoinclient.KucoinClientApplication;
+import org.example.kucoinclient.comm.ApplicationContextProvider;
 import org.example.kucoinclient.pojo.Currency;
 import org.example.kucoinclient.util.RedisUtil;
 import org.json.JSONException;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.SpringApplication;
+import org.springframework.context.ApplicationContext;
+import org.springframework.context.ConfigurableApplicationContext;
+import org.springframework.stereotype.Component;
 
 import javax.websocket.*;
 import java.io.UnsupportedEncodingException;
@@ -55,7 +62,7 @@
         this.executorService = Executors.newScheduledThreadPool(1); // 创建调度线程池
     }
 
-    public void start() {
+    public void start() throws Exception {
         try {
             connect(); // 连接到 WebSocket 服务器
             if (session == null) {
@@ -71,7 +78,8 @@
             }
 
         } catch (Exception e) {
-            log.error("kucoin ws 连接过程中发生异常: ", e); // 捕获并记录异常
+            log.error("kucoin ws 连接过程中发生异常:" + e.getMessage(), e); // 记录连接过程中发生的异常
+           throw e;
         } finally {
             executorService.shutdown(); // 关闭调度线程池
         }
@@ -144,34 +152,34 @@
             ObjectMapper mapper = new ObjectMapper(); // 创建 ObjectMapper 实例
 
             // 空值检查,避免存储 null 值到 Redis
-            if (resultMap.get("bids") != null) {
-                Object bidsObj = resultMap.get("bids");
-                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", bidsList.get(0));
-                    pvMap.put("V", bidsList.get(1));
-                    hashMap.put("bids",pvMap);
-                }
-            }
-            if (resultMap.get("asks") != null) {
+            if (resultMap.get("bids") != null && resultMap.get("asks") != null) {
+                Object asksObj = resultMap.get("bids");
+                Object bidsObj = resultMap.get("asks");
 
-                Object asksObj = resultMap.get("asks");
-                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", asksList.get(0));
-                    pvMap.put("V", asksList.get(1));
-                    hashMap.put("asks",pvMap);
-                }
-            }
+                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);
+                    }
 
-            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 (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失败"); // 输出处理失败日志
@@ -189,46 +197,46 @@
     }
 
     @OnClose
-    public void onClose() {
+    public void onClose() throws Exception {
         log.info("kucoin ws 连接已关闭,尝试重新连接..."); // 输出连接关闭日志
-        handleConnectionClosedOrError(); // 处理连接关闭或错误
+        throw new Exception();
     }
 
     @OnError
-    public void onError(Throwable throwable) {
+    public void onError(Throwable throwable) throws Exception {
         log.error("kucoin ws 发生错误: ", throwable); // 输出错误日志
-        handleConnectionClosedOrError(); // 处理连接关闭或错误
+        throw new Exception();
     }
 
-    private void handleConnectionClosedOrError() {
-        synchronized (lock) { // 同步块,确保线程安全
-            if (!reconnecting) {
-                reconnecting = true; // 状态标记为重连中
-                executorService.execute(this::attemptReconnect); // 执行重连操作
-            }
-        }
-    }
-
-    private void attemptReconnect() {
-        try {
-            log.info("kucoin ws 开始重连"); // 输出重连开始日志
-            connect(); // 尝试重新连接
-            log.info("kucoin ws 重连成功"); // 输出重连成功日志
-        } catch (Exception e) {
-            log.error("kucoin ws 重连失败: ", e); // 输出重连失败日志
-        } finally {
-            synchronized (lock) { // 同步块
-                reconnecting = false; // 状态标记为未重连
-                scheduleReconnect(); // 重新调度重连任务
-            }
-        }
-    }
-
-    private void scheduleReconnect() {
-        if (!executorService.isShutdown()) { // 检查线程池是否已经关闭
-            executorService.schedule(this::attemptReconnect, RECONNECT_DELAY, TimeUnit.SECONDS); // 调度重连
-        }
-    }
+//    private void handleConnectionClosedOrError() {
+//        synchronized (lock) { // 同步块,确保线程安全
+//            if (!reconnecting) {
+//                reconnecting = true; // 状态标记为重连中
+//                executorService.execute(this::attemptReconnect); // 执行重连操作
+//            }
+//        }
+//    }
+//
+//    private void attemptReconnect() {
+//        try {
+//            log.info("kucoin ws 开始重连"); // 输出重连开始日志
+//            connect(); // 尝试重新连接
+//            log.info("kucoin ws 重连成功"); // 输出重连成功日志
+//        } catch (Exception e) {
+//            log.error("kucoin ws 重连失败: ", e); // 输出重连失败日志
+//        } finally {
+//            synchronized (lock) { // 同步块
+//                reconnecting = false; // 状态标记为未重连
+//                scheduleReconnect(); // 重新调度重连任务
+//            }
+//        }
+//    }
+//
+//    private void scheduleReconnect() {
+//        if (!executorService.isShutdown()) { // 检查线程池是否已经关闭
+//            executorService.schedule(this::attemptReconnect, RECONNECT_DELAY, TimeUnit.SECONDS); // 调度重连
+//        }
+//    }
 
     private void sendPing() {
         try {

--
Gitblit v1.9.3