websocket服务端实现-蒲公英云

websocket服务端实现

- 日理万妓 2023-10-09 17:23 178阅读 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无需配置)：

<bean class="org.springframework.web.socket.messaging.WebSocketAnnotationMethodMessageHandler">
    <constructor-arg ref="messageConverter" />
    <property name="validator" ref="validator" />
</bean>