Java WebSocket基于注解配置通路ServletAPI

配置類

package com.xyes.music.websocket;

import com.sun.xml.internal.bind.v2.runtime.reflect.opt.Const;
import com.xyes.music.context.Consts;

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;

/**
 1. Created by 郭文梁 on 2017/10/17.
 2. 擷取httpSession到WebSocket
 */
public class GetHttpSessionConfigure extends Configurator {
    @Override
    public void modifyHandshake(ServerEndpointConfig sec, HandshakeRequest request, HandshakeResponse response) {
        super.modifyHandshake(sec, request, response);
        HttpSession session = (HttpSession) request.getHttpSession();
        sec.getUserProperties().put(Consts.USER_IN_PROPERTIES_KEY, session);
    }
}
           

1.配置類中，在用戶端發起連接配接時向ServerEndpointConfig的UserProperties中添加要擷取的資源

WebSocket類

@OnOpen
    public void onOpen(Session session, EndpointConfig config) {
        session = session;
        initSession(config);
        SESSIONS.add(session);
    }

    private void initSession(EndpointConfig config) {
        httpSession = (HttpSession) config.getUserProperties().get(Consts.USER_IN_PROPERTIES_KEY);
        user = (User) httpSession.getAttribute(Consts.USER_IN_SESSION_KEY);
    }
           

在WebSocket中就可以從UserProperties中擷取對象

*另外推薦本人網站 MQTT實作音樂推送

小魚音樂