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); } }这个加测试

嗯，用户让我分析这个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。
接下来，测试的话，需要考虑以下几个方面：

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

由于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中。
总结测试点：

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

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

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

此外，对于onClose中的bug，需要修改代码，例如：
原onClose方法：
webSocketMap.remove(userId);
应该改为：
Set set = webSocketMap.get(userId);
if (set != null) {
set.remove(this);
if (set.isEmpty()) {
webSocketMap.remove(userId);
}
}
这样才能正确移除当前实例，而不是整个用户的所有连接。测试时需要验证这一点。
用户提供的代码中的onClose方法确实存在这个问题，所以测试时需要特别注意，并建议用户修正这个bug。
总结，测试这个WebSocketServer类需要覆盖正常连接、多连接管理、消息发送、连接关闭处理、异常情况以及潜在的并发问题。同时，需要修正现有的onClose方法中的错误，以确保正确管理连接。### 代码分析与测试方案
一、代码分析
这是一个基于 javax.websocket 的 WebSocket 服务端实现，支持多终端连接和消息推送，关键逻辑如下：

核心功能

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

潜在问题

线程安全问题：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 生命周期。
步骤：

启动内嵌 Web 服务器。
模拟多个客户端连接同一 userId。
验证连接数、消息收发、异常处理等。

代码片段：
@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。

三、代码修正建议

修复 onClose 方法修正连接关闭逻辑，仅移除当前实例而非整个用户：
@OnClose
public void onClose() {
Set<WebSocketServer> connections = webSocketMap.get(userId);
if (connections != null) {
connections.remove(this);
if (connections.isEmpty()) {
webSocketMap.remove(userId);
}
subOnlineCount();
}
}
登录后复制

增强线程安全将 webSocketMap 替换为 ConcurrentHashMap，并对关键操作加锁：
private static Map<String, Set<WebSocketServer>> webSocketMap = new ConcurrentHashMap<>();
登录后复制

完善 JSON 解析在 onMessage 中补充 JSON 解析逻辑：
@OnMessage
public void onMessage(String message, Session session) {
try {
JSONObject json = JSON.parseObject(message);
// 处理业务逻辑
} catch (Exception e) {
log.error("消息解析失败: {}", message);
}
}
登录后复制

四、总结
通过单元测试、集成测试、并发测试和异常测试的组合，可全面覆盖以下场景：

单用户多终端连接管理
消息广播的正确性
线程安全性保障
异常场景的健壮性

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