Springboot+WebSocket實現(xiàn)一對一聊天和公告的示例代碼
更新時間：2021年04月26日 11:09:38 作者：一個頭發(fā)茂密的程序員
這篇文章主要介紹了Springboot+WebSocket實現(xiàn)一對一聊天和公告的示例代碼，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
1.POM文件導入Springboot整合websocket的依賴
<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服務端實現(xiàn)
@ServerEndpoint 注解聲明為一個WebSocket服務，訪問地址為/chat/{username}，@Component將其注冊為Spring的一個組件，交給Spring進行管理
@ServerEndpoint("/chat/{username}")
@Component
@Slf4j
public class WebSocket {
//注入dao或者service，注意：因為dao層接口和service層都是單例的Bean
//webSocket 不是單例的，所以在注入dao或者service時，需要用set方法對其進行注入，保證每一個都是獨立的
private static ChatMapper chatMapper;
//參數(shù)中的ChatMapper 是 單例池中的ChatMapper
@Autowired
public void setChatMapper(ChatMapper chatMapperBean){
WebSocket.chatMapper = chatMapperBean;
}
//當前連接數(shù)
private static int onLinePersonNum;
//定義為Map結構，key值代表用戶名稱或其他唯一標識，Value代表對應的WebSocket連接。
//ConcurrentHashMap 保證線程安全，用來存放每個客戶端對應的WebSocket對象
private static Map<String,WebSocket> webSocketMap = new ConcurrentHashMap<String, WebSocket>();
//用戶名
private String username;
//當前httpSession
private Session session;
/**
* 打開鏈接
* @param username
* @param session
*/
@OnOpen
public void openConnect(@PathParam("username")String username, Session session){
this.session = session;
this.username = username;
//在線連接數(shù)+1
onlinePerNumAdd();
//用戶名和當前用戶WebSocket對象放進Map中
webSocketMap.put(this.username,this);
log.info("{}連接服務器成功。。。。",this.username);
}
/**
* 關閉連接
* @param username
* @param session
* @PathParam 用來獲取路徑中的動態(tài)參數(shù)Key值
*/
@OnClose
public void closeConnect(@PathParam("username")String username, Session session){
webSocketMap.remove(username);
//在線連接數(shù)-1
onlinePerNumSub();
log.info("{} 斷開連接。。。",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);
}
/**
* 點對點發(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 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("連接成功！！")
}
//連接關閉的回調方法
websocket.onclose = function(){
// setMessageInnerHTML("close");
}
//監(jiān)聽窗口關閉事件，當窗口關閉時，主動去關閉websocket連接，防止連接還沒斷開就關閉窗口，server端會拋異常。
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("服務器連接成功！！")
}
//連接關閉的回調方法
websocket.onclose = function(){
// setMessageInnerHTML("close");
}
//監(jiān)聽窗口關閉事件，當窗口關閉時，主動去關閉websocket連接，防止連接還沒斷開就關閉窗口，server端會拋異常。
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>
以上就是具體的代碼實現(xiàn)，對于如果用戶離線，websocket斷開連接的情況，可以采用持久化的存儲方式。例如使用mysql關系型數(shù)據(jù)庫或Redis緩存等等保存用戶的讀取狀態(tài)，當用戶登錄后查詢用戶是否有未讀消息，然后進行推送。
到此這篇關于Springboot+WebSocket實現(xiàn)一對一聊天和公告的示例代碼的文章就介紹到這了,更多相關Springboot WebSocket一對一聊天內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！
您可能感興趣的文章:
相關文章
這篇文章主要介紹了Java中的while無限循環(huán)結構及實例,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-01-01
詳解Elasticsearch如何實現(xiàn)簡單的腳本排序
Elasticsearch?是位于?Elastic?Stack?核心的分布式搜索和分析引擎，可以為所有類型的數(shù)據(jù)提供近乎實時的搜索和分析。本文主要介紹了Elasticsearch如何實現(xiàn)簡單的腳本排序，感興趣的可以了解一下2023-01-01
本篇文章是對在Java中如何獲取Spring中配置的bean進行了詳細的分析介紹，需要的朋友參考下2013-07-07
SpringBoot自帶了validation工具可以從后端對前端傳來的參數(shù)進行校驗,本文給大家介紹SpringBoot參數(shù)校驗及用法,感興趣的朋友跟隨小編一起看看吧2023-10-10
這篇文章主要為大家詳細介紹了Java字母大小寫轉換的方法，具有一定的參考價值，感興趣的小伙伴們可以參考一下2017-06-06
Spring Cloud Alibaba 是阿里巴巴提供的一站式微服務開發(fā)解決方案。而 Nacos 作為 Spring Cloud Alibaba 的核心組件之一，提供了兩個非常重要的功能：注冊中心和配置中心，我們今天來了解和實現(xiàn)一下二者2022-08-08
這篇文章主要介紹了解決mapper接口無法映射mapper.xml的問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-06-06
這篇文章主要介紹了深入理解java1.8之supplier，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-12-12
工作中很多時候需要用到合并兩個List并去除其中的重復內容,這是一個很簡單的操作,實現(xiàn)的方法也多種多樣,這篇文章主要給大家介紹了關于Java合并兩個List后并去掉重復項的兩種做法,需要的朋友可以參考下2023-10-10
這篇文章主要給大家介紹了關于Java給實體每一個字段賦默認值的相關資料,在編程過程中有時會出現(xiàn)這樣一種情況,在查詢無結果時我們需要給實體賦默認值,需要的朋友可以參考下2023-09-09