1

zj

2024-10-11 9ed28183ca391eadb7b5cb2f3160e3ec852da124

 bitgetsClient/src/main/java/org/example/bitgetsclient/wsClient/BitgetClient.java

@@ -8,13 +8,22 @@
import com.google.gson.Gson;
import com.google.gson.JsonSyntaxException;
import lombok.extern.slf4j.Slf4j;
import org.example.bitgetsclient.BitgetsClientApplication;
import org.example.bitgetsclient.comm.ApplicationContextProvider;
import org.example.bitgetsclient.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.context.annotation.Bean;
import org.springframework.stereotype.Component;
import org.springframework.util.CollectionUtils;

import javax.websocket.*;
import java.io.IOException;
import java.lang.reflect.Type;
import java.math.BigDecimal;
import java.net.URI;
import java.nio.ByteBuffer;
import java.util.Arrays;
@@ -24,6 +33,7 @@
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

@ClientEndpoint
@Slf4j
@@ -53,7 +63,7 @@



    public void start() {
    public void start()  {
        try {
            connect(); // 尝试连接
            if (session == null) {
@@ -65,7 +75,6 @@
            executorService.scheduleAtFixedRate(this::sendPing, PING_INTERVAL, PING_INTERVAL, TimeUnit.MILLISECONDS);
            // 发送订阅消息
            session.getBasicRemote().sendText(subscriptions); // 发送订阅信息

            synchronized (this) {
                this.wait(); // 等待 WebSocket 消息到来
            }
@@ -105,17 +114,39 @@

                // 确保 dataNode 不为 null，以避免空指针异常
                if (dataNode != null) {
                    // 存储数据到 HashMap
                    Map<String, Object> bidAskMap = new HashMap<>(); // 变量命名更具描述性
                    bidAskMap.put("bids", dataNode.get("bids")); // 获取并存储 bids
                    bidAskMap.put("asks", dataNode.get("asks")); // 获取并存储 asks
                    Map<String, Object> hashMap = new HashMap<>(); // 变量命名更具描述性

                    Map<String, Object> argMap = gson.fromJson(map.get("arg").toString(), new TypeToken<Map<String, Object>>() {}.getType()); // 解析 arg 为 Map
                    String key = "bitget" + argMap.get("instId"); // 构建 Redis 存储的键
                    String asks = dataNode.get("bids").toString();
                    String bids = dataNode.get("asks").toString();

                    // 存储到 Redis，使用 ObjectMapper 转换为 JSON 字符串
                    String jsonData = objectMapper.writeValueAsString(bidAskMap); // 先将 HashMap 转换为 JSON 字符串
                    RedisUtil.set(key, jsonData); // 存储数据
                    Type listType = new TypeToken<List<List<String>>>(){}.getType();
                    List<List<String>> bidsList = gson.fromJson(bids, listType);
                    List<List<String>> asksList = gson.fromJson(asks, listType);

                    if(!bidsList.isEmpty()  && !bidsList.get(0).isEmpty() && !asksList.isEmpty() && !asksList.get(0).isEmpty()){
                        if (!bidsList.isEmpty()  && !bidsList.get(0).isEmpty()) {
                            List<String> bidsStringList = bidsList.get(0);
                            HashMap<String, Object> pvMap = new HashMap<>(); // 创建新的 HashMap 保存 bids 和 asks
                            pvMap.put("p", new BigDecimal(bidsStringList.get(0)).toPlainString());
                            pvMap.put("v", new BigDecimal(bidsStringList.get(1)).toPlainString());
                            hashMap.put("bids", pvMap); // 获取并存储 bids
                        }

                        if (!asksList.isEmpty() && !asksList.get(0).isEmpty()) {
                            List<String> asksStringList = asksList.get(0); // 修正此处为 asksList
                            HashMap<String, Object> pvMap = new HashMap<>(); // 创建新的 HashMap 保存 bids 和 asks
                            pvMap.put("p", new BigDecimal(asksStringList.get(0)).toPlainString());
                            pvMap.put("v", new BigDecimal(asksStringList.get(1)).toPlainString());
                            hashMap.put("asks", pvMap); // 获取并存储 asks
                        }

                        Map<String, Object> argMap = gson.fromJson(map.get("arg").toString(), new TypeToken<Map<String, Object>>() {}.getType()); // 解析 arg 为 Map
                        String key = "bitget" + argMap.get("instId"); // 构建 Redis 存储的键

                        // 存储到 Redis，使用 ObjectMapper 转换为 JSON 字符串
                        String jsonData = objectMapper.writeValueAsString(hashMap); // 先将 HashMap 转换为 JSON 字符串
                        RedisUtil.set(key, jsonData); // 存储数据
                    }
                }
            }
        } catch (JsonSyntaxException e) {
@@ -143,24 +174,25 @@
    }

    @OnClose
    public void onClose() {
    public void onClose()  {
        log.info("bitget ws 连接已关闭，尝试重新连接..."); // 记录连接关闭的信息
        handleConnectionClosedOrError(); // 处理连接关闭事件
        handleConnectionClosedOrError();
    }

    @OnError
    public void onError(Throwable throwable) {
        log.error("bitget ws 发生错误: " + throwable.getMessage(), throwable); // 记录错误信息
        handleConnectionClosedOrError(); // 处理错误事件
        handleConnectionClosedOrError();
    }

    private void handleConnectionClosedOrError() {
        synchronized (lock) {
            if (!reconnecting) {
                reconnecting = true; // 开始重连
                executorService.execute(this::attemptReconnect); // 执行重连操作
        executorService.execute(() -> {
            try {
                attemptReconnect();
            } catch (Exception e) {
                throw new RuntimeException(e);
            }
        }
        });
    }

    private void attemptReconnect() {
@@ -179,7 +211,6 @@
            }
        } else {
            log.error("超过最大重连次数，停止重连"); // 超过最大重连次数
            reconnecting = false; // 重连状态重置
        }
    }