SpringBoot WebSocket 消息交互

import lombok.extern.slf4j.Slf4j;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.support.HttpSessionHandshakeInterceptor;import javax.servlet.http.HttpSession;import java.util.Map;@Slf4jpublic class WebSocketHandlerIntereptor extends HttpSessionHandshakeInterceptor {    @Override    public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,                                   Map<String, Object> attributes) throws Exception {        log.info("Before Handshake");        if (request instanceof ServletServerHttpRequest) {            ServletServerHttpRequest servletRequest = (ServletServerHttpRequest) request;            HttpSession session = servletRequest.getServletRequest().getSession();            if (session != null) {                //使用 idcard 区分 WebSocketHandler，以便定向发送消息【一般直接保存 user 实体】                String idcard = (String) session.getAttribute("idcard");                if (idcard != null) {                    attributes.put("WEBSOCKET_IDCARD", idcard);                }            }        }        return super.beforeHandshake(request, response, wsHandler, attributes);    }    @Override    public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,                               Exception ex) {        super.afterHandshake(request, response, wsHandler, ex);    }}

3.2 WebSocket前端 

WebSocket前端

4. 总结