websocket服务端实现

- 日理万妓 2023-10-09 17:23 52阅读 0赞

方式一：

import javax.websocket.*;
    import javax.websocket.server.ServerEndpoint;
    import java.io.IOException;
    
    @ServerEndpoint("/websocket")
    public class WebSocketServer {
        
    
        @OnOpen
        public void onOpen(Session session) {
        
            System.out.println("WebSocket opened: " + session.getId());
        }
    
        @OnMessage
        public void onMessage(String message, Session session) {
        
            System.out.println("WebSocket message received: " + message);
            try {
        
                session.getBasicRemote().sendText("Server received message: " + message);
            } catch (IOException e) {
        
                e.printStackTrace();
            }
        }
    
        @OnClose
        public void onClose(Session session) {
        
            System.out.println("WebSocket closed: " + session.getId());
        }
    
        @OnError
        public void onError(Session session, Throwable throwable) {
        
            System.out.println("WebSocket error: " + session.getId() + ", " + throwable.getMessage());
        }
    }

在这个示例中，使用了 Java 的 WebSocket API 来实现 WebSocket 服务端。具体来说，使用了 @ServerEndpoint 注解来定义 WebSocket 服务端的端点，即 URI。在这个示例中，定义了一个端点 /websocket。

在类中，使用了四个注解来处理 WebSocket 的生命周期事件：@OnOpen、@OnMessage、@OnClose 和 @OnError。其中，@OnOpen 表示 WebSocket 连接建立时触发的事件，@OnMessage 表示收到消息时触发的事件，@OnClose 表示连接关闭时触发的事件，@OnError 表示出错时触发的事件。

在代码中，分别实现了这四个事件的处理函数。在 @OnOpen 函数中，输出了一个连接已建立的提示信息。在 @OnMessage 函数中，输出了收到的消息，并通过 session.getBasicRemote().sendText() 方法将消息发送回客户端。在 @OnClose 函数中，输出了一个连接已关闭的提示信息。在 @OnError 函数中，输出了出错的提示信息。

需要注意的是，这个示例中并没有处理多个客户端连接的情况。在实际应用中，您需要使用数据结构来管理多个客户端的连接，并实现相应的线程安全措施。

方式二：

import org.springframework.stereotype.Component;
    import org.springframework.web.socket.CloseStatus;
    import org.springframework.web.socket.TextMessage;
    import org.springframework.web.socket.WebSocketSession;
    import org.springframework.web.socket.handler.AbstractWebSocketHandler;
    import org.springframework.web.socket.messaging.SessionConnectedEvent;
    import org.springframework.web.socket.messaging.SessionDisconnectEvent;
    import org.springframework.web.socket.messaging.SessionUnsubscribeEvent;
    
    @Component
    public class WebSocketServer extends AbstractWebSocketHandler {
        
    
        @EventListener
        public void handleSessionConnected(SessionConnectedEvent event) {
        
            WebSocketSession session = (WebSocketSession) event.getSource();
            System.out.println("WebSocket connected: " + session.getId());
        }
    
        @EventListener
        public void handleSessionDisconnect(SessionDisconnectEvent event) {
        
            WebSocketSession session = (WebSocketSession) event.getSource();
            System.out.println("WebSocket disconnected: " + session.getId());
        }
    
        @EventListener
        public void handleSessionUnsubscribe(SessionUnsubscribeEvent event) {
        
            WebSocketSession session = (WebSocketSession) event.getSource();
            System.out.println("WebSocket unsubscribed: " + session.getId());
        }
    
        @Override
        protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        
            System.out.println("WebSocket message received: " + message.getPayload());
            session.sendMessage(new TextMessage("Server received message: " + message.getPayload()));
        }
    
        @Override
        public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        
            System.out.println("WebSocket connection closed: " + session.getId());
        }
    }

在这个示例中，创建了一个名为 WebSocketServer 的类，并使用了 @Component 注解将其声明为一个 Spring 组件。同时，继承了 AbstractWebSocketHandler 类，并实现了其中的 handleTextMessage() 和 afterConnectionClosed() 方法，用于处理收到消息和连接关闭的事件。

除此之外，使用了三个 @EventListener 注解来分别处理 WebSocket 连接建立、断开和取消订阅的事件。在这些处理函数中，可以使用 event.getSource() 方法获取 WebSocketSession 对象，并输出相应的提示信息。

需要注意的是，为了使用 @EventListener 注解，需要在 Spring 配置文件中添加如下配置(springboot无需配置)：

另外，需要在配置类上添加 @EnableWebSocket 注解来启用 WebSocket 功能。