Java WebSocket基于注解配置访问ServletAPI

来源:互联网 发布:法兰绒衬衫怎么洗 知乎 编辑:程序博客网 时间:2024/06/06 00:52

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实现音乐推送
小鱼音乐