设备管理

airport-admin/src/main/java/com/sundot/airport/web/controller/equipment/EquipmentLedgerController.java

@@ -21,6 +21,7 @@ import com.sundot.airport.equipment.domain.EquipmentLedger;
 import com.sundot.airport.equipment.service.IEquipmentLedgerService;
 import com.sundot.airport.common.utils.poi.ExcelUtil;
 import com.sundot.airport.common.core.page.TableDataInfo;
+import org.springframework.web.multipart.MultipartFile;
 
 /**
  * 设备台账Controller
@@ -107,4 +108,26 @@ public class EquipmentLedgerController extends BaseController {
     public AjaxResult remove(@PathVariable Long[] ids) {
         return toAjax(equipmentLedgerService.deleteEquipmentLedgerByIds(ids));
     }
+
+    /**
+     * 导入设备台账列表
+     */
+    @PreAuthorize("@ss.hasPermi('blocked:rate:import')")
+    @Log(title = "导入设备台账列表", businessType = BusinessType.IMPORT)
+    @PostMapping("/importData")
+    public AjaxResult importData(MultipartFile file, boolean updateSupport) throws Exception {
+        ExcelUtil<EquipmentLedger> util = new ExcelUtil<>(EquipmentLedger.class);
+        List<EquipmentLedger> list = util.importExcel(file.getInputStream());
+        String message = equipmentLedgerService.importData(list, updateSupport);
+        return success(message);
+    }
+
+    /**
+     * 获取导入模板
+     */
+    @PostMapping("/importTemplate")
+    public void importTemplate(HttpServletResponse response) {
+        ExcelUtil<EquipmentLedger> util = new ExcelUtil<>(EquipmentLedger.class);
+        util.importTemplateExcel(response, "设备台账数据导入模板");
+    }
 }

airport-equipment/src/main/java/com/sundot/airport/equipment/domain/EquipmentLedger.java

