springboot中使用websocket簡單例子

1、springboot中使用 websocket簡單例子gradle中添加依賴，引入websocket支持compile("org.springframework.boot:spring-boot-starter-websocket:$springBootVersion")啟用 websocketpackage cn.xiaojf.today.ws.configuration;importcn.xiaojf.today.ws.handler.CountWebSocketHandler;erceptor.Handshake

2、Interceptor;import org.springframework.context.ApplicationContext;import org.springframework.context.annotation.Bean;importorg.springframework.context.annotation.Configuration;importorg.springframework.web.socket.config.annotation.EnableWebSocket;importorg.springframework.web.socket.config.annotatio

3、n.WebSocketConfigurer;importorg.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;importorg.springframework.web.socket.server.standard.ServerEndpointExporter;/* websocket 配置* author xiaojf 2017/3/2 9:50.*/ConfigurationEnableWebSocketpublic class WebSocketConfig implements WebSock

4、etConfigurer Beanpublic ServerEndpointExporterserverEndpointExporter(ApplicationContext context) return new ServerEndpointExporter();Overridepublic voidregisterWebSocketHandlers(WebSocketHandlerRegistryregistry) registry.addHandler(newCountWebSocketHandler(),"/web/count").addInterceptors(n

5、ewHandshakeInterceptor();消息攔截處理package erceptor;importorg.springframework.http.server.ServerHttpRequest;importorg.springframework.http.server.ServerHttpResponse;importorg.springframework.web.socket.WebSocketHandler;importorg.springframework.web.socket.server.support.HttpSession

6、HandshakeInterceptor;import java.util.Map;/* 消息攔截處理類* author xiaojf 2017/3/2 10:36.*/public class HandshakeInterceptor extends HttpSessionHandshakeInterceptor Overridepublic boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,Map<String,O

7、bject> attributes) throws Exception /解決 The extension x-webkit-deflate-frame is not supported 問題if(request.getHeaders().containsKey("Sec-WebSocket-Exte nsions") request.getHeaders().set("Sec-WebSocket-Extensions", "permessage-deflate");System.out.println("Be

8、fore Handshake");return super.beforeHandshake(request, response, wsHandler, attributes);Overridepublic void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,Exception ex) System.out.println("After Handshake");super.afterHandshake(reques

9、t, response,wsHandler, ex);消息處理相應(yīng)類package cn.xiaojf.today.ws.handler;import com.alibaba.druid.support.json.JSONUtils;import mons.collections.map.HashedMap;import org.springframework.stereotype.Component;import org.springframework.web.socket.CloseStatus;import org.springframework.web.socket.TextMessa

10、ge;importorg.springframework.web.socket.WebSocketSession;importorg.springframework.web.socket.handler.TextWebSocketHandler;import java.io.IOException;import java.util.*;/* author xiaojf 2017/3/2 9:55.*/Componentpublic class CountWebSocketHandler extends TextWebSocketHandler private static long count

11、 = 0;private static Map<String,WebSocketSession> sessionMap = new HashedMap();Overrideprotected voidhandleTextMessage(WebSocketSession session, TextMessage message) throws Exception /session.sendMessage(newTextMessage(session.getPrincipal().getName()+",你是第 "+ (sessionMap.size

12、() + "位訪客 "); /p2pObject parse =JSONUtils.parse(message.getPayload();Collection<WebSocketSession> sessions = sessionMap.values();for (WebSocketSession ws : sessions) / 廣播 ws.sendMessage(message);sendMessage(sessionMap.keySet(),"你好 ");Overridepublic voidafterConnecti

13、onEstablished(WebSocketSession session)throws Exception sessionMap.put(session.getPrincipal().getName(),session);super.afterConnectionEstablished(session);Overridepublic voidafterConnectionClosed(WebSocketSession session,CloseStatus status) throws Exception sessionMap.remove(session.getPrincipal().g

14、etName(); super.afterConnectionClosed(session, status);/* 發(fā)送消息* author xiaojf 2017/3/2 11:43*/public static void sendMessage(String username,String message) throws IOException sendMessage(Arrays.asList(username),Arrays.asList(message);/* 發(fā)送消息* author xiaojf 2017/3/2 11:43*/public static voidsendMess

15、age(Collection<String> acceptorList,String message) throws IOException sendMessage(acceptorList,Arrays.asList(message);/* 發(fā)送消息， p2p 群發(fā)都支持* author xiaojf 2017/3/2 11:43*/public static voidsendMessage(Collection<String> acceptorList,Collection<String> msgList)

16、 throws IOException if (acceptorList != null && msgList !=null) for (String acceptor : acceptorList) WebSocketSession session =sessionMap.get(acceptor);if (session != null) for (String msg : msgList) session.sendMessage(newTextMessage(msg.getBytes();前端 js ，需要依賴jqueryrequire("jquery","formUtil", function ($) var url = 'ws:/'+window.location.host+'/web/count' var ws = new WebSocket(url); ws.onopen = function(e)/ ws.send('hello');ws.onmessage = function(e) alert(e.data);ws.onerror = function(e) aler