WebSocket的集成 === > JEECG BOOT 增加websocket 旨在服務端主動向客戶端推送數據，實現系統向在線用戶推送消息，可群發，可對指定用戶發送 #### jeecg boot 集成 websocket 步驟 ##### （1）maven依賴 ``` <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> </dependency> ``` #### （2）WebSocket配置類 ``` package org.jeecg.config; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.web.socket.server.standard.ServerEndpointExporter; @Configuration public class WebSocketConfig { /** * 注入ServerEndpointExporter， * 這個bean會自動注冊使用了@ServerEndpoint注解聲明的Websocket endpoint */ @Bean public ServerEndpointExporter serverEndpointExporter() { return new ServerEndpointExporter(); } } ``` #### (3) WebSocket操作類 > 通過該類WebSocket可以進行群推送以及單點推送 ``` package org.jeecg.modules.message.websocket; import java.util.HashMap; import java.util.Map; import java.util.concurrent.CopyOnWriteArraySet; import javax.websocket.OnClose; import javax.websocket.OnMessage; import javax.websocket.OnOpen; import javax.websocket.Session; import javax.websocket.server.PathParam; import javax.websocket.server.ServerEndpoint; import org.springframework.stereotype.Component; import lombok.extern.slf4j.Slf4j; @Component @Slf4j @ServerEndpoint("/websocket/{userId}") public class WebSocket { private Session session; private static CopyOnWriteArraySet<WebSocket> webSockets =new CopyOnWriteArraySet<>(); private static Map<String,Session> sessionPool = new HashMap<String,Session>(); @OnOpen public void onOpen(Session session, @PathParam(value="userId")String userId) { try { this.session = session; webSockets.add(this); sessionPool.put(userId, session); log.info("【websocket消息】有新的連接，總數為:"+webSockets.size()); } catch (Exception e) { } } @OnClose public void onClose() { try { webSockets.remove(this); log.info("【websocket消息】連接斷開，總數為:"+webSockets.size()); } catch (Exception e) { } } @OnMessage public void onMessage(String message) { log.info("【websocket消息】收到客戶端消息:"+message); } // 此為廣播消息 public void sendAllMessage(String message) { log.info("【websocket消息】廣播消息:"+message); for(WebSocket webSocket : webSockets) { try { if(webSocket.session.isOpen()) { webSocket.session.getAsyncRemote().sendText(message); } } catch (Exception e) { e.printStackTrace(); } } } // 此為單點消息 public void sendOneMessage(String userId, String message) { Session session = sessionPool.get(userId); if (session != null&&session.isOpen()) { try { log.info("【websocket消息】 單點消息:"+message); session.getAsyncRemote().sendText(message); } catch (Exception e) { e.printStackTrace(); } } } // 此為單點消息(多人) public void sendMoreMessage(String[] userIds, String message) { for(String userId:userIds) { Session session = sessionPool.get(userId); if (session != null&&session.isOpen()) { try { log.info("【websocket消息】 單點消息:"+message); session.getAsyncRemote().sendText(message); } catch (Exception e) { e.printStackTrace(); } } } } } ``` ### 前端中VUE使用WebSocket ``` <script> import store from '@/store/' export default { data() { return { } }, mounted() { //初始化websocket this.initWebSocket() }, destroyed: function () { // 離開頁面生命周期函數 this.websocketclose(); }, methods: { initWebSocket: function () { // WebSocket與普通的請求所用協議有所不同，ws等同于http，wss等同于https var userId = store.getters.userInfo.id; var url = window._CONFIG['domianURL'].replace("https://","ws://").replace("http://","ws://")+"/websocket/"+userId; this.websock = new WebSocket(url); this.websock.onopen = this.websocketonopen; this.websock.onerror = this.websocketonerror; this.websock.onmessage = this.websocketonmessage; this.websock.onclose = this.websocketclose; }, websocketonopen: function () { console.log("WebSocket連接成功"); }, websocketonerror: function (e) { console.log("WebSocket連接發生錯誤"); }, websocketonmessage: function (e) { var data = eval("(" + e.data + ")"); //處理訂閱信息 if(data.cmd == "topic"){ //TODO 系統通知 }else if(data.cmd == "user"){ //TODO 用戶消息 } }, websocketclose: function (e) { console.log("connection closed (" + e.code + ")"); } } } </script> ```