SpringBoot集成websocket（1）|（websocket客户端实现）

深碍√TFBOYSˉ_ 2023-10-10 17:49 93阅读 0赞

## SpringBoot集成websocket（1）|（websocket客户端实现） ##

--------------------

### 文章目录  SpringBoot集成websocket（1）|（websocket客户端实现）  @\[TOC\]  前言  一、websocket客户端依赖引入  二、websocket代码实现  1.WSClient客户端实现  2.WSClient客户端使用实例  三、实际使用场景  1.多WSClient连接使用  总结 ###

章节

## 前言 ##

WebSocket使得客户端和服务器之间的数据交换变得更加简单，允许服务端主动向客户端推送数据。在WebSocket API中，浏览器和服务器只需要完成一次握手，两者之间就直接可以创建持久性的连接，并进行双向数据传输。

## 一、websocket客户端依赖引入 ##

> pom依赖如下

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

## 二、websocket代码实现 ##

### 1.WSClient客户端实现 ###

> 可以定义了websocket的client类，其他地方如果需要使用ws连接，直接注入即可使用

import lombok.extern.slf4j.Slf4j;
    import org.java_websocket.client.WebSocketClient;
    import org.java_websocket.drafts.Draft_6455;
    import org.java_websocket.handshake.ServerHandshake;
    import org.springframework.stereotype.Component;
    
    import java.net.URI;
    @Slf4j
    @Component
    public class WSClient {
        
    
        public WebSocketClient webSocketClient(String url) {
        
            try {
        
                WebSocketClient webSocketClient = new WebSocketClient(new URI(url),new Draft_6455()) {
        
                    @Override
                    public void onOpen(ServerHandshake handshakedata) {
        
                        log.info("[websocket] 连接成功");
                    }
    
                    @Override
                    public void onMessage(String message) {
        
                        log.info("[websocket] 收到消息={}",message);
    
                    }
                    @Override
                    public void onClose(int code, String reason, boolean remote) {
        
                        log.info("[websocket] 退出连接");
                    }
    
                    @Override
                    public void onError(Exception ex) {
        
                        log.info("[websocket] 连接错误={}",ex.getMessage());
                    }
                };
                webSocketClient.connect();
                return webSocketClient;
            } catch (Exception e) {
        
                e.printStackTrace();
            }
            return null;
        }
    }

### 2.WSClient客户端使用实例 ###

> 定义了怎么使用WSClient类建立一个ws连接

@Api(tags = {
        "websocket测试"})
    @RestController
    @RequestMapping("/ws")
    public class SendMessageCtrl {
        
    
    
        @Autowired
        private WSClient wsClient;
    
        @GetMapping(value = "hello")
        public void hello(String name) throws InterruptedException {
        
            WebSocketClient client = this.wsClient.webSocketClient("ws://localhost:8905/ws/test");
            if (client != null){
        
                for (int i = 0; i < 10; i++) {
        
                    Thread.sleep(1000);
                    client.send("客户端发送消息"+i);
                }
            }
        }
    }

## 三、实际使用场景 ##

### 1.多WSClient连接使用 ###

> 上述是一个简单的demo，能够通过客户端连接websocket接口，但实际场景可能是很多用户访问本地的服务，改服务需要调用远程的ws接口，这个时候怎么区分用户间独立的建立websocket连接呢

> 思路：定义一个全局变量，用于存放websocket的客户端，客户端触发发送消息事件，在全局变量里面获取对应的client进行消息发送

@Api(tags = {
        "websocket测试"})
    @RestController
    @RequestMapping("/ws")
    public class SendMessageCtrl {
        
    
    
        @Autowired
        private WSClient wsClient;
    
        private static ConcurrentHashMap<String, WebSocketClient> wsClientSet = new ConcurrentHashMap<>();
    
    
        @GetMapping(value = "hello")
        public void hello(String name) {
        
            WebSocketClient client;
            if (wsClientSet.containsKey(name)) {
        
                client = wsClientSet.get(name);
            } else {
        
                client = this.wsClient.webSocketClient("ws://localhost:8905/ws/" + name);
                wsClientSet.put(name, client);
            }
            if (client != null) {
        
                client.send("请根据卖火柴的小女孩生成一个类似的故事");
            }
        }
    }

## 总结 ##

以上就是SpringBoot集成websocket客户端的内容，用于后端服务调用websocket请求。具体业务逻辑执行在代码中自己实现，这里只是给了具体的demo和实例，实际中运用场景还需要自己琢磨。