test HttpSessionHandshakeInterceptor

Author: netbuffer

src/main/java/cn/netbuffer/springboot/websocket/demo/config/SpringWebSocketConfig.java

@@ -8,6 +8,7 @@
 import org.springframework.web.socket.WebSocketHandler;
 import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
 import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
+import org.springframework.web.socket.server.support.HttpSessionHandshakeInterceptor;
 
 @Slf4j
 @Configuration
@@ -16,7 +17,10 @@ public class SpringWebSocketConfig implements WebSocketConfigurer {
     @Override
     public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
 //        registry.addHandler(buildTextHandler(), "/ws").setAllowedOrigins("*");
-        registry.addHandler(buildMessageHandler(), "/ws").setAllowedOrigins("*");
+        registry.addHandler(buildMessageHandler(), "/ws")
+                //复制HttpSession中的属性到WebSocketSession中
+                .addInterceptors(new HttpSessionHandshakeInterceptor())
+                .setAllowedOrigins("*");
     }
 
     @Bean

src/main/java/cn/netbuffer/springboot/websocket/demo/controller/HttpSessionController.java

@@ -0,0 +1,20 @@
+package cn.netbuffer.springboot.websocket.demo.controller;
+
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+import javax.servlet.http.HttpSession;
+
+@Slf4j
+@RestController
+@RequestMapping("/http/session")
+public class HttpSessionController {
+
+    @PostMapping("set")
+    public void setSessionValue(HttpSession httpSession, String key, String value) {
+        httpSession.setAttribute(key, value);
+        log.debug("setSessionValue key={} value={}", key, value);
+    }
+
+}

src/main/java/cn/netbuffer/springboot/websocket/demo/websocket/handler/MessageHandler.java

@@ -12,6 +12,7 @@ public void afterConnectionEstablished(WebSocketSession session) throws Exceptio
         super.afterConnectionEstablished(session);
         SessionManager.put(session.getId(), session);
         log.info("session[{}] afterConnectionEstablished", session.getId());
+        log.info("get WebSocketSession attributes={}",session.getAttributes());
     }
 
     @Override

src/main/resources/templates/index.html

@@ -23,6 +23,22 @@
                 alert("send success");
             });
     }
+
+    function set_session() {
+        var key = prompt("请输入key");
+        var value = prompt("请输入value");
+        fetch("/http/session/set", {
+            body: ({"key": key, "value": value}),
+            method: 'POST'
+        })
+            .then(function (response) {
+                return response;
+            })
+            .then(function (response) {
+                console.log(`send msg success`);
+                alert("send success");
+            });
+    }
 </script>
 </body>
 </html>