From d2b1a724f2a480fca2bd1b92e4d4a175eb634875 Mon Sep 17 00:00:00 2001
From: rain <1679827795@qq.com>
Date: Wed, 14 Jan 2026 09:19:42 +0800
Subject: [PATCH] 设备配置表信息
---
drone-service/drone-fw/src/main/java/org/sxkj/fw/area/service/impl/FwDefenseZoneServiceImpl.java | 69 ++++++++++++++--------------------
1 files changed, 29 insertions(+), 40 deletions(-)
diff --git a/drone-service/drone-fw/src/main/java/org/sxkj/fw/area/service/impl/FwDefenseZoneServiceImpl.java b/drone-service/drone-fw/src/main/java/org/sxkj/fw/area/service/impl/FwDefenseZoneServiceImpl.java
index 8927af5..ee824b0 100644
--- a/drone-service/drone-fw/src/main/java/org/sxkj/fw/area/service/impl/FwDefenseZoneServiceImpl.java
+++ b/drone-service/drone-fw/src/main/java/org/sxkj/fw/area/service/impl/FwDefenseZoneServiceImpl.java
@@ -21,19 +21,19 @@
import org.sxkj.common.utils.HeaderUtils;
import org.sxkj.fw.area.dto.FwDefenseZoneDTO;
import org.sxkj.fw.area.entity.FwAreaDivideEntity;
+import org.sxkj.fw.area.entity.FwDefenseSceneEntity;
import org.sxkj.fw.area.entity.FwDefenseZoneEntity;
import org.sxkj.fw.area.vo.FwDefenseZoneVO;
import org.sxkj.fw.area.excel.FwDefenseZoneExcel;
import org.sxkj.fw.area.mapper.FwDefenseZoneMapper;
import org.sxkj.fw.area.service.IFwAreaDivideService;
+import org.sxkj.fw.area.service.IFwDefenseSceneService;
import org.sxkj.fw.area.service.IFwDefenseZoneService;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import org.springblade.core.mp.base.BaseServiceImpl;
import javax.annotation.Resource;
-import java.math.BigDecimal;
-import java.math.RoundingMode;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
@@ -54,6 +54,9 @@
@Resource
private IFwAreaDivideService fwAreaDivideService;
+
+ @Resource
+ private IFwDefenseSceneService fwDefenseSceneService;
@Override
public IPage<FwDefenseZoneVO> selectFwDefenseZonePage(IPage<FwDefenseZoneVO> page, FwDefenseZoneDTO fwDefenseZone) {
@@ -76,39 +79,42 @@
return;
}
Set<Long> sceneIdSet = new HashSet<>();
- Set<Long> areaIdSet = new HashSet<>();
Map<Long, List<Long>> zoneSceneMap = new HashMap<>();
- Map<Long, List<Long>> zoneAreaMap = new HashMap<>();
for (FwDefenseZoneVO record : records) {
List<Long> sceneIds = parseIdList(record.getDefenseSceneIds());
if (record.getId() != null) {
zoneSceneMap.put(record.getId(), sceneIds);
}
sceneIdSet.addAll(sceneIds);
- List<Long> areaIds = parseIdList(record.getAreaDivideIds());
- if (record.getId() != null) {
- zoneAreaMap.put(record.getId(), areaIds);
- }
- areaIdSet.addAll(areaIds);
}
- if (sceneIdSet.isEmpty() && areaIdSet.isEmpty()) {
+ Map<Long, List<Long>> sceneAreaMap = new HashMap<>();
+ Set<Long> sceneAreaIdSet = new HashSet<>();
+ if (!sceneIdSet.isEmpty()) {
+ List<FwDefenseSceneEntity> sceneList = fwDefenseSceneService.listByIds(sceneIdSet);
+ for (FwDefenseSceneEntity scene : sceneList) {
+ if (scene == null || scene.getId() == null) {
+ continue;
+ }
+ List<Long> sceneAreaIds = parseIdList(scene.getAreaDivideIds());
+ sceneAreaMap.put(scene.getId(), sceneAreaIds);
+ sceneAreaIdSet.addAll(sceneAreaIds);
+ }
+ }
+ if (sceneIdSet.isEmpty()) {
for (FwDefenseZoneVO record : records) {
record.setSceneCount(0);
record.setDeviceCount(0);
record.setAreaCount(0);
- record.setZoneArea(BigDecimal.ZERO);
}
return;
}
- Map<Long, BigDecimal> areaSizeMap = new HashMap<>();
Map<Long, List<Long>> areaDeviceMap = new HashMap<>();
- if (!areaIdSet.isEmpty()) {
- List<FwAreaDivideEntity> areaList = fwAreaDivideService.listByIds(areaIdSet);
+ if (!sceneAreaIdSet.isEmpty()) {
+ List<FwAreaDivideEntity> areaList = fwAreaDivideService.listByIds(sceneAreaIdSet);
for (FwAreaDivideEntity area : areaList) {
if (area == null || area.getId() == null) {
continue;
}
- areaSizeMap.put(area.getId(), area.getAreaSize());
areaDeviceMap.put(area.getId(), parseIdList(area.getDeviceIds()));
}
}
@@ -120,21 +126,22 @@
}
Set<Long> uniqueSceneIds = new HashSet<>(sceneIds);
record.setSceneCount(uniqueSceneIds.size());
- List<Long> zoneAreaIds = zoneAreaMap.get(record.getId());
- if (zoneAreaIds == null) {
- zoneAreaIds = Collections.emptyList();
+ Set<Long> sceneAreaIds = new HashSet<>();
+ for (Long sceneId : uniqueSceneIds) {
+ List<Long> areaIds = sceneAreaMap.get(sceneId);
+ if (areaIds != null) {
+ sceneAreaIds.addAll(areaIds);
+ }
}
- Set<Long> areaIds = new HashSet<>(zoneAreaIds);
Set<Long> deviceIds = new HashSet<>();
- for (Long areaId : areaIds) {
+ for (Long areaId : sceneAreaIds) {
List<Long> areaDeviceIds = areaDeviceMap.get(areaId);
if (areaDeviceIds != null) {
deviceIds.addAll(areaDeviceIds);
}
}
- record.setAreaCount(areaIds.size());
+ record.setAreaCount(sceneAreaIds.size());
record.setDeviceCount(deviceIds.size());
- record.setZoneArea(sumAreaSize(areaIds, areaSizeMap));
}
}
@@ -159,24 +166,6 @@
}
return idList;
}
-
- private BigDecimal sumAreaSize(Set<Long> areaIds, Map<Long, BigDecimal> areaSizeMap) {
- if (areaIds == null || areaIds.isEmpty()) {
- return BigDecimal.ZERO;
- }
- BigDecimal total = BigDecimal.ZERO;
- for (Long areaId : areaIds) {
- BigDecimal areaSize = areaSizeMap.get(areaId);
- if (areaSize != null) {
- total = total.add(areaSize);
- }
- }
- if (BigDecimal.ZERO.compareTo(total) == 0) {
- return BigDecimal.ZERO;
- }
- return total.setScale(2, RoundingMode.HALF_UP);
- }
-
@Override
public List<FwDefenseZoneExcel> exportFwDefenseZone(Wrapper<FwDefenseZoneEntity> queryWrapper) {
--
Gitblit v1.9.3