From 8dedfeac9458b0d1f419a012361940b89285a9f8 Mon Sep 17 00:00:00 2001
From: liusuyi <1951119284@qq.com>
Date: Mon, 08 Dec 2025 13:15:16 +0800
Subject: [PATCH] 优化
---
ard-work/src/main/java/com/ard/work/service/impl/ArdDroneInfoServiceImpl.java | 112 ++++++++++++++++++++++++++++++++++++++++----------------
1 files changed, 80 insertions(+), 32 deletions(-)
diff --git a/ard-work/src/main/java/com/ard/work/service/impl/ArdDroneInfoServiceImpl.java b/ard-work/src/main/java/com/ard/work/service/impl/ArdDroneInfoServiceImpl.java
index ed3a12e..fe42391 100644
--- a/ard-work/src/main/java/com/ard/work/service/impl/ArdDroneInfoServiceImpl.java
+++ b/ard-work/src/main/java/com/ard/work/service/impl/ArdDroneInfoServiceImpl.java
@@ -1,28 +1,36 @@
package com.ard.work.service.impl;
import com.alibaba.fastjson2.JSONObject;
+import com.ard.common.annotation.DataScope;
+import com.ard.common.constant.CacheConstants;
+import com.ard.common.core.redis.RedisCache;
import com.ard.common.exception.ServiceException;
import com.ard.common.utils.DateUtils;
import com.ard.common.utils.uuid.IdUtils;
import com.ard.system.authorize.service.AuthorizeService;
+import com.ard.work.domain.dto.BoxSettingsDTO;
import com.ard.work.domain.entity.ArdDroneInfo;
+import com.ard.work.domain.entity.ArdServerInfo;
import com.ard.work.domain.receiver.BoxInfoReceiver;
+import com.ard.work.domain.receiver.BoxStreamUrlReceiver;
import com.ard.work.mapper.ArdDroneInfoMapper;
import com.ard.work.service.IArdDroneInfoService;
+import com.ard.work.service.IArdServerInfoService;
import com.ard.work.service.ITyService;
-import com.ard.work.util.UtilMethod;
import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.util.*;
+import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;
/**
* 无人机盒子Service业务层处理
*
- * @author ruoyi
+ * @author ard
* @date 2025-03-04
*/
@Slf4j
@@ -38,7 +46,11 @@
@Resource
private AuthorizeService authorizeService;
- UtilMethod utilMethod = new UtilMethod();
+ @Resource
+ private RedisCache redisCache;
+
+ @Autowired
+ private IArdServerInfoService ardServerInfoService;
/**
* 查询无人机盒子
@@ -65,12 +77,25 @@
/**
* 查询无人机盒子列表
*
- * @param ArdDroneInfo 无人机盒子
+ * @param ardDroneInfo 无人机盒子
* @return 无人机盒子
*/
@Override
- public List<ArdDroneInfo> selectArdDroneList(ArdDroneInfo ArdDroneInfo) {
- return ArdDroneInfoMapper.selectArdDroneList(ArdDroneInfo);
+ @DataScope(deptAlias = "d", userAlias = "u")
+ public List<ArdDroneInfo> selectArdDroneList(ArdDroneInfo ardDroneInfo) {
+ List<ArdDroneInfo> ardDroneInfos = ArdDroneInfoMapper.selectArdDroneList(ardDroneInfo);
+ // 使用传统的 for 循环避免 forEach 带来的性能开销
+ for (ArdDroneInfo uav : ardDroneInfos) {
+ try {
+ BoxStreamUrlReceiver boxLiveUrl = tyService.getBoxLiveUrl(uav.getBoxSn());
+ if (boxLiveUrl != null) {
+ uav.setWebrtcUrl(boxLiveUrl.getRtc());
+ }
+ } catch (Exception e) {
+ log.error("获取设备 {} 的直播地址失败", uav.getBoxSn(), e);
+ }
+ }
+ return ardDroneInfos;
}
/**
@@ -94,6 +119,11 @@
*/
@Override
public int updateArdDrone(ArdDroneInfo ArdDroneInfo) {
+ Boolean b =
+ tyService.updateBoxSettings(BoxSettingsDTO.builder().boxSn(ArdDroneInfo.getBoxSn()).boxName(ArdDroneInfo.getBoxName()).build());
+ if (!b) {
+ throw new ServiceException("修改无人机盒子名称失败");
+ }
ArdDroneInfo.setUpdateTime(DateUtils.getNowDate());
return ArdDroneInfoMapper.updateArdDrone(ArdDroneInfo);
}
@@ -123,7 +153,7 @@
/**
* 批量新增无人机盒子
*
- * @param boxSn 盒子编码数组
+ * @param boxSn 盒子编码数组
* @param deptId 部门ID
* @return 成功插入的记录数
*/
@@ -138,9 +168,6 @@
throw new IllegalArgumentException("部门ID不能为空");
}
try {
-
- //utilMethod.authorizationVerification(boxSn.length);
-
// 1. 检查已存在的无人机盒子
List<ArdDroneInfo> existingDrones = getExistingDrones(boxSn);
// 2. 授权数量验证
@@ -153,6 +180,14 @@
log.error("批量新增无人机盒子失败, boxSn: {}, deptId: {}", Arrays.toString(boxSn), deptId, e);
throw new ServiceException("批量新增无人机盒子失败: " + e.getMessage());
}
+ }
+
+ /**
+ * 修改无人机状态
+ */
+ @Override
+ public int updateDroneStatus(String serverId, String status) {
+ return ArdDroneInfoMapper.updateDroneStatus(serverId, status);
}
/**
@@ -239,37 +274,50 @@
*/
private int batchInsertDrones(List<ArdDroneInfo> drones) {
if (drones.isEmpty()) {
- log.info("没有需要插入的无人机信息");
return 0;
}
-
- int result = ArdDroneInfoMapper.insertArdDroneLs(drones);
- log.info("成功插入 {} 条无人机记录", result);
- return result;
+ return ArdDroneInfoMapper.insertArdDroneLs(drones);
}
/**
* 同步盒子属性
- *
- * @return 结果
*/
- @Scheduled(fixedRate = 1000 * 10)
+ @Scheduled(fixedRate = 1000 * 5)
public void syncBoxInfo() {
- List<ArdDroneInfo> ArdDroneeListEntity = selectArdDroneList(new ArdDroneInfo());
- // 将本地列表转换为Map，方便快速查找
- Map<String, ArdDroneInfo> localArdDroneMap = ArdDroneeListEntity.stream()
- .collect(Collectors.toMap(ArdDroneInfo::getBoxSn, drone -> drone));
- //遍历接口最新获取的无人机盒子信息
- tyService.getBoxList().forEach(remoteArdDrone -> {
- ArdDroneInfo localArdDroneInfo = localArdDroneMap.get(remoteArdDrone.getBoxSn());
- if (localArdDroneInfo != null) {
- if (!localArdDroneInfo.isSameAs(remoteArdDrone)) {
- // 修改无人机（只更新变化的属性）
- updateChangedAttributes(localArdDroneInfo, remoteArdDrone);
- updateArdDrone(localArdDroneInfo);
- }
+ ArdServerInfo ardServerInfo = new ArdServerInfo();
+ ardServerInfo.setUseDataScope(false);
+ List<ArdServerInfo> servers = ardServerInfoService.selectArdServerInfoList(ardServerInfo);
+ for (ArdServerInfo server : servers) {
+ ArdDroneInfo droneInfo = new ArdDroneInfo();
+ droneInfo.setServerId(server.getId());
+ droneInfo.setUseDataScope(false);
+ List<ArdDroneInfo> localList = selectArdDroneList(droneInfo);
+ Map<String, ArdDroneInfo> localMap = localList.stream()
+ .collect(Collectors.toMap(ArdDroneInfo::getBoxSn, drone -> drone));
+ List<BoxInfoReceiver> boxList = tyService.getBoxList(server.getId());
+ if (boxList == null) {
+ updateArdDrone(ArdDroneInfo.builder().serverId(server.getId()).onLine(0).build());
+ continue;
}
- });
+ boxList.forEach(remote -> {
+ String boxSn = remote.getBoxSn();
+
+ // ✅ 维护在线状态缓存
+ // 只有在线的无人机才存 Redis
+ String redisKey = CacheConstants.UAV_ONLINE_KEY + boxSn;
+ if (remote.getOnLine() == 1) {
+ redisCache.setCacheObject(redisKey, remote, 30, TimeUnit.SECONDS);
+ } else {
+ redisCache.deleteObject(CacheConstants.UAV_ONLINE_KEY + boxSn);
+ }
+ // 数据库同步
+ ArdDroneInfo local = localMap.get(boxSn);
+ if (local != null && !local.isSameAs(remote)) {
+ updateChangedAttributes(local, remote);
+ updateArdDrone(local);
+ }
+ });
+ }
}
// 更新变化的属性
--
Gitblit v1.9.3