我最近參與了一個項目，我們需要捕獲Websocket請求的http會話ID –原因是要確定使用相同基礎HTTP會話的Websocket會話的數量。

這樣做的方法基于利用新的spring-session模塊的示例，并在此處進行描述。

捕獲http會話ID的技巧是理解在瀏覽器和服務器之間建立Websocket連接之前，有一個通過HTTP協商的握手階段，并且會話ID在此握手階段傳遞給服務器。

Spring Websocket支持提供了一種注冊HandShakeInterceptor的好方法，該方法可用于捕獲http會話ID并將其設置在子協議（通常為STOMP）標頭中。 首先，這是捕獲會話ID并將其設置為標頭的方法：

public class HttpSessionIdHandshakeInterceptor implements HandshakeInterceptor {@Overridepublic boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {if (request instanceof ServletServerHttpRequest) {ServletServerHttpRequest servletRequest = (ServletServerHttpRequest) request;HttpSession session = servletRequest.getServletRequest().getSession(false);if (session != null) {attributes.put("HTTPSESSIONID", session.getId());}}return true;}public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Exception ex) {} }

并在Spring Websocket支持下注冊此HandshakeInterceptor：

@Configuration @EnableWebSocketMessageBroker public class WebSocketDefaultConfig extends AbstractWebSocketMessageBrokerConfigurer {@Overridepublic void configureMessageBroker(MessageBrokerRegistry config) {config.enableSimpleBroker("/topic/", "/queue/");config.setApplicationDestinationPrefixes("/app");}@Overridepublic void registerStompEndpoints(StompEndpointRegistry registry) {registry.addEndpoint("/chat").withSockJS().setInterceptors(httpSessionIdHandshakeInterceptor());}@Beanpublic HttpSessionIdHandshakeInterceptor httpSessionIdHandshakeInterceptor() {return new HttpSessionIdHandshakeInterceptor();}}

現在，會話ID是STOMP標頭的一部分，可以將其作為STOMP標頭抓取，以下是在向服務器注冊訂閱時抓取的示例：

@Component public class StompSubscribeEventListener implements ApplicationListener<SessionSubscribeEvent> {private static final Logger logger = LoggerFactory.getLogger(StompSubscribeEventListener.class);@Overridepublic void onApplicationEvent(SessionSubscribeEvent sessionSubscribeEvent) {StompHeaderAccessor headerAccessor = StompHeaderAccessor.wrap(sessionSubscribeEvent.getMessage());logger.info(headerAccessor.getSessionAttributes().get("HTTPSESSIONID").toString());} }

或者可以從將websocket消息作為MessageHeaders參數處理的控制器方法中獲取：

@MessageMapping("/chats/{chatRoomId}")public void handleChat(@Payload ChatMessage message, @DestinationVariable("chatRoomId") String chatRoomId, MessageHeaders messageHeaders, Principal user) {logger.info(messageHeaders.toString());this.simpMessagingTemplate.convertAndSend("/topic/chats." + chatRoomId, "[" + getTimestamp() + "]:" + user.getName() + ":" + message.getMessage());}

• 這是實現此模式的完整工作示例。

翻譯自: https://www.javacodegeeks.com/2014/11/spring-boot-based-websocket-application-and-capturing-http-session-id.html

總結

以上是生活随笔為你收集整理的基于Spring Boot的WebSocket应用程序并捕获HTTP会话ID的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。