From 298655c560cd54e088d6e11affda9d169f5e2b5f Mon Sep 17 00:00:00 2001
From: Administrator <admin>
Date: Thu, 26 Aug 2021 16:34:00 +0800
Subject: [PATCH] 导入接口修改

---
 src/main/java/org/springblade/modules/system/service/impl/UserServiceImpl.java |  107 ++++++++++++++++++++++++++++++++++++++++++++---------
 1 files changed, 89 insertions(+), 18 deletions(-)

diff --git a/src/main/java/org/springblade/modules/system/service/impl/UserServiceImpl.java b/src/main/java/org/springblade/modules/system/service/impl/UserServiceImpl.java
index 5dc0476..8657224 100644
--- a/src/main/java/org/springblade/modules/system/service/impl/UserServiceImpl.java
+++ b/src/main/java/org/springblade/modules/system/service/impl/UserServiceImpl.java
@@ -304,28 +304,40 @@
 		data.forEach(userExcel -> {
 			User user = Objects.requireNonNull(BeanUtil.copy(userExcel, User.class));
 			// 设置租户ID
-			if (!AuthUtil.isAdministrator() || StringUtil.isBlank(user.getTenantId())) {
-				user.setTenantId(AuthUtil.getTenantId());
-			}
-			if (null != userExcel.getStatus() && userExcel.getStatus() != "") {
-				if (userExcel.getStatus().equals("在职")) {
-					user.setStatus(1);
-				} else {
-					user.setStatus(2);
+			user.setTenantId("000000");
+			//默认在职
+			user.setStatus(1);
+			//判断是否持证
+			if (null != userExcel.getHold() && userExcel.getHold() != "") {
+				if (userExcel.getHold().equals("是")) {
+					user.setHold("1");
+					//分配保安角色
+					Role role = new Role();
+					role.setRoleAlias("保安");
+					Role oneRole = roleService.getOne(Condition.getQueryWrapper(role));
+					user.setRoleId(oneRole.getId().toString());
+				}
+				if(userExcel.getHold().equals("否")) {
+					user.setHold("2");
+					Role role = new Role();
+					role.setRoleAlias("未持证保安");
+					Role oneRole = roleService.getOne(Condition.getQueryWrapper(role));
+					user.setRoleId(oneRole.getId().toString());
 				}
 			}
-			if (userExcel.getSex().equals("男性")) {
-				user.setSex(1);
-			} else {
-				user.setSex(2);
+			if (null!=userExcel.getSex()) {
+				if (userExcel.getSex().equals("男")) {
+					user.setSex(1);
+				}
+				if (userExcel.getSex().equals("女")) {
+					user.setSex(2);
+				}
 			}
-			user.setRoleId("1412226235153731586");
-			String deptname = user.getDeptId();
-			String id = userDeptService.selectIn(deptname);
-			user.setDeptId(id);
+			//设置部门id
+			user.setDeptId(userDeptService.selectIn(user.getDeptId()));
 			//获取默认密码配置
-			String initPassword = ParamCache.getValue(DEFAULT_PARAM_PASSWORD);
-			user.setPassword(initPassword);
+			user.setPassword(ParamCache.getValue(DEFAULT_PARAM_PASSWORD));
+			//新增
 			this.submit(user);
 		});
 	}
@@ -523,4 +535,63 @@
 		return baseMapper.selectUser();
 	}
 
+	/**
+	 * 导入保安员
+	 * @param data
+	 * @param isCovered
+	 */
+	@Override
+	@Transactional(rollbackFor = Exception.class)
+	public void importSecurity(List<UserExcel> data, Boolean isCovered) {
+		data.forEach(userExcel -> {
+			User user = Objects.requireNonNull(BeanUtil.copy(userExcel, User.class));
+			// 设置租户ID
+			user.setTenantId("000000");
+			//默认在职
+			user.setStatus(1);
+			//判断是否持证
+			if (null != userExcel.getHold() && userExcel.getHold() != "") {
+				if (userExcel.getHold().equals("是")) {
+					user.setHold("1");
+					//分配保安角色
+					Role role = new Role();
+					role.setRoleAlias("保安");
+					Role oneRole = roleService.getOne(Condition.getQueryWrapper(role));
+					user.setRoleId(oneRole.getId().toString());
+				}
+				if(userExcel.getHold().equals("否")) {
+					user.setHold("2");
+					Role role = new Role();
+					role.setRoleAlias("未持证保安");
+					Role oneRole = roleService.getOne(Condition.getQueryWrapper(role));
+					user.setRoleId(oneRole.getId().toString());
+				}
+			}
+			if (null!=userExcel.getSex()) {
+				if (userExcel.getSex().equals("男")) {
+					user.setSex(1);
+				}
+				if (userExcel.getSex().equals("女")) {
+					user.setSex(2);
+				}
+			}
+			//设置部门id
+			user.setDeptId(userDeptService.selectIn(user.getDeptId()));
+			//获取默认密码配置
+			user.setPassword(ParamCache.getValue(DEFAULT_PARAM_PASSWORD));
+			//新增
+			this.submit(user);
+		});
+	}
+
+
+	/**
+	 * 根据保安员编码查询保安信息
+	 * @param securityNumber 保安证编码
+	 * @return
+	 */
+	@Override
+	public UserVO getUserInfoBySecurityNumber(String securityNumber) {
+		return baseMapper.getUserInfoBySecurityNumber(securityNumber);
+	}
 }

--
Gitblit v1.9.3