Spring Boot MVC应用中如何保留并复用STOMP会话？

对于多页面MVC应用每次打开页面都重复创建STOMP连接的问题，有几个实用方案可以实现连接复用：

一、使用浏览器共享Worker（推荐） #

共享Worker是浏览器提供的跨页面共享进程能力，同源下的所有页面都能和同一个Worker实例通信。我们可以把STOMP连接逻辑放在共享Worker中，所有页面通过Worker收发通知，这样整个浏览器会话里只会存在一个STOMP连接。

实现步骤： #

1. 创建共享Worker文件（notification-worker.js）：

let stompClient = null;
let retryNotificationListener = true;
const connectedPages = new Set();

// 初始化STOMP连接
function initStompConnection() {
    const importSocket = new SockJS("/stomp/user");
    stompClient = Stomp.over(importSocket);

    stompClient.connect({}, (frame) => {
        console.log('Worker Connected to ' + frame);
        // 订阅通知主题
        stompClient.subscribe('/user/topic/notifications', (message) => {
            // 把消息推送给所有连接的页面
            connectedPages.forEach(port => port.postMessage(JSON.parse(message.body)));
        });
    }, (message) => {
        console.log('Worker connection error:', message);
        if (message instanceof Object) {
            const body = JSON.parse(message.body);
            if (body.statusCode === 401) retryNotificationListener = false;
        } else if (message.includes('Whoops! Lost connection to') && retryNotificationListener) {
            setTimeout(() => initStompConnection(), 15000);
        }
    });
}

// 监听页面连接
self.addEventListener('connect', (e) => {
    const port = e.ports[0];
    connectedPages.add(port);

    // 若未创建连接则初始化
    if (!stompClient) {
        initStompConnection();
    }

    // 接收页面发来的指令（比如订阅额外主题）
    port.addEventListener('message', (msg) => {
        if (msg.data.type === 'subscribe') {
            stompClient.subscribe(msg.data.topic, (message) => {
                port.postMessage(JSON.parse(message.body));
            });
        }
    });

    port.start();

    // 页面关闭时移除对应连接
    port.addEventListener('close', () => {
        connectedPages.delete(port);
        // 无页面连接时断开STOMP连接
        if (connectedPages.size === 0 && stompClient) {
            stompClient.disconnect();
            stompClient = null;
        }
    });
});

// 初始化连接
initStompConnection();

2. 页面端修改连接逻辑：

// 页面中连接共享Worker
function startListening() {
    if (!window.SharedWorker) {
        console.error('浏览器不支持SharedWorker');
        // 降级到原有连接逻辑
        originalStartListening();
        return;
    }

    const worker = new SharedWorker('notification-worker.js');
    worker.port.start();

    // 接收Worker推送的通知
    worker.port.addEventListener('message', (msg) => {
        // 复用原有通知处理逻辑
        handleNotification(msg.data);
    });

    // 页面卸载时关闭端口
    window.addEventListener('beforeunload', () => {
        worker.port.close();
    });
}

// 保留原有逻辑作为降级方案
function originalStartListening() {
    const importSocket = new SockJS("/stomp/user");
    const notificationStompClient = Stomp.over(importSocket);

    notificationStompClient.connect({}, (frame) => {
        console.log('Connected to ' + frame);
        subscribeToNotifications(notificationStompClient);
    }, (message) => {
        console.log(message);
        if (message instanceof Object) {
            const body = JSON.parse(message.body);
            if (body.statusCode ===  401) retryNotificationListener = false;
        } else if (message.includes('Whoops! Lost connection to') && retryNotificationListener)
            setTimeout(() => originalStartListening(), 15000);
    });
}

// 原有的通知处理函数
function handleNotification(notification) {
    // 替换成你实际的通知处理逻辑
    console.log('收到通知:', notification);
}

二、利用会话标识复用连接（依赖服务器支持） #

如果你的ActiveMQ Artemis服务器支持STOMP会话复用（比如通过持久化会话ID），可以尝试以下方式：

• 首次连接成功后，把STOMP会话ID存储到sessionStorage中
• 新页面加载时，先读取sessionStorage里的会话ID，尝试用这个ID重新连接
• 注意：这种方式需要服务器配置允许会话复用，否则可能无法成功，可靠性不如共享Worker

示例代码片段：

let retryNotificationListener = true;

function startListening() {
    const sessionId = sessionStorage.getItem('stomp-session-id');
    const importSocket = new SockJS("/stomp/user");
    const notificationStompClient = Stomp.over(importSocket);

    // 尝试复用会话ID
    const connectHeaders = sessionId ? {'session-id': sessionId} : {};

    notificationStompClient.connect(connectHeaders, (frame) => {
        console.log('Connected to ' + frame);
        // 存储服务器返回的新会话ID
        const newSessionId = frame.headers['session-id'];
        if (newSessionId) {
            sessionStorage.setItem('stomp-session-id', newSessionId);
        }
        subscribeToNotifications(notificationStompClient);
    }, (message) => {
        console.log(message);
        if (message instanceof Object) {
            const body = JSON.parse(message.body);
            if (body.statusCode === 401) {
                retryNotificationListener = false;
                sessionStorage.removeItem('stomp-session-id');
            }
        } else if (message.includes('Whoops! Lost connection to') && retryNotificationListener) {
            setTimeout(() => startListening(), 15000);
        }
    });
}

三、迁移为单页应用（SPA） #

如果业务允许，把多页面应用改成SPA（比如用Vue/React结合Spring Boot），SPA只有一个页面实例，自然只会创建一个STOMP连接，从根源上解决问题。不过这个方案需要重构前端代码，成本较高。

内容的提问来源于stack exchange，提问作者Alain Cruz