SpringBoot建立websocket连接

介绍

WebSocket 协议是一种基于TCP的协议，用于在客户端和服务器之间建立持久连接，并且可以在这个连接上实时地交换数据。WebSocket协议有自己的握手协议，用于建立连接，也有自己的数据传输格式。

当客户端发送一个 WebSocket 请求时，服务器将发送一个协议响应以确认请求。在握手期间，客户端和服务器将协商使用的协议版本、支持的子协议、支持的扩展选项等。一旦握手完成，连接将保持打开状态，客户端和服务器就可以在连接上实时地传递数据。

WebSocket 协议使用的是双向数据传输，即客户端和服务器都可以在任意时间向对方发送数据，而不需要等待对方的请求。它支持二进制数据和文本数据，可以自由地在它们之间进行转换。

总之，WebSocket协议是一种可靠的、高效的、双向的、持久的通信协议，它适用于需要实时通信的Web应用程序，如在线游戏、实时聊天等。

使用

导入依赖

        <dependency>
            <groupId>org.java-websocket</groupId>
            <artifactId>Java-WebSocket</artifactId>
            <version>1.5.2</version>
        </dependency>

工具

package com.seekgenjs.position.util;


import org.java_websocket.handshake.ServerHandshake;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Timer;
import java.util.TimerTask;


@SuppressWarnings("all")
@Component
public class WebSocketClient {

    Logger logger = LoggerFactory.getLogger(WebSocketClient.class);


    private final String URI = "wss://xxxx/wss";
    private final String User_Name = "xxx";
    private final String Act_Login = "xxx";
    private final String Act_Device = "xxxx";

    @Resource
    MyWebSocketClient myWebSocketClient;


    //项目启动建立连接
    @PostConstruct
    public void init() {

        try {
            URI uri = new URI(URI);
            org.java_websocket.client.WebSocketClient client = new org.java_websocket.client.WebSocketClient(uri) {
                @Override
                public void onOpen(ServerHandshake hand) {

                    logger.debug("{}_Connected to WebSocket server", URI);

                    send(login());

                    timer();

                }

                private void timer() {
                    // Send heartbeat packet every 30 seconds
                    logger.debug("{}_开始执行定时任务，Send heartbeat packet every 30 seconds", System.currentTimeMillis());

                    Timer timer = new Timer();
                    timer.schedule(new TimerTask() {
                        @Override
                        public void run() {
                            try {
                                //发送心跳的业务代码
                                send(sendHeartbeat());
                            } catch (Exception e) {
                                //print the error log
                                e.printStackTrace();
                            }

                        }
                    }, 1000L * 3, 1000L * 25);
                }

                @Override
                public void onMessage(String message) {
                    System.out.println(message);
                    handleMessage(message);
                }

                @Override
                public void onClose(int code, String reason, boolean remote) {
                    System.out.println("Connection closed");
                    logger.debug("{}_Connection closed", System.currentTimeMillis());
                }

                @Override
                public void one rror(Exception ex) {
                    ex.printStackTrace();
                }

            };
            client.connect();
        } catch (URISyntaxException e) {
            e.printStackTrace();
        }
    }


    //登录
    private String login() {
        HashMap<String, Object> map = new HashMap<>();
        map.put("act", Act_Login);
        map.put("user_name", User_Name);
        map.put("access_token", myWebSocketClient.getToken());
        return JSON.toJSONString(map);

    }

    //心跳包
    private String sendHeartbeat() {
        logger.debug("{}_Send heartbeat packet", System.currentTimeMillis());
        HashMap<String, Object> map = new HashMap<>();
        map.put("act", Act_Device);
        return JSON.toJSONString(map);
    }


    /**
     * 处理接收数据
     *
     * @param message
     */
    public void handleMessage(String message) {
        System.out.println("{}_Received message: " + message);
       
    }


}