- 資訊首頁(yè) > 開(kāi)發(fā)技術(shù) >
- Springboot+WebSocket實(shí)現一對一聊天和公告的示例代碼
1.POM文件導入Springboot整合websocket的依賴(lài)
<dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> <version>2.1.6.RELEASE</version> </dependency>
2.注冊WebSocket的Bean交給Spring容器管理
@Configuration public class WebSocketServiceConfig { @Bean public ServerEndpointExporter serverEndpointExporter() { return new ServerEndpointExporter(); } }
3.WebSocket服務(wù)端實(shí)現
@ServerEndpoint 注解聲明為一個(gè)WebSocket服務(wù),訪(fǎng)問(wèn)地址為/chat/{username},@Component將其注冊為Spring的一個(gè)組件,交給Spring進(jìn)行管理
@ServerEndpoint("/chat/{username}") @Component @Slf4j public class WebSocket { //注入dao或者service,注意:因為dao層接口和service層都是單例的Bean //webSocket 不是單例的,所以在注入dao或者service時(shí),需要用set方法對其進(jìn)行注入,保證每一個(gè)都是獨立的 private static ChatMapper chatMapper; //參數中的ChatMapper 是 單例池中的ChatMapper @Autowired public void setChatMapper(ChatMapper chatMapperBean){ WebSocket.chatMapper = chatMapperBean; } //當前連接數 private static int onLinePersonNum; //定義為Map結構,key值代表用戶(hù)名稱(chēng)或其他唯一標識,Value代表對應的WebSocket連接。 //ConcurrentHashMap 保證線(xiàn)程安全,用來(lái)存放每個(gè)客戶(hù)端對應的WebSocket對象 private static Map<String,WebSocket> webSocketMap = new ConcurrentHashMap<String, WebSocket>(); //用戶(hù)名 private String username; //當前httpSession private Session session; /** * 打開(kāi)鏈接 * @param username * @param session */ @OnOpen public void openConnect(@PathParam("username")String username, Session session){ this.session = session; this.username = username; //在線(xiàn)連接數+1 onlinePerNumAdd(); //用戶(hù)名和當前用戶(hù)WebSocket對象放進(jìn)Map中 webSocketMap.put(this.username,this); log.info("{}連接服務(wù)器成功。。。。",this.username); } /** * 關(guān)閉連接 * @param username * @param session * @PathParam 用來(lái)獲取路徑中的動(dòng)態(tài)參數Key值 */ @OnClose public void closeConnect(@PathParam("username")String username, Session session){ webSocketMap.remove(username); //在線(xiàn)連接數-1 onlinePerNumSub(); log.info("{} 斷開(kāi)連接。。。",username); } /** * 錯誤提示 */ @OnError public void errorConnect(Session session, Throwable error){ log.error("websocket連接異常:{}",error.getMessage()); } @OnMessage public void send(String message, Session session) throws IOException { ObjectMapper objectMapper = new ObjectMapper(); Map map = objectMapper.readValue(message, Map.class); sendMessage(map.get("username").toString(),message); } /** * 點(diǎn)對點(diǎn)發(fā)送 * @param username * @param message * @throws IOException */ private void sendMessage(String username,String message) throws IOException { WebSocket webSocket = webSocketMap.get(username); webSocket.session.getBasicRemote().sendText(message); } /** * 廣播類(lèi)型發(fā)送 * @param message * @throws IOException */ private void sendMessage(String message) throws IOException { Set<String> keys = webSocketMap.keySet(); for (String key : keys) { WebSocket webSocket = webSocketMap.get(key); webSocket.sendMessage(message); } } private synchronized static void onlinePerNumAdd(){ WebSocket.onLinePersonNum ++; } private synchronized static void onlinePerNumSub(){ WebSocket.onLinePersonNum --; } private synchronized static int getOnLinePerNum(){ return WebSocket.onLinePersonNum; } }
4.webSocket客戶(hù)端
chat1.html
<!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>Title</title> <!-- <script src="https://heerey525.github.io/layui-v2.4.3/layui/layui.js"></script>--> <script src="https://cdn.bootcss.com/jquery/3.4.1/jquery.js"></script> </head> <body> <!--<button id="user1" onclick="connect()">連接</button>--> <input id="link" type="text"/> <input id="sendMsg" type="text"/> <button onclick="send()">發(fā)送</button> <div id="message"> </div> </body> <script type="text/javascript"> var websocket = null; // function connect() { //判斷當前瀏覽器是否支持WebSocket ,主要此處要更換為自己的地址 if('WebSocket' in window){ websocket = new WebSocket("ws://localhost:8089/chat/bbb"); } else{ alert('Not support websocket') } //連接發(fā)生錯誤的回調方法 websocket.onerror = function(){ // setMessageInnerHTML("error"); }; //連接成功建立的回調方法 websocket.onopen = function(event){ console.log("連接成功?。?!") // setMessageInnerHTML("open"); $("#link").val("連接成功??!") } //連接關(guān)閉的回調方法 websocket.onclose = function(){ // setMessageInnerHTML("close"); } //監聽(tīng)窗口關(guān)閉事件,當窗口關(guān)閉時(shí),主動(dòng)去關(guān)閉websocket連接,防止連接還沒(méi)斷開(kāi)就關(guān)閉窗口,server端會(huì )拋異常。 window.onbeforeunload = function(){ websocket.close(); } //發(fā)送消息 function send(){ websocket.send("aaa"); // onmessage(); } //接收到消息的回調方法 // function onmessage(){ websocket.onmessage = function(event){ console.log(event.data) // setMessageInnerHTML(event.data); $("#message").append("<h1>"+ event.data + "</h1>") // } } </script> </html>
chat2.html
<!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>Title</title> <!-- <script src="https://heerey525.github.io/layui-v2.4.3/layui/layui.js"></script>--> <script src="https://cdn.bootcss.com/jquery/3.4.1/jquery.js"></script> </head> <body> <!--<button id="user1" onclick="connect()">連接</button>--> <input id="link" type="text"/> <input id="sendMsg" type="text"/> <button onclick="send()">發(fā)送</button> <div id="message"> </div> </body> <script type="text/javascript"> var websocket = null; // function connect() { //判斷當前瀏覽器是否支持WebSocket ,主要此處要更換為自己的地址 if('WebSocket' in window){ websocket = new WebSocket("ws://localhost:8089/pushMsg/aaa"); } else{ alert('Not support websocket') } //連接發(fā)生錯誤的回調方法 websocket.onerror = function(){ // setMessageInnerHTML("error"); }; //連接成功建立的回調方法 websocket.onopen = function(event){ console.log("連接成功?。?!") // setMessageInnerHTML("open"); $("#link").val("服務(wù)器連接成功??!") } //連接關(guān)閉的回調方法 websocket.onclose = function(){ // setMessageInnerHTML("close"); } //監聽(tīng)窗口關(guān)閉事件,當窗口關(guān)閉時(shí),主動(dòng)去關(guān)閉websocket連接,防止連接還沒(méi)斷開(kāi)就關(guān)閉窗口,server端會(huì )拋異常。 window.onbeforeunload = function(){ websocket.close(); } //發(fā)送消息 function send(){ websocket.send("bbb"); // onmessage(); } //接收到消息的回調方法 // function onmessage(){ websocket.onmessage = function(event){ console.log(event.data) // setMessageInnerHTML(event.data); $("#message").append("<h1>"+ event.data + "</h1>") // } } </script> </html>
以上就是具體的代碼實(shí)現,對于如果用戶(hù)離線(xiàn),websocket斷開(kāi)連接的情況,可以采用持久化的存儲方式。例如使用mysql關(guān)系型數據庫或Redis緩存等等保存用戶(hù)的讀取狀態(tài),當用戶(hù)登錄后查詢(xún)用戶(hù)是否有未讀消息,然后進(jìn)行推送。
到此這篇關(guān)于Springboot+WebSocket實(shí)現一對一聊天和公告的示例代碼的文章就介紹到這了,更多相關(guān)Springboot WebSocket一對一聊天內容請搜索腳本之家以前的文章或繼續瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
免責聲明:本站發(fā)布的內容(圖片、視頻和文字)以原創(chuàng )、來(lái)自互聯(lián)網(wǎng)轉載和分享為主,文章觀(guān)點(diǎn)不代表本網(wǎng)站立場(chǎng),如果涉及侵權請聯(lián)系QQ:712375056 進(jìn)行舉報,并提供相關(guān)證據,一經(jīng)查實(shí),將立刻刪除涉嫌侵權內容。
Copyright ? 2009-2021 56dr.com. All Rights Reserved. 特網(wǎng)科技 特網(wǎng)云 版權所有 珠海市特網(wǎng)科技有限公司 粵ICP備16109289號
域名注冊服務(wù)機構:阿里云計算有限公司(萬(wàn)網(wǎng)) 域名服務(wù)機構:煙臺帝思普網(wǎng)絡(luò )科技有限公司(DNSPod) CDN服務(wù):阿里云計算有限公司 中國互聯(lián)網(wǎng)舉報中心 增值電信業(yè)務(wù)經(jīng)營(yíng)許可證B2
建議您使用Chrome、Firefox、Edge、IE10及以上版本和360等主流瀏覽器瀏覽本網(wǎng)站