基于Java对STOMP服务端进行测试

一、目标

上一节我们说了构建SpringBoot + WebSocket+STOMP指定推送消息。我们这一节对它进行测试。

我们预期的并发目标：

• 支持10000+ 用户
• 每个用户同时发布 500条数据

我们准备的环境：

• -Xms512m
• -Xmx4096m
• CPU 12核 20线程
• 内存 16G

二、服务端改动

之前的服务端我们是这样设置的 config.enableSimpleBroker(),这种设置属于入门级使用，它非常简单但仅支持 STOMP 命令的子集（无确认、收据等）。

但我们测试时要求确认订阅状态，来确定订阅是否成功。所以必须对之前的配置类进行改动，来满足要求。

客户端订阅确认配置：

 @Bean
    public ApplicationListener<SessionSubscribeEvent> webSocketEventListener(
            final AbstractSubscribableChannel clientOutboundChannel) {
   
        return event -> {
   
            Message<byte[]> message = event.getMessage();
            StompHeaderAccessor stompHeaderAccessor = StompHeaderAccessor.wrap(message);
            if (stompHeaderAccessor.getReceipt() != null) {
   
                stompHeaderAccessor.setHeader("stompCommand", StompCommand.RECEIPT);
                stompHeaderAccessor.setReceiptId(stompHeaderAccessor.getReceipt());
                clientOutboundChannel.send(
                        MessageBuilder.createMessage(new byte[0], stompHeaderAccessor.getMessageHeaders()));
            }
        };
    }

三、测试程序

3.1 依赖

  <dependency>
            <groupId>org.eclipse.jetty.websocket</groupId>
            <artifactId>websocket-client</artifactId>
            <version>9.4.48.v20220622</version>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>org.eclipse.jetty.websocket</groupId>
            <artifactId>websocket-server</artifactId>
			<version>9.4.48.v20220622</version>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>org.eclipse.jetty</groupId>
            <artifactId>jetty-webapp</artifactId>
			<version>9.4.48.v20220622</version>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>org.eclipse.jetty</groupId>
            <artifactId>jetty-client</artifactId>
			<version>9.4.48.v20220622</version>
            <scope>test</scope>
        </dependency>

3.1 用户量和消息数

	//用户量
	private static final int NUMBER_OF_USERS = 10000;

	//消息数
	private static final int BROADCAST_MESSAGE_COUNT = 500;

3.2 测试端口是否可用

		String host = "localhost";
		if (args.length > 0) {
   
			host = args[0];
		}

		int port = 6060;
		if (args.length > 1)