From b0625931bc528429d592f642bdd759e7dbdfb1c3 Mon Sep 17 00:00:00 2001
From: Administrator <admin>
Date: Tue, 19 Oct 2021 22:21:07 +0800
Subject: [PATCH] 1.保安员新增接口修改,单个新增保安员默认密码为身份证号后6位,账号为身份证号, 2.考试成绩可以按天导出 3.个人图片批量导入接口修改 4.保安员批量导入修改,如有重复的账号导入的跳过,导入完成后返回提示信息

---
 src/main/java/org/springblade/modules/system/service/impl/UserServiceImpl.java |  201 +++++++++++++++++++++++++++++++++++++------------
 1 files changed, 151 insertions(+), 50 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 7b9fe05..cff531d 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
@@ -17,19 +17,17 @@
 package org.springblade.modules.system.service.impl;
 
 
-import com.baomidou.mybatisplus.core.conditions.Wrapper;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.google.zxing.WriterException;
 import lombok.AllArgsConstructor;
-import org.springblade.common.cache.DictCache;
-import org.springblade.common.cache.ParamCache;
+import org.apache.commons.lang3.StringUtils;
 import org.springblade.common.cache.SysCache;
-import org.springblade.common.cache.UserCache;
 import org.springblade.common.constant.AgeUtil;
 import org.springblade.common.constant.CommonConstant;
 import org.springblade.common.constant.TenantConstant;
-import org.springblade.common.enums.DictEnum;
+import org.springblade.common.utils.QRCodeUtil;
 import org.springblade.core.log.exception.ServiceException;
 import org.springblade.core.mp.base.BaseServiceImpl;
 import org.springblade.core.mp.support.Condition;
@@ -42,29 +40,28 @@
 import org.springblade.core.tool.utils.*;
 import org.springblade.modules.FTP.FtpUtil;
 import org.springblade.modules.auth.enums.UserEnum;
-import org.springblade.modules.dispatcher.service.IDispatcherService;
-import org.springblade.modules.dispatcher.vo.DispatcherUnitVO;
 import org.springblade.modules.dispatcher.vo.DispatcherVO;
 import org.springblade.modules.information.entity.Information;
 import org.springblade.modules.information.service.IInformationService;
-import org.springblade.modules.jurisdiction.entity.Jurisdiction;
 import org.springblade.modules.system.entity.*;
+import org.springblade.modules.system.excel.QrCodeExcel;
 import org.springblade.modules.system.excel.UserExcel;
 import org.springblade.modules.system.mapper.UserMapper;
 import org.springblade.modules.system.node.TreeNode;
-import org.springblade.modules.system.service.IRoleService;
-import org.springblade.modules.system.service.IUserDeptService;
-import org.springblade.modules.system.service.IUserOauthService;
-import org.springblade.modules.system.service.IUserService;
+import org.springblade.modules.system.service.*;
 import org.springblade.modules.system.vo.UserVO;
 import org.springblade.modules.system.wrapper.UserWrapper;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import java.io.FileNotFoundException;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.UnsupportedEncodingException;
+import java.net.URLEncoder;
 import java.text.SimpleDateFormat;
 import java.util.*;
