From ff968e2cfb132eb86c70e94e99b9937a8a7b15e9 Mon Sep 17 00:00:00 2001
From: xieb <vip_xiaobin810@163.com>
Date: Mon, 08 Apr 2024 13:58:56 +0800
Subject: [PATCH] 直播
---
src/main/java/com/dji/sample/component/websocket/service/impl/SendMessageServiceImpl.java | 48 +++++++++++++++++++++++++++++++++++++++---------
1 files changed, 39 insertions(+), 9 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 8ee6131..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
@@ -3,13 +3,17 @@
import com.dji.sample.component.websocket.config.ConcurrentWebSocketSession;
import com.dji.sample.component.websocket.model.CustomWebSocketMessage;
import com.dji.sample.component.websocket.service.ISendMessageService;
+import com.dji.sample.component.websocket.service.IWebSocketManageService;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
+import org.springframework.util.StringUtils;
import org.springframework.web.socket.TextMessage;
import java.io.IOException;
import java.util.Collection;
+import java.util.Objects;
/**
* @author sean.zhou
@@ -19,6 +23,12 @@
@Service
@Slf4j
public class SendMessageServiceImpl implements ISendMessageService {
+
+ @Autowired
+ private ObjectMapper mapper;
+
+ @Autowired
+ private IWebSocketManageService webSocketManageService;
@Override
public void sendMessage(ConcurrentWebSocketSession session, CustomWebSocketMessage message) {
@@ -33,7 +43,6 @@
return;
}
- ObjectMapper mapper = new ObjectMapper();
session.sendMessage(new TextMessage(mapper.writeValueAsBytes(message)));
} catch (IOException e) {
@@ -50,17 +59,17 @@
try {
- ObjectMapper mapper = new ObjectMapper();
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) {
@@ -69,4 +78,25 @@
e.printStackTrace();
}
}
-}
\ No newline at end of file
+
+ @Override
+ public void sendBatch(String workspaceId, Integer userType, String bizCode, Object data) {
+ if (!StringUtils.hasText(workspaceId)) {
+ throw new RuntimeException("项目id不存在："+workspaceId);
+ }
+ Collection<ConcurrentWebSocketSession> sessions = Objects.isNull(userType) ?
+ webSocketManageService.getValueWithWorkspace(workspaceId) :
+ webSocketManageService.getValueWithWorkspaceAndUserType(workspaceId, userType);
+
+ this.sendBatch(sessions, CustomWebSocketMessage.builder()
+ .data(data)
+ .timestamp(System.currentTimeMillis())
+ .bizCode(bizCode)
+ .build());
+ }
+
+ @Override
+ public void sendBatch(String workspaceId, String bizCode, Object data) {
+ this.sendBatch(workspaceId, null, bizCode, data);
+ }
+}
--
Gitblit v1.9.3