SpringBoot webSocket实现发送广播、点对点消息和Android接收

以下是“SpringBoot webSocket实现发送广播、点对点消息和Android接收”的完整攻略，包含两个示例说明。

简介

WebSocket是一种在Web浏览器和Web服务器之间进行全双工通信的协议。SpringBoot提供了对WebSocket的支持，可以轻松地实现WebSocket通信。本教程将介绍如何使用SpringBoot实现WebSocket通信，并提供相应的示例说明。

步骤1：添加WebSocket依赖

在SpringBoot应用程序中，可以使用WebSocket依赖来实现WebSocket通信。在pom.xml文件中添加以下依赖：

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

步骤2：创建WebSocket配置类

在SpringBoot应用程序中，可以使用@EnableWebSocket注解启用WebSocket，并使用@Configuration注解配置WebSocket。在WebSocketConfig类中添加以下配置：

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(new MyWebSocketHandler(), "/websocket").setAllowedOrigins("*");
    }
}

在这个示例中，我们使用@EnableWebSocket注解启用WebSocket，并使用@Configuration注解配置WebSocket。我们创建了一个名为MyWebSocketHandler的WebSocket处理程序，并使用setAllowedOrigins方法指定允许的来源。

示例1：实现发送广播和点对点消息

以下是一个实现发送广播和点对点消息的示例：

1. 创建WebSocket处理程序

public class MyWebSocketHandler extends TextWebSocketHandler {
    private static final Map<String, WebSocketSession> sessions = new ConcurrentHashMap<>();

    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        sessions.put(session.getId(), session);
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        sessions.remove(session.getId());
    }

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        String payload = message.getPayload();
        JSONObject jsonObject = new JSONObject(payload);
        String type = jsonObject.getString("type");
        String content = jsonObject.getString("content");
        if ("broadcast".equals(type)) {
            broadcast(content);
        } else if ("p2p".equals(type)) {
            String to = jsonObject.getString("to");
            sendTo(to, content);
        }
    }

    private void broadcast(String content) throws IOException {
        for (WebSocketSession session : sessions.values()) {
            session.sendMessage(new TextMessage(content));
        }
    }

    private void sendTo(String to, String content) throws IOException {
        WebSocketSession session = sessions.get(to);
        if (session != null) {
            session.sendMessage(new TextMessage(content));
        }
    }
}

在这个示例中，我们创建了一个名为MyWebSocketHandler的WebSocket处理程序。我们使用afterConnectionEstablished方法将WebSocket会话添加到会话列表中，使用afterConnectionClosed方法将WebSocket会话从会话列表中删除。我们使用handleTextMessage方法处理文本消息，根据消息类型发送广播或点对点消息。

2. 测试

在浏览器中打开两个或多个WebSocket客户端，连接到WebSocket服务器。在其中一个客户端中发送广播消息：

{
    "type": "broadcast",
    "content": "Hello, world!"
}

在其他客户端中，将收到相同的广播消息。在其中一个客户端中发送点对点消息：

{
    "type": "p2p",
    "to": "session-id",
    "content": "Hello, world!"
}

其中，session-id是要发送消息的WebSocket会话的ID。在指定的WebSocket会话中，将收到相同的点对点消息。

示例2：Android接收WebSocket消息

以下是一个Android应用程序接收WebSocket消息的示例：

1. 添加WebSocket依赖

在Android应用程序中，可以使用WebSocket依赖来接收WebSocket消息。在build.gradle文件中添加以下依赖：

implementation 'org.java-websocket:Java-WebSocket:1.5.1'

2. 创建WebSocket客户端

public class MyWebSocketClient extends WebSocketClient {
    private final Handler handler;

    public MyWebSocketClient(URI serverUri, Handler handler) {
        super(serverUri);
        this.handler = handler;
    }

    @Override
    public void onOpen(ServerHandshake handshakedata) {
        Log.d("WebSocket", "onOpen");
    }

    @Override
    public void onMessage(String message) {
        Log.d("WebSocket", "onMessage: " + message);
        Message msg = handler.obtainMessage();
        msg.obj = message;
        handler.sendMessage(msg);
    }

    @Override
    public void onClose(int code, String reason, boolean remote) {
        Log.d("WebSocket", "onClose");
    }

    @Override
    public void onError(Exception ex) {
        Log.d("WebSocket", "onError", ex);
    }
}

在这个示例中，我们创建了一个名为MyWebSocketClient的WebSocket客户端。我们使用onOpen方法处理WebSocket打开事件，使用onMessage方法处理WebSocket消息事件，使用onClose方法处理WebSocket关闭事件，使用onError方法处理WebSocket错误事件。

3. 连接WebSocket服务器

URI uri = new URI("ws://localhost:8080/websocket");
MyWebSocketClient client = new MyWebSocketClient(uri, new Handler() {
    @Override
    public void handleMessage(Message msg) {
        String message = (String) msg.obj;
        // 处理WebSocket消息
    }
});
client.connect();

在这个示例中，我们创建了一个名为MyWebSocketClient的WebSocket客户端，并使用connect方法连接到WebSocket服务器。我们使用Handler类处理WebSocket消息。在handleMessage方法中，我们可以处理WebSocket消息。