在 Java EE 7 中按 id 查找 websocket 会话

Question

我已阅读 Java EE 文档，但对我来说还不清楚一件事。根据API，找到另一个会话的唯一方法是这样的代码：（假设我们有其他会话的标识符）：

import javax.websocket.Session;
...
private static Session findOtherSessionById(Session user, String id) {
    for (Session session : user.getOpenSessions()) {
        if (id.equals(session.getId())) {
            return session;
        }
    }
    return null;
}

但是当我们有成千上万的用户时，这段代码就会成为性能瓶颈。

那么，有没有办法在不使用自己的 ConcurrentHashMap 的情况下通过 id 快速获取 Session？或者也许某些应用程序服务器对此具有未记录的功能（对我来说 Wildfly 会很棒）？

Answer 1

你可以这样做：

Map<String, Session> map = new HashMap<>();
static Map<String, Session> peers = Collections.synchronizedMap(map);

@OnOpen
public void onOpen(Session session) {
   peers.add(session.getId(), session);
}

@OnClose
public void onClose(Session session) {
    peers.remove(session.getId());
}

private static Session findOtherSessionById(Session user, String id) {
    if (peers.containsKey(user.getId()) {
        return peers.get(user.getId());
    }
}