如何使用Java服务器发送消息到特定的websocket连接

Question

我是WebSockets新手。

我已经在WebSockets做了一个简单的服务器-客户端聊天。

现在我正在尝试使客户端-服务器-客户端聊天应用程序。

我有一个问题，在java服务器中我们如何发送消息到特定的WebSocket连接。

如果用户a想发送消息给用户b。

那么我如何管理用户b正在使用这个或那个连接，或者向那个特定的连接发送消息?

我在谷歌上搜索了太多，但没有找到任何好的

Answer 1

您必须为此设计一个体系结构。

当客户端与服务器建立连接(打开WebSocket)时，服务器必须将连接保存在某个数据结构中(无论您如何识别与正在使用的Java后端连接)，该数据结构将取决于您要做的事情。一个好的标识符应该是用户提供的ID(就像连接到同一服务器的另一个对等端尚未选择的昵称)。否则，只需使用套接字对象作为唯一标识符，并且在列出前端的其他用户时，将它们与其唯一标识符相关联，以便客户端可以向特定的对等端发送消息。

如果客户端要与另一个特定的客户端聊天，那么HashMap将是一个很好的数据结构选择，因为您可以将客户端的唯一ID映射到套接字，并在哈希表中找到0(1)的条目。

如果你想从客户端广播消息到所有其他客户端，尽管HashMap也可以很好地工作(与HashMap.values()类似)，你可以使用简单的List，将传入的消息发送到除了原始发送者之外的所有连接的客户端。

当然，当您失去与客户端的连接时，您也希望从数据结构中删除客户端，这很容易使用WebSocket(您正在使用的Java框架应该在socket关闭时回叫您)。

下面是一个使用Jetty 9 WebSocket(和JDK 7)的(几乎完整的)示例:

package so.example;
import java.io.IOException;
import java.util.HashMap;
import org.eclipse.jetty.websocket.api.Session;
import org.eclipse.jetty.websocket.api.annotations.OnWebSocketClose;
import org.eclipse.jetty.websocket.api.annotations.OnWebSocketConnect;
import org.eclipse.jetty.websocket.api.annotations.OnWebSocketMessage;
import org.eclipse.jetty.websocket.api.annotations.WebSocket;
@WebSocket
public class MyWebSocket {
    private final static HashMap<String, MyWebSocket> sockets = new HashMap<>();
    private Session session;
    private String myUniqueId;
    private String getMyUniqueId() {
        // unique ID from this class' hash code
        return Integer.toHexString(this.hashCode());
    }
    @OnWebSocketConnect
    public void onConnect(Session session) {
        // save session so we can send
        this.session = session;
        // this unique ID
        this.myUniqueId = this.getMyUniqueId();
        // map this unique ID to this connection
        MyWebSocket.sockets.put(this.myUniqueId, this);
        // send its unique ID to the client (JSON)
        this.sendClient(String.format("{"msg": "uniqueId", "uniqueId": "%s"}",
                this.myUniqueId));
        // broadcast this new connection (with its unique ID) to all other connected clients
        for (MyWebSocket dstSocket : MyWebSocket.sockets.values()) {
            if (dstSocket == this) {
                // skip me
                continue;
            }
            dstSocket.sendClient(String.format("{"msg": "newClient", "newClientId": "%s"}",
                    this.myUniqueId));
        }
    }
    @OnWebSocketMessage
    public void onMsg(String msg) {
        /*
         * process message here with whatever JSON library or protocol you like
         * to get the destination unique ID from the client and the actual message
         * to be sent (not shown). also, make sure to escape the message string
         * for further JSON inclusion. 
         */
        String destUniqueId = ...;
        String escapedMessage = ...;
        // is the destination client connected?
        if (!MyWebSocket.sockets.containsKey(destUniqueId)) {
            this.sendError(String.format("destination client %s does not exist", destUniqueId));
            return;
        }
        // send message to destination client
        this.sendClient(String.format("{"msg": "message", "destId": "%s", "message": "%s"}",
                destUniqueId, escapedMessage));
    }
    @OnWebSocketClose
    public void onClose(Session session, int statusCode, String reason) {
        if (MyWebSocket.sockets.containsKey(this.myUniqueId)) {
            // remove connection
            MyWebSocket.sockets.remove(this.myUniqueId);
            // broadcast this lost connection to all other connected clients
            for (MyWebSocket dstSocket : MyWebSocket.sockets.values()) {
                if (dstSocket == this) {
                    // skip me
                    continue;
                }
                dstSocket.sendClient(String.format("{"msg": "lostClient", "lostClientId": "%s"}",
                        this.myUniqueId));
            }
        }
    }
    private void sendClient(String str) {
        try {
            this.session.getRemote().sendString(str);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    private void sendError(String err) {
        this.sendClient(String.format("{"msg": "error", "error": "%s"}", err));
    }
}

代码是不言自明的。关于JSON格式和解析，Jetty在org.eclipse.jetty.util.ajax包中提供了一些有趣的实用程序。

还要注意，如果你的WebSocket服务器框架不是线程安全的，你需要同步数据结构以确保没有数据损坏(这里是MyWebSocket.sockets)。