基于spring實現(xiàn)websocket實時推送實例
更新時間：2018年03月30日 09:49:11 作者：云游遍天下
這篇文章主要為大家詳細(xì)介紹了基于spring實現(xiàn)websocket實時推送實例，具有一定的參考價值，感興趣的小伙伴們可以參考一下
基于spring框架來寫的，websocket實時推送例子，具體內(nèi)容如下
第一步：自己搭建一個springmvc項目，很簡單，網(wǎng)上百度都有；pom文件添加以下：
<!-- WebSocket --> <dependency> <groupId>org.springframework</groupId> <artifactId>spring-websocket</artifactId> <version>4.2.4.RELEASE</version> </dependency> <dependency> <groupId>org.springframework</groupId> <artifactId>spring-messaging</artifactId> <version>4.2.4.RELEASE</version> </dependency>
我的spring版本是4.2.4的，所以websocket也是4.2.4的；websocket最好和spring版本保持一致
第二步：編寫消息處理器
/**
* Project Name:springRabbitMQ
* File Name:MyMessageHandler.java
* Package Name:com.zsy.websocket
* Date:2018年1月31日上午11:10:03
* Copyright (c) 2018, zhaoshouyun All Rights Reserved.
*
*/
package com.zsy.websocket;
import java.io.IOException;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import org.apache.commons.lang3.StringUtils;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.WebSocketMessage;
import org.springframework.web.socket.WebSocketSession;
/**
* ClassName: MyMessageHandler
* Function: 實現(xiàn)webscoket接口
* date: 2018年1月31日 上午11:10:03
* @author zhaoshouyun
* @version
* @since JDK 1.7
*/
public class MyMessageHandler implements WebSocketHandler {
//用戶key
public static final String USER_KEY = "current_user";
/**
* userMap:存儲用戶連接webscoket信息
* @since JDK 1.7
*/
private final static Map<String, WebSocketSession> userMap;
static {
userMap = new ConcurrentHashMap<String,WebSocketSession>(30);
}
/**
* 關(guān)閉websocket時調(diào)用該方法
* @see org.springframework.web.socket.WebSocketHandler#afterConnectionClosed(org.springframework.web.socket.WebSocketSession, org.springframework.web.socket.CloseStatus)
*/
@Override
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
String userId = this.getUserId(session);
if(StringUtils.isNoneBlank(userId)){
userMap.remove(userId);
System.err.println("該" + userId +"用戶已成功關(guān)閉");
}else{
System.err.println("關(guān)閉時，獲取用戶id為空");
}
}
/**
* 建立websocket連接時調(diào)用該方法
* @see org.springframework.web.socket.WebSocketHandler#afterConnectionEstablished(org.springframework.web.socket.WebSocketSession)
*/
@Override
public void afterConnectionEstablished(WebSocketSession session) throws Exception {
String userId = this.getUserId(session);
if(StringUtils.isNoneBlank(userId)){
userMap.put(userId, session);
session.sendMessage(new TextMessage("建立WebSocket連接成功！"));
}
}
/**
* 客戶端調(diào)用websocket.send時候，會調(diào)用該方法,進(jìn)行數(shù)據(jù)通信
* @see org.springframework.web.socket.WebSocketHandler#handleMessage(org.springframework.web.socket.WebSocketSession, org.springframework.web.socket.WebSocketMessage)
*/
@Override
public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {
String msg = message.toString();
String userId = this.getUserId(session);
System.err.println("該"+userId+"用戶發(fā)送的消息是："+msg);
message = new TextMessage("服務(wù)端已經(jīng)接收到消息，msg="+msg);
session.sendMessage(message);
}
/**
* 傳輸過程出現(xiàn)異常時，調(diào)用該方法
* @see org.springframework.web.socket.WebSocketHandler#handleTransportError(org.springframework.web.socket.WebSocketSession, java.lang.Throwable)
*/
@Override
public void handleTransportError(WebSocketSession session, Throwable e) throws Exception {
WebSocketMessage<String> message = new TextMessage("異常信息："+e.getMessage());
session.sendMessage(message);
}
/**
*
* @see org.springframework.web.socket.WebSocketHandler#supportsPartialMessages()
*/
@Override
public boolean supportsPartialMessages() {
return false;
}
/**
* sendMessageToUser:發(fā)給指定用戶
* @author zhaoshouyun
* @param userId
* @param contents
* @since JDK 1.7
*/
public void sendMessageToUser(String userId,String contents) {
WebSocketSession session = userMap.get(userId);
if(session !=null && session.isOpen()) {
try {
TextMessage message = new TextMessage(contents);
session.sendMessage(message);
} catch (IOException e) {
e.printStackTrace();
}
}
}
/**
* sendMessageToAllUsers:發(fā)給所有的用戶
* @author zhaoshouyun
* @param contents
* @since JDK 1.7
*/
public void sendMessageToAllUsers(String contents) {
Set<String> userIds = userMap.keySet();
for(String userId: userIds) {
this.sendMessageToUser(userId, contents);
}
}
/**
* getUserId:獲取用戶id
* @author zhaoshouyun
* @param session
* @return
* @since JDK 1.7
*/
private String getUserId(WebSocketSession session){
try {
String userId = (String)session.getAttributes().get(USER_KEY);
return userId;
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
}
第三步：編寫websocket相關(guān)配置，當(dāng)然可以在xml配置；我現(xiàn)在沒有使用xml配置，使用代碼配置，需要在xml里添加掃描包<context:component-scan base-package="com.zsy.websocket" />
/**
* Project Name:springRabbitMQ
* File Name:WebSocketConfig.java
* Package Name:com.zsy.websocket
* Date:2018年1月31日下午1:10:33
* Copyright (c) 2018, zhaoshouyun All Rights Reserved.
*
*/
/**
* Project Name:springRabbitMQ
* File Name:WebSocketConfig.java
* Package Name:com.zsy.websocket
* Date:2018年1月31日下午1:10:33
* Copyright (c) 2018, zhaoshouyun All Rights Reserved.
*
*/
package com.zsy.websocket;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
/**
* ClassName: WebSocketConfig
* Function: TODO ADD FUNCTION.
* date: 2018年1月31日 下午1:10:33
* @author zhaoshouyun
* @version
* @since JDK 1.7
*/
@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
/**
* 注冊handle
* @see org.springframework.web.socket.config.annotation.WebSocketConfigurer#registerWebSocketHandlers(org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry)
*/
@Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
registry.addHandler(myHandler(), "/testHandler").addInterceptors(new WebSocketInterceptor());
registry.addHandler(myHandler(), "/socketJs/testHandler").addInterceptors(new WebSocketInterceptor()).withSockJS();
}
@Bean
public WebSocketHandler myHandler(){
return new MyMessageHandler();
}
}
第四步：編寫websocket適配器
package com.zsy.websocket;
import java.util.Map;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.http.server.ServletServerHttpRequest;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.server.support.HttpSessionHandshakeInterceptor;
/**
* ClassName: WebSocketInterceptor
* Function: TODO ADD FUNCTION.
* date: 2018年1月31日 上午11:42:34
* @author zhaoshouyun
* @version
* @since JDK 1.7
*/
public class WebSocketInterceptor extends HttpSessionHandshakeInterceptor {
/**
* TODO 簡單描述該方法的實現(xiàn)功能（可選）.
* @see org.springframework.web.socket.server.support.HttpSessionHandshakeInterceptor#beforeHandshake(org.springframework.http.server.ServerHttpRequest, org.springframework.http.server.ServerHttpResponse, org.springframework.web.socket.WebSocketHandler, java.util.Map)
*/
@Override
public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,
Map<String, Object> attributes) throws Exception {
if(request instanceof ServletServerHttpRequest){
ServletServerHttpRequest serverHttpRequest = (ServletServerHttpRequest)request;
//獲取參數(shù)
String userId = serverHttpRequest .getServletRequest().getParameter("userId");
attributes.put(MyMessageHandler.USER_KEY, userId);
}
return true;
}
}
第五步對應(yīng)的js：
<%@ page language="java" contentType="text/html; charset=ISO-8859-1"
pageEncoding="ISO-8859-1"%>
<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN" "http://www.w3.org/TR/html4/loose.dtd">
<html>
<head>
<meta http-equiv="Content-Type" content="text/html; charset=ISO-8859-1">
<title>Insert title here</title>
<script type="text/javascript">
var websocket;
// 首先判斷是否 支持 WebSocket
if('WebSocket' in window) {
websocket = new WebSocket("ws://localhost:8085/springTest/testHandler?userId=zhaoshouyun");
} else if('MozWebSocket' in window) {
websocket = new MozWebSocket("ws://localhost:8085/springTest/testHandler?userId=zhaoshouyun");
} else {
websocket = new SockJS("http://localhost:8085/springTest/socketJs/testHandler?userId=zhaoshouyun");
}
// 打開連接時
websocket.onopen = function(evnt) {
console.log(" websocket.onopen ");
};
// 收到消息時
websocket.onmessage = function(evnt) {
alert(evnt.data);
};
websocket.onerror = function(evnt) {
console.log(" websocket.onerror ");
};
websocket.onclose = function(evnt) {
console.log(" websocket.onclose ");
};
function say(){
//客戶端主動發(fā)消息
websocket.send(document.getElementById('msg').value);
}
</script>
</head>
<body>
<input type="text" value="" id="msg"><button onclick="say()"></button>
</body>
</html>
第六步測試：
package com.zsy.test.controller;
import java.util.HashMap;
import java.util.Map;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import com.zsy.websocket.MyMessageHandler;
/**
* ClassName: TestController
* Function: TODO ADD FUNCTION.
* date: 2017年12月14日 上午11:11:23
* @author zhaoshouyun
* @version
* @since JDK 1.7
*/
@Controller
public class TestController {
@Autowired
MyMessageHandler handler;
@RequestMapping("/get")
public String get(){
return "index";
}
@ResponseBody
@RequestMapping("/get1")
public String send(String name){
handler.sendMessageToUser("zhaoshouyun", "服務(wù)端發(fā)送的內(nèi)容:"+name);
return "success";
}
}
以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。
相關(guān)文章
SpringBoot+SpringSecurity 不攔截靜態(tài)資源的實現(xiàn)
這篇文章主要介紹了SpringBoot+SpringSecurity 不攔截靜態(tài)資源的實現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-09-09
Java網(wǎng)絡(luò)IO模型詳解(BIO、NIO、AIO)
Java支持BIO、NIO和AIO三種網(wǎng)絡(luò)IO模型,BIO是同步阻塞模型,適用于連接數(shù)較少的場景,NIO是同步非阻塞模型,適用于處理多個連接,支持自JDK1.4起,AIO是異步非阻塞模型,適用于異步操作多的場景,文中通過代碼介紹的非常詳細(xì),需要的朋友可以參考下2024-10-10
java -D參數(shù)設(shè)置系統(tǒng)屬性無效問題及解決
這篇文章主要介紹了java -D參數(shù)設(shè)置系統(tǒng)屬性無效問題及解決方案，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-12-12
如何在Spring?Boot框架中使用攔截器實現(xiàn)URL限制
在Spring?Boot框架中,您可以使用攔截器（Interceptor）來控制限制URL列表,本文通過一個簡單的示例給大家介紹Spring?Boot?攔截器實現(xiàn)URL限制的操作方法,感興趣的朋友跟隨小編一起看看吧2023-08-08
在Java中,我們經(jīng)常需要將數(shù)組轉(zhuǎn)換為List來方便地進(jìn)行操作,Arrays.asList()方法是一種常見的方式,本文主要介紹了Java中Arrays.asList()需要注意的坑,具有一定的參考價值,感興趣的可以了解一下2023-08-08
這篇文章主要介紹了 Android中幾種圖片特效的處理的實現(xiàn)方法的相關(guān)資料,這里有放大縮小圖片，獲得圓角圖片，獲得帶倒影圖片的幾種方法，需要的朋友可以參考下2017-08-08
這篇文章主要介紹了Java中的Semaphore信號量簡析,Semaphore：信號量,用來限制能同時訪問共享資源的線程上限,使用Semaphore實現(xiàn)簡單連接池,對比享元模式下的實現(xiàn)(用wait和notify),性能和可讀性要更好,需要的朋友可以參考下2023-12-12
防止未登錄用戶操作—基于struts2攔截器的簡單實現(xiàn)
下面小編就為大家?guī)硪黄乐刮吹卿浻脩舨僮鳌趕truts2攔截器的簡單實現(xiàn)。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-10-10
Java中dubbo+zookeeper微服務(wù)架構(gòu)簡介
Apache Dubbo是一款高性能的 Java RPC 框架，這篇文章主要介紹了Java中dubbo+zookeeper微服務(wù)架構(gòu),需要的朋友可以參考下2021-09-09
貪心算法是指，在對問題求解時，總是做出在當(dāng)前看來是最好的選擇。也就是說，不從整體最優(yōu)上加以考慮，他所做出的是在某種意義上的局部最優(yōu)解，枚舉法的本質(zhì)就是從所有候選答案中去搜索正確的解，枚舉算法簡單粗暴，他暴力的枚舉所有可能，盡可能地嘗試所有的方法2022-04-04