在SSM项目中配置websocket

最近在ssm项目中配置了websocket，踩了很多坑，来分享一下

本文暂不提供发送消息等内容的代码逻辑（后续也许会补充），如果你直接复制这类可能会对配置造成更大的麻烦（博主就是复制了他人逻辑导致连接后秒断开连接，排查了很长时间），建议看完本文配置成功后，去其他文章或者github寻找相应代码逻辑

1.Maven依赖

首先是jackson的依赖

		com.fasterxml.jackson.corejackson-annotations2.3.0com.fasterxml.jackson.corejackson-core2.3.1com.fasterxml.jackson.corejackson-databind2.3.3

然后是spring关于websocket的依赖

		org.springframeworkspring-websocket${spring.version}org.springframeworkspring-messaging${spring.version}

这里的${spring.version}需要注意，一定要和自己的spring版本匹配，所以推荐用这种形式，而不是直接用x.x.x.RELEASE

另外在spring4.0.5.RELEASE之前与其以后，session获取参数的方法有些不同，需要注意

本文使用的是

4.2.5.RELEASE

注意：请删除所有关于javax的依赖，将本地的tomcat引入库中，否则可能会出现类型转换异常导致连接报错（你可以继续下面的步骤，如果在启动测试时出现该问题，再回来尝试解决）

2.配置类

首先在我们的项目中创建如下结构

当然也可以不使用这样的结构，如果你的项目中本来有config、Interceptor、handler包，可以将这些类放入其中

注意以下代码没有给出包名，如需复制请注意

首先是WebSocketConfig.java

import org.springframework.stereotype.Component;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;import javax.annotation.Resource;/*** WebScoket配置处理器*/
@Component
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {@ResourceMyWebSocketHandler handler;@ResourceHandShake handShake;public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {//浏览器支持websocketregistry.addHandler(handler, "/ws").addInterceptors(handShake);//浏览器不支持的话，需要socketjs引入支持registry.addHandler(handler, "/ws/sockjs").addInterceptors(handShake).withSockJS();}}

这个类的作用是配置websocket的链接地址，分为浏览器支持websocket和不支持websocket的情况，不支持可以引入sockjs（某些环境下使用这行代码会导致无法连接，不放心可以删去）

websocket连接地址是

ws://ip:端口号/项目名称/ws

ws与上述配置类相对应

这个连接地址是可以以?的形式传参数的，例如

"ws://ip:端口号/项目名称/ws?userId="+userId

第二个类是HandShake.java

import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.http.server.ServletServerHttpRequest;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.server.HandshakeInterceptor;import javax.servlet.http.HttpSession;
import java.util.Map;/*** Socket建立连接（握手）和断开*/
@Component
public class HandShake implements HandshakeInterceptor {@Overridepublic boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map attributes) throws Exception {// 1.获取sessionServletServerHttpRequest servletRequest = (ServletServerHttpRequest) request;HttpSession session = servletRequest.getServletRequest().getSession(false);// 2.从session中获取uidLong uid = (Long) session.getAttribute("uid");// 3.做正向判断，如果uid是空值直接返回false，拒绝握手if (uid == null){return false;}// 4.输出日志System.out.println("Websocket:用户[ID:" + uid + "]准备进行握手");// 5.将用户id存入mapattributes.put("uid", uid);// 6.可以进行握手return true;}@Overridepublic void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Exception exception) {System.out.println("after hand");}}

这个类是websocket的握手拦截器，会拦截websocket链接地址的请求，进行握手，不建议在该类中做业务处理

最后是MyWebSocketHandler.java

import org.springframework.stereotype.Component;
import org.springframework.web.socket.*;
import java.io.IOException;/*** Socket处理器*/
@Component
public class MyWebSocketHandler implements WebSocketHandler {@Overridepublic void afterConnectionEstablished(WebSocketSession session)throws Exception {System.out.println("ConnectionEstablished");}@Overridepublic void handleMessage(WebSocketSession session, WebSocketMessage message) throws Exception {}/*** 消息传输错误处理*/@Overridepublic void handleTransportError(WebSocketSession session,Throwable exception) throws Exception {System.out.println("TransportError");}/*** 关闭连接后*/@Overridepublic void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {System.out.println("ConnectionClosed");}@Overridepublic boolean supportsPartialMessages() {return false;}
}

这个类是对websocket的各种状态作出相应的处理，如开头所说，本文不会给出相应的逻辑，防止出现错误，这里真的很坑

大体思路就是通过方法中的WebSocketSession获取到session中的用户id，再注入自己的userService（或者其他service层）去完成业务，同时可以在本类中使用ConcurrentHashMap或者直接开一个线程来实现同时在线人数等功能

发送信息的逻辑推荐对信息进行封装，构建实体类

此外你还可以创建一个相应的Controller，通过注入MyWebSocketHandler来建立相应功能的请求地址

import com.webchat.websocket.MyWebSocketHandler;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;@RestController
@RequestMapping("/websocket")
public class WebSocketController {@AutowiredMyWebSocketHandler handler;}

3.JS部分

最后是js部分

		let websocket = new WebSocket("ws://ip:端口号/项目名称/ws?userId="+userId);websocket.onmessage = function(event) {console.log(event.data)};websocket.onopen = function (event){console.log("连接成功")}websocket.onerror = function (event){console.log("连接错误")}websocket.onclose = function (event){console.log('websocket 断开: ' + event.code + ' ' + event.reason + ' ' + event.wasClean)}

注意更改地址

4.测试结果

测试结果如下


至此，ssm项目中配置websocket就完成了

如果在测试中出现其他错误，可以根据输出的event.code来匹配错误原因，如果连接后直接断开并显示1011请检查后端代码规范性