java使用Websocket獲取HttpSession出現的問題與解決

websocket的寫法就不多說了，主要記一記其中出現的問題

1.獲取不到httpSession

解決辦法：先重寫握手方法，將httpsession放入ServerEndpointConfig.getUserProperties()方法返回的map中

import javax.servlet.http.HttpSession;
import javax.websocket.HandshakeResponse;
import javax.websocket.server.HandshakeRequest;
import javax.websocket.server.ServerEndpointConfig;
import javax.websocket.server.ServerEndpointConfig.Configurator;

/**
 * 繼承websocket配置類，將httpsession放入ServerEndpointConfig的map中
 * 從而達到使websocket對象可以訪問到httpsession中的對象
 */
public class GetHttpSessionConfigurator extends Configurator{

    /**
     * 重寫修改握手方法
     * @param sec
     * @param request
     * @param response
     */
    @Override
    public void modifyHandshake(ServerEndpointConfig sec, HandshakeRequest request, HandshakeResponse response) {
        HttpSession httpSession = (HttpSession)request.getHttpSession();
        sec.getUserProperties().put(HttpSession.class.getName(),httpSession);
    }
}

然后在注解@ServerEndpoint加入configurator的配置

 @ServerEndpoint(value="/websocket",configurator=GetHttpSessionConfigurator.class) 

此時已經可以獲取到httpsession了。

然而如果我沒有登錄，反而需要臨時登錄這種情況呢？根據上述demo，會發現根本沒辦法連接上，一直是close狀態

下面是對這個問題的解決

2、未登錄用戶連接WebSocket，一直close狀態的情況解決：

解決方法是建立個請求監聽器

@WebListener
public class RequestListener implements ServletRequestListener {
    
    public void requestInitialized(ServletRequestEvent sre)  { 
        //將所有request請求都攜帶上httpSession
        ((HttpServletRequest) sre.getServletRequest()).getSession();
        
    }
    public RequestListener() {
        // TODO Auto-generated constructor stub
    }

    public void requestDestroyed(ServletRequestEvent arg0)  { 
         // TODO Auto-generated method stub
    }
}

問題解決。