package com.ruoyi.web.controller.webSocket;
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import java.net.URI;
|
import java.util.concurrent.CountDownLatch;
|
import java.util.concurrent.TimeUnit;
|
|
/**
|
* 简单的WebSocket客户端实现
|
* 用于测试WebSocket服务端功能
|
*/
|
@Slf4j
|
public class SimpleWebSocketClient implements WebSocketClientTest.WebSocketClient {
|
|
private final String serverUrl;
|
private final WebSocketClientTest.WebSocketClient.MessageHandler messageHandler;
|
private WebSocketConnection connection;
|
private volatile boolean connected = false;
|
|
public SimpleWebSocketClient(String serverUrl, WebSocketClientTest.WebSocketClient.MessageHandler messageHandler) {
|
this.serverUrl = serverUrl;
|
this.messageHandler = messageHandler;
|
}
|
|
@Override
|
public void connect() throws Exception {
|
log.info("正在连接到WebSocket服务器: {}", serverUrl);
|
|
// 这里使用模拟的WebSocket连接
|
// 在实际项目中，您可以使用Java-WebSocket库或其他WebSocket客户端库
|
connection = new WebSocketConnection(serverUrl, messageHandler);
|
connection.connect();
|
|
// 等待连接建立
|
int retryCount = 0;
|
while (!connected && retryCount < 10) {
|
Thread.sleep(500);
|
retryCount++;
|
}
|
|
if (!connected) {
|
throw new RuntimeException("连接WebSocket服务器超时");
|
}
|
}
|
|
@Override
|
public void send(String message) {
|
if (connection != null && connected) {
|
connection.send(message);
|
} else {
|
log.warn("WebSocket未连接，无法发送消息: {}", message);
|
}
|
}
|
|
@Override
|
public void close() {
|
if (connection != null) {
|
connection.close();
|
connected = false;
|
}
|
}
|
|
/**
|
* 模拟的WebSocket连接类
|
*/
|
private static class WebSocketConnection {
|
private final String serverUrl;
|
private final WebSocketClientTest.WebSocketClient.MessageHandler messageHandler;
|
private volatile boolean isConnected = false;
|
|
public WebSocketConnection(String serverUrl, WebSocketClientTest.WebSocketClient.MessageHandler messageHandler) {
|
this.serverUrl = serverUrl;
|
this.messageHandler = messageHandler;
|
}
|
|
public void connect() {
|
// 模拟连接过程
|
new Thread(() -> {
|
try {
|
Thread.sleep(1000); // 模拟连接延迟
|
isConnected = true;
|
log.info("WebSocket连接已建立");
|
messageHandler.onOpen();
|
} catch (InterruptedException e) {
|
Thread.currentThread().interrupt();
|
messageHandler.onError(e);
|
}
|
}).start();
|
}
|
|
public void send(String message) {
|
if (isConnected) {
|
log.info("发送消息: {}", message);
|
// 模拟服务器响应
|
simulateServerResponse(message);
|
}
|
}
|
|
public void close() {
|
isConnected = false;
|
log.info("WebSocket连接已关闭");
|
messageHandler.onClose(1000, "正常关闭");
|
}
|
|
/**
|
* 模拟服务器响应
|
*/
|
private void simulateServerResponse(String message) {
|
new Thread(() -> {
|
try {
|
Thread.sleep(100); // 模拟网络延迟
|
|
// 解析消息类型并模拟响应
|
if (message.contains("\"type\":\"auth\"")) {
|
if (message.contains("\"token\"")) {
|
// 有token的认证响应
|
String response = "{\"type\":\"auth_success\",\"appUserId\":\"test-user-001\",\"nickName\":\"测试用户\",\"message\":\"认证成功\",\"timestamp\":\"" +
|
java.time.LocalDateTime.now() + "\"}";
|
messageHandler.onMessage(response);
|
} else {
|
// 无token的认证响应
|
String response = "{\"type\":\"auth_success\",\"appUserId\":\"test-user-001\",\"nickName\":\"测试用户\",\"message\":\"认证成功（无Token）\",\"timestamp\":\"" +
|
java.time.LocalDateTime.now() + "\"}";
|
messageHandler.onMessage(response);
|
}
|
} else if (message.contains("\"type\":\"ping\"")) {
|
// 心跳响应
|
String response = "{\"type\":\"pong\",\"appUserId\":\"test-user-001\",\"timestamp\":\"" +
|
java.time.LocalDateTime.now() + "\"}";
|
messageHandler.onMessage(response);
|
} else if (message.contains("\"type\":\"business\"")) {
|
// 业务消息响应
|
String response = "{\"type\":\"message_response\",\"appUserId\":\"test-user-001\",\"originalMessage\":" +
|
message + ",\"serverTime\":\"" + java.time.LocalDateTime.now() + "\"}";
|
messageHandler.onMessage(response);
|
}
|
} catch (Exception e) {
|
messageHandler.onError(e);
|
}
|
}).start();
|
}
|
}
|
}
|
