临时

vital-signs/src/main/java/com/rax/vital/config/WebSocketConfig.java

@@ -1,47 +1,39 @@
 package com.rax.vital.config;
 
-import com.rax.vital.interceptor.WSChannelInterceptor;
-import org.springframework.beans.factory.annotation.Autowired;
+import com.rax.vital.handler.ChatHandler;
+import com.rax.vital.handler.MedicineHandler;
+import com.rax.vital.interceptor.WebSocketInterceptors;
+import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
-import org.springframework.messaging.simp.config.ChannelRegistration;
-import org.springframework.messaging.simp.config.MessageBrokerRegistry;
-import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
-import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
-import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
-import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
+import org.springframework.web.socket.WebSocketHandler;
+import org.springframework.web.socket.config.annotation.EnableWebSocket;
+import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
+import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
+import org.springframework.web.socket.server.HandshakeInterceptor;
+import org.springframework.web.socket.server.support.HttpSessionHandshakeInterceptor;
 
+@EnableWebSocket
 @Configuration
-@EnableWebSocketMessageBroker
-public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
-
-	@Autowired
-	private WSChannelInterceptor wsChannelInterceptor;
-
+public class WebSocketConfig implements WebSocketConfigurer {
 	@Override
-	public void registerStompEndpoints(StompEndpointRegistry registry) {
-		registry.addEndpoint("/rax/chat", "/rax/SurgeryData")
+	public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
+		registry.addHandler(medicineHandler(), "/vitalSignsMedicine")
+				.addHandler(chatHandler(), "/chatRoom")
+				.addInterceptors(webSocketHandshakeInterceptor())
 				.setAllowedOrigins("*");
 	}
 
-	@Override
-	public void configureMessageBroker(MessageBrokerRegistry registry) {
-		// 第一个值表示客户端发送心跳消息的间隔时间，第二个值表示服务端发送心跳消息的间隔时间
-		long [] heartbeat = {60000, 60000};
-		ThreadPoolTaskScheduler threadPoolTaskScheduler = new ThreadPoolTaskScheduler();
-		threadPoolTaskScheduler.initialize();
-		registry.enableSimpleBroker("/topic").setTaskScheduler(threadPoolTaskScheduler).setHeartbeatValue(heartbeat);
-		registry.setApplicationDestinationPrefixes("/front");
-		registry.setUserDestinationPrefix("/topic/user");
+	@Bean
+	public WebSocketHandler medicineHandler() {
+		return new MedicineHandler();
 	}
 
-	/**
-	 * stomp未登录验证
-	 *
-	 * @param registration
-	 */
-	@Override
-	public void configureClientInboundChannel(ChannelRegistration registration) {
-		registration.interceptors(wsChannelInterceptor);
+	@Bean
+	public WebSocketHandler chatHandler() {
+		return new ChatHandler();
 	}
 
+	public HandshakeInterceptor webSocketHandshakeInterceptor() {
+		return new WebSocketInterceptors();
+	}
 }

vital-signs/src/main/java/com/rax/vital/config/WebSocketStompConfig.java

@@ -0,0 +1,47 @@
+package com.rax.vital.config;
+
+import com.rax.vital.interceptor.WSChannelInterceptor;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.messaging.simp.config.ChannelRegistration;
+import org.springframework.messaging.simp.config.MessageBrokerRegistry;
+import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
+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 {
+
+	@Autowired
+	private WSChannelInterceptor wsChannelInterceptor;
+
+	@Override
+	public void registerStompEndpoints(StompEndpointRegistry registry) {
+		registry.addEndpoint("/rax/chat", "/rax/SurgeryData")
+				.setAllowedOrigins("*");
+	}
+
+	@Override
+	public void configureMessageBroker(MessageBrokerRegistry registry) {
+		// 第一个值表示客户端发送心跳消息的间隔时间，第二个值表示服务端发送心跳消息的间隔时间
+		long [] heartbeat = {60000, 60000};
+		ThreadPoolTaskScheduler threadPoolTaskScheduler = new ThreadPoolTaskScheduler();
+		threadPoolTaskScheduler.initialize();
+		registry.enableSimpleBroker("/topic").setTaskScheduler(threadPoolTaskScheduler).setHeartbeatValue(heartbeat);
+		registry.setApplicationDestinationPrefixes("/front");
+		registry.setUserDestinationPrefix("/topic/user");
+	}
+
+	/**
+	 * stomp未登录验证
+	 *
+	 * @param registration
+	 */
+	@Override
+	public void configureClientInboundChannel(ChannelRegistration registration) {
+		registration.interceptors(wsChannelInterceptor);
+	}
+
+}

vital-signs/src/main/java/com/rax/vital/handler/ChatHandler.java

@@ -0,0 +1,33 @@
+package com.rax.vital.handler;
+
+import org.springframework.web.socket.CloseStatus;
+import org.springframework.web.socket.WebSocketHandler;
+import org.springframework.web.socket.WebSocketMessage;
+import org.springframework.web.socket.WebSocketSession;
+
+public class ChatHandler implements WebSocketHandler {
+	@Override
+	public void afterConnectionEstablished(WebSocketSession session) throws Exception {
+
+	}
+
+	@Override
+	public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {
+
+	}
+
+	@Override
+	public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
+
+	}
+
+	@Override
+	public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
+
+	}
+
+	@Override
+	public boolean supportsPartialMessages() {
+		return false;
+	}
+}

vital-signs/src/main/java/com/rax/vital/handler/MedicineHandler.java

@@ -0,0 +1,33 @@
+package com.rax.vital.handler;
+
+import org.springframework.web.socket.CloseStatus;
+import org.springframework.web.socket.WebSocketHandler;
+import org.springframework.web.socket.WebSocketMessage;
+import org.springframework.web.socket.WebSocketSession;
+
+public class MedicineHandler implements WebSocketHandler {
+	@Override
+	public void afterConnectionEstablished(WebSocketSession session) throws Exception {
+		session.getPrincipal();
+	}
+
+	@Override
+	public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {
+
+	}
+
+	@Override
+	public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
+
+	}
+
+	@Override
+	public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
+
+	}
+
+	@Override
+	public boolean supportsPartialMessages() {
+		return false;
+	}
+}

vital-signs/src/main/java/com/rax/vital/interceptor/WebSocketInterceptors.java

@@ -0,0 +1,25 @@
+package com.rax.vital.interceptor;
+
+import org.springframework.http.server.ServerHttpRequest;
+import org.springframework.http.server.ServerHttpResponse;
+import org.springframework.http.server.ServletServerHttpRequest;
+import org.springframework.web.socket.WebSocketHandler;
+import org.springframework.web.socket.server.HandshakeInterceptor;
+
+import java.util.Map;
+
+public class WebSocketInterceptors  implements HandshakeInterceptor {
+	@Override
+	public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {
+		if (request instanceof ServletServerHttpRequest) {
+			ServletServerHttpRequest servletServerHttpRequest = (ServletServerHttpRequest) request;
+			servletServerHttpRequest.
+		}
+		return false;
+	}
+
+	@Override
+	public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Exception exception) {
+
+	}
+}