方式一:
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>
另外,需要在配置类上添加 @EnableWebSocket 注解来启用 WebSocket 功能。