load tester beta

src/main/java/com/eactive/testmaster/config/ServletConfig.java

@@ -9,20 +9,21 @@ import org.springframework.context.annotation.Configuration;
 
 @Configuration
 public class ServletConfig {
-	@Value("${server.port.http}")
-	private int serverPortHttp;
 
-	@Bean
-	public ServletWebServerFactory serverFactory() {
-		TomcatServletWebServerFactory tomcat = new TomcatServletWebServerFactory();
-		tomcat.addAdditionalTomcatConnectors(createStandardConnector());
-		return tomcat;
-	}
+    @Value("${server.port.http}")
+    private int serverPortHttp;
 
-	private Connector createStandardConnector() {
-		Connector connector = new Connector("org.apache.coyote.http11.Http11NioProtocol");
-		connector.setPort(serverPortHttp);
-		return connector;
-	}
+    @Bean
+    public ServletWebServerFactory serverFactory() {
+        TomcatServletWebServerFactory tomcat = new TomcatServletWebServerFactory();
+        tomcat.addAdditionalTomcatConnectors(createStandardConnector());
+        return tomcat;
+    }
+
+    private Connector createStandardConnector() {
+        Connector connector = new Connector("org.apache.coyote.http11.Http11NioProtocol");
+        connector.setPort(serverPortHttp);
+        return connector;
+    }
 
 }

src/main/java/com/eactive/testmaster/config/WebSocketStompConfig.java

@@ -0,0 +1,48 @@
+package com.eactive.testmaster.config;
+
+import org.springframework.context.annotation.Configuration;
+import org.springframework.messaging.simp.config.ChannelRegistration;
+import org.springframework.messaging.simp.config.MessageBrokerRegistry;
+import org.springframework.messaging.simp.stomp.StompHeaderAccessor;
+import org.springframework.messaging.support.ChannelInterceptor;
+import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
+import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
+import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
+
+@Configuration
+@EnableWebSocketMessageBroker
+public class WebSocketStompConfig implements WebSocketMessageBrokerConfigurer {
+
+    @Override
+    public void registerStompEndpoints(StompEndpointRegistry registry) {
+        registry.addEndpoint("/load-test-result").withSockJS();
+    }
+
+
+    @Override
+    public void configureMessageBroker(MessageBrokerRegistry registry) {
+        registry.enableSimpleBroker("/topic");
+        registry.setApplicationDestinationPrefixes("/app");
+    }
+    @Override
+    public void configureClientInboundChannel(ChannelRegistration registration) {
+        registration.interceptors(new ChannelInterceptor() {
+            @Override
+            public void postSend(org.springframework.messaging.Message<?> message, org.springframework.messaging.MessageChannel channel, boolean sent) {
+                StompHeaderAccessor accessor = StompHeaderAccessor.wrap(message);
+                switch (accessor.getCommand()) {
+                    case CONNECT:
+                        // Log or handle session connect
+                        System.out.println("STOMP Connect [sessionId: " + accessor.getSessionId() + "]");
+                        break;
+                    case DISCONNECT:
+                        // Log or handle session disconnect
+                        System.out.println("STOMP Disconnect [sessionId: " + accessor.getSessionId() + "]");
+                        break;
+                    default:
+                        break;
+                }
+            }
+        });
+    }
+}