From 6322f4c190dc9b41572e59e671dd3ecc005253b9 Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Wed, 27 Mar 2024 10:03:04 +0800
Subject: [PATCH] 图斑
---
src/main/java/com/dji/sample/map/controller/WorkspaceElementController.java | 42 +++++++++---------------------------------
1 files changed, 9 insertions(+), 33 deletions(-)
diff --git a/src/main/java/com/dji/sample/map/controller/WorkspaceElementController.java b/src/main/java/com/dji/sample/map/controller/WorkspaceElementController.java
index 1ad52b8..1970479 100644
--- a/src/main/java/com/dji/sample/map/controller/WorkspaceElementController.java
+++ b/src/main/java/com/dji/sample/map/controller/WorkspaceElementController.java
@@ -3,8 +3,6 @@
import com.dji.sample.common.model.CustomClaim;
import com.dji.sample.common.model.ResponseResult;
import com.dji.sample.component.websocket.model.BizCodeEnum;
-import com.dji.sample.component.websocket.model.CustomWebSocketMessage;
-import com.dji.sample.component.websocket.model.WebSocketManager;
import com.dji.sample.component.websocket.service.ISendMessageService;
import com.dji.sample.map.model.dto.*;
import com.dji.sample.map.service.IWorkspaceElementService;
@@ -76,14 +74,8 @@
// Notify all WebSocket connections in this workspace to be updated when an element is created.
elementService.getElementByElementId(elementCreate.getId())
- .ifPresent(groupElement ->
- sendMessageService.sendBatch(
- WebSocketManager.getValueWithWorkspace(workspaceId),
- CustomWebSocketMessage.<GroupElementDTO>builder()
- .timestamp(System.currentTimeMillis())
- .bizCode(BizCodeEnum.MAP_ELEMENT_CREATE.getCode())
- .data(groupElement)
- .build()));
+ .ifPresent(groupElement -> sendMessageService.sendBatch(
+ workspaceId, BizCodeEnum.MAP_ELEMENT_CREATE.getCode(), groupElement));
return ResponseResult.success(new ConcurrentHashMap<>(Map.of("id", elementCreate.getId())));
}
@@ -112,14 +104,8 @@
// Notify all WebSocket connections in this workspace to update when there is an element update.
elementService.getElementByElementId(elementId)
- .ifPresent(groupElement ->
- sendMessageService.sendBatch(
- WebSocketManager.getValueWithWorkspace(workspaceId),
- CustomWebSocketMessage.<GroupElementDTO>builder()
- .timestamp(System.currentTimeMillis())
- .bizCode(BizCodeEnum.MAP_ELEMENT_UPDATE.getCode())
- .data(groupElement)
- .build()));
+ .ifPresent(groupElement -> sendMessageService.sendBatch(
+ workspaceId, BizCodeEnum.MAP_ELEMENT_UPDATE.getCode(), groupElement));
return response;
}
@@ -141,16 +127,11 @@
// Notify all WebSocket connections in this workspace to update when an element is deleted.
if (ResponseResult.CODE_SUCCESS == response.getCode()) {
elementOpt.ifPresent(element ->
- sendMessageService.sendBatch(
- WebSocketManager.getValueWithWorkspace(workspaceId),
- CustomWebSocketMessage.<WebSocketElementDelDTO>builder()
- .timestamp(System.currentTimeMillis())
- .bizCode(BizCodeEnum.MAP_ELEMENT_DELETE.getCode())
- .data(WebSocketElementDelDTO.builder()
+ sendMessageService.sendBatch(workspaceId, BizCodeEnum.MAP_ELEMENT_DELETE.getCode(),
+ WebSocketElementDelDTO.builder()
.elementId(elementId)
.groupId(element.getGroupId())
- .build())
- .build()));
+ .build()));
}
return response;
}
@@ -170,14 +151,9 @@
// Notify all WebSocket connections in this workspace to update when elements are deleted.
if (ResponseResult.CODE_SUCCESS == response.getCode()) {
- sendMessageService.sendBatch(
- WebSocketManager.getValueWithWorkspace(workspaceId),
- CustomWebSocketMessage.builder()
- .timestamp(System.currentTimeMillis())
- .bizCode(BizCodeEnum.MAP_GROUP_REFRESH.getCode())
+ sendMessageService.sendBatch(workspaceId, BizCodeEnum.MAP_GROUP_REFRESH.getCode(),
// Group ids that need to re-request data
- .data(new ConcurrentHashMap<String, String[]>(Map.of("ids", new String[]{groupId})))
- .build());
+ new ConcurrentHashMap<String, String[]>(Map.of("ids", new String[]{groupId})));
}
return response;
}
--
Gitblit v1.9.3