---

代码结构

步骤

1.建立handler

代码如下：

package xxxx;import net.sf.json.JSONObject;
import org.jeecgframework.p3.core.logger.Logger;
import org.jeecgframework.p3.core.logger.LoggerFactory;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.AbstractWebSocketHandler;import java.io.IOException;
import java.util.*;@Component
public class WebSocketHandler extends AbstractWebSocketHandler {private static final ArrayList<WebSocketSession> users;private static final HashMap<String, List<WebSocketSession>> sessionlist;private Logger logger = LoggerFactory.getLogger(WebSocketHandler.class);static {users = new ArrayList<WebSocketSession>();sessionlist=new HashMap<String, List<WebSocketSession>>();}//客户端连接时添加session@Overridepublic void afterConnectionEstablished(WebSocketSession session) throws Exception {users.add(session);}//客户端连接时移除session,因为做了群发处理，这里通过for循环把list集合的断开连接session移除@Overridepublic void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {Iterator<Map.Entry<String, List<WebSocketSession>>> entries = sessionlist.entrySet().iterator(); while (entries.hasNext()) { Map.Entry<String, List<WebSocketSession>> entrylist = entries.next(); List<WebSocketSession> entrylistvalue=entrylist.getValue();String key=entrylist.getKey();if(entrylistvalue!=null){for(int i=0;i<entrylistvalue.size();i++){if(session.getId().equals(entrylistvalue.get(i).getId())){entrylistvalue.remove(i);sessionlist.put(key, entrylistvalue);} }}}users.remove(session);}//连接后接收客户端的数据并处理@Overrideprotected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {//获取客户端的string类型的数据String joStr=message.getPayload();//把String类型数据转为json对象JSONObject jsonObject = JSONObject.fromObject(joStr);//通过getString("")分别取出里面的信息String socketid = jsonObject.getString("socketid");List<WebSocketSession> sessionlists=sessionlist.get(socketid);//这里根据json数据的socketid存放sessionif(sessionlists==null) {super.handleTextMessage(session, message);List<WebSocketSession> list=new ArrayList<WebSocketSession>();list.add(session);sessionlist.put(socketid,list);}else{boolean flag=false;for(WebSocketSession sessions:sessionlists){if(sessions.getId().equals(session.getId())){flag=true;}}if(!flag){sessionlists.add(session);sessionlist.put(socketid,sessionlists);}}//返回自定义数据给指定session客户端TextMessage messages=new TextMessage("1001");sendMessgaeToUsers(messages,session);//提示，如果页面有多个页面需要用到websocket并且业务逻辑不一样。处理数据可根据客户端发送的json的socketid不同来进行不同的数据处理，建议分别编写业务逻辑方法，代码会看起来简洁很多}//群发消息public void sendMessgaeToUsers(TextMessage message, String socketid) {List<WebSocketSession> sessionlists=sessionlist.get(socketid);if(sessionlists!=null){for(WebSocketSession session:sessionlists){if(session!=null) {if(session.isOpen()) {try {session.sendMessage(message);} catch (IOException e) {e.printStackTrace();}}}}}}//根据session单发消息public void sendMessgaeToUsers(TextMessage message, WebSocketSession session) {if(session!=null) {if(session.isOpen()) {try {session.sendMessage(message);} catch (IOException e) {e.printStackTrace();}}}}
}

2.建立Interceptor拦截器

代码如下：

package xxxxx;import org.jeecgframework.p3.core.logger.Logger;
import org.jeecgframework.p3.core.logger.LoggerFactory;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.server.HandshakeInterceptor;import java.util.Map;@Component
public class WebSocketHandshakeInterceptor implements HandshakeInterceptor {private static final Logger logger = LoggerFactory.getLogger(WebSocketHandshakeInterceptor.class);@Overridepublic boolean beforeHandshake(ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse, WebSocketHandler webSocketHandler, Map<String, Object> map) throws Exception {return true;}@Overridepublic void afterHandshake(ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse, WebSocketHandler webSocketHandler, Exception e) {}}

3.拦截器配置

代码如下：

