From fa7479cd93d2b87c3a62a7c954c1833df34f3f0c Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Sat, 03 Aug 2024 17:50:00 +0800
Subject: [PATCH] 国土调查注册对接,完成注册接口,完成SM2和SM4加密解密方法

---
 src/main/java/com/dji/sample/component/websocket/service/impl/SendMessageServiceImpl.java |   14 ++++++++------
 1 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/main/java/com/dji/sample/component/websocket/service/impl/SendMessageServiceImpl.java b/src/main/java/com/dji/sample/component/websocket/service/impl/SendMessageServiceImpl.java
index a54b97f..727df3a 100644
--- a/src/main/java/com/dji/sample/component/websocket/service/impl/SendMessageServiceImpl.java
+++ b/src/main/java/com/dji/sample/component/websocket/service/impl/SendMessageServiceImpl.java
@@ -62,12 +62,14 @@
             TextMessage data = new TextMessage(mapper.writeValueAsBytes(message));
 
             for (ConcurrentWebSocketSession session : sessions) {
-                if (!session.isOpen()) {
-                    session.close();
-                    log.debug("This session is closed.");
-                    return;
+                if (!Objects.isNull(session)) {
+                    if (!session.isOpen()) {
+                        session.close();
+                        log.debug("This session is closed.");
+                        return;
+                    }
+                    session.sendMessage(data);
                 }
-                session.sendMessage(data);
             }
 
         } catch (IOException e) {
@@ -80,7 +82,7 @@
     @Override
     public void sendBatch(String workspaceId, Integer userType, String bizCode, Object data) {
         if (!StringUtils.hasText(workspaceId)) {
-            throw new RuntimeException("项目id不存在;Workspace ID does not exist.");
+            throw new RuntimeException("项目id不存在:"+workspaceId);
         }
         Collection<ConcurrentWebSocketSession> sessions = Objects.isNull(userType) ?
                 webSocketManageService.getValueWithWorkspace(workspaceId) :

--
Gitblit v1.9.3