Springboot+WebSocket實(shí)現(xiàn)一對(duì)一聊天和公告的示例代碼
1.POM文件導(dǎo)入Springboot整合websocket的依賴
<dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> <version>2.1.6.RELEASE</version></dependency>
2.注冊(cè)WebSocket的Bean交給Spring容器管理
@Configurationpublic class WebSocketServiceConfig { @Bean public ServerEndpointExporter serverEndpointExporter() {return new ServerEndpointExporter(); }}
3.WebSocket服務(wù)端實(shí)現(xiàn)
@ServerEndpoint 注解聲明為一個(gè)WebSocket服務(wù)，訪問(wèn)地址為/chat/{username}，@Component將其注冊(cè)為Spring的一個(gè)組件，交給Spring進(jìn)行管理
@ServerEndpoint('/chat/{username}')@Component@Slf4jpublic class WebSocket { //注入dao或者service，注意：因?yàn)閐ao層接口和service層都是單例的Bean //webSocket 不是單例的，所以在注入dao或者service時(shí)，需要用set方法對(duì)其進(jìn)行注入，保證每一個(gè)都是獨(dú)立的 private static ChatMapper chatMapper; //參數(shù)中的ChatMapper 是 單例池中的ChatMapper @Autowired public void setChatMapper(ChatMapper chatMapperBean){WebSocket.chatMapper = chatMapperBean; } //當(dāng)前連接數(shù) private static int onLinePersonNum; //定義為Map結(jié)構(gòu)，key值代表用戶名稱或其他唯一標(biāo)識(shí)，Value代表對(duì)應(yīng)的WebSocket連接。 //ConcurrentHashMap 保證線程安全，用來(lái)存放每個(gè)客戶端對(duì)應(yīng)的WebSocket對(duì)象 private static Map<String,WebSocket> webSocketMap = new ConcurrentHashMap<String, WebSocket>(); //用戶名 private String username; //當(dāng)前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;//在線連接數(shù)+1onlinePerNumAdd();//用戶名和當(dāng)前用戶WebSocket對(duì)象放進(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)參數(shù)Key值 */ @OnClose public void closeConnect(@PathParam('username')String username, Session session){webSocketMap.remove(username);//在線連接數(shù)-1onlinePerNumSub();log.info('{} 斷開(kāi)連接。。。',username); } /** * 錯(cuò)誤提示 */ @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)對(duì)點(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); } /** * 廣播類型發(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客戶端
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 onclick='connect()'>連接</button>--><input type='text'/><input type='text'/><button onclick='send()'>發(fā)送</button><div id='message'></div></body><script type='text/javascript'> var websocket = null; // function connect() { //判斷當(dāng)前瀏覽器是否支持WebSocket ,主要此處要更換為自己的地址 if(’WebSocket’ in window){ websocket = new WebSocket('ws://localhost:8089/chat/bbb'); } else{ alert(’Not support websocket’) } //連接發(fā)生錯(cuò)誤的回調(diào)方法 websocket.onerror = function(){ // setMessageInnerHTML('error'); }; //連接成功建立的回調(diào)方法 websocket.onopen = function(event){ console.log('連接成功！！！') // setMessageInnerHTML('open'); $('#link').val('連接成功！！') } //連接關(guān)閉的回調(diào)方法 websocket.onclose = function(){ // setMessageInnerHTML('close'); } //監(jiān)聽(tīng)窗口關(guān)閉事件，當(dāng)窗口關(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(); } //接收到消息的回調(diào)方法 // 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 onclick='connect()'>連接</button>--><input type='text'/><input type='text'/><button onclick='send()'>發(fā)送</button><div id='message'></div></body><script type='text/javascript'> var websocket = null; // function connect() {//判斷當(dāng)前瀏覽器是否支持WebSocket ,主要此處要更換為自己的地址if(’WebSocket’ in window){ websocket = new WebSocket('ws://localhost:8089/pushMsg/aaa');}else{ alert(’Not support websocket’)}//連接發(fā)生錯(cuò)誤的回調(diào)方法websocket.onerror = function(){ // setMessageInnerHTML('error');};//連接成功建立的回調(diào)方法websocket.onopen = function(event){ console.log('連接成功！！！') // setMessageInnerHTML('open'); $('#link').val('服務(wù)器連接成功！！')}//連接關(guān)閉的回調(diào)方法websocket.onclose = function(){ // setMessageInnerHTML('close');}//監(jiān)聽(tīng)窗口關(guān)閉事件，當(dāng)窗口關(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(); } //接收到消息的回調(diào)方法 // function onmessage(){websocket.onmessage = function(event){ console.log(event.data) // setMessageInnerHTML(event.data); $('#message').append('<h1>'+ event.data + '</h1>')// } }</script></html>
以上就是具體的代碼實(shí)現(xiàn)，對(duì)于如果用戶離線，websocket斷開(kāi)連接的情況，可以采用持久化的存儲(chǔ)方式。例如使用mysql關(guān)系型數(shù)據(jù)庫(kù)或Redis緩存等等保存用戶的讀取狀態(tài)，當(dāng)用戶登錄后查詢用戶是否有未讀消息，然后進(jìn)行推送。
到此這篇關(guān)于Springboot+WebSocket實(shí)現(xiàn)一對(duì)一聊天和公告的示例代碼的文章就介紹到這了,更多相關(guān)Springboot WebSocket一對(duì)一聊天內(nèi)容請(qǐng)搜索好吧啦網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持好吧啦網(wǎng)！
相關(guān)文章：
1. Java GZip 基于內(nèi)存實(shí)現(xiàn)壓縮和解壓的方法2. Springboot 全局日期格式化處理的實(shí)現(xiàn)3. 利用CSS制作3D動(dòng)畫4. .Net加密神器Eazfuscator.NET?2023.2?最新版使用教程5. jsp+servlet簡(jiǎn)單實(shí)現(xiàn)上傳文件功能（保存目錄改進(jìn)）6. JAMon(Java Application Monitor)備忘記7. 完美解決vue 中多個(gè)echarts圖表自適應(yīng)的問(wèn)題8. SpringBoot+TestNG單元測(cè)試的實(shí)現(xiàn)9. 存儲(chǔ)于xml中需要的HTML轉(zhuǎn)義代碼10. idea配置jdk的操作方法
網(wǎng)公網(wǎng)安備