From 7ec3683c8e41460f4bb0bd3a6677198742313e2b Mon Sep 17 00:00:00 2001
From: doum <doum>
Date: 星期二, 02 六月 2026 14:34:57 +0800
Subject: [PATCH] 新增智能电表、空调管理

---
 server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/YwDeviceServiceImpl.java |  329 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 328 insertions(+), 1 deletions(-)

diff --git a/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/YwDeviceServiceImpl.java b/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/YwDeviceServiceImpl.java
index dda5508..4754f5a 100644
--- a/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/YwDeviceServiceImpl.java
+++ b/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/YwDeviceServiceImpl.java
@@ -9,14 +9,19 @@
 import com.doumee.core.utils.Constants;
 import com.doumee.core.utils.DateUtil;
 import com.doumee.core.utils.Utils;
+import com.doumee.core.annotation.excel.ExcelImporter;
+import com.doumee.dao.admin.request.DeviceImport;
+import com.doumee.dao.business.*;
 import com.doumee.dao.business.YwDeviceMapper;
 import com.doumee.dao.business.YwDeviceRecordMapper;
+import com.doumee.dao.business.join.MemberJoinMapper;
 import com.doumee.dao.business.model.*;
 import com.doumee.dao.business.vo.YwDeviceCateDataVO;
 import com.doumee.dao.business.vo.YwDeviceDataVO;
 import com.doumee.dao.business.vo.YwDeviceParentCateDataVO;
 import com.doumee.dao.business.vo.YwDeviceStatusDataVO;
 import com.doumee.dao.system.MultifileMapper;
+import com.doumee.dao.system.SystemUserMapper;
 import com.doumee.dao.system.model.Multifile;
 import com.doumee.dao.system.model.SystemUser;
 import com.doumee.service.business.YwDeviceRecordService;
@@ -31,8 +36,11 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.CollectionUtils;
+import org.springframework.web.multipart.MultipartFile;
 
+import java.text.SimpleDateFormat;
 import java.util.*;
 import java.util.stream.Collectors;
 
@@ -57,6 +65,22 @@
     @Autowired
     private RedisTemplate<String,Object> redisTemplate;
 
+    @Autowired
+    private CategoryMapper categoryMapper;
+    @Autowired
+    private YwProjectMapper ywProjectMapper;
+    @Autowired
+    private YwBuildingMapper ywBuildingMapper;
+    @Autowired
+    private YwFloorMapper ywFloorMapper;
+    @Autowired
+    private YwRoomMapper ywRoomMapper;
+    @Autowired
+    private MemberJoinMapper memberJoinMapper;
+    @Autowired
+    private SystemUserMapper systemUserMapper;
+
+    private static final int IMPORT_EXCEL_ROW_OFFSET = 3;
 
     @Override
     public Integer create(YwDevice ywDevice) {
@@ -430,14 +454,317 @@
         ywDeviceStatusDataVO.setAmount(ywDeviceList.size());
         ywDeviceStatusDataVO.setWorkAmount(ywDeviceList.stream().filter(i->Objects.nonNull(i.getStatus())&&Constants.equalsInteger(i.getStatus(),Constants.ZERO)).collect(Collectors.toList()).size());
         ywDeviceStatusDataVO.setExceptionAmount(ywDeviceList.stream().filter(i->Objects.nonNull(i.getStatus())&&Constants.equalsInteger(i.getStatus(),Constants.ONE)).collect(Collectors.toList()).size());
-        ywDeviceStatusDataVO.setErrAmount(ywDeviceList.stream().filter(i->Objects.nonNull(i.getStatus())&&Constants.equalsInteger(i.getStatus(),Constants.THREE)).collect(Collectors.toList()).size());
+        ywDeviceStatusDataVO.setErrAmount(ywDeviceList.stream().filter(i->Objects.nonNull(i.getStatus())&&Constants.equalsInteger(i.getStatus(),Constants.TWO)).collect(Collectors.toList()).size());
         return ywDeviceStatusDataVO;
 
     }
 
