From e60ea14897e4512efe505b8b3aea7cb2e49df39f Mon Sep 17 00:00:00 2001
From: xieb <vip_xiaobin810@163.com>
Date: Tue, 17 Oct 2023 18:10:31 +0800
Subject: [PATCH] 用户操作日志
---
src/main/java/com/dji/sample/media/service/impl/MediaServiceImpl.java | 133 +++++++++++++++++++++++---------------------
1 files changed, 70 insertions(+), 63 deletions(-)
diff --git a/src/main/java/com/dji/sample/media/service/impl/MediaServiceImpl.java b/src/main/java/com/dji/sample/media/service/impl/MediaServiceImpl.java
index d14b3de..77dd969 100644
--- a/src/main/java/com/dji/sample/media/service/impl/MediaServiceImpl.java
+++ b/src/main/java/com/dji/sample/media/service/impl/MediaServiceImpl.java
@@ -1,17 +1,17 @@
package com.dji.sample.media.service.impl;
-import com.dji.sample.common.error.CommonErrorEnum;
import com.dji.sample.common.model.ResponseResult;
-import com.dji.sample.component.mqtt.model.*;
+import com.dji.sample.component.mqtt.model.ChannelName;
+import com.dji.sample.component.mqtt.model.CommonTopicReceiver;
+import com.dji.sample.component.mqtt.model.MapKeyConst;
import com.dji.sample.component.mqtt.service.IMessageSenderService;
import com.dji.sample.component.redis.RedisConst;
import com.dji.sample.component.redis.RedisOpsUtils;
import com.dji.sample.component.websocket.model.BizCodeEnum;
-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.dji.sample.manage.model.dto.DeviceDTO;
import com.dji.sample.manage.model.enums.UserTypeEnum;
+import com.dji.sample.manage.service.IDeviceRedisService;
import com.dji.sample.manage.service.IDeviceService;
import com.dji.sample.media.model.FileUploadCallback;
import com.dji.sample.media.model.FileUploadDTO;
@@ -22,9 +22,9 @@
import com.dji.sample.wayline.model.dto.WaylineJobDTO;
import com.dji.sample.wayline.service.IWaylineJobService;
import com.fasterxml.jackson.databind.ObjectMapper;
+import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.integration.annotation.ServiceActivator;
-import org.springframework.integration.mqtt.support.MqttHeaders;
import org.springframework.messaging.MessageHeaders;
import org.springframework.stereotype.Service;
@@ -40,6 +40,7 @@
* @date 2021/12/9
*/
@Service
+@Slf4j
public class MediaServiceImpl implements IMediaService {
@Autowired
@@ -61,7 +62,7 @@
private ISendMessageService sendMessageService;
@Autowired
- private IWebSocketManageService webSocketManageService;
+ private IDeviceRedisService deviceRedisService;
@Override
public Boolean fastUpload(String workspaceId, String fingerprint) {
@@ -91,68 +92,75 @@
}
- @Override
- @ServiceActivator(inputChannel = ChannelName.INBOUND_EVENTS_FILE_UPLOAD_CALLBACK, outputChannel = ChannelName.OUTBOUND)
- public void handleFileUploadCallBack(CommonTopicReceiver receiver) {
+ /**
+ * Handle media files messages reported by dock.
+ * 处理由dock报告的媒体文件消息。
+ * @param receiver
+ * @return
+ */
+ @ServiceActivator(inputChannel = ChannelName.INBOUND_EVENTS_FILE_UPLOAD_CALLBACK, outputChannel = ChannelName.OUTBOUND_EVENTS)
+ public CommonTopicReceiver handleFileUploadCallBack(CommonTopicReceiver receiver) {
FileUploadCallback callback = objectMapper.convertValue(receiver.getData(), FileUploadCallback.class);
- String topic = TopicConst.THING_MODEL_PRE + TopicConst.PRODUCT + receiver.getGateway()
- + TopicConst.EVENTS_SUF + TopicConst._REPLY_SUF;
- CommonTopicResponse<RequestsReply> data = CommonTopicResponse.<RequestsReply>builder()
- .timestamp(System.currentTimeMillis())
- .method(EventsMethodEnum.FILE_UPLOAD_CALLBACK.getMethod())
- .data(RequestsReply.success())
- .tid(receiver.getTid())
- .bid(receiver.getBid())
- .build();
-
if (callback.getResult() != ResponseResult.CODE_SUCCESS) {
- messageSenderService.publish(topic, data);
- return;
+ log.error("媒体文件上传失败；Media file upload failed!");
+ return null;
}
String jobId = callback.getFile().getExt().getFlightId();
+ Optional<DeviceDTO> deviceOpt = deviceRedisService.getDeviceOnline(receiver.getGateway());
MediaFileCountDTO mediaFileCount = (MediaFileCountDTO) RedisOpsUtils.hashGet(RedisConst.MEDIA_FILE_PREFIX + receiver.getGateway(), jobId);
// duplicate data
- if (receiver.getBid().equals(mediaFileCount.getBid()) && receiver.getTid().equals(mediaFileCount.getTid())) {
- System.out.println("相同" + receiver.getBid() + "\t tid:" + receiver.getTid());
- messageSenderService.publish(topic, data);
- return;
+ if (deviceOpt.isEmpty()
+ || (Objects.nonNull(mediaFileCount) && receiver.getBid().equals(mediaFileCount.getBid())
+ && receiver.getTid().equals(mediaFileCount.getTid()))) {
+ return receiver;
}
- Optional<WaylineJobDTO> jobOpt = waylineJobService.getJobByJobId(jobId);
+
+ DeviceDTO device = deviceOpt.get();
+ Optional<WaylineJobDTO> jobOpt = waylineJobService.getJobByJobId(device.getWorkspaceId(), jobId);
if (jobOpt.isPresent()) {
boolean isSave = parseMediaFile(callback, jobOpt.get());
if (!isSave) {
- data.setData(RequestsReply.error(CommonErrorEnum.ILLEGAL_ARGUMENT));
+ log.error("保存文件到数据库失败，请手动检查数据；Failed to save the file to the database, please check the data manually.");
+ return null;
}
}
- messageSenderService.publish(topic, data);
-
- notifyUploadedCount(mediaFileCount, receiver, jobId);
+ notifyUploadedCount(mediaFileCount, receiver, jobId, device);
+ return receiver;
}
/**
* update the uploaded count and notify web side
+ * 更新上传的计数并通知web端
* @param mediaFileCount
* @param receiver
* @param jobId
*/
- private void notifyUploadedCount(MediaFileCountDTO mediaFileCount, CommonTopicReceiver receiver, String jobId) {
+ private void notifyUploadedCount(MediaFileCountDTO mediaFileCount, CommonTopicReceiver receiver, String jobId, DeviceDTO dock) {
+ // Do not notify when files that do not belong to the route are uploaded.
+ //上传不属于该路线的文件时不进行通知。
+ if (Objects.isNull(mediaFileCount)) {
+ return;
+ }
mediaFileCount.setBid(receiver.getBid());
mediaFileCount.setTid(receiver.getTid());
mediaFileCount.setUploadedCount(mediaFileCount.getUploadedCount() + 1);
String key = RedisConst.MEDIA_FILE_PREFIX + receiver.getGateway();
// After all the files of the job are uploaded, delete the media file key.
+ //待作业的所有文件上传完成后，删除媒体文件密钥。
if (mediaFileCount.getUploadedCount() >= mediaFileCount.getMediaCount()) {
RedisOpsUtils.hashDel(key, new String[]{jobId});
// After uploading, delete the key with the highest priority.
+ //上传完成后，删除优先级最高的密钥。
String highestKey = RedisConst.MEDIA_HIGHEST_PRIORITY_PREFIX + receiver.getGateway();
- if (jobId.equals(Objects.requireNonNullElse(RedisOpsUtils.get(highestKey), ""))) {
+ if (RedisOpsUtils.checkExist(highestKey) &&
+ jobId.equals(((MediaFileCountDTO) RedisOpsUtils.get(highestKey)).getJobId())) {
RedisOpsUtils.del(highestKey);
}
@@ -163,13 +171,9 @@
RedisOpsUtils.hashSet(key, jobId, mediaFileCount);
}
- DeviceDTO dock = (DeviceDTO) RedisOpsUtils.get(RedisConst.DEVICE_ONLINE_PREFIX + receiver.getGateway());
- sendMessageService.sendBatch(webSocketManageService.getValueWithWorkspaceAndUserType(dock.getWorkspaceId(), UserTypeEnum.WEB.getVal()),
- CustomWebSocketMessage.builder()
- .bizCode(BizCodeEnum.FILE_UPLOAD_CALLBACK.getCode())
- .timestamp(System.currentTimeMillis())
- .data(mediaFileCount)
- .build());
+ //通过websocket把数据发送给web
+ sendMessageService.sendBatch(dock.getWorkspaceId(), UserTypeEnum.WEB.getVal(),
+ BizCodeEnum.FILE_UPLOAD_CALLBACK.getCode(), mediaFileCount);
}
private Boolean parseMediaFile(FileUploadCallback callback, WaylineJobDTO job) {
@@ -184,40 +188,43 @@
return fileService.saveFile(job.getWorkspaceId(), callback.getFile()) > 0;
}
- @Override
- @ServiceActivator(inputChannel = ChannelName.INBOUND_EVENTS_HIGHEST_PRIORITY_UPLOAD_FLIGHT_TASK_MEDIA, outputChannel = ChannelName.OUTBOUND)
- public void handleHighestPriorityUploadFlightTaskMedia(CommonTopicReceiver receiver, MessageHeaders headers) {
+ /**
+ * Handles the highest priority message about media uploads.
+ * @param receiver
+ * @param headers
+ * @return
+ */
+ @ServiceActivator(inputChannel = ChannelName.INBOUND_EVENTS_HIGHEST_PRIORITY_UPLOAD_FLIGHT_TASK_MEDIA, outputChannel = ChannelName.OUTBOUND_EVENTS)
+ public CommonTopicReceiver handleHighestPriorityUploadFlightTaskMedia(CommonTopicReceiver receiver, MessageHeaders headers) {
Map map = objectMapper.convertValue(receiver.getData(), Map.class);
if (map.isEmpty() || !map.containsKey(MapKeyConst.FLIGHT_ID)) {
- return;
+ return null;
}
String dockSn = receiver.getGateway();
String jobId = map.get(MapKeyConst.FLIGHT_ID).toString();
String key = RedisConst.MEDIA_HIGHEST_PRIORITY_PREFIX + dockSn;
- Object preJobId = RedisOpsUtils.get(key);
+ MediaFileCountDTO countDTO = new MediaFileCountDTO();
+ if (RedisOpsUtils.checkExist(key)) {
+ countDTO = (MediaFileCountDTO) RedisOpsUtils.get(key);
+ if (jobId.equals(countDTO.getJobId())) {
+ RedisOpsUtils.setWithExpire(key, countDTO, RedisConst.DEVICE_ALIVE_SECOND * 5);
+ return null;
+ }
- RedisOpsUtils.setWithExpire(key, jobId,
- RedisConst.DEVICE_ALIVE_SECOND * 5);
+ countDTO.setPreJobId(countDTO.getJobId());
+ }
+ countDTO.setJobId(jobId);
- DeviceDTO dock = (DeviceDTO) RedisOpsUtils.get(RedisConst.DEVICE_ONLINE_PREFIX + dockSn);
+ RedisOpsUtils.setWithExpire(key, countDTO, RedisConst.DEVICE_ALIVE_SECOND * 5);
- sendMessageService.sendBatch(webSocketManageService.getValueWithWorkspaceAndUserType(dock.getWorkspaceId(), UserTypeEnum.WEB.getVal()),
- CustomWebSocketMessage.builder()
- .timestamp(System.currentTimeMillis())
- .bizCode(BizCodeEnum.HIGHEST_PRIORITY_UPLOAD_FLIGHT_TASK_MEDIA.getCode())
- .data(MediaFileCountDTO.builder()
- .preJobId(Objects.nonNull(preJobId) ? preJobId.toString() : null)
- .jobId(jobId).build())
- .build());
+ Optional<DeviceDTO> deviceOpt = deviceRedisService.getDeviceOnline(receiver.getGateway());
+ if (deviceOpt.isEmpty()) {
+ return null;
+ }
+ sendMessageService.sendBatch(deviceOpt.get().getWorkspaceId(), UserTypeEnum.WEB.getVal(),
+ BizCodeEnum.HIGHEST_PRIORITY_UPLOAD_FLIGHT_TASK_MEDIA.getCode(), countDTO);
- messageSenderService.publish(headers.get(MqttHeaders.RECEIVED_TOPIC) + TopicConst._REPLY_SUF,
- CommonTopicResponse.builder()
- .data(RequestsReply.success())
- .method(receiver.getMethod())
- .timestamp(System.currentTimeMillis())
- .bid(receiver.getBid())
- .tid(receiver.getTid())
- .build());
+ return receiver;
}
}
--
Gitblit v1.9.3