Java如何连接多个服务端websocket，并区分每个服务端发过来的消息

在Java中，连接多个WebSocket服务器并处理来自每个服务器的不同消息，通常需要借助第三方库如`javax.websocket`或第三方框架如`Spring Websocket`。这里简述一下基本步骤：

1. **添加依赖**：如果你使用Java的标准库，需要先导入`java.websocket`包。如果使用Spring，则需引入相应的Spring WebSocket依赖。

2. **创建客户端连接**：对于每个服务端，你需要创建一个`javax.websocket.ClientEndpoint`（标准库）或者`WebsocketClient`（Spring），并配置URL地址。

// 标准库示例
ClientEndpointConfig config = ClientEndpointConfig.Builder.create()
   .endpointURI("ws://service1.com/websocket")
   .build();

ClientEndpoint client = new MyWebSocketClient();
client.connect(config);

// Spring 示例
WebsocketStompClient stompClient = new WebsocketStompClient(new SimpleUrlConnectionHttpClient());
stompClient.setUri(new URI("ws://service1.com/websocket"));
StompSession session = stompClient.connect().get();

3. **区分消息**：一旦连接建立，你可以通过监听`MessageReceivedEvent`或`TextMessage`、`BinaryMessage`等事件来区分不同服务端发送的消息。比如，可以使用消息头的信息或者自定义标识字段来标记：

// 标准库示例
@OnMessage
public void onMessage(String message, Session session) {
    if (session.getHandshakeHeaders().containsKey("ServiceId")) { // 或者检查其他自定义头信息
        String serviceId = session.getHandshakeHeaders().getFirst("ServiceId");
        handleMessage(message, serviceId);
    }
}

// Spring 示例
@OnMessage
public void onMessage(StompHeaderAccessor headers, Payload payload) {
    String serviceId = headers.getString("service_id"); // 假设消息头中有service_id字段
    handleMessage(payload.getData(), serviceId);
}

4. **处理消息函数**：为每个服务端创建单独的处理函数，根据`serviceId`来区分和处理相应的服务消息。

private void handleMessage(String message, String serviceId) {
    switch (serviceId) {
        case "service1":
            // 处理来自service1的消息
            break;
        case "service2":
            // 处理来自service2的消息
            break;
        default:
            // 对于未知服务，可以记录异常或忽略
    }
}