From 832d80c13f96155264fc35ffc8b1136dbdf5946b Mon Sep 17 00:00:00 2001
From: Administrator <admin>
Date: Thu, 12 Aug 2021 21:38:54 +0800
Subject: [PATCH] 公司装备导入

---
 src/main/java/org/springblade/modules/equipage/service/impl/EquipageServiceImpl.java |   61 ++++++++++++++++++++++++++++++
 1 files changed, 61 insertions(+), 0 deletions(-)

diff --git a/src/main/java/org/springblade/modules/equipage/service/impl/EquipageServiceImpl.java b/src/main/java/org/springblade/modules/equipage/service/impl/EquipageServiceImpl.java
index d55061a..3a849d4 100644
--- a/src/main/java/org/springblade/modules/equipage/service/impl/EquipageServiceImpl.java
+++ b/src/main/java/org/springblade/modules/equipage/service/impl/EquipageServiceImpl.java
@@ -4,18 +4,30 @@
 
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import lombok.AllArgsConstructor;
 import org.springblade.modules.equipage.entity.Equipage;
+import org.springblade.modules.equipage.excel.EquipageExcel;
 import org.springblade.modules.equipage.mapper.EquipageMapper;
 import org.springblade.modules.equipage.service.EquipageService;
 import org.springblade.modules.equipage.vo.EquipageVo;
+import org.springblade.modules.system.service.IUserDeptService;
+import org.springblade.modules.system.service.IUserService;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.util.List;
 
 /**
  * 装备服务实现类
  * @author zhongrj
  */
 @Service
+@AllArgsConstructor
 public class EquipageServiceImpl extends ServiceImpl<EquipageMapper, Equipage> implements EquipageService {
+
+	private final IUserService userService;
+
+	private final IUserDeptService userDeptService;
 
 	@Override
 	public IPage<EquipageVo> selectEquipagePage(IPage<EquipageVo> page, EquipageVo rquipage) {
@@ -30,4 +42,53 @@
 	public EquipageVo selectEquipageInfo(Equipage rquipage) {
 		return baseMapper.selectEquipageInfo(rquipage);
 	}
+
+	/**
+	 * 保安公司装备导入
+	 * @param data 装备数据集合
+	 * @param isCovered
+	 */
+	@Override
+	@Transactional(rollbackFor = Exception.class)
+	public void importEquipage(List<EquipageExcel> data, Boolean isCovered) {
+		if (data.size()>0){
+			//数据导入
+			data.forEach(equipageExcel -> {
+				Equipage equipage = new Equipage();
+				//使用责任人身份证号匹配userId
+				if (null!=equipageExcel.getIdCardNo() && equipageExcel.getIdCardNo()!=""){
+					equipage.setUserId(userService.getUserInfoByIdCardNo(equipageExcel.getIdCardNo()).getId());
+					equipage.setPersonInCharge(userService.getUserInfoByIdCardNo(equipageExcel.getIdCardNo()).getRealName());
+				}
+				//使用保安公司名称匹配组织机构id
+				if (null!=equipageExcel.getDeptName() && equipageExcel.getDeptName()!=""){
+					equipage.setDeptId(Long.parseLong(userDeptService.selectIn(equipageExcel.getDeptName())));
+				}
+
+				if (null!=equipageExcel.getBrand() && equipageExcel.getBrand()!="") {
+					equipage.setBrand(equipageExcel.getBrand());
+				}
+
+				if (null!=equipageExcel.getMode() && equipageExcel.getMode()!="") {
+					equipage.setMode(equipageExcel.getMode());
+				}
+
+				if (null!=equipageExcel.getName() && equipageExcel.getName()!="") {
+					equipage.setName(equipageExcel.getName());
+				}
+
+				if (null!=equipageExcel.getNumber() && equipageExcel.getNumber()!="") {
+					equipage.setNumber(equipageExcel.getNumber());
+				}
+
+				if (null!=equipageExcel.getSpecifications() && equipageExcel.getSpecifications()!="") {
+					equipage.setSpecifications(equipageExcel.getSpecifications());
+				}
+
+				//插入数据
+				this.save(equipage);
+
+			});
+		}
+	}
 }

--
Gitblit v1.9.3