SpringBoot+WebSocket實現簡單的數據推送

問題背景

為什麼要要用websocket呢？websocket相對於傳統http協議有什麼優勢呢？

• http協議有一個缺陷，就是通信只能由客戶端發起，伺服器返回數據，不能做到伺服器主動向客戶端推送。這就導致了一個弊端，假如我要獲得伺服器連續的狀態變化，那麼我只能每隔一段時間向伺服器發起請求，也就是所謂的輪詢，這就導致了大量的資源浪費。由此，websocket應運而生，來解決這一痛點。
• websocket使用tcp連接保持全雙工通信協議，伺服器端比較容易實現；
• 數據格式比較輕量，通信比較高效；
• 既能發送文本，也能發送二進位數據，沒有同源限制；
• wbsocket響應報文與http不同，如下

websocket響應報文

好了，簡單的了解一下，我們來看看springboot+websocket實現推送的過程，以下是個簡單的demo。

搭建過程

• 首先我們初始化一個spring boot項目，大家可以根據自己的習慣去官網下載或者用idea來生成，這裡我不多說，主要的是引入websocket依賴，如下圖；

websocket依賴

• 編寫websocket配置類，這裡直接new 一個ServerEndpointExporter就行了

@Componentpublic class WebSocketConfig { @Bean public ServerEndpointExporter serverEndpointExporter() { return new ServerEndpointExporter(); }}

• 編寫websocket推送服務端

@Component@ServerEndpoint(value = &34;)public class LoggingWSServer { private static final Logger LOGGER = LoggerFactory.getLogger(LoggingWSServer.class); private static Map<String, Session> sessionMap = new ConcurrentHashMap<>(); private static Gson gson = new Gson(); private static Map<String,Object> map = new ConcurrentHashMap<>(); @OnOpen public void onOpen(Session session) { new Thread(() -> { // 這裡大家可以根據業務來優化，使用線程池等手段 while(sessionMap.get(session.getId()) != null) { try { List<AaaServer> list = aaaServerService.findList(); if(session.isOpen()) { send(session, gson.toJson(list)); } Thread.sleep(5000); } catch (InterruptedException e) { e.printStackTrace(); } } }).start(); } @OnClose public void onClose(Session session) { sessionMap.remove(session.getId()); LOGGER.info(&34; , session.getId()); } @OnMessage public void onMessage(String message,Session session) { LOGGER.info(&34;,message); map.put(&34;,message); send(session,gson.toJson(map)); } private void send(Session session, String toJson) { try { session.getBasicRemote().sendText(toJson); } catch (IOException e) { LOGGER.error(&34;,e.getMessage()); } } }

服務端api解釋：

• OnOpen:在這個端點連接時被調用；
• OnClose:與OnOpen相反，連接斷開時被調用
• OnMessage：服務端收到客戶端的信息時被調用
• OnError：當然還有這個，上面沒有列出，這是發送錯誤時被調用

• 編寫前端頁面，這裡只展示js

//websocket對象 var websocket = null; //判斷當前瀏覽器是否支持WebSocket if (&39; in window) { //動態獲取域名或ip var hostname = window.location.hostname; port = window.location.port; websocket = new WebSocket(&34;+hostname+&34; + port + &34;); } else { console.error(&34;); } //連接發生錯誤的回調方法 websocket.onerror = function (e) { console.error(&34; + e); }; //連接成功建立的回調方法 websocket.onopen = function () { console.log(&34;) }; //接收到消息的回調方法 websocket.onmessage = function (event) { console.log(&34;, event.data ); var data = JSON.parse(event.data); if (data.push) { // 內容 var temp = template(&39;,{data : data.push}); console.log(&34; + temp); document.getElementById(&34;).innerHTML = temp; } if(data.ret) { console.log(data.ret) alert(data.ret); } } //發送消息 $(&send&39;39;).val()); });

前端頁面注意：

• 編寫的時候一定要注意一個地方，就是new WebSocket 的地址，一定要和服務端推送的那個地址相同，不然的話，這裡接收不到推送過來的數據。
• 另外，有些瀏覽器可能不支持websocket，也沒關係，可以使用sockjs或者stomp.js，這是基於websocket的上層協議，大家可以自行去了解以下。不過要注意，sockjs 處理的url是http或者https，不在是以ws開頭的。

結語

今天就簡單的介紹到這裡，有需要這個demo的，可以關注一下小編，後續小編會把代碼上傳到gitee，https://gitee.com/bigqianqian/springboot-websocket，可以了解下。下篇文章小編將會介紹websocket點對點推送和廣播，喜歡的朋友點個關注唄[奮鬥][奮鬥][奮鬥]，也可以動動手評論下，留下你的理解，小編與你共同成長！