+    @Override
+    @Transactional(rollbackFor = {BusinessException.class, Exception.class})
+    public String importBatch(MultipartFile file, LoginUserInfo loginUserInfo) {
+        List<DeviceImport> dataList;
+        try {
+            ExcelImporter ie = new ExcelImporter(file, 1, 0);
+            dataList = ie.getDataList(DeviceImport.class, null);
+        } catch (Exception e) {
+            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝瀵煎叆鏂囦欢瑙f瀽澶辫触锛岃妫�鏌ヨ〃鏍兼牸寮忥紒");
+        }
+        if (CollectionUtils.isEmpty(dataList)) {
+            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝褰曞叆鏁版嵁涓虹┖锛�");
+        }
+        DeviceImportCache cache = buildImportCache();
+        List<YwDevice> insertList = new ArrayList<>();
+        List<YwDevice> updateList = new ArrayList<>();
+        Set<String> fileCodes = new HashSet<>();
+        Date now = new Date();
+        for (int i = 0; i < dataList.size(); i++) {
+            DeviceImport row = dataList.get(i);
+            if (isImportBlankRow(row)) {
+                continue;
+            }
+            YwDevice device = checkImportRow(row, i, cache, fileCodes, loginUserInfo, now);
+            if (device.getId() != null) {
+                updateList.add(device);
+            } else {
+                insertList.add(device);
+            }
+        }
+        if (CollectionUtils.isEmpty(insertList) && CollectionUtils.isEmpty(updateList)) {
+            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝褰曞叆鏈夋晥鏁版嵁涓虹┖锛�");
+        }
+        if (!CollectionUtils.isEmpty(insertList)) {
+            ywDeviceMapper.insert(insertList);
+        }
+        for (YwDevice device : updateList) {
+            ywDeviceMapper.updateById(device);
+        }
+        return String.format("鎿嶄綔鎴愬姛锛屾湰娆″鍏� 鏂板%d涓澶囷紝鏇存柊%d涓澶�", insertList.size(), updateList.size());
+    }
 
+    private boolean isImportBlankRow(DeviceImport row) {
+        return StringUtils.isBlank(row.getCode())
+                && StringUtils.isBlank(row.getName())
+                && StringUtils.isBlank(row.getProjectName())
+                && StringUtils.isBlank(row.getRoomPath());
+    }
 
