WebSocketJavaExample.java

package org.example.ws;

import com.alibaba.fastjson2.JSONArray;
import com.alibaba.fastjson2.JSONObject;
import jakarta.annotation.PostConstruct;
import jakarta.websocket.*;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import java.io.IOException;
import java.net.URI;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;

@ClientEndpoint
@Slf4j
@Component
public class WebsocketExample {

    //本地session通道
    private static Session session;

    //wss连接地址 business可以为stock、crypto、common；apikey为您的凭证
    private static final String WS_URL = "wss://data.infoway.io/ws?business=crypto&apikey=yourApikey";

    @PostConstruct
    public void connectAll() {
        try {
            //建立WEBSOCKET连接
            connect(WS_URL);
            //开启自动重连
            startReconnection(WS_URL);
        } catch (Exception e) {
            log.error("Failed to connect to " + WS_URL + ": " + e.getMessage());
        }
    }

    //自动重连机制，会开启一个定时线程判断连接是否断开，断开自动重连
    private void startReconnection(String s) {
        ScheduledExecutorService usExecutor = Executors.newScheduledThreadPool(1);
        Runnable usTask = () -> {
            if (session == null || !session.isOpen()) {
                log.debug("reconnection...");
                connect(s);
            }
        };
        usExecutor.scheduleAtFixedRate(usTask, 1000, 10000, TimeUnit.MILLISECONDS);
    }

    //建立WEBSOCKET连接具体实现
    private void connect(String s) {
        try {
            WebSocketContainer container = ContainerProvider.getWebSocketContainer();
            session = container.connectToServer(WebsocketExample.class, URI.create(s));
        } catch (DeploymentException | IOException e) {
            log.error("Failed to connect to the server: {}", e.getMessage());
        }
    }

    //WEBSOCKET连接建立成功后会执行下面的方法
    @OnOpen
    public void onOpen(Session session) throws IOException, InterruptedException {
        //WEBSOCKET连接建立成功会执行该方法
        System.out.println("Connection opened: " + session.getId());

        // 发送实时成交明细订阅请求
        JSONObject tradeSendObj = new JSONObject();
        //参考WEBSOCKET API 里面的不同请求的协议号，1000为订阅实时交易明细数据，文档：https://infoway.readme.io/reference/subscribe-trade#/
        tradeSendObj.put("code", 10000);
        //自定义trace
        tradeSendObj.put("trace", "01213e9d-90a0-426e-a380-ebed633cba7a");
        //封装订阅请求实体，json格式
        JSONObject data = new JSONObject();
        //订阅BTCUSDT
        data.put("codes", "BTCUSDT");
        tradeSendObj.put("data", data);
        //发送实时成交明细订阅请求
        session.getBasicRemote().sendText(tradeSendObj.toJSONString());

        //-----------------------------------------------------------------//
        //不同请求之间间隔一段时间
        Thread.sleep(5000);

        //发送实时盘口数据订阅请求，文档：https://infoway.readme.io/reference/subscribe-depth#/
        JSONObject depthSendObj = new JSONObject();
        //参考WEBSOCKET API 里面的不同请求的协议号，1003为订阅实时盘口数据
        depthSendObj.put("code", 10003);
        //自定义trace
        depthSendObj.put("trace", "01213e9d-90a0-426e-a380-ebed633cba7a");
        //封装订阅请求实体，json格式
        depthSendObj.put("data", data);
        //发送实时成交明细订阅请求
        session.getBasicRemote().sendText(depthSendObj.toJSONString());


        //-----------------------------------------------------------------//
        //不同请求之间间隔一段时间
        Thread.sleep(5000);

        //发送实时k线数据订阅请求，文档：https://infoway.readme.io/reference/subscribe-candles#/
        JSONObject klineSendObj = new JSONObject();
        //参考WEBSOCKET API 里面的不同请求的协议号，1006为订阅K线数据
        klineSendObj.put("code", 10006);
        //自定义trace
        klineSendObj.put("trace", "01213e9d-90a0-426e-a380-ebed633cba7a");
        //封装订阅请求实体，json格式
        JSONObject klineData = new JSONObject();
        JSONArray klineDataArray = new JSONArray();

        //封装订阅1分钟k线的实体
        JSONObject kline1minObj = new JSONObject();
        //klineType，可以查看文档：https://infoway.readme.io/reference/k%E7%BA%BF%E7%B1%BB%E5%9E%8B#/
        kline1minObj.put("type", 1);
        kline1minObj.put("codes", "BTCUSDT");
        klineDataArray.add(kline1minObj);
        klineData.put("arr", klineDataArray);

        klineSendObj.put("data", klineData);
        //发送实时成交明细订阅请求
        session.getBasicRemote().sendText(klineSendObj.toJSONString());

        //定时发送心跳任务，文档：https://infoway.readme.io/reference/heartbeat#/
        ScheduledExecutorService pingExecutor = Executors.newScheduledThreadPool(1);
        Runnable pingTask = WebsocketExample::ping;
        pingExecutor.scheduleAtFixedRate(pingTask, 30, 30, TimeUnit.SECONDS);

    }

    @OnMessage
    public void onMessage(String message, Session session) {
        //会接收INFOWAY服务端返回的数据，包含订阅成功或失败的提示，以及正式的行情数据推送
        try {
            System.out.println("Message received: " + message);
        } catch (Exception e) {
        }
    }

    @OnClose
    public void onClose(Session session, CloseReason reason) {
        //WEBSOCKET连接关闭会回调该方法
        System.out.println("Connection closed: " + session.getId() + ", reason: " + reason);
    }

    @OnError
    public void onError(Throwable error) {
        error.printStackTrace();
    }

    //持续性发送心跳，防止服务端长时间检查不到心跳请求关闭连接
    public static void ping() {
        try {
            JSONObject jsonObject = new JSONObject();
            jsonObject.put("code", 10010);
            jsonObject.put("trace", "01213e9d-90a0-426e-a380-ebed633cba7a");
            session.getBasicRemote().sendText(jsonObject.toJSONString());
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

}