SpringBoot在WebSocket長連接中獲取到HttpSession
Websocket是通過http協議握手后升級成為長連接，在握手的時候，可以讀取到客戶端http請求的所有信息，自然也包括 HttpSession。
自定義配置類，繼承 Configurator ，覆寫modifyHandshake方法
ServerEndpointConfigImpl.Configurator，提供了一個modifyHandshake方法，可以在完成握手后，讀取客戶端的請求，以及修改對客戶端的響應。HandshakeRequest 類已經提供了獲取HttpSession的接口。
import javax.servlet.http.HttpSession;
import javax.websocket.HandshakeResponse;
import javax.websocket.server.HandshakeRequest;
import javax.websocket.server.ServerEndpointConfig;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import io.undertow.websockets.jsr.ServerEndpointConfigImpl;
public class ServerEndpointConfigurator extends ServerEndpointConfigImpl.Configurator {
private static final Logger LOGGER = LoggerFactory.getLogger(ServerEndpointConfigurator.class);
@Override
public void modifyHandshake(ServerEndpointConfig sec, HandshakeRequest request, HandshakeResponse response) {
// 嘗試獲取到當前Websocket鏈接的HttpSession
HttpSession httpSession = (HttpSession)request.getHttpSession();
LOGGER.info("session={}", httpSession);
if (httpSession != null) {
// session id
LOGGER.info("sessionId={}", httpSession.getId());
// 讀取session域中存儲的數據
LOGGER.info("name={}", httpSession.getAttribute("name"));
}
super.modifyHandshake(sec, request, response);
}
}
在@ServerEndpoint指定Configurator實現類
通過 @ServerEndpoint 注解的 configurator 指定Configurator實現類
import java.io.IOException;
import javax.websocket.CloseReason;
import javax.websocket.EndpointConfig;
import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;
import io.springboot.twitter.websocket.ServerEndpointConfigurator;
@ServerEndpoint(value = "/channel/test",
configurator = ServerEndpointConfigurator.class // 指定服務端的配置類
)
public class TestChannel {
@OnMessage
public void onMessage(String message) throws IOException {
}
@OnOpen
public void onOpen(Session session, EndpointConfig endpointConfig) throws IOException {
}
@OnClose
public void onClose(CloseReason closeReason) {
}
@OnError
public void onError(Throwable throwable) throws IOException {
}
}
測試
通過Controller渲染視圖，並且創建Session
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.servlet.ModelAndView;
@RestController
@RequestMapping("/test")
public class TestController {
private static final Logger LOGGER = LoggerFactory.getLogger(TestController.class);
@GetMapping
public Object test (HttpServletRequest request) {
HttpSession httpSesion = request.getSession();
LOGGER.info("創建新的httpSession={}", httpSesion.getId());
// 存儲數據到Session域
httpSesion.setAttribute("name", "SpringBoot中文社區");
return new ModelAndView("test/test");
}
}
視圖（HTML）中加載的JS客戶端
const {protocol, host} = window.location;
const websocket = new WebSocket(`${protocol === 'https:' ? 'wss:': 'ws:'}//${host}/channel/test`);
websocket.onmessage = e => {
const message = JSON.parse(e.data);
console.log('收到消息:', message);
}
websocket.onclose = e => {
let {code, reason} = e;
console.log(`鏈接斷開:code=${code}, reason=${reason}`);
}
websocket.onopen = () => {
console.log(`鏈接建立...`);
}
websocket.onerror = e => {
console.log('鏈接異常:', e);
}
控制台日志
可以從日志中看到，在Controller創建了HttpSession，並且在Websocket鏈接握手中獲取到了HttpSession，並且成功讀取到了存儲的數據。
2020-07-03 14:09:20.322 DEBUG 2004 --- [ XNIO-1 task-1] o.s.web.servlet.DispatcherServlet : GET "/test", parameters={}
2020-07-03 14:09:20.327 DEBUG 2004 --- [ XNIO-1 task-1] s.w.s.m.m.a.RequestMappingHandlerMapping : Mapped to io.springboot.twitter.web.controller.TestController#test(HttpServletRequest)
2020-07-03 14:09:20.348 DEBUG 2004 --- [ XNIO-1 task-1] io.undertow.session : Created session with id I2d7ub_oKNcOB7QSjl1xQ59_uBkgjvWhiTX9BmXk for exchange HttpServerExchange{ GET /test}
2020-07-03 14:09:20.349 INFO 2004 --- [ XNIO-1 task-1] i.s.t.web.controller.TestController : 創建新的httpSession=I2d7ub_oKNcOB7QSjl1xQ59_uBkgjvWhiTX9BmXk
2020-07-03 14:09:20.431 DEBUG 2004 --- [ XNIO-1 task-1] o.s.w.s.v.ContentNegotiatingViewResolver : Selected 'text/html' given [text/html, application/xhtml+xml, image/webp, image/apng, application/xml;q=0.9, application/signed-exchange;v=b3;q=0.9, */*;q=0.8]
2020-07-03 14:09:20.432 DEBUG 2004 --- [ XNIO-1 task-1] o.s.w.s.view.freemarker.FreeMarkerView : View name 'test/test', model {}
2020-07-03 14:09:20.433 DEBUG 2004 --- [ XNIO-1 task-1] o.s.w.s.view.freemarker.FreeMarkerView : Rendering [test/test.ftl]
2020-07-03 14:09:20.470 DEBUG 2004 --- [ XNIO-1 task-1] o.s.web.servlet.DispatcherServlet : Completed 200 OK
2020-07-03 14:09:20.661 DEBUG 2004 --- [ XNIO-1 task-1] io.undertow.request.security : Attempting to authenticate /static/js/test.js, authentication required: false
2020-07-03 14:09:20.661 DEBUG 2004 --- [ XNIO-1 task-1] io.undertow.request.security : Authentication outcome was NOT_ATTEMPTED with method io.undertow.security.impl.CachedAuthenticatedSessionMechanism@51c2d817 for /static/js/test.js
2020-07-03 14:09:20.661 DEBUG 2004 --- [ XNIO-1 task-1] io.undertow.request.security : Authentication result was ATTEMPTED for /static/js/test.js
2020-07-03 14:09:20.662 DEBUG 2004 --- [ XNIO-1 task-1] o.s.web.servlet.DispatcherServlet : GET "/static/js/test.js", parameters={}
2020-07-03 14:09:20.675 DEBUG 2004 --- [ XNIO-1 task-1] o.s.w.s.handler.SimpleUrlHandlerMapping : Mapped to ResourceHttpRequestHandler ["classpath:/static/", "/"]
2020-07-03 14:09:20.694 DEBUG 2004 --- [ XNIO-1 task-1] o.s.web.servlet.DispatcherServlet : Completed 200 OK
2020-07-03 14:09:20.701 DEBUG 2004 --- [ XNIO-1 task-1] io.undertow.request.security : Attempting to authenticate /channel/test, authentication required: false
2020-07-03 14:09:20.701 DEBUG 2004 --- [ XNIO-1 task-1] io.undertow.request.security : Authentication outcome was NOT_ATTEMPTED with method io.undertow.security.impl.CachedAuthenticatedSessionMechanism@51c2d817 for /channel/test
2020-07-03 14:09:20.701 DEBUG 2004 --- [ XNIO-1 task-1] io.undertow.request.security : Authentication result was ATTEMPTED for /channel/test
2020-07-03 14:09:20.707 DEBUG 2004 --- [ XNIO-1 task-1] io.undertow.request : Upgrading request HttpServerExchange{ GET /channel/test}
2020-07-03 14:09:20.712 INFO 2004 --- [ XNIO-1 task-1] i.s.t.w.ServerEndpointConfigurator : session=io.undertow.servlet.spec.HttpSessionImpl@570c2102
2020-07-03 14:09:20.712 INFO 2004 --- [ XNIO-1 task-1] i.s.t.w.ServerEndpointConfigurator : sessionId=I2d7ub_oKNcOB7QSjl1xQ59_uBkgjvWhiTX9BmXk
2020-07-03 14:09:20.712 INFO 2004 --- [ XNIO-1 task-1] i.s.t.w.ServerEndpointConfigurator : name=SpringBoot中文社區
2020-07-03 14:10:20.744 DEBUG 2004 --- [ XNIO-1 I/O-1] io.undertow.request : Timing out idle connection from /0:0:0:0:0:0:0:1:14058