package xxxxx;import org.jeecgframework.p3.core.logger.Logger;
import org.jeecgframework.p3.core.logger.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;
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 org.springframework.web.socket.server.standard.ServletServerContainerFactoryBean;
import com.xxxxx.util.socket.*;import javax.annotation.Resource;@Configuration
@EnableWebSocket
public class WebSocketConfig extends WebMvcConfigurerAdapter implements WebSocketConfigurer {private static final Logger logger = LoggerFactory.getLogger(WebSocketConfig.class);@Resourceprivate WebSocketHandler webSocketHandler;@Resourceprivate WebSocketHandshakeInterceptor webSocketHandshakeInterceptor;@Overridepublic void registerWebSocketHandlers(WebSocketHandlerRegistry webSocketHandlerRegistry) {webSocketHandlerRegistry.addHandler(webSocketHandler,"/webSocketHandler.ws").addInterceptors(webSocketHandshakeInterceptor);webSocketHandlerRegistry.addHandler(webSocketHandler,"/sockjs/webSocketHandler.ws").addInterceptors(webSocketHandshakeInterceptor).withSockJS();}@Beanpublic ServletServerContainerFactoryBean createWebSocketContainer() {ServletServerContainerFactoryBean container = new ServletServerContainerFactoryBean();container.setMaxTextMessageBufferSize(8192*4);container.setMaxBinaryMessageBufferSize(8192*4);return container;}
}

3.拦截器配置

代码如下：

		var socket; //websocket的实例var lockReconnect = false; //避免重复连接getwebsocket(); //加载WebSocketfunction getwebsocket() { //新建websocket的函数 页面初始化 断开连接时重新调用return false;var wsUrl = 'ws://127.0.0.1:8080/demo/webSocketHandler.ws';socket = new WebSocket(wsUrl);socket.onerror = function(event) {console.log('websocket服务出错了');reconnect(wsUrl);};socket.onclose = function(event) {console.log('websocket服务关闭了');reconnect(wsUrl);};socket.onopen = function(event) {console.log('open');socket.send("{\n" +"\"socketid\": \"Websocket\"}\n");heartCheck.reset().start(); //传递信息};socket.onmessage = function(event) {//如果获取到消息，心跳检测重置//拿到任何消息都说明当前连接是正常的//接受消息后的UI变化var data = JSON.parse(event.data);console.log('websocket服务获得数据了:' + data);if (data != "1001") {//这里定义了data='1001'时，为心跳返回数据，//所以非1001时为业务逻辑处理代码if (data == "101") {console.log("hello!wrold");}};heartCheck.reset().start();};}function reconnect(url) {if (lockReconnect) return;lockReconnect = true;//没连接上会一直重连，设置延迟避免请求过多setTimeout(function() {getwebsocket();lockReconnect = false;}, 2000);}//心跳检测var heartCheck = {timeout: 60000, //60秒timeoutObj: null,serverTimeoutObj: null,reset: function() {clearTimeout(this.timeoutObj);clearTimeout(this.serverTimeoutObj);return this;},start: function() {var self = this;this.timeoutObj = setTimeout(function() {//这里发送一个心跳，后端收到后，返回一个心跳消息，//onmessage拿到返回的心跳就说明连接正常socket.send("{\n" +"\"socketid\": \"Websocket\"}\n");self.serverTimeoutObj = setTimeout(function() { //如果超过一定时间还没重置，说明后端主动断开了socket.close(); //如果onclose会执行reconnect，我们执行ws.close()就行了.如果直接执行reconnect 会触发onclose导致重连两次}, self.timeout)}, this.timeout)}}

4.进行web.xml配置

代码如下：
下列代码表示遇到以.ws的请求时做转发

<servlet-mapping><servlet-name>springMvc</servlet-name><url-pattern>*.ws</url-pattern></servlet-mapping>

5.进行spring-mvc.xml配置

代码如下：
本质HttpRequestHandlerAdapter是HttpRequestHandler的适配器，最终调用HttpRequestHandler的handleRequest方法。
1.）添加适配器

<bean class="org.springframework.web.servlet.mvc.HttpRequestHandlerAdapter"/>

2.）对WebSocket的请求路径不做拦截。

<mvc:exclude-mapping path="/*.ws"/>

总结

以上配置完成，估计就可以接通websocket了