session中保存map集合

在Web开发中，特别是在基于Java Servlets或者Spring MVC这样的框架中，`session`通常用于存储用户的状态信息，而Map集合是一个非常方便的数据结构，可以用于存放键值对。如果你想在`session`中保存一个Map集合，你可以这样做：

1. 创建Map对象：

Map<String, Object> sessionMap = new HashMap<>();

这里，你可以选择HashMap、LinkedHashMap或者其他类型的Map，取决于你的具体需求。

2. 添加元素到Map：

sessionMap.put("key1", "value1");
sessionMap.put("key2", "value2");

3. 将Map添加到Session：

HttpSession session = request.getSession(); // 获取当前请求的session
session.setAttribute("myMap", sessionMap); // 存储Map到session

4. 需要在后续访问时，可以从session中取出Map：

Map<String, Object> retrievedMap = (Map<String, Object>) session.getAttribute("myMap");

注意，为了性能考虑，建议检查Map是否为空或已被移除再进行获取，避免不必要的null pointer异常。

相关问题

hibernate查询结果map集合

可以使用 Hibernate 提供的 `ResultTransformer` 接口来将查询结果转换成 Map 集合。

以下是一个示例代码：

Query query = session.createQuery("SELECT u.id, u.name FROM User u");
query.setResultTransformer(Transformers.ALIAS_TO_ENTITY_MAP);
List<Map<String, Object>> result = query.list();

在这个示例中，我们使用 Hibernate 的 `createQuery` 方法来创建一个查询，查询语句是 `SELECT u.id, u.name FROM User u`，它会返回 `id` 和 `name` 两列数据。

接着，我们调用 `setResultTransformer` 方法来设置结果转换器，将结果转换成 `Map` 集合。`Transformers.ALIAS_TO_ENTITY_MAP` 是一个预定义的结果转换器，它将结果转换成一个 `Map` 集合，键是列别名，值是列值。

最后，我们调用 `list` 方法来执行查询，并将查询结果保存到一个 `List` 集合中，每个元素都是一个 `Map` 集合。