-
-import static org.springblade.common.constant.CommonConstant.DEFAULT_PARAM_PASSWORD;
+import java.util.concurrent.atomic.AtomicBoolean;
 
 /**
  * 服务实现类
@@ -81,6 +78,7 @@
 	private final IRoleService roleService;
 	private final BladeTenantProperties tenantProperties;
 	private final IInformationService iInformationService;
+	private final IDeptService deptService;
 
 	@Override
 	@Transactional(rollbackFor = Exception.class)
@@ -105,7 +103,8 @@
 		if (Func.isNotEmpty(user.getPassword())) {
 			user.setPassword(DigestUtil.encrypt(user.getPassword()));
 		}
-		Integer userCount = baseMapper.selectCount(Wrappers.<User>query().lambda().eq(User::getTenantId, tenantId).eq(User::getAccount, user.getAccount()));
+//		Integer userCount = baseMapper.selectCount(Wrappers.<User>query().lambda().eq(User::getTenantId, tenantId).eq(User::getAccount, user.getAccount()));
+		Integer userCount = baseMapper.selectCountAccount(user.getAccount());
 		if (userCount > 0 && Func.isEmpty(user.getId())) {
 			throw new ServiceException(StringUtil.format("当前用户 [{}] 已存在!", user.getAccount()));
 		}
@@ -381,55 +380,109 @@
 	}
 
 	@Override
-	@Transactional(rollbackFor = Exception.class)
+//	@Transactional(rollbackFor = Exception.class)
 	public void importUser(List<UserExcel> data, Boolean isCovered,String deptId) {
+		//将不能导入的保安员账号存起来
+		List<String> errorList = new ArrayList<>();
+		//导入状态,默认为true ,如果有一个出现问题则为 false
+		AtomicBoolean status = new AtomicBoolean(true);
 		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");
-				}
-				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
 			String deptIds = userDeptService.selectIn(user.getDeptId());
 			if (null!=deptIds && !deptIds.equals("")) {
 				if (null != deptId && !deptId.equals("")) {
 					if (!deptId.equals(deptIds)) {
-						throw new ServiceException("不能导入不是本公司的保安员数据!");
+						throw new ServiceException("导入失败!不能导入不是本公司的保安员数据!");
 					}
 				}
 				user.setDeptId(deptIds);
+			}else {
+				//如果deptIds 为空,则说明还没有改公司
+				throw new ServiceException("导入失败!公司名:["+user.getDeptId()+"]不存在!");
 			}
-			//设置账号
-			user.setAccount(user.getCardid());
-			//获取默认密码配置
-			user.setPassword(user.getCardid().substring(user.getCardid().length() - 6));
+			//判断当前用户是否已在本单位,如果是的更新数据
+			User user1 = new User();
+			user1.setAccount(user.getCardid());
+//			System.out.println("user.getCardid() = " + user.getCardid());
+			user1.setIsDeleted(0);
+			user1.setStatus(1);
+			User user2 = this.getOne(Condition.getQueryWrapper(user1));
+			if (null==user2){
+				//用户不存在,去新增
+				// 设置租户ID
+				user.setTenantId("000000");
+				//默认在职
+				user.setStatus(1);
+				//判断是否持证
+				if (null != userExcel.getHold() && userExcel.getHold() != "") {
+					if (userExcel.getHold().equals("是")) {
+						user.setHold("1");
+					}
+					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);
+					}
+				}
+
+				//设置账号
+				user.setAccount(user.getCardid());
+				//获取默认密码配置
+				user.setPassword(user.getCardid().substring(user.getCardid().length() - 6));
 //			user.setPassword(ParamCache.getValue(DEFAULT_PARAM_PASSWORD));
-			//新增
-			this.submit(user);
+				//新增
+				this.submit(user);
+			}else {
+				//匹配组织机构是否一致,如果不一致
+				if(!user2.getDeptId().equals(user.getDeptId())){
+					Dept dept = deptService.getById(user2.getDeptId());
+//					throw new ServiceException("用户:"+"["+user.getCardid()+"]"+"已在其他单位存在!");
+					status.set(false);
+					//加入集合
+					errorList.add(user.getCardid());
+					//forEach 只能使用 return 跳出本次循环
+					return;
+				}else {
+					//如果是一致,则更新用户数据
+					//判断是否持证
+					if (null != userExcel.getHold() && userExcel.getHold() != "") {
+						if (userExcel.getHold().equals("是") && userExcel.getSecuritynumber()!=null && !userExcel.getSecuritynumber().equals("")) {
+							user2.setHold("1");
+							//更新保安证编号
+							user2.setSecuritynumber(user.getSecuritynumber());
+						}
+						if (userExcel.getHold().equals("否")) {
+							user2.setHold("2");
+						}
+					}
+					//更新用户数据
+					this.updateById(user2);
+					String s1 =
+						"update blade_user set hold = " + "'" + user2.getHold() + "'"
+							+ ",securitynumber = " + "'" + user2.getSecuritynumber() + "'"
+							+ " " + "where id = " + "'" + user2.getId() + "'";
+					FtpUtil.sqlFileUpload(s1);
+				}
+			}
 		});
+		//如果所有数据导入有一个异常
+		if (!status.get()){
+			String errorAccount = StringUtils.join(errorList, "\\\n");
+			throw new ServiceException("用户:["+errorAccount+"]导入失败!已在其他单位存在!");
+		}
 	}
 
 //	@Override
@@ -775,4 +828,52 @@
 	public User getUserInfo(String username) {
 		return baseMapper.getUserInfo(username);
 	}
+
+	/**
+	 * 生成二维码
+	 * @param data
+	 * @param isCovered
+	 * @param deptId
+	 */
+	@Override
+	public void importQrCode(List<QrCodeExcel> data, Boolean isCovered, String deptId) {
+		String url = "http://223.82.109.183:2080/securityInfo.html";
+		data.forEach(qrCodeExcel -> {
+			if (null!=qrCodeExcel.getSecuritynumber()) {
+				String encoded = null;
+				try {
+					//中文字符串编码
+					encoded = URLEncoder.encode(qrCodeExcel.getSecuritynumber(),"UTF-8");
+				} catch (UnsupportedEncodingException e) {
+					e.printStackTrace();
+				}
+				//url 拼接
+				String content = url + "?securityNumber=" + encoded;
+				byte[] qrCodeImage = new byte[0];
+				try {
+					//生成二维码字节流
+					qrCodeImage = QRCodeUtil.getQRCodeImage(content, 350, 350);
+				} catch (WriterException e) {
+					e.printStackTrace();
+				} catch (IOException e) {
+					e.printStackTrace();
+				}
+				String path = "D:\\QrCode\\";
+				FileOutputStream fileOutputStream = null;
+				try {
+					fileOutputStream = new FileOutputStream(path+qrCodeExcel.getSecuritynumber()+".png");
+				} catch (FileNotFoundException e) {
+					e.printStackTrace();
+				}
+				try {
+					//图片下载到本地
+					fileOutputStream.write(qrCodeImage,0,qrCodeImage.length);
+					fileOutputStream.flush();
+					fileOutputStream.close();
+				} catch (IOException e) {
+					e.printStackTrace();
+				}
+			}
+		});
+	}
 }

--
Gitblit v1.9.3