Spring boot 下WebSocket的使用

男娘i 2021-12-17 02:45 206阅读 0赞

前端js代码：

第二步导入websocket依赖：

<dependency>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-starter-websocket</artifactId>
            </dependency>

第三步写WebSocketConfig：

@Component
    public class WebSocketConfig {
    
        @Bean
        public ServerEndpointExporter serverEndpointExporter(){
            return new ServerEndpointExporter();
        }
    }

第四步编写webSocket：

@Component
    @ServerEndpoint("/webSocket")
    @Slf4j
    public class WebSocket {
    
        private Session session;
    
        private static CopyOnWriteArrayList<WebSocket> webSockets = new CopyOnWriteArrayList<>();
    
        @OnOpen
        public void opOpen(Session session){
            this.session = session;
            webSockets.add(this);
            log.info("【websocket消息】有新的连接，总数:{}",webSockets.size());
        }
    
        @OnClose
        public void onClose(){
            webSockets.remove(this);
            log.info("【websocket消息】连接断开，总数:{}",webSockets.size());
        }
    
        @OnMessage
        public void onMessage(String message){
            log.info("【websocket消息】收到客户端发来的消息：{}",message);
        }
    
        public void sendMessage(String message){
            for(WebSocket webSocket:webSockets){
                log.info("【websocket消息】广播消息，message={}",message);
                try {
                    webSocket.session.getBasicRemote().sendText(message);
                }catch (Exception e){
                    e.printStackTrace();
                }
            }
        }
    }

最后引入依赖，然后使用websocket发送消息给客户端：

@Autowired
        private WebSocket webSocket;

//发送websocket消息
    webSocket.sendMessage(orderDTO.getOrderId());