@@ -36,7 +36,7 @@ public class EquipmentLedger extends BaseEntity {
     private Long id;
 
     /** 设备编号 */
-    @Excel(name = "设备编号")
+    @Excel(name = "设备编号", type = Excel.Type.EXPORT)
     private String equipmentCode;
 
     /** 设备名称 */
@@ -90,7 +90,7 @@ public class EquipmentLedger extends BaseEntity {
     private String initialAcceptanceStatus;
 
     /** 使用状态 */
-    @Excel(name = "使用状态")
+    @Excel(name = "使用状态", readConverterExp = "IN_USE=在用,SEND_FOR_REPAIR=寄修,LEND=借出,DEACTIVATE=停用,NOT_ENABLED=未启用,SUSPEND_USE=暂停使用", combo = "在用,寄修,借出,停用,未启用,暂停使用")
     private String usageStatus;
 
     /** 报废日期 */
@@ -120,7 +120,7 @@ public class EquipmentLedger extends BaseEntity {
     private Date nextInspectionDueDate;
 
     /** 定/自检小组组长ID */
-    @Excel(name = "定/自检小组组长ID")
+    @Excel(name = "定/自检小组组长ID", type = Excel.Type.EXPORT)
     private Long inspectionTeamLeaderId;
 
     /** 定/自检小组组长名称 */
@@ -128,7 +128,7 @@ public class EquipmentLedger extends BaseEntity {
     private String inspectionTeamLeaderName;
 
     /** 定/自检小组组员1ID */
-    @Excel(name = "定/自检小组组员1ID")
+    @Excel(name = "定/自检小组组员1ID", type = Excel.Type.EXPORT)
     private Long inspectionTeamMember1Id;
 
     /** 定/自检小组组员1名称 */
@@ -136,7 +136,7 @@ public class EquipmentLedger extends BaseEntity {
     private String inspectionTeamMember1Name;
 
     /** 定/自检小组组员2ID */
-    @Excel(name = "定/自检小组组员2ID")
+    @Excel(name = "定/自检小组组员2ID", type = Excel.Type.EXPORT)
     private Long inspectionTeamMember2Id;
 
     /** 定/自检小组组员2名称 */

airport-equipment/src/main/java/com/sundot/airport/equipment/service/IEquipmentLedgerService.java

@@ -80,4 +80,13 @@ public interface IEquipmentLedgerService extends IService<EquipmentLedger> {
      * @return 设备台账集合
      */
     public List<EquipmentLedger> selectTwoWeeksUpcomingInspectionList();
+
+    /**
+     * 导入设备台账列表
+     *
+     * @param list          数据列表
+     * @param updateSupport 是否更新支持
+     * @return 导入结果信息
+     */
+    public String importData(List<EquipmentLedger> list, boolean updateSupport);
 }

airport-equipment/src/main/java/com/sundot/airport/equipment/service/impl/EquipmentLedgerServiceImpl.java

@@ -1,6 +1,8 @@
 package com.sundot.airport.equipment.service.impl;
 
 import java.util.List;
+import java.util.Map;
+import java.util.stream.Collectors;
 
 import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.util.ObjUtil;
@@ -8,6 +10,8 @@ import cn.hutool.core.util.StrUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.sundot.airport.common.constant.Constants;
+import com.sundot.airport.common.core.domain.entity.SysDictData;
+import com.sundot.airport.common.core.domain.entity.SysUser;
 import com.sundot.airport.common.core.redis.RedisSerialGenerator;
 import com.sundot.airport.common.domain.BaseAttachment;
 import com.sundot.airport.common.enums.BaseAttachmentSourceTypeEnum;
@@ -19,6 +23,8 @@ import com.sundot.airport.common.utils.DateUtils;
 import com.sundot.airport.common.utils.EquipmentDateUtils;
 import com.sundot.airport.equipment.domain.EquipmentInspectionRecord;
 import com.sundot.airport.equipment.service.IEquipmentInspectionRecordService;
+import com.sundot.airport.system.service.ISysDictDataService;
+import com.sundot.airport.system.service.ISysUserService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import com.sundot.airport.equipment.mapper.EquipmentLedgerMapper;
@@ -42,6 +48,10 @@ public class EquipmentLedgerServiceImpl extends ServiceImpl<EquipmentLedgerMappe
     private IBaseAttachmentService baseAttachmentService;
     @Autowired
     private RedisSerialGenerator redisSerialGenerator;
+    @Autowired
+    private ISysDictDataService sysDictDataService;
+    @Autowired
+    private ISysUserService sysUserService;
 
     /**
      * 查询设备台账
@@ -268,4 +278,107 @@ public class EquipmentLedgerServiceImpl extends ServiceImpl<EquipmentLedgerMappe
     public List<EquipmentLedger> selectTwoWeeksUpcomingInspectionList() {
         return equipmentLedgerMapper.selectTwoWeeksUpcomingInspectionList();
     }
+
+    /**
+     * 导入设备台账数据
+     *
+     * @param list            数据列表
+     * @param isUpdateSupport 是否更新支持，如果已存在，则进行更新数据
+     * @return 结果
+     */
+    @Transactional(rollbackFor = Exception.class)
+    @Override
+    public String importData(List<EquipmentLedger> list, boolean isUpdateSupport) {
+        if (CollUtil.isEmpty(list)) {
+            throw new ServiceException("导入设备台账数据不能为空！");
+        }
+
+        List<SysUser> sysUserListAll = sysUserService.selectUserAll();
+        Map<String, Long> sysUserMap = sysUserListAll.stream().collect(Collectors.toMap(SysUser::getNickName, SysUser::getUserId, (oldValue, newValue) -> newValue));
+
+        SysDictData equipmentUsageStatusQuery = new SysDictData();
+        equipmentUsageStatusQuery.setDictType("equipment_usage_status");
+        List<SysDictData> equipmentUsageStatusList = sysDictDataService.selectDictDataList(equipmentUsageStatusQuery);
+        Map<String, String> equipmentUsageStatusMap = equipmentUsageStatusList.stream().collect(Collectors.toMap(SysDictData::getDictLabel, SysDictData::getDictValue));
+
+        int successNum = 0;
+        int failureNum = 0;
+        StringBuilder successMsg = new StringBuilder();
+        StringBuilder failureMsg = new StringBuilder();
+
+        for (EquipmentLedger data : list) {
+            // 根据名称填充ID字段
+            fillIdsByName(data, sysUserMap, equipmentUsageStatusMap);
+            try {
+                if (ObjUtil.isNull(data.getEquipmentSerialNumber())) {
+                    failureNum++;
+                    failureMsg.append("<br/>" + failureNum + "、设备序列号【" + data.getEquipmentSerialNumber() + "】、设备名称【" + data.getEquipmentName() + "】设备序列号不能为空");
+                    continue;
+                }
+                if (ObjUtil.isNull(data.getEquipmentName())) {
+                    failureNum++;
+                    failureMsg.append("<br/>" + failureNum + "、设备序列号【" + data.getEquipmentSerialNumber() + "】、设备名称【" + data.getEquipmentName() + "】设备名称不能为空");
+                    continue;
+                }
+
+                // 查询是否已存在（根据【设备序列号】唯一）
+                EquipmentLedger queryParam = new EquipmentLedger();
+                queryParam.setEquipmentSerialNumber(data.getEquipmentSerialNumber());
+                List<EquipmentLedger> existingList = equipmentLedgerMapper.selectEquipmentLedgerList(queryParam);
+
+                if (CollUtil.isEmpty(existingList)) {
+                    // 新增
+                    data.setEquipmentCode(redisSerialGenerator.generate(Constants.EQUIPMENT_LEDGER));
+                    data.setCreateTime(DateUtils.getNowDate());
+                    equipmentLedgerMapper.insertEquipmentLedger(data);
+                    successNum++;
+                    successMsg.append("<br/>" + successNum + "、设备序列号【" + data.getEquipmentSerialNumber() + "】、设备名称【" + data.getEquipmentName() + "】导入成功");
+                } else if (isUpdateSupport) {
+                    // 更新
+                    EquipmentLedger old = existingList.get(0);
+                    data.setId(old.getId());
+                    data.setEquipmentCode(old.getEquipmentCode());
+                    data.setUpdateTime(DateUtils.getNowDate());
+                    equipmentLedgerMapper.updateEquipmentLedger(data);
+                    successNum++;
+                    successMsg.append("<br/>" + successNum + "、设备序列号【" + data.getEquipmentSerialNumber() + "】、设备名称【" + data.getEquipmentName() + "】更新成功");
+                } else {
+                    failureNum++;
+                    failureMsg.append("<br/>" + failureNum + "、设备序列号【" + data.getEquipmentSerialNumber() + "】、设备名称【" + data.getEquipmentName() + "】已存在");
+                }
+            } catch (Exception e) {
+                failureNum++;
+                String msg = "<br/>" + failureNum + "、设备序列号【" + data.getEquipmentSerialNumber() + "】、设备名称【" + data.getEquipmentName() + "】导入失败：";
+                failureMsg.append(msg + e.getMessage());
+            }
+        }
+
+        if (failureNum > 0) {
+            failureMsg.insert(0, "很抱歉，导入失败！共 " + failureNum + " 条数据格式不正确，错误如下：");
+            throw new ServiceException(failureMsg.toString());
+        } else {
+            successMsg.insert(0, "恭喜您，数据已全部导入成功！共 " + successNum + " 条，数据如下：");
+        }
+        return successMsg.toString();
+    }
+
+    /**
+     * 根据名称填充ID字段
+     *
+     * @param data 速率数据
+     */
+    private void fillIdsByName(EquipmentLedger data, Map<String, Long> sysUserMap, Map<String, String> equipmentUsageStatusMap) {
+        // 定/自检小组组长
+        if (ObjUtil.isNotNull(data.getInspectionTeamLeaderName())) {
+            data.setInspectionTeamLeaderId(sysUserMap.get(data.getInspectionTeamLeaderName()));
+        }
+        // 定/自检小组组员1
+        if (ObjUtil.isNotNull(data.getInspectionTeamMember1Name())) {
+            data.setInspectionTeamMember1Id(sysUserMap.get(data.getInspectionTeamMember1Name()));
+        }
+        // 定/自检小组组员2
+        if (ObjUtil.isNotNull(data.getInspectionTeamMember2Name())) {
+            data.setInspectionTeamMember2Id(sysUserMap.get(data.getInspectionTeamMember2Name()));
+        }
+    }
 }