springboot+websocket+rabbitmq整合实现消息实时
SpringBoot+WebSocket+RabbitMQ整合实现消息实时
========>可能影响项目运行，访问不到静态资源文件，整合WebSocket也可参考:
https://www.cnblogs.com/yu-si/articles/15075737.html
WebSocket+RabbitMQ基于Spring boot实现
1.配置
maven导包
SpringBoot 与webSocket的关联
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
SpringBoot与Rabbit MQ的关联
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-amqp</artifactId>
</dependency>
配置application.yml文件
spring:
rabbitmq:
host: 127.0.0.1
username: guest
password: guest
#虚拟主机
virtual-host: /
port: 5672
编写RabbitMQ服务端
配置服务端的config
@Configuration
public class SendConfig {
@Bean("queue")
public Queue queueMsg(){
return new Queue("queue");
}
@Bean("exchange")
public TopicExchange exchange(){
return new TopicExchange("exchange");
}
@Bean
Binding bindingExchangeAndQueuemsg(@Qualifier("queue") Queue queue, @Qualifier("exchange") TopicExchange exchange){
return BindingBuilder.bind(queue).to(exchange).with("a");
}
}
编写服务端的Controller
import org.springframework.amqp.core.AmqpTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
@RestController
public class Controller {
@Autowired
private AmqpTemplate amqpTemplate;
@RequestMapping("/send")
public void SennerMsg(String msg){
amqpTemplate.convertAndSend("exchange","a",msg);
}
}
到这启动MQ后运行程序便可以看到有一个新的邮件
或者用命令查看
编写MQ的客户端（与WebSocket结合）
配置Websocket—做请求拦截
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.HandshakeInterceptor;
import java.util.Map;
public class Hank implements HandshakeInterceptor{
@Override
public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse serverHttpResponse, WebSocketHandler webSocketHandler, Map<String, Object> map) throws Exception {
String jspCode = ((ServletServerHttpRequest) request).getServletRequest().getParameter("jspCode");
if (jspCode != null) {
map.put("jspCoe", jspCode);
} else {
return false;
}
return true;
}
@Override
public void afterHandshake(ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse, WebSocketHandler webSocketHandler, Exception e) {
}
}
配置Websocket的Handler
@Component
public class MyWebSocketHandler implements WebSocketHandler {
private static final Map<String ,WebSocketSession> userMap=new HashMap<>();
@Override
public void afterConnectionEstablished(WebSocketSession webSocketSession) throws Exception {
String jspCode = (String )webSocketSession.getAttributes().get("jspCode");
//if(userMap.get(jspCode)==null){
System.out.println(jspCode);
userMap.put(jspCode,webSocketSession);
//}
}
@Override
public void handleMessage(WebSocketSession webSocketSession, WebSocketMessage<?> webSocketMessage) throws Exception {
}
@Override
public void handleTransportError(WebSocketSession webSocketSession, Throwable throwable) throws Exception {
}
@Override
public void afterConnectionClosed(WebSocketSession webSocketSession, CloseStatus closeStatus) throws Exception {
}
@Override
public boolean supportsPartialMessages() {
return false;
}
public void sendMsgToJsp(final TextMessage message, String type) throws Exception{
Iterator<Map.Entry<String ,WebSocketSession>> it=userMap.entrySet().iterator();
while (it.hasNext()){
final Map.Entry<String ,WebSocketSession> entry=it.next();
System.out.println(entry.getValue().isOpen());
System.out.println(entry.getKey().contains(type));
if(entry.getValue().isOpen()&&entry.getKey().contains(type)){
new Thread(new Runnable() {
@Override
public void run() {
try {
if(entry.getValue().isOpen()){
entry.getValue().sendMessage(message);
}
}catch (IOException e){
e.printStackTrace();
}
}
}).start();
}
}
}
}
配置WebSocket的config
@Component
@EnableWebMvc
@EnableWebSocket
public class WebSocketConfig extends WebMvcConfigurerAdapter implements WebSocketConfigurer {
@Resource
private MyWebSocketHandler handler;
@Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
registry.addHandler(handler, "/wsMy").addInterceptors(new HandshakeInterceptor() {
@Override
public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse serverHttpResponse, WebSocketHandler webSocketHandler, Map<String, Object> map) throws Exception {
String jspCode = ((ServletServerHttpRequest) request).getServletRequest().getParameter("jspCode");
if (jspCode != null) {
map.put("jspCode", jspCode);
} else {
return false;
}
return true;
}
@Override
public void afterHandshake(ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse, WebSocketHandler webSocketHandler, Exception e) {
}
});
}
}
配置客户端的cortorller（二者关联的部分）
@Autowired
public MyWebSocketHandler handler;
@RabbitListener(queues = "queue")
public void Recive(String msg) throws Exception{
handler.sendMsgToJsp(new TextMessage(msg), "A");
}
到此便可以启动测试了
另附上前端页面
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Title</title>
</head>
<body>
</body>
<script type="text/javascript">
//1.创建websocket客户端
var wsServerUrl = 'ws://127.0.0.1/';
var limitConnect = 3; // 断线重连次数
var timeConnect = 0;
webSocketInit(wsServerUrl);
//socket初始化链接
function webSocketInit(wsServerUrl) {
// 首先判断是否 支持 WebSocket
if ('WebSocket' in window) {
websocket = new WebSocket("ws://localhost:80/wsMy?jspCode=AA");
} else if ('MozWebSocket' in window) {
websocket = new MozWebSocket("ws://localhost:80/wsMy?jspCode=AA");
} else {
websocket = new SockJS("ws://localhost:80/wsMy?jspCode=AA");
}
// 打开连接时
websocket.onopen = function (event) {
console.log("已连接TCP服务器");
};
// 收到消息时
websocket.onmessage = function (event) {
console.log("收到一条消息" + event.data);
alert(event.data);
};
websocket.onerror = function (event) {
console.log("服务器报错：");
reconnect()
};
websocket.onclose = function (event) {
console.log('服务器已经断开');
reconnect()
};
// 重连
function reconnect(wsServerUrl) {
// lockReconnect加锁，防止onclose、onerror两次重连
if (limitConnect > 0) {
if (localStorage.getItem('lockReconnect') != true) {
localStorage.setItem("lockReconnect", 1);
limitConnect--;
timeConnect++;
console.log("第" + timeConnect + "次重连");
// 进行重连
setTimeout(function () {
webSocketInit(wsServerUrl);
localStorage.removeItem("lockReconnect");
}, 2000);
}
} else {
console.log("TCP连接已超时");
}
}
// 心跳 * 回应
setInterval(function () {
websocket.send('');
}, 1000 * 100);
}
</script>
</html>
原文链接：https://blog.csdn.net/lilin0800/article/details/80884950