+    private DeviceImportCache buildImportCache() {
+        DeviceImportCache cache = new DeviceImportCache();
+        List<Category> categories = categoryMapper.selectList(new QueryWrapper<Category>().lambda()
+                .eq(Category::getIsdeleted, Constants.ZERO)
+                .eq(Category::getType, Constants.FIVE));
+        Map<Integer, Category> categoryById = categories.stream()
+                .collect(Collectors.toMap(Category::getId, c -> c, (a, b) -> a));
+        for (Category category : categories) {
+            if (category.getParentId() == null || category.getParentId() <= 0) {
+                continue;
+            }
+            Category parent = categoryById.get(category.getParentId());
+            if (parent == null || StringUtils.isBlank(parent.getName()) || StringUtils.isBlank(category.getName())) {
+                continue;
+            }
+            cache.categoryPathMap.put(parent.getName() + "/" + category.getName(), category.getId());
+        }
+        List<YwProject> projects = ywProjectMapper.selectList(new QueryWrapper<YwProject>().lambda()
+                .eq(YwProject::getIsdeleted, Constants.ZERO));
+        for (YwProject project : projects) {
+            if (StringUtils.isNotBlank(project.getName())) {
+                cache.projectNameMap.put(project.getName(), project.getId());
+            }
+        }
+        List<YwBuilding> buildings = ywBuildingMapper.selectList(new QueryWrapper<YwBuilding>().lambda()
+                .eq(YwBuilding::getIsdeleted, Constants.ZERO));
+        for (YwBuilding building : buildings) {
+            if (building.getProjectId() != null && StringUtils.isNotBlank(building.getName())) {
+                cache.buildingMap.put(building.getProjectId() + "|" + building.getName(), building.getId());
+            }
+        }
+        List<YwFloor> floors = ywFloorMapper.selectList(new QueryWrapper<YwFloor>().lambda()
+                .eq(YwFloor::getIsdeleted, Constants.ZERO));
+        for (YwFloor floor : floors) {
+            if (floor.getBuildingId() != null && StringUtils.isNotBlank(floor.getName())) {
+                cache.floorMap.put(floor.getBuildingId() + "|" + floor.getName(), floor.getId());
+            }
+        }
+        List<YwRoom> rooms = ywRoomMapper.selectList(new QueryWrapper<YwRoom>().lambda()
+                .eq(YwRoom::getIsdeleted, Constants.ZERO));
+        for (YwRoom room : rooms) {
+            if (room.getProjectId() == null || room.getBuildingId() == null || room.getFloor() == null) {
+                continue;
+            }
+            if (StringUtils.isNotBlank(room.getCode())) {
+                cache.roomMap.put(roomKey(room.getProjectId(), room.getBuildingId(), room.getFloor(), room.getCode()), room.getId());
+            }
+            if (StringUtils.isNotBlank(room.getRoomNum())) {
+                cache.roomMap.put(roomKey(room.getProjectId(), room.getBuildingId(), room.getFloor(), room.getRoomNum()), room.getId());
+            }
+        }
+        List<YwDevice> devices = ywDeviceMapper.selectList(new QueryWrapper<YwDevice>().lambda()
+                .eq(YwDevice::getIsdeleted, Constants.ZERO)
+                .select(YwDevice::getId, YwDevice::getCode));
+        for (YwDevice device : devices) {
+            if (StringUtils.isNotBlank(device.getCode())) {
+                cache.existingCodeIdMap.put(device.getCode(), device.getId());
+            }
+        }
+        List<Member> members = memberJoinMapper.selectJoinList(Member.class, new MPJLambdaWrapper<Member>()
+                .select(Member::getId, Member::getName)
+                .leftJoin(Company.class, Company::getId, Member::getCompanyId)
+                .eq(Member::getIsdeleted, Constants.ZERO)
+                .eq(Company::getType, Constants.ONE));
+        if (!CollectionUtils.isEmpty(members)) {
+            List<Integer> memberIds = members.stream().map(Member::getId).filter(Objects::nonNull).collect(Collectors.toList());
+            if (!CollectionUtils.isEmpty(memberIds)) {
+                List<SystemUser> systemUsers = systemUserMapper.selectList(new QueryWrapper<SystemUser>().lambda()
+                        .eq(SystemUser::getDeleted, Boolean.FALSE)
+                        .in(SystemUser::getMemberId, memberIds));
+                Map<Integer, Integer> memberUserMap = systemUsers.stream()
+                        .filter(u -> u.getMemberId() != null)
+                        .collect(Collectors.toMap(SystemUser::getMemberId, SystemUser::getId, (a, b) -> a));
+                for (Member member : members) {
+                    if (StringUtils.isBlank(member.getName())) {
+                        continue;
+                    }
+                    Integer userId = memberUserMap.get(member.getId());
+                    if (userId == null) {
+                        continue;
+                    }
+                    if (cache.internalUserNameMap.containsKey(member.getName())) {
+                        cache.duplicateInternalUserNames.add(member.getName());
+                    }
+                    cache.internalUserNameMap.put(member.getName(), userId);
+                }
+            }
+        }
+        return cache;
+    }
 
