Pig Cloud遇到websocket不能实现同一个用户不同浏览器接受到广播的消息解决方案

自定义SecuritySessionKeyGenerator类,为每个客户端连接建立唯一的key

java 复制代码
package com.pig4cloud.plugin.websocket.custom;

import com.pig4cloud.plugin.websocket.holder.SessionKeyGenerator;
import org.springframework.web.socket.WebSocketSession;

import java.util.UUID;
public class SecuritySessionKeyGenerator implements SessionKeyGenerator {
	@Override
	public Object sessionKey(WebSocketSession webSocketSession) {
		Object user = webSocketSession.getAttributes().get("USER_KEY_ATTR_NAME");
		// 添加随机后缀使每个连接键唯一
		return user != null ? user + "-" + UUID.randomUUID().toString() : null;
	}
}

重写WebSocketSessionHolder类方法

java 复制代码
package com.pig4cloud.plugin.websocket.holder;

import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CopyOnWriteArraySet;

import org.springframework.web.socket.WebSocketSession;

public final class WebSocketSessionHolder {
	// 仅修改这一行:将Value改为Set类型
	private static final Map<String, Set<WebSocketSession>> USER_SESSION_MAP = new ConcurrentHashMap<>();

	private WebSocketSessionHolder() {
	}

	public static void addSession(Object sessionKey, WebSocketSession session) {
		// 修改为支持多会话的添加方式
		USER_SESSION_MAP.computeIfAbsent(sessionKey.toString(),
				k -> new CopyOnWriteArraySet<>()).add(session);
	}

	public static void removeSession(Object sessionKey) {
		// 移除时不再自动清除所有会话
		Set<WebSocketSession> sessions = USER_SESSION_MAP.get(sessionKey.toString());
		if (sessions != null) {
			sessions.removeIf(s -> !s.isOpen()); // 只移除已关闭的连接
			if (sessions.isEmpty()) {
				USER_SESSION_MAP.remove(sessionKey.toString());
			}
		}
	}

	// 保持原有方法签名不变(兼容现有调用)
	public static WebSocketSession getSession(Object sessionKey) {
		Set<WebSocketSession> sessions = USER_SESSION_MAP.get(sessionKey.toString());
		return sessions != null ? sessions.stream().findFirst().orElse(null) : null;
	}

	// 新增方法:获取用户的所有会话
	public static Set<WebSocketSession> getSessions(Object sessionKey) {
		return USER_SESSION_MAP.getOrDefault(sessionKey.toString(), Collections.emptySet());
	}

	// 保持原有方法不变
	public static Collection<WebSocketSession> getSessions() {
		List<WebSocketSession> allSessions = new ArrayList<>();
		USER_SESSION_MAP.values().forEach(allSessions::addAll);
		return allSessions;
	}

	// 保持原有方法不变
	public static Set<String> getSessionKeys() {
		return USER_SESSION_MAP.keySet();
	}
}
相关推荐
你好,赵志伟2 小时前
传输层协议 TCP
网络·网络协议·tcp/ip
茯苓gao2 小时前
CAN总线学习(四)错误处理 STM32CAN外设一
网络·笔记·stm32·单片机·学习
从零开始的ops生活2 小时前
【Day 68】Zabbix-自动监控-Web检测-分布式监控
linux·网络·zabbix
遗憾随她而去.4 小时前
Uni-App 页面跳转监控实战:快速定位路由问题
前端·网络·uni-app
小胖西瓜4 小时前
[解决方案] 回顾一下业务中的网络技术演化
网络
计算机毕业设计指导4 小时前
基于Django的内部网络资产发现与管理工具
网络·python·网络安全·django
Gss7775 小时前
Nginx 核心安全配置总结
网络·nginx·安全
DokiDoki之父6 小时前
web核心—HTTP
前端·网络协议·http
汪汪大队u6 小时前
IS-IS核心解析:驱动现代网络的隐形力量
网络·智能路由器