Springboot-WebSocket获取HttpSession问题

换了新工作,第一个任务就是和这个有关，以前没接触过，没办法，各种度娘、谷哥，大部分都是只言片语，要么就是特定的配置环境还不贴配置……，踩坑无数，遂整理成笔记

WebSocket协议

WebSocket是一种在单个TCP连接上进行全双工通讯的协议。WebSocket通信协议于2011年被IETF定为标准RFC 6455，并由RFC7936补充规范。WebSocket API也被W3C定为标准。

WebSocket使得客户端和服务器之间的数据交换变得更加简单，允许服务端主动向客户端推送数据。在WebSocket API中，浏览器和服务器只需要完成一次握手，两者之间就直接可以创建持久性的连接，并进行双向数据传输

STOMP协议

STOMP是面向文本的消息传送协议。STOMP客户端与支持STOMP协议的消息代理进行通信。STOMP使用不同的命令，如连接，发送，订阅，断开等进行通信。

具体参考：官方介绍

SockJS

SockJS是一个JavaScript库，提供跨浏览器JavaScript的API，创建了一个低延迟、全双工的浏览器和web服务器之间通信通道

以上内容出自维基百科和百度百科

环境配置

SpringBoot2.0全家桶
特别说明一下，现在很多服务器都支持websocket,我这次写代码用的是SpringBoot内置的tomcat

应用场景

如果只是单纯的配置一个建立连接发送消息，并不难,具体客户端编写参考这个菜鸟教程-websocket
我需要解决的问题是：当客户端和服务器建立连接时，将登录的用户信息从httpSession中取出,这个问题的难点在于，websocket的请求和http请求完全不相关，所以没有办法直接获取HttpSession

实施思路

服务器端的编码实现，具体有两套方案
1. 使用继承的方式，这种方式，我在看Spring官方介绍文档时貌似也是这种方式实现的
2. 使用注解的方式：很蛋疼的是我们的项目使用的是这种方式
具体思路
1. 虽然websocket的请求和http请求完全不相关,但是如果基于注解的话，EndPoint支持读取一个配置
2. 我当时的想法就是在配置中拦截或者获取HttpSession,事实证明思路是正确的，但是当我按照这个思路去百度谷歌发现获取的一律都是null，然后我就崩溃了

具体编码实现

只贴核心代码,完整项目在本文的底部,我放在了github上

先上核心的服务器端消息处理类

/**

 * WebSocket主要的消息类

 * @author 侯叶飞

 */

 //onfigurator = WebsocketConfig.class 该属性就是我上面提到我们可以自己配置的东西

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

@Component

@Slf4j

public class WebSocket {

	/*每个浏览器连接都会有一个新的会话对象*/

	private Session session;

	/*用来存储每个会话的session,静态的不会被实例化*/

	private static CopyOnWriteArraySet<WebSocket> webSocketSets = new CopyOnWriteArraySet<>();

	/**

	 * 主要用来监听连接建立，config用来获取WebsocketConfig中的配置信息

	 * @param session

	 * @param config

	 */

	@OnOpen

	public void onOpen(Session session, EndpointConfig config) {

		log.info("config:{}", config.getUserProperties().get("name"));

		log.info("session:{}", config.getUserProperties().get("sessionid"));

		this.session = session;

		webSocketSets.add(this);

		log.info("【websocket消息】有新的连接, 总数:{}", webSocketSets.size());

	}

	@OnClose

	public void onClose() {

		webSocketSets.remove(this);

		log.info("【websocket消息】连接断开, 总数:{}", webSocketSets.size());

	}

	@OnError

	public void onError(Throwable e, Session session) {

		webSocketSets.remove(this);

		log.info("【websocket消息】连接出错或未关闭socket：" + e.getMessage());

	}

	@OnMessage

	public void onMessage(String message, Session session) {

		for(WebSocket ws:webSocketSets){

			ws.session.getAsyncRemote().sendText("广播："+message);

		}

		log.info("【websocket消息】收到客户端发来的消息:{}", message);

	}

}

下面代码就是核心的配置类

/**

 * 主要的配置类

 *  本类必须要继承Configurator,因为@ServerEndpoint注解中的config属性只接收这个类型

 * @author 侯叶飞

 *

 */

@Configuration

@Slf4j

public class WebsocketConfig extends ServerEndpointConfig.Configurator {

	private static final String HttpSession = null;

	/* 修改握手,就是在握手协议建立之前修改其中携带的内容 */

	@Override

	public void modifyHandshake(ServerEndpointConfig sec, HandshakeRequest request, HandshakeResponse response) {

		/*如果没有监听器,那么这里获取到的HttpSession是null*/

		StandardSessionFacade ssf = (StandardSessionFacade) request.getHttpSession();

		if (ssf != null) {

			HttpSession session = (HttpSession) request.getHttpSession();

			sec.getUserProperties().put("sessionid", session);

			log.info("获取到的SessionID：{}",session.getId());

		}

		sec.getUserProperties().put("name", "小强");

		super.modifyHandshake(sec, request, response);

	}

	@Bean

	public ServerEndpointExporter serverEndpointExporter() {

	    //这个对象说一下，貌似只有服务器是tomcat的时候才需要配置,具体我没有研究

		return new ServerEndpointExporter();

	}

}

仅仅有上面的配置获取的肯定是null,至于原因，网上说法不一，我也不确定，解决方案如下

/**

 * 监听器类:主要任务是用ServletRequest将我们的HttpSession携带过去

 * @author 侯叶飞

 */

@Component //此注解千万千万不要忘记，它的主要作用就是将这个监听器纳入到Spring容器中进行管理,相当于注册监听吧

@Slf4j

public class RequestListener implements ServletRequestListener {

    @Override

    public void requestInitialized(ServletRequestEvent sre)  {

    	//将所有request请求都携带上httpSession

        HttpSession session = ((HttpServletRequest) sre.getServletRequest()).getSession();

        log.info("将所有request请求都携带上httpSession {}",session.getId());

    }

    public RequestListener() {}  

    @Override

    public void requestDestroyed(ServletRequestEvent arg0)  {}

}

此外在我谷歌的过程中有博客提到需要添加一个WebListener的注解，结果发现基于springboot的话，如果不加也不影响

GitHub项目地址：Demo

以上代码纯属个人研究，如果有错误的地方，各位留言或者发邮件都可以！

巴特西