src/main/java/com/campus/water/service/AreaService.java

@@ -1,14 +1,11 @@
+// src/main/java/com/campus/water/service/AreaService.java
 package com.campus.water.service;
 
 import com.campus.water.entity.Area;
-import com.campus.water.entity.Device;
 import com.campus.water.mapper.AreaRepository;
 import com.campus.water.entity.Admin;
 import com.campus.water.mapper.AdminRepository;
-import com.campus.water.mapper.DeviceRepository;
-import com.campus.water.mapper.DeviceTerminalMappingRepository;
 import com.campus.water.security.RoleConstants;
-import com.campus.water.entity.vo.AreaDeviceStatsVO;
 import org.springframework.beans.factory.annotation.Autowired;
 import java.util.List;
 import org.springframework.stereotype.Service;
@@ -17,7 +14,6 @@ import org.springframework.transaction.annotation.Transactional;
 import java.time.LocalDateTime;
 import java.util.List;
 import java.util.Optional;
-import java.util.stream.Collectors;
 
 /**
  * 区域管理服务类
@@ -31,17 +27,10 @@ public class AreaService {
     // 新增：管理员仓库注入
     private final AdminRepository adminRepository;
 
-    // 新增：注入现有设备、终端映射Repository（无新增Mapper方法）
-    private final DeviceRepository deviceRepository;
-    private final DeviceTerminalMappingRepository deviceTerminalMappingRepository;
-
-    public AreaService(AreaRepository areaRepository, AdminRepository adminRepository,
-                       DeviceRepository deviceRepository,
-                       DeviceTerminalMappingRepository deviceTerminalMappingRepository) {
+    // 改造构造器，添加AdminRepository参数
+    public AreaService(AreaRepository areaRepository, AdminRepository adminRepository) {
         this.areaRepository = areaRepository;
         this.adminRepository = adminRepository;
-        this.deviceRepository = deviceRepository;
-        this.deviceTerminalMappingRepository = deviceTerminalMappingRepository;
     }
 
     /**
@@ -107,7 +96,14 @@ public class AreaService {
         existingArea.setUpdatedTime(LocalDateTime.now());
 
         // 5. 保存修改
-        return areaRepository.save(existingArea);
+        Area updatedArea = areaRepository.save(existingArea);
+
+        // 6. 更新关联管理员的areaId
+        if (existingArea.getManager() != null && !existingArea.getManager().trim().isEmpty()) {
+            bindAdminToArea(existingArea.getManager(), updatedArea.getAreaId());
+        }
+
+        return updatedArea;
     }
 
     /**
@@ -207,8 +203,11 @@ public class AreaService {
         String adminId = area.getManager();
         if (adminId != null && !adminId.trim().isEmpty()) {
             // 校验管理员是否存在
-            Admin admin = adminRepository.findById(adminId)
-                    .orElseThrow(() -> new RuntimeException("区域管理员不存在，ID：" + adminId));
+            Optional<Admin> adminOpt = adminRepository.findByAdminId(adminId);
+            if (adminOpt.isEmpty()) {
+                throw new RuntimeException("区域管理员不存在，ID：" + adminId);
+            }
+            Admin admin = adminOpt.get();
 
             // 校验管理员角色是否为区域管理员
             if (!RoleConstants.ROLE_AREA_ADMIN.equals(admin.getRole().name())) {
@@ -224,112 +223,15 @@ public class AreaService {
      */
     private void bindAdminToArea(String adminId, String areaId) {
         if (adminId != null && !adminId.trim().isEmpty() && areaId != null) {
-            Admin admin = adminRepository.findById(adminId).get(); // 已在前序校验，无需再次处理空值
-            admin.setAreaId(areaId); // 给管理员设置关联的区域ID（Admin实体需有areaId字段）
-            adminRepository.save(admin);
-        }
-    }
-
-
-    /**
-     * 获取指定片区的设备统计（校区：自身数据；市区：自身+下属所有校区汇总数据）
-     * @param areaId 片区ID（市区/校园）
-     * @return 片区设备统计结果（适配现有实体）
-     */
-    @Transactional(readOnly = true) // 只读事务提升查询性能，不修改数据
-    public AreaDeviceStatsVO getAreaDeviceStats(String areaId) {
-        // 1. 校验片区是否存在，获取片区基础信息
-        Area targetArea = getAreaById(areaId);
-        AreaDeviceStatsVO statsVO = new AreaDeviceStatsVO();
-
-        // 2. 填充片区基础信息（复用Area实体字段）
-        statsVO.setAreaId(targetArea.getAreaId());
-        statsVO.setAreaName(targetArea.getAreaName());
-        statsVO.setAreaTypeDesc(targetArea.getAreaType().getDesc());
-
-        // 3. 区分片区类型执行统计（核心逻辑，复用现有Mapper方法）
-        if (Area.AreaType.campus.equals(targetArea.getAreaType())) {
-            // 校区：仅统计自身关联的设备/终端
-            fillSelfDeviceStats(statsVO, areaId);
-        } else if (Area.AreaType.zone.equals(targetArea.getAreaType())) {
-            // 市区：统计自身 + 下属所有校区的汇总数据
-            fillCityTotalDeviceStats(statsVO, areaId);
-        }
-
-        return statsVO;
-    }
-
-    /**
-     * 填充单个片区（校区/市区自身）的设备统计数据（复用现有Mapper的count方法）
-     * @param statsVO 统计结果VO
-     * @param areaId  片区ID
-     */
-    private void fillSelfDeviceStats(AreaDeviceStatsVO statsVO, String areaId) {
-        // 复用DeviceRepository.countByAreaIdAndDeviceType（已存在，无需新增）
-        long waterMakerCount = deviceRepository.countByAreaIdAndDeviceType(
-                areaId, Device.DeviceType.water_maker); // 假设Device枚举包含WATER_MAKER/WATER_SUPPLY
-        long waterSupplyCount = deviceRepository.countByAreaIdAndDeviceType(
-                areaId, Device.DeviceType.water_supply);
-
-        // 复用DeviceTerminalMappingRepository.countByAreaId（已存在，无需新增）
-        long terminalCount = deviceTerminalMappingRepository.countByAreaId(areaId);
-
-        // 赋值到VO（强转int，若数量过大可改为long，兼容现有数据）
-        statsVO.setWaterMakerCount((int) waterMakerCount);
-        statsVO.setWaterSupplyCount((int) waterSupplyCount);
-        statsVO.setTerminalCount((int) terminalCount);
-    }
-
-    /**
-     * 填充市区的总设备统计数据（仅汇总下属所有校区，市区自身无设备/终端）
-     * @param statsVO 统计结果VO
-     * @param cityId  市区ID
-     */
-    private void fillCityTotalDeviceStats(AreaDeviceStatsVO statsVO, String cityId) {
-        // 第一步：获取市区下属所有校区的ID列表（复用现有getCampusesByCityId方法）
-        List<Area> campusList = getCampusesByCityId(cityId);
-        List<String> campusIds = campusList.stream()
-                .map(Area::getAreaId)
-                .collect(Collectors.toList());
-
-        // 第二步：汇总下属所有校区的设备/终端数量（循环复用现有count方法，适配现有Mapper）
-        long campusWaterMaker = 0;
-        long campusWaterSupply = 0;
-        long campusTerminal = 0;
-        if (!campusIds.isEmpty()) {
-            for (String campusId : campusIds) {
-                campusWaterMaker += deviceRepository.countByAreaIdAndDeviceType(campusId, Device.DeviceType.water_maker);
-                campusWaterSupply += deviceRepository.countByAreaIdAndDeviceType(campusId, Device.DeviceType.water_supply);
-                campusTerminal += deviceTerminalMappingRepository.countByAreaId(campusId);
+            Optional<Admin> adminOpt = adminRepository.findByAdminId(adminId);
+            if (adminOpt.isPresent()) {
+                Admin admin = adminOpt.get();
+                admin.setAreaId(areaId); // 给管理员设置关联的区域ID
+                adminRepository.save(admin);
+            } else {
+                throw new RuntimeException("管理员不存在，ID：" + adminId);
             }
         }
-
-        // 第三步：直接赋值校区汇总数据（市区自身无数据）
-        statsVO.setWaterMakerCount((int) campusWaterMaker);
-        statsVO.setWaterSupplyCount((int) campusWaterSupply);
-        statsVO.setTerminalCount((int) campusTerminal);
-    }
-
-    /**
-     * 获取指定市区下所有校区的单独设备统计（每个校区各自的统计结果，不汇总）
-     * @param cityId 市区ID
-     * @return 校区统计列表（复用现有方法，无新增依赖）
-     */
-    @Transactional(readOnly = true)
-    public List<AreaDeviceStatsVO> getCampusDeviceStatsUnderCity(String cityId) {
-        // 1. 校验市区存在且类型正确
-        Area cityArea = getAreaById(cityId);
-        if (!Area.AreaType.zone.equals(cityArea.getAreaType())) {
-            throw new RuntimeException("指定区域不是市区，无法查询下属校区统计");
-        }
-
-        // 2. 获取下属所有校区（复用现有方法）
-        List<Area> campusList = getCampusesByCityId(cityId);
-
-        // 3. 逐个生成校区统计结果（复用getAreaDeviceStats方法）
-        return campusList.stream()
-                .map(campus -> getAreaDeviceStats(campus.getAreaId()))
-                .collect(Collectors.toList());
     }
 
 }