package com.sqx.controller; import javafx.beans.binding.Bindings; import lombok.extern.slf4j.Slf4j; import org.springframework.stereotype.Component; import javax.websocket.*; import javax.websocket.server.PathParam; import javax.websocket.server.ServerEndpoint; import java.io.IOException; import java.util.HashMap; import java.util.HashSet; import java.util.Map; import java.util.Set; @ServerEndpoint("/webSocket/{userId}") @Component @Slf4j public class WebSocketServer { /** * 记录当前在线连接数 */ private static int onlineCount = 0; /** * 存放每个客户端对应的MyWebSocket对象, 使用map对象便于根据userId来获取对应的WebSocket */ private static Map<String, Set<WebSocketServer>> webSocketMap = new HashMap<>(); /** * 与客户端的连接会话，需要通过它来给客户端发送数据 */ private Session session; /** * 客户端连接会话对应的用户Id */ private String userId; /** * 判断当前userId是否已创建连接 * @param userId * @return */ public boolean checkOnline(String userId){ for (String key : webSocketMap.keySet()) { if(userId.equals(key)){ return true; } } return false; } /** * 连接建立成功调用的方法 */ @OnOpen public void onOpen(Session session, @PathParam("userId") String userId) { this.session = session; this.userId = userId; //在线数加1 addOnlineCount(); //根据该用户当前是否已经在别的终端登录进行添加操作 if (webSocketMap.containsKey(this.userId)) { Set<WebSocketServer> set = webSocketMap.get(this.userId); set.add(this); webSocketMap.put(this.userId,set); }else{ Set<WebSocketServer> addUserSet = new HashSet<>(); addUserSet.add(this); webSocketMap.put(this.userId, addUserSet); } } /** * 连接关闭调用的方法 */ @OnClose public void onClose() { if(webSocketMap.containsKey(userId)){ webSocketMap.remove(userId); //在线数减1 subOnlineCount(); } } /** * 收到客户端消息后调用的方法 * * @param message 客户端发送过来的消息 * @param session 客户端的连接会话 */ @OnMessage public void onMessage(String message, Session session) { try{ //todo 需要添加 JSONObject json = JSON.parseObject(message); }catch (Exception e){ e.printStackTrace(); } } /** * 发生错误时调用的方法 * * @param session 客户端的连接会话 * @param error 异常错误 */ @OnError public void onError(Session session, Throwable error) { } /** * @Title: sendMessageToUser * : 发送消息给用户下的所有终端 * @param @param userId 用户id * @param @param message 发送的消息 * @param @return 发送成功返回true，反则返回false */ public static Boolean sendMessageToUser(String userId,String message){ if (webSocketMap.containsKey(userId)) { for (WebSocketServer ws : webSocketMap.get(userId)) { try { ws.session.getBasicRemote().sendText(message); } catch (IOException e) { e.printStackTrace(); return false; } } return true; } return false; } private static synchronized int getOnlineCount() { return WebSocketServer.onlineCount; } private static synchronized void addOnlineCount() { WebSocketServer.onlineCount++; } private static synchronized void subOnlineCount() { WebSocketServer.onlineCount--; } @Override public int hashCode() { return super.hashCode(); } @Override public boolean equals(Object obj) { return super.equals(obj); } }这个加测试

<think>嗯，用户让我分析这个Java WebSocket的代码，并且要加测试。首先，我需要理解这段代码的结构和功能。看起来这是一个使用Spring和JavaFX的WebSocket服务器端实现，可能用于处理多个客户端的连接，并且支持同一用户多个终端的连接。用户可能想测试这个WebSocket控制器的各个方法，比如连接建立、关闭、消息处理以及错误处理等。

首先，我应该检查代码的结构。类上有@ServerEndpoint注解，说明这是一个WebSocket端点，路径是/webSocket/{userId}，所以每个连接都会带一个用户ID。使用了@Component，说明由Spring管理，但可能需要注意是否配置了正确的ServerEndpointExporter，不过用户的问题是关于测试，可能不需要深入这里。

接下来，代码中有静态的webSocketMap，用来按用户ID存储WebSocketServer实例的集合。每个用户可能有多个连接（比如多个浏览器标签或设备），所以用Set来保存。onlineCount记录在线连接数，不过这里可能有线程安全问题，因为多个线程可能同时修改这个变量，但代码里用了synchronized方法，所以应该没问题。

然后看@OnOpen方法，当有连接打开时，会将当前实例添加到对应userId的集合中。如果用户已经存在，就添加到现有集合，否则新建一个。这里需要注意的是，当同一个用户多次连接时，是否正确处理了多个实例。同时，onClose方法里，当连接关闭时，直接从webSocketMap中移除整个userId对应的集合，这可能有问题，因为一个用户可能有多个连接，关闭一个连接应该只移除当前实例，而不是整个用户的所有连接。例如，如果用户A有两个连接，当其中一个关闭时，代码里直接webSocketMap.remove(userId)，这会导致另一个连接也被移除，这显然不对。这可能是个bug，用户可能在测试时发现这个问题，需要修正。

接下来，@OnMessage方法处理收到的消息，但当前只是捕获异常，没有实际处理消息，可能用户需要在这里添加业务逻辑，比如解析JSON消息。用户提到需要添加JSONObject的解析，但被注释掉了，可能这也是测试需要注意的地方，比如发送不同格式的消息是否会引发异常。

sendMessageToUser方法用于向特定用户的所有终端发送消息，遍历该用户的WebSocketServer实例，逐个发送消息。这里如果发送过程中某个连接出现异常，比如IOException，会返回false，但其他连接可能已经发送成功，这里的错误处理可能需要优化，比如记录失败的连接，而不是直接返回false。

接下来，测试的话，需要考虑以下几个方面：

1. 单元测试：测试各个方法的功能，比如onOpen是否正确添加用户连接，onClose是否正确移除，sendMessageToUser是否能正确发送消息等。
2. 集成测试：测试整个WebSocket的生命周期，包括连接、消息收发、关闭等。
3. 并发测试：多个用户同时连接，或者同一用户多个连接，是否处理正确。
4. 异常测试：发送错误格式的消息，断开连接时的处理等。

由于WebSocket通常需要与客户端交互，测试可能需要模拟客户端。可以使用JUnit和Spring的测试框架，结合WebSocketClient来模拟连接。比如使用Spring Boot的Test框架，或者像Tomcat的WebSocket客户端。

首先，单元测试方面，可能需要使用Mock对象来模拟Session等对象。例如，测试onOpen方法时，传入一个模拟的Session和userId，然后验证webSocketMap中是否添加了该实例。但webSocketMap是静态的，需要注意测试之间的隔离，避免状态残留。可以使用@Before或@After方法清理静态变量。

但单元测试可能不够，因为很多功能涉及网络交互，集成测试可能更有效。例如，启动嵌入式Web服务器，然后使用WebSocketClient连接，发送消息，验证是否收到响应。

另外，注意到代码中使用的是javax.websocket的注解，而Spring的WebSocket支持可能有所不同，需要确认是否正确配置。不过用户代码中用了@ServerEndpoint和@Component，可能已经通过配置ServerEndpointExporter来注册端点。

在测试onClose方法时，当前代码中的逻辑可能有问题。当用户的一个连接关闭时，应该只从该用户的Set中移除当前实例，而不是删除整个userId的条目。例如，假设用户A有两个连接，关闭其中一个，webSocketMap.get(userId)应该还保留另一个连接。但当前代码在onClose中直接webSocketMap.remove(userId)，这会导致所有连接都被移除。这可能是个严重的错误，需要修正。正确的做法应该是从Set中移除当前实例，并且如果Set为空，再移除userId的条目。因此，在测试时，需要验证当多个连接存在时，关闭一个是否不影响其他连接。

例如，在测试用例中，模拟两个连接同一个userId的客户端，然后关闭其中一个，检查webSocketMap中该用户的Set是否还包含另一个实例。如果代码没有正确处理，测试会失败，从而发现问题。

另一个测试点是onlineCount的增减是否正确。每次onOpen时onlineCount加1，onClose时减1。需要测试多个连接情况下，计数是否正确。

对于sendMessageToUser方法，测试是否能正确发送消息给所有用户的连接。可以模拟多个客户端连接，然后调用该方法，检查每个客户端是否收到消息。同时，测试当某个连接已关闭时，发送是否会失败，并处理异常。

可能还需要测试线程安全性，因为WebSocket是多线程环境，多个连接同时操作webSocketMap和onlineCount是否会导致竞态条件。虽然代码中对onlineCount的操作是synchronized，但webSocketMap的操作没有同步，可能存在并发问题。例如，在onOpen中，检查webSocketMap.containsKey(userId)后，另一个线程可能同时进行相同的操作，导致重复添加或覆盖。应该考虑使用ConcurrentHashMap或者同步块来保证线程安全。

此外，equals和hashCode方法被覆盖为默认的Object实现，这可能有问题。因为WebSocketServer实例被存储在Set中，如果两个不同的实例被认为是相等的，会导致无法正确存储多个连接。但当前代码中的equals是使用super.equals，即对象地址比较，所以每个实例都是唯一的，这样Set可以正确存储多个实例。不过，是否需要根据userId或其他属性来重写equals和hashCode？例如，如果两个WebSocketServer实例的userId相同，是否应该视为相等？当前代码中可能不需要，因为每个连接都是独立的实例，即使同一用户的不同连接，实例也不同。所以当前的equals实现是正确的，但需要确认业务逻辑是否允许同一用户的多个连接存在，这在当前代码中是允许的，所以没问题。

在测试中，需要模拟多个连接同一userId的情况，检查webSocketMap中的Set是否正确添加了多个实例。例如，两个不同的WebSocketServer实例被添加到同一个userId的Set中。

总结测试点：

1. 单用户单连接：连接建立后，webSocketMap中是否正确记录，onlineCount增加。
2. 同一用户多连接：多个连接是否都被添加到Set中，onlineCount是否正确累加。
3. 连接关闭：关闭一个连接后，是否从Set中移除，onlineCount减少，其他连接不受影响。
4. 发送消息：向用户发送消息，所有连接的客户端是否都能收到。
5. 异常处理：发送消息时出现IOException，是否正确处理并返回false。
6. 线程安全测试：多线程同时连接和断开，检查数据一致性。
7. JSON解析测试：在onMessage中添加JSON解析后，发送合法和非法消息，检查是否处理正确。

在编写测试时，可以使用JUnit 5，Mockito来模拟Session对象，以及使用Spring Boot的WebSocket测试工具。例如，使用@SpringBootTest启动嵌入式容器，然后使用WebSocketClient来建立连接。

例如，一个集成测试的步骤可能如下：

1. 启动测试服务器。
2. 客户端1连接到/webSocket/user1。
3. 验证webSocketMap中user1的Set大小为1，onlineCount为1。
4. 客户端2连接到/webSocket/user1。
5. 验证Set大小为2，onlineCount为2。
6. 调用sendMessageToUser("user1", "test")，验证两个客户端是否收到消息。
7. 关闭客户端1，验证Set大小减为1，onlineCount减为1。
8. 关闭客户端2，验证webSocketMap中user1的条目被移除，onlineCount为0。

此外，对于onClose中的bug，需要修改代码，例如：

原onClose方法：
webSocketMap.remove(userId);

应该改为：
Set<WebSocketServer> set = webSocketMap.get(userId);
if (set != null) {
set.remove(this);
if (set.isEmpty()) {
webSocketMap.remove(userId);
}
}

这样才能正确移除当前实例，而不是整个用户的所有连接。测试时需要验证这一点。

用户提供的代码中的onClose方法确实存在这个问题，所以测试时需要特别注意，并建议用户修正这个bug。

总结，测试这个WebSocketServer类需要覆盖正常连接、多连接管理、消息发送、连接关闭处理、异常情况以及潜在的并发问题。同时，需要修正现有的onClose方法中的错误，以确保正确管理连接。</think>### 代码分析与测试方案

#### 一、代码分析
这是一个基于 `javax.websocket` 的 WebSocket 服务端实现，支持多终端连接和消息推送，关键逻辑如下：

1. **核心功能**
- **连接管理**：通过 `Map<String, Set<WebSocketServer>>` 按 `userId` 管理多个终端连接。
- **消息推送**：支持向同一用户的所有终端广播消息（如 `sendMessageToUser` 方法）。
- **在线计数**：通过静态变量 `onlineCount` 记录全局在线连接数。

2. **潜在问题**
- **线程安全问题**：`webSocketMap` 使用非线程安全的 `HashMap`，多线程并发操作可能引发数据不一致。
- **连接关闭逻辑错误**：`onClose` 方法直接移除整个 `userId` 对应的连接集合，导致同一用户的其他终端连接被错误清理。
- **JSON 解析未实现**：`onMessage` 方法中的 JSON 解析逻辑被注释，需补充完整。

---

#### 二、测试方案

##### 1. 单元测试（JUnit + Mockito）
**目标**：验证单个方法的逻辑正确性。

**测试用例示例**：

@Mock
private Session mockSession;

@Test
public void testOnOpenAddsUserToMap() {
    WebSocketServer server = new WebSocketServer();
    server.onOpen(mockSession, "user1");
    
    assertTrue(webSocketMap.containsKey("user1"));
    assertEquals(1, webSocketMap.get("user1").size());
}

##### 2. 集成测试（Spring Boot Test + WebSocketClient）
**目标**：验证完整的 WebSocket 生命周期。

**步骤**：
1. 启动内嵌 Web 服务器。
2. 模拟多个客户端连接同一 `userId`。
3. 验证连接数、消息收发、异常处理等。

**代码片段**：

@SpringBootTest(webEnvironment = WebEnvironment.RANDOM_PORT)
public class WebSocketIntegrationTest {
    
    @LocalServerPort
    private int port;

    @Test
    public void testMultiClientConnection() throws Exception {
        WebSocketClient client1 = new StandardWebSocketClient();
        WebSocketSession session1 = client1.doHandshake(new TextWebSocketHandler(), "ws://localhost:" + port + "/webSocket/user1").get();

        WebSocketClient client2 = new StandardWebSocketClient();
        WebSocketSession session2 = client2.doHandshake(new TextWebSocketHandler(), "ws://localhost:" + port + "/webSocket/user1").get();

        assertTrue(WebSocketServer.sendMessageToUser("user1", "Hello"));
        // 验证两个客户端是否收到消息
    }
}

##### 3. 并发测试
**目标**：验证线程安全性。

**工具**：`CountDownLatch` + 多线程模拟并发连接。

**测试逻辑**：

@Test
public void testConcurrentConnections() throws InterruptedException {
    int threads = 100;
    ExecutorService executor = Executors.newFixedThreadPool(threads);
    CountDownLatch latch = new CountDownLatch(threads);

    for (int i = 0; i < threads; i++) {
        executor.submit(() -> {
            WebSocketServer server = new WebSocketServer();
            server.onOpen(mockSession, "user1");
            latch.countDown();
        });
    }

    latch.await();
    assertEquals(threads, WebSocketServer.getOnlineCount());
}

##### 4. 异常测试
**目标**：验证错误处理逻辑。

**用例**：
- 发送非法 JSON 格式消息，检查是否抛出预期异常。
- 模拟断开连接后发送消息，验证 `sendMessageToUser` 是否返回 `false`。

---

#### 三、代码修正建议
1. **修复 `onClose` 方法**
修正连接关闭逻辑，仅移除当前实例而非整个用户：

   @OnClose
   public void onClose() {
       Set<WebSocketServer> connections = webSocketMap.get(userId);
       if (connections != null) {
           connections.remove(this);
           if (connections.isEmpty()) {
               webSocketMap.remove(userId);
           }
           subOnlineCount();
       }
   }

2. **增强线程安全**
将 `webSocketMap` 替换为 `ConcurrentHashMap`，并对关键操作加锁：

   private static Map<String, Set<WebSocketServer>> webSocketMap = new ConcurrentHashMap<>();

3. **完善 JSON 解析**
在 `onMessage` 中补充 JSON 解析逻辑：

   @OnMessage
   public void onMessage(String message, Session session) {
       try {
           JSONObject json = JSON.parseObject(message);
           // 处理业务逻辑
       } catch (Exception e) {
           log.error("消息解析失败: {}", message);
       }
   }

---

#### 四、总结
通过单元测试、集成测试、并发测试和异常测试的组合，可全面覆盖以下场景：
- 单用户多终端连接管理
- 消息广播的正确性
- 线程安全性保障
- 异常场景的健壮性

建议优先修复 `onClose` 的逻辑错误和线程安全问题，再逐步补充测试用例。