+    private String roomKey(Integer projectId, Integer buildingId, Integer floorId, String roomName) {
+        return projectId + "|" + buildingId + "|" + floorId + "|" + roomName;
+    }
+
+    private YwDevice checkImportRow(DeviceImport row, int index, DeviceImportCache cache, Set<String> fileCodes,
+                                    LoginUserInfo loginUserInfo, Date now) {
+        int rowNum = index + IMPORT_EXCEL_ROW_OFFSET;
+        if (StringUtils.isBlank(row.getCode())) {
+            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝绗�" + rowNum + "琛岃澶囩紪鍙蜂笉鑳戒负绌猴紝璇锋鏌ヨ〃鏍煎唴瀹癸紒");
+        }
+        if (StringUtils.isBlank(row.getName())) {
+            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝绗�" + rowNum + "琛岃澶囧悕绉颁笉鑳戒负绌猴紝璇锋鏌ヨ〃鏍煎唴瀹癸紒");
+        }
+        if (StringUtils.isBlank(row.getProjectName())) {
+            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝绗�" + rowNum + "琛屾墍灞為」鐩笉鑳戒负绌猴紝璇锋鏌ヨ〃鏍煎唴瀹癸紒");
+        }
+        if (StringUtils.isBlank(row.getRoomPath())) {
+            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝绗�" + rowNum + "琛屽叧鑱旀埧婧愪笉鑳戒负绌猴紝璇锋鏌ヨ〃鏍煎唴瀹癸紒");
+        }
+        String code = row.getCode().trim();
+        if (fileCodes.contains(code)) {
+            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝绗�" + rowNum + "琛岃澶囩紪鍙枫��" + code + "銆戦噸澶嶏紝璇锋鏌ヨ〃鏍煎唴瀹癸紒");
+        }
+        fileCodes.add(code);
+        Integer existingId = cache.existingCodeIdMap.get(code);
+
+        Integer cateId = null;
+        if (StringUtils.isNotBlank(row.getCategoryPath())) {
+            String categoryPath = row.getCategoryPath().trim();
+            String[] cateParts = categoryPath.split("/");
+            if (cateParts.length < 2 || StringUtils.isBlank(cateParts[1])) {
+                throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝绗�" + rowNum + "琛岃澶囧垎绫汇��" + categoryPath + "銆戞牸寮忎笉姝g‘锛岃浣跨敤涓�绾�/浜岀骇鏍煎紡锛�");
+            }
+            cateId = cache.categoryPathMap.get(categoryPath);
+            if (cateId == null) {
+                throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝绗�" + rowNum + "琛岃澶囧垎绫汇��" + categoryPath + "銆戞湭鎵惧埌锛岃妫�鏌ヨ〃鏍煎唴瀹癸紒");
+            }
+        }
+
+        Integer projectId = cache.projectNameMap.get(row.getProjectName().trim());
+        if (projectId == null) {
+            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝绗�" + rowNum + "琛屾墍灞為」鐩��" + row.getProjectName() + "銆戞湭鎵惧埌锛岃妫�鏌ヨ〃鏍煎唴瀹癸紒");
+        }
+
+        String[] roomParts = row.getRoomPath().trim().split("/");
+        if (roomParts.length < 3 || StringUtils.isAnyBlank(roomParts[0], roomParts[1], roomParts[2])) {
+            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝绗�" + rowNum + "琛屽叧鑱旀埧婧愩��" + row.getRoomPath() + "銆戞牸寮忎笉姝g‘锛岃浣跨敤妤煎畤/妤煎眰/鎴挎簮鏍煎紡锛�");
+        }
+        Integer buildingId = cache.buildingMap.get(projectId + "|" + roomParts[0].trim());
+        if (buildingId == null) {
+            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝绗�" + rowNum + "琛屽叧鑱旀埧婧愩��" + row.getRoomPath() + "銆戞湭鎵惧埌锛岃妫�鏌ヨ〃鏍煎唴瀹癸紒");
+        }
+        Integer floorId = cache.floorMap.get(buildingId + "|" + roomParts[1].trim());
+        if (floorId == null) {
+            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝绗�" + rowNum + "琛屽叧鑱旀埧婧愩��" + row.getRoomPath() + "銆戞湭鎵惧埌锛岃妫�鏌ヨ〃鏍煎唴瀹癸紒");
+        }
+        Integer roomId = cache.roomMap.get(roomKey(projectId, buildingId, floorId, roomParts[2].trim()));
+        if (roomId == null) {
+            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝绗�" + rowNum + "琛屽叧鑱旀埧婧愩��" + row.getRoomPath() + "銆戞湭鎵惧埌锛岃妫�鏌ヨ〃鏍煎唴瀹癸紒");
+        }
+
+        Integer status = parseImportStatus(row.getStatusText(), rowNum);
+
+        Integer userId = resolveInternalUserId(row.getAdminUserName(), rowNum, "璁惧绠$悊鍛�", cache);
+        Integer maintenanceUserId = resolveInternalUserId(row.getMaintenanceUserName(), rowNum, "缁翠繚璐熻矗浜�", cache);
+
+        YwDevice device = new YwDevice();
+        device.setCode(code);
+        device.setName(row.getName().trim());
+        device.setCateId(cateId);
+        device.setModelNo(StringUtils.trimToNull(row.getModelNo()));
+        device.setUserId(userId);
+        device.setAddr(StringUtils.trimToNull(row.getAddr()));
+        device.setProjectId(projectId);
+        device.setBuildingId(buildingId);
+        device.setFloorId(floorId);
+        device.setRoomId(roomId);
+        device.setBuyDate(parseImportDate(row.getBuyDate(), rowNum, "璐叆鏃堕棿"));
+        device.setStatus(status);
+        device.setContent(StringUtils.trimToNull(row.getContent()));
+        device.setSupplier(StringUtils.trimToNull(row.getSupplier()));
+        device.setSupplierLinker(StringUtils.trimToNull(row.getSupplierLinker()));
+        device.setSupplierPhone(StringUtils.trimToNull(row.getSupplierPhone()));
+        device.setMaintenanceUserId(maintenanceUserId);
+        device.setMaintenanceOverDate(parseImportDate(row.getMaintenanceOverDate(), rowNum, "缁翠繚鍒版湡鏃�"));
+        device.setMaintenanceContent(StringUtils.trimToNull(row.getMaintenanceContent()));
+        if (existingId != null) {
+            device.setId(existingId);
+            device.setEditor(loginUserInfo.getId());
+            device.setEditDate(now);
+        } else {
+            device.setCreator(loginUserInfo.getId());
+            device.setCreateDate(now);
+            device.setIsdeleted(Constants.ZERO);
+        }
+        return device;
+    }
+
+    private Integer resolveInternalUserId(String userName, int rowNum, String fieldLabel, DeviceImportCache cache) {
+        if (StringUtils.isBlank(userName)) {
+            return null;
+        }
+        String name = userName.trim();
+        if (cache.duplicateInternalUserNames.contains(name)) {
+            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝绗�" + rowNum + "琛�" + fieldLabel + "銆�" + name + "銆戝瓨鍦ㄩ噸鍚嶏紝璇锋鏌ヨ〃鏍煎唴瀹癸紒");
+        }
+        Integer userId = cache.internalUserNameMap.get(name);
+        if (userId == null) {
+            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝绗�" + rowNum + "琛�" + fieldLabel + "銆�" + name + "銆戞湭鎵惧埌锛岃妫�鏌ヨ〃鏍煎唴瀹癸紒");
+        }
+        return userId;
+    }
+
+    private Integer parseImportStatus(String statusText, int rowNum) {
+        if (StringUtils.isBlank(statusText)) {
+            return Constants.ZERO;
+        }
+        String text = statusText.trim();
+        if ("姝e父".equals(text)) {
+            return Constants.ZERO;
+        }
+        if ("鎹熷潖".equals(text)) {
+            return Constants.ONE;
+        }
+        if ("鎶ュ簾".equals(text)) {
+            return Constants.TWO;
+        }
+        throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "瀵逛笉璧凤紝绗�" + rowNum + "琛岃澶囩姸鎬併��" + statusText + "銆戜笉姝g‘锛屼粎鏀寔姝e父/鎹熷潖/鎶ュ簾锛�");
+    }
+
+    private Date parseImportDate(String dateText, int rowNum, String fieldLabel) {
+        if (StringUtils.isBlank(dateText)) {
+            return null;
+        }
+        String text = dateText.trim();
+        if (StringUtils.endsWith(text, ".0")) {
+            text = StringUtils.substringBefore(text, ".0");
+        }
+        Date date = DateUtil.parseFromFormats(text);
+        if (date != null) {
+            return date;
+        }
+        try {
+            return new SimpleDateFormat("yyyy-MM-dd").parse(text);
+        } catch (Exception ignored) {
+            // try excel serial number below
+        }
+        try {
+            return org.apache.poi.ss.usermodel.DateUtil.getJavaDate(Double.parseDouble(text));
+        } catch (Exception ignored) {
+            // fall through to error
+        }
+        throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(),
+                "瀵逛笉璧凤紝绗�" + rowNum + "琛�" + fieldLabel + "銆�" + dateText + "銆戞牸寮忎笉姝g‘锛岃浣跨敤yyyy-MM-dd鏍煎紡锛�");
+    }
+
+    private static class DeviceImportCache {
+        private final Map<String, Integer> categoryPathMap = new HashMap<>();
+        private final Map<String, Integer> projectNameMap = new HashMap<>();
+        private final Map<String, Integer> buildingMap = new HashMap<>();
+        private final Map<String, Integer> floorMap = new HashMap<>();
+        private final Map<String, Integer> roomMap = new HashMap<>();
+        private final Map<String, Integer> existingCodeIdMap = new HashMap<>();
+        private final Map<String, Integer> internalUserNameMap = new HashMap<>();
+        private final Set<String> duplicateInternalUserNames = new HashSet<>();
+    }
 
 }
 

--
Gitblit v1.9.3