WebSocket实现系统后台消息实时通知功能

import javax.websocket.OnClose;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;
import java.io IOException;
import java.util.concurrent.CopyOnWriteArrayList;

@ServerEndpoint("/websocket")
public class WebSocketServer {

    private static CopyOnWriteArrayList<Session> sessions = new CopyOnWriteArrayList<>();

    @OnOpen
    public void onOpen(Session session) {
        sessions.add(session);
    }

    @OnMessage
    public void onMessage(String message, Session session) {
        // 处理收到的消息
    }

    @OnClose
    public void onClose(Session session) {
        sessions.remove(session);
    }

    public static void broadcastMessage(String message) {
        for (Session session : sessions) {
            try {
                session.getBasicRemote().sendText(message);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.messaging.simp.SimpMessagingTemplate;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class MessageController {

    @Autowired
    private SimpMessagingTemplate messagingTemplate;

    @PostMapping("/send-message")
    public void sendMessageToUser(@RequestBody MessageDto message) {
        // 在此处，您可以执行任何逻辑以准备要发送的消息内容
        String messageContent = message.getContent();

        // 向指定用户发送消息
        messagingTemplate.convertAndSendToUser(message.getRecipient(), "/queue/messages", messageContent);
    }
}

public class MessageDto {
    private String recipient;
    private String content;

    // Getters and setters
}

const socket = new WebSocket("ws://example.com/websocket");

socket.onopen = () => {
    console.log("WebSocket连接已建立");
};

socket.onmessage = (event) => {
    const message = event.data;
    // 使用弹窗通知用户
    showNotification(message);
};

socket.onclose = () => {
    console.log("WebSocket连接已关闭");
};

function showNotification(message) {
    // 使用浏览器的通知API或自定义弹窗组件来通知用户
    // 以下是一个简单的示例，使用浏览器通知API
    if ("Notification" in window) {
        if (Notification.permission === "granted") {
            new Notification("新消息", { body: message });
        } else if (Notification.permission !== "denied") {
            Notification.requestPermission().then((permission) => {
                if (permission === "granted") {
                    new Notification("新消息", { body: message });
                }
            });
        }
    }
}