[文末贈(zèng)書]手把手教你SpringBoot整合websocket實(shí)現(xiàn)消息推送
點(diǎn)擊上方 Java學(xué)習(xí)之道，選擇 設(shè)為星標(biāo)
來源: segmentfault.com/a/1190000011908831
作者: kunm
依賴
springboot基礎(chǔ)環(huán)境,請參考springboot文檔 maven依賴
????????????org.springframework.boot
????????????spring-boot-starter-websocket
????????
環(huán)境準(zhǔn)備
準(zhǔn)備常量類
//webSocket相關(guān)配置
????//鏈接地址
????public?static?String?WEBSOCKETPATHPERFIX?=?"/ws-push";
????public?static?String?WEBSOCKETPATH?=?"/endpointWisely";
????//消息代理路徑
????public?static?String?WEBSOCKETBROADCASTPATH?=?"/topic";
????//前端發(fā)送給服務(wù)端請求地址
????public?static?final?String?FORETOSERVERPATH?=?"/welcome";
????//服務(wù)端生產(chǎn)地址,客戶端訂閱此地址以接收服務(wù)端生產(chǎn)的消息
????public?static?final?String?PRODUCERPATH?=?"/topic/getResponse";
????//點(diǎn)對點(diǎn)消息推送地址前綴
????public?static?final?String?P2PPUSHBASEPATH?=?"/user";
????//點(diǎn)對點(diǎn)消息推送地址后綴,最后的地址為/user/用戶識(shí)別碼/msg
????public?static?final?String?P2PPUSHPATH?=?"/msg";
接收前端消息實(shí)體
public?class?WiselyMessage?{
????private?String?name;
????public?String?getName()?{
????????return?name;
????}
????public?void?setName(String?name)?{
????????this.name?=?name;
????}
}
后臺(tái)發(fā)送消息實(shí)體
private?String?responseMessage;
????public?WiselyResponse(String?responseMessage){
????????this.responseMessage?=?responseMessage;
????}
????public?String?getResponseMessage()?{
????????return?responseMessage;
????}
????public?void?setResponseMessage(String?responseMessage)?{
????????this.responseMessage?=?responseMessage;
????}
配置websocket
@Configuration
//?@EnableWebSocketMessageBroker注解用于開啟使用STOMP協(xié)議來傳輸基于代理（MessageBroker）的消息，這時(shí)候控制器（controller）
//?開始支持@MessageMapping,就像是使用@requestMapping一樣。
@EnableWebSocketMessageBroker
public?class?WebSocketConfig?extends?AbstractWebSocketMessageBrokerConfigurer?{
????@Override
????public?void?registerStompEndpoints(StompEndpointRegistry?stompEndpointRegistry)?{
????????//注冊一個(gè)Stomp的節(jié)點(diǎn)（endpoint）,并指定使用SockJS協(xié)議。
????????stompEndpointRegistry.addEndpoint(Constant.WEBSOCKETPATH).withSockJS();
????}
????@Override
????public?void?configureMessageBroker(MessageBrokerRegistry?registry)?{
????????//服務(wù)端發(fā)送消息給客戶端的域,多個(gè)用逗號(hào)隔開
????????registry.enableSimpleBroker(Constant.WEBSOCKETBROADCASTPATH,?Constant.P2PPUSHBASEPATH);
????????//定義一對一推送的時(shí)候前綴
????????registry.setUserDestinationPrefix(Constant.P2PPUSHBASEPATH);
????????//定義websoket前綴
????????registry.setApplicationDestinationPrefixes(Constant.WEBSOCKETPATHPERFIX);
????}
}
service
@Service
public?class?WebSocketService?{
????@Autowired
????private?SimpMessagingTemplate?template;
????/**
?????*?廣播
?????*?發(fā)給所有在線用戶
?????*
?????*?@param?msg
?????*/
????public?void?sendMsg(WiselyResponse?msg)?{
????????template.convertAndSend(Constant.PRODUCERPATH,?msg);
????}
????/**
?????*?發(fā)送給指定用戶
?????*?@param?users
?????*?@param?msg
?????*/
????public?void?send2Users(List?users,?WiselyResponse?msg)?{
????????users.forEach(userName?->?{
????????????template.convertAndSendToUser(userName,?Constant.P2PPUSHPATH,?msg);
????????});
????}
}
控制器
@Controller
public?class?WsController?{
????@Resource
????WebSocketService?webSocketService;
????@MessageMapping(Constant.FORETOSERVERPATH)//@MessageMapping和@RequestMapping功能類似，用于設(shè)置URL映射地址，瀏覽器向服務(wù)器發(fā)起請求，需要通過該地址。
????@SendTo(Constant.PRODUCERPATH)//如果服務(wù)器接受到了消息，就會(huì)對訂閱了@SendTo括號(hào)中的地址傳送消息。
????public?WiselyResponse?say(WiselyMessage?message)?throws?Exception?{
????????List?users?=?Lists.newArrayList();
????????users.add("d892bf12bf7d11e793b69c5c8e6f60fb");//此處寫死只是為了方便測試,此值需要對應(yīng)頁面中訂閱個(gè)人消息的userId
????????webSocketService.send2Users(users,?new?WiselyResponse("admin?hello"));
????????return?new?WiselyResponse("Welcome,?"?+?message.getName()?+?"!");
????}
}
頁面
"http://www.thymeleaf.org">
????"UTF-8"?/>
????Spring?Boot+WebSocket+廣播式
"disconnect()">
????
????????"connect"?onclick="connect();">連接
????????"disconnect"?disabled="disabled"?onclick="disconnect();">斷開連接
????
????"conversationDiv">
????????type="text"?id="name"?/>
????????"sendName"?onclick="sendName();">發(fā)送
????????"response">
????????"response1">
????