如何用springboot和RabbitMQ实现发送通知前端
时间: 2024-02-24 22:53:45 浏览: 118
使用Spring Boot和RabbitMQ发送通知到前端可以采用WebSocket技术。WebSocket是一种在单个TCP连接上进行全双工通信的技术,它可以让服务器主动向客户端推送消息。
以下是使用Spring Boot和RabbitMQ发送通知到前端的示例代码:
首先,需要在Spring Boot中添加WebSocket支持,可以在pom.xml文件中添加以下依赖:
```xml
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
```
然后,创建一个WebSocket配置类,例如WebSocketConfig:
```java
@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
@Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
registry.addHandler(new NotificationHandler(), "/notifications");
}
}
```
在上面的代码中,我们创建了一个WebSocket处理程序NotificationHandler,并将其注册到路径“/notifications”上。
接下来,创建NotificationHandler类,用于接收来自RabbitMQ的通知并将其发送到前端:
```java
public class NotificationHandler extends TextWebSocketHandler {
private RabbitTemplate rabbitTemplate;
public NotificationHandler(RabbitTemplate rabbitTemplate) {
this.rabbitTemplate = rabbitTemplate;
}
@Override
public void afterConnectionEstablished(WebSocketSession session) throws Exception {
String userId = getUserIdFromSession(session);
if (userId != null) {
rabbitTemplate.convertAndSend("notification.exchange", "notification." + userId, "connected");
}
}
@Override
protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
// do nothing
}
private String getUserIdFromSession(WebSocketSession session) {
// extract user ID from session
return null;
}
}
```
在上面的代码中,我们继承了TextWebSocketHandler类,并在afterConnectionEstablished方法中使用RabbitTemplate将“connected”消息发送到名为notification.exchange的交换机中,使用用户ID作为路由键。
最后,我们需要在前端页面中创建一个WebSocket连接,并订阅通知:
```javascript
const webSocket = new WebSocket("ws://localhost:8080/notifications");
webSocket.onopen = function() {
// send user ID to server
const userId = getUserId();
webSocket.send(userId);
};
webSocket.onmessage = function(event) {
// handle notification
console.log(event.data);
};
function getUserId() {
// get user ID from session
return null;
}
```
在上面的代码中,我们使用WebSocket连接到路径“/notifications”,并在连接成功后将用户ID发送到服务器。当服务器有新的通知时,WebSocket会接收到onmessage事件,并在其中处理通知。
这是一个简单的示例,说明如何使用Spring Boot和RabbitMQ发送通知到前端。当然,在实际应用程序中可能需要更复杂的逻辑和处理。
阅读全文