From 923a5e9fdfcfcc765606bce6fa7f99cb3717dcbe Mon Sep 17 00:00:00 2001
From: guoshilong <123456>
Date: Wed, 15 Nov 2023 14:05:06 +0800
Subject: [PATCH] 图斑飞行
---
src/main/java/com/dji/sample/component/websocket/service/impl/SendMessageServiceImpl.java | 16 +++++++++-------
1 files changed, 9 insertions(+), 7 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 b605cf9..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("Workspace ID does not exist.");
+ throw new RuntimeException("项目id不存在："+workspaceId);
}
Collection<ConcurrentWebSocketSession> sessions = Objects.isNull(userType) ?
webSocketManageService.getValueWithWorkspace(workspaceId) :
@@ -97,4 +99,4 @@
public void sendBatch(String workspaceId, String bizCode, Object data) {
this.sendBatch(workspaceId, null, bizCode, data);
}
-}
\ No newline at end of file
+}
--
Gitblit v1.9.3