吉安感知网项目-后端
rain
2026-01-14 d2b1a724f2a480fca2bd1b92e4d4a175eb634875
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) {