1. 程式人生 > >WebSocket之獲取HttpSession

WebSocket之獲取HttpSession

由於WebSocket與Http協議的不同,故在使用常用的HttpSession方面就存在了一些問題。通過google翻閱到了在onOpen方法下使用HttpSession的方法。

新建一個GetHttpSessionConfigurator類並繼承Configurator類

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;
/*
 * 獲取HttpSession
 * 
 */

public class GetHttpSessionConfigurator extends Configurator {

    @Override
    public void modifyHandshake(ServerEndpointConfig sec,
            HandshakeRequest request, HandshakeResponse response) {
        // TODO Auto-generated method stub
        HttpSession httpSession=(HttpSession) request.getHttpSession();
        sec.getUserProperties().put(HttpSession.class.getName(),httpSession);
    }
    
}

然後在@ServerEndpoint註解裡面新增configurator屬性

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

在onOpen方法里加入引數EndpointConfig config即可獲取HttpSession

 @OnOpen
    public void onOpen(Session session,EndpointConfig config) {
        HttpSession httpSession= (HttpSession) config.getUserProperties().get(HttpSession.class.getName());
        System.out.println( httpSession.getAttribute("name"));
        sessionMap.put(session.getId(), session);
    }