今天就跟大家聊聊有關(guān)springboot中怎么整合WebSocket，可能很多人都不太了解，為了讓大家更加了解，小編給大家總結(jié)了以下內(nèi)容，希望大家根據(jù)這篇文章可以有所收獲。

網(wǎng)站建設(shè)哪家好，找創(chuàng)新互聯(lián)建站！專注于網(wǎng)頁(yè)設(shè)計(jì)、網(wǎng)站建設(shè)、微信開發(fā)、小程序開發(fā)、集團(tuán)企業(yè)網(wǎng)站建設(shè)等服務(wù)項(xiàng)目。為回饋新老客戶創(chuàng)新互聯(lián)還提供了昂昂溪免費(fèi)建站歡迎大家使用！

1、依賴引入

要使用 websocket 關(guān)鍵是@ServerEndpoint這個(gè)注解，該注解是 javaee 標(biāo)準(zhǔn)中的注解,tomcat7 及以上已經(jīng)實(shí)現(xiàn)了,如果使用傳統(tǒng)方法將 war 包部署到 tomcat 中，只需要引入如下 javaee 標(biāo)準(zhǔn)依賴即可：

javax
javaee-api
7.0
provided

如使用 springboot 內(nèi)置容器,無(wú)需引入，springboot 已經(jīng)做了包含。我們只需引入如下依賴即可：

org.springframework.boot
spring-boot-starter-websocket
1.5.3.RELEASE
pom

2、注入 Bean

首先注入一個(gè)ServerEndpointExporterBean,該 Bean 會(huì)自動(dòng)注冊(cè)使用@ServerEndpoint 注解申明的 websocket endpoint。代碼如下：

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

3、申明 endpoint

建立MyWebSocket.java類，在該類中處理 websocket 邏輯

@ServerEndpoint(value = "/websocket") //接受websocket請(qǐng)求路徑
@Component //注冊(cè)到spring容器中
public class MyWebSocket {
//保存所有在線socket連接
private static Map webSocketMap = new LinkedHashMap<>();
//記錄當(dāng)前在線數(shù)目
private static int count=0;
//當(dāng)前連接（每個(gè)websocket連入都會(huì)創(chuàng)建一個(gè)MyWebSocket實(shí)例
private Session session;
private Logger log = LoggerFactory.getLogger(this.getClass());
//處理連接建立
@OnOpen
public void onOpen(Session session){
this.session=session;
webSocketMap.put(session.getId(),this);
addCount();
log.info("新的連接加入：{}",session.getId());
}
//接受消息
@OnMessage
public void onMessage(String message,Session session){
log.info("收到客戶端{(lán)}消息：{}",session.getId(),message);
try{
this.sendMessage("收到消息："+message);
}catch (Exception e){
e.printStackTrace();
}
}
//處理錯(cuò)誤
@OnError
public void onError(Throwable error,Session session){
log.info("發(fā)生錯(cuò)誤{},{}",session.getId(),error.getMessage());
}
//處理連接關(guān)閉
@OnClose
public void onClose(){
webSocketMap.remove(this.session.getId());
reduceCount();
log.info("連接關(guān)閉:{}",this.session.getId());
}
//群發(fā)消息
//發(fā)送消息
public void sendMessage(String message) throws IOException {
this.session.getBasicRemote().sendText(message);
}
//廣播消息
public static void broadcast(){
MyWebSocket.webSocketMap.forEach((k,v)->{
try{
v.sendMessage("這是一條測(cè)試廣播");
}catch (Exception e){
}
});
}
//獲取在線連接數(shù)目
public static int getCount(){
return count;
}
//操作count，使用synchronized確保線程安全
public static synchronized void addCount(){
MyWebSocket.count++;
}
public static synchronized void reduceCount(){
MyWebSocket.count--;
}
}

4、客戶的實(shí)現(xiàn)

客戶端使用 h6 原生 websocket，部分瀏覽器可能不支持。代碼如下：

測(cè)試

三、測(cè)試

建立一個(gè) controller 測(cè)試群發(fā)，代碼如下：

@RestController
public class HomeController {
@GetMapping("/broadcast")
public void broadcast(){
MyWebSocket.broadcast();
}
}

然后打開上面的 html，可以看到瀏覽器和服務(wù)器都輸出連接成功的信息：

瀏覽器：
Event {isTrusted: true, type: "open", target: WebSocket, currentTarget: WebSocket, eventPhase: 2, …}
服務(wù)端：
2018-08-01 14:05:34.727 INFO 12708 --- [nio-8080-exec-1] com.fxb.h6websocket.MyWebSocket : 新的連接加入：0

點(diǎn)擊測(cè)試按鈕，可在服務(wù)端看到如下輸出：

2018-08-01 15:00:34.644 INFO 12708 --- [nio-8080-exec-6] com.fxb.h6websocket.MyWebSocket : 收到客戶端2消息：這是一個(gè)測(cè)試數(shù)據(jù)

再次打開 html 頁(yè)面，這樣就有兩個(gè) websocket 客戶端，然后在瀏覽器訪問(wèn)localhost:8080/broadcast測(cè)試群發(fā)功能,每個(gè)客戶端都會(huì)輸出如下信息：

MessageEvent {isTrusted: true, data: "這是一條測(cè)試廣播", origin: "ws://localhost:8080", lastEventId: "", source: null, …}

看完上述內(nèi)容，你們對(duì)springboot中怎么整合WebSocket有進(jìn)一步的了解嗎？如果還想了解更多知識(shí)或者相關(guān)內(nèi)容，請(qǐng)關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道，感謝大家的支持。