这个demo主要是利用Java多线程来测试WebSocket通信。首先，创建一个WebSocket服务器和客户端，然后使用多线程来模拟多个客户端同时连接服务器进行通信。通过多线程测试，可以验证WebSocket通信的并发性能和稳定性。同时，可以通过多线程测试来模拟不同场景下的并发请求，以便对WebSocket服务器进行压力测试和性能优化。整个demo会包括创建WebSocket服务器和客户端的代码，以及多线程测试的代码实现。通过这个demo，可以更好地了解WebSocket通信的多线程测试方法和实现。
@ClientEndpoint
public class WebSocketClient {private static final AtomicInteger connectedCount = new AtomicInteger(0);private static final LongAdder firstFrameLatencySum = new LongAdder();private static final CountDownLatch latch = new CountDownLatch(20);private static final int MESSAGE_SIZE = 1024;private static final int CONCURRENCY_LEVEL = 20;private static final AtomicLong startTime = new AtomicLong(0);@OnOpenpublic void onOpen(Session session) {connectedCount.incrementAndGet();}@OnMessagepublic void onMessage(String message) {startTime.set(System.nanoTime());System.out.println(Thread.currentThread().getName() + ": " + message);}@OnClosepublic void onClose(Session session, CloseReason closeReason) {connectedCount.decrementAndGet();}public static void main(String[] args) throws URISyntaxException, Exception {for (int i = 0; i < CONCURRENCY_LEVEL; i++) {WebSocketContainer container = ContainerProvider.getWebSocketContainer();Session session = container.connectToServer(WebSocketClient.class, new URI("ws://you_url"));session.getBasicRemote().sendBinary(ByteBuffer.wrap(generateMessage()));latch.countDown();}latch.await(100, TimeUnit.SECONDS);System.out.println("Connected clients: " + connectedCount.get());System.out.println("Average first frame latency: " + (firstFrameLatencySum.sum() / CONCURRENCY_LEVEL) + " nanoseconds");}private static byte[] generateMessage() {try {FileInputStream fileInputStream = new FileInputStream("D:\\smn.wav");byte[] fileData = new byte[fileInputStream.available()];fileInputStream.read(fileData);fileInputStream.close();return ArrayUtils.subarray(fileData, 0, 300000);} catch (IOException e) {throw new RuntimeException(e);}}
}
)
)
