websocket协议是用于前后端长链接交互的技术,此技术多用于交互不断的场景,比如说类似于微信。QQ两者或者多者之间的交互;
websocket的前端四个注解 对应于后端的四个注解方法,对应触发时间就会对应后端接收消息,
@OnOpen,链接成功交互初始化
@OnClose,关闭事件
@OnMessage,消息事件
@OnError 异常事件
本文只是简单的记录一下项目中用到的场景,并没有详细记录,需要了解的可以查看官方文档;
配置包:
![](/i/l/?n=22&i=blog/1108133/202210/1108133-20221026115053764-197511268.png)
websocket java类:
@Component @Slf4j @ServerEndpoint("/websocket/{coreCode}/{employeeCode}")//请求案例:// 接口路径 ws://localhost:8087/webSocket/userCode; public class WebsocketService { private Session session;//与某个客户端的连接会话,需要通过它来给客户端发送数据 //concurrent包的线程安全Set,用来存放每个客户端对应的MyWebSocket对象。 //虽然@Component默认是单例模式的,但springboot还是会为每个websocket连接初始化一个bean,所以可以用一个静态set保存起来。 // 注:底下WebSocket是当前类名 private static CopyOnWriteArraySet<WebsocketService> webSockets = new CopyOnWriteArraySet<>(); // 用来存在线连接数 private static Map<String, Session> sessionPool = new HashMap<String, Session>(); // 用来存储当前用户登录的归属核企 private static Map<String, String> coreMap = new HashMap<String, String>(); private String employeeCode; //每次链接访问的时候,都是一个新的请求,所以这个变量是不会覆盖的 /** * 链接成功调用的方法 */ @OnOpen public void onOpen(Session session, @PathParam(value = "coreCode") String coreCode, @PathParam(value = "employeeCode") String employeeCode) { try { this.session = session; this.employeeCode = employeeCode; webSockets.add(this); sessionPool.put(employeeCode, session); if (coreMap.containsKey((employeeCode))) { coreMap.remove(employeeCode); //加入map } coreMap.put(employeeCode, coreCode); log.info("【websocket消息】有新的连接,总数为:" + webSockets.size()); } catch (Exception e) { log.error("websocket链接失败:", e.getMessage()); } } /** * 链接关闭调用的方法
*此方法关闭事件,不管是关闭浏览器还是直接退出或者超时回话都会触发这个接口
*/ @OnClose public void onClose() { try { webSockets.remove(this); if (coreMap.containsKey(employeeCode)) { coreMap.remove(employeeCode); } if(sessionPool.containsKey(employeeCode)){ sessionPool.remove(employeeCode); } log.info("【websocket消息】连接断开,总数为:" + webSockets.size()); } catch (Exception e) { log.error("websocket关闭失败:", e.getMessage()); } } /** * 收到客户端消息后调用的方法 * * @param message */ @OnMessage public void onMessage(String message) { log.info("【websocket消息】收到客户端消息:" + message); } /** * 发送错误时的处理 * * @param session * @param error */ @OnError public void one rror(Session session, Throwable error) { log.error("用户错误,原因:" + error.getMessage()); error.printStackTrace(); } /** * 全部人员消息 * * @param message */ public void sendAllMessage(String message) { log.info("【websocket消息】全部人员消息:" + message); for (WebsocketService webSocket : webSockets) { try { if (webSocket.session.isOpen()) { webSocket.session.getAsyncRemote().sendText(message); } } catch (Exception e) { e.printStackTrace(); log.error("全部人员发送消息失败:", e.getMessage()); } } } /** * 点对点发送指定人消息 * * @param employeeCode * @param message */ public void sendOneMessage(String employeeCode, String message) { Session session = sessionPool.get(employeeCode); if (session != null && session.isOpen()) { try { log.info("【websocket消息】 点对点发送指定人消息:" + message); session.getAsyncRemote().sendText(message); } catch (Exception e) { e.printStackTrace(); log.error("点对点发送指定人消息:", e.getMessage()); } } } /** * 发送指定多个人员消息 * * @param employeeCodes * @param message */ public void sendMoreMessage(String[] employeeCodes, String message) { for (String employeeCode : employeeCodes) { Session session = sessionPool.get(employeeCode); if (session != null && session.isOpen()) { try { log.info("【websocket消息】 发送指定多个人员消息:" + message); session.getAsyncRemote().sendText(message); } catch (Exception e) { e.printStackTrace(); log.error("发送指定多个人员消息:", e.getMessage()); } } } } /** * 点对点发送指定场景人员编码消息 * * @param centres * @param configTemplates */ public void sendScenetOneMessage(List<InformationCentre> centres, InformationConfigTemplate configTemplates, String code) { log.info("当前存储的用户编码和核企的关系集合:{},session集合:{}",coreMap,sessionPool); centres.stream().forEach(item -> { String value = coreMap.get(item.getEmployeeCode()); if(StrUtil.isNotBlank(value)){ //判断是否是全部核企还是部分核企,如果是部分核企,当前登录人员所属核企消息发送对象是否包含 if ("PART".equals(configTemplates.getScope()) && StrUtil.isNotBlank(configTemplates.getCoreStrand())) { List<InformationCofigCore> informationCofigCores = JSONUtil.toList(configTemplates.getCoreStrand(), InformationCofigCore.class); List<String> coreList = informationCofigCores.stream().map(InformationCofigCore::getCoreCode).collect(Collectors.toList()); String coreCodeMap = coreMap.get(item.getEmployeeCode()); if (!coreList.contains(coreCodeMap)) { return; } } if (InformationEnum.INTERNALINFORMATIONCHANNEL.code.equals(code)) { //发送给当前登录人消息 Session session = sessionPool.get(item.getEmployeeCode()); if (session != null && session.isOpen()) { try { log.info("【websocket消息】 点对点发送指定人消息:" + item); session.getAsyncRemote().sendText(JSONUtil.toJsonStr(item)); } catch (Exception e) { e.printStackTrace(); log.error("点对点发送指定人消息:", e.getMessage()); } } } } }); } }
标签:websocket,log,前后,session,message,employeeCode,交互,String From: https://www.cnblogs.com/Generalwake/p/16827810.html