1

zj

2024-08-07 8ed60795a7c278f9699616f72eb05ce49800ba6f

 geteClient/src/main/java/org/example/geteclient/wsClinet/GateClient.java

@@ -6,9 +6,16 @@
import com.google.gson.Gson;
import com.google.gson.JsonSyntaxException;
import lombok.extern.slf4j.Slf4j;
import org.example.geteclient.GeteClientApplication;
import org.example.geteclient.comm.ApplicationContextProvider;
import org.example.geteclient.pojo.Currency;
import org.example.geteclient.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.math.BigDecimal;
@@ -46,7 +53,7 @@
        this.executorService = Executors.newScheduledThreadPool(1); // 创建一个定时任务调度线程池
    }

    public void start() {
    public void start() throws Exception {
        try {
            connect(); // 尝试连接 WebSocket
            if (session == null) { // 如果连接失败，session 仍然为 null
@@ -69,6 +76,7 @@

        } catch (Exception e) {
            log.error("gate ws 连接过程中发生异常: " + e.getMessage(), e); // 记录连接过程中发生的异常
            throw e;
        } finally {
            executorService.shutdown(); // 确保定时任务调度服务被关闭
        }
@@ -92,8 +100,6 @@
    private static final String RESULT_KEY = "result"; // 定义结果键的常量
    private static final String BIDS_KEY = "bids"; // 定义 bids 的常量
    private static final String ASKS_KEY = "asks"; // 定义 asks 的常量
    private static final String p = "p"; // 定义 asks 的常量
    private static final String v = "v"; // 定义 asks 的常量
    private static final String S_KEY = "s"; // 定义 s 的常量

    @OnMessage
@@ -110,28 +116,30 @@
                Object asksObj = resultMap.get(BIDS_KEY);
                Object bidsObj = resultMap.get(ASKS_KEY);

                if (asksObj instanceof List && !((List<?>) asksObj).isEmpty()) {
                    List<?> asksList = (List<?>) asksObj;
                    String[][] dataArray = gson.fromJson(gson.toJson(asksList), String[][].class);
                    HashMap<String, Object> pvMap = new HashMap<>(); // 创建新的 HashMap 保存 bids 和 asks
                    String[] asksData = dataArray[0];
                    pvMap.put(p, new BigDecimal(asksData[0]));
                    pvMap.put(v, new BigDecimal(asksData[1]));
                    hashMap.put(BIDS_KEY, pvMap); // 放入 bids 数据
                }
                if(asksObj instanceof List && !((List<?>) asksObj).isEmpty() && bidsObj instanceof List && !((List<?>) bidsObj).isEmpty()){
                    if (asksObj instanceof List && !((List<?>) asksObj).isEmpty()) {
                        List<?> asksList = (List<?>) asksObj;
                        String[][] dataArray = gson.fromJson(gson.toJson(asksList), String[][].class);
                        HashMap<String, Object> pvMap = new HashMap<>(); // 创建新的 HashMap 保存 bids 和 asks
                        String[] asksData = dataArray[0];
                        pvMap.put("p", new BigDecimal(asksData[0]).toPlainString());
                        pvMap.put("v", new BigDecimal(asksData[1]).toPlainString());
                        hashMap.put(ASKS_KEY, pvMap); // 放入 bids 数据
                    }

                if (bidsObj instanceof List && !((List<?>) bidsObj).isEmpty()) {
                    List<?> bidsList = (List<?>) bidsObj;
                    String[][] dataArray = gson.fromJson(gson.toJson(bidsList), String[][].class);
                    String[] bidsData = dataArray[0];
                    HashMap<String, Object> pvMap = new HashMap<>(); // 创建新的 HashMap 保存 bids 和 asks
                    pvMap.put(p, new BigDecimal(bidsData[0]));
                    pvMap.put(v, new BigDecimal(bidsData[1]));
                    hashMap.put(ASKS_KEY,pvMap);
                }
                    if (bidsObj instanceof List && !((List<?>) bidsObj).isEmpty()) {
                        List<?> bidsList = (List<?>) bidsObj;
                        String[][] dataArray = gson.fromJson(gson.toJson(bidsList), String[][].class);
                        String[] bidsData = dataArray[0];
                        HashMap<String, Object> pvMap = new HashMap<>(); // 创建新的 HashMap 保存 bids 和 asks
                        pvMap.put("p", new BigDecimal(bidsData[0]).toPlainString());
                        pvMap.put("v", new BigDecimal(bidsData[1]).toPlainString());
                        hashMap.put(BIDS_KEY,pvMap);
                    }

                String key = "gate" + resultMap.get(S_KEY); // 生成 Redis 键
                RedisUtil.set(key.replace("_", ""), gson.toJson(hashMap)); // 存入 Redis，使用 Gson 进行序列化
                    String key = "gate" + resultMap.get(S_KEY); // 生成 Redis 键
                    RedisUtil.set(key.replace("_", ""), gson.toJson(hashMap)); // 存入 Redis，使用 Gson 进行序列化
                }
            }
        } catch (JsonSyntaxException e) {
            log.error("JSON 解析异常：" + e.getMessage(), e); // 记录 JSON 解析异常
@@ -141,52 +149,53 @@
    }



    @OnClose
    public void onClose() {
    public void onClose() throws Exception {
        log.info("gate ws 连接已关闭，尝试重新连接..."); // 连接关闭日志
        handleConnectionClosedOrError(); // 处理连接关闭或错误
        throw new Exception();
    }

    @OnError
    public void onError(Throwable throwable) {
    public void onError(Throwable throwable) throws Exception {
        log.error("gate ws 发生错误: " + throwable.getMessage(), throwable); // 记录错误日志
        handleConnectionClosedOrError(); // 处理连接关闭或错误
        throw new Exception();
    }

    private void handleConnectionClosedOrError() {
        synchronized (lock) { // 进入同步块以防止并发重连
            if (!reconnecting) { // 检查当前是否已经在重连
                reconnecting = true; // 设置 reconnecting 为 true 表示开始重连
                executorService.execute(this::attemptReconnect); // 使用 execute 方法立即执行重连
            }
        }
    }

    private void attemptReconnect() {
        boolean doReconnect = true; // 是否进行重连的标志
        try {
            log.info("gate ws 开始重连"); // 开始重连日志
            connect(); // 尝试重新连接
            log.info("gate ws 重连成功"); // 重连成功日志
        } catch (Exception e) {
            log.error("gate ws 重连失败", e); // 重连失败记录日志
            doReconnect = false; // 标记不再继续重连
        } finally {
            synchronized (lock) { // 进入同步块
                if (doReconnect) {
                    scheduleReconnect(); // 如果需要继续重连，调度重连任务
                } else {
                    reconnecting = false; // 重连结束，标记重连状态为 false
                }
            }
        }
    }

    private void scheduleReconnect() {
        if (!executorService.isShutdown()) { // 确保调度服务未关闭
            executorService.schedule(this::attemptReconnect, 5, TimeUnit.SECONDS); // 重连调度，5秒后尝试重新连接
        }
    }
//    private void handleConnectionClosedOrError() {
//        synchronized (lock) { // 进入同步块以防止并发重连
//            if (!reconnecting) { // 检查当前是否已经在重连
//                reconnecting = true; // 设置 reconnecting 为 true 表示开始重连
//                executorService.execute(this::attemptReconnect); // 使用 execute 方法立即执行重连
//            }
//        }
//    }
//
//    private void attemptReconnect() {
//        boolean doReconnect = true; // 是否进行重连的标志
//        try {
//            log.info("gate ws 开始重连"); // 开始重连日志
//            connect(); // 尝试重新连接
//            log.info("gate ws 重连成功"); // 重连成功日志
//        } catch (Exception e) {
//            log.error("gate ws 重连失败", e); // 重连失败记录日志
//            doReconnect = false; // 标记不再继续重连
//        } finally {
//            synchronized (lock) { // 进入同步块
//                if (doReconnect) {
//                    scheduleReconnect(); // 如果需要继续重连，调度重连任务
//                } else {
//                    reconnecting = false; // 重连结束，标记重连状态为 false
//                }
//            }
//        }
//    }
//
//    private void scheduleReconnect() {
//        if (!executorService.isShutdown()) { // 确保调度服务未关闭
//            executorService.schedule(this::attemptReconnect, 5, TimeUnit.SECONDS); // 重连调度，5秒后尝试重新连接
//        }
//    }

    private void sendPing() {
        try {