From 3d356c8409b3c73e904b9a1ce866c342ccfea10c Mon Sep 17 00:00:00 2001
From: Administrator <admin>
Date: Fri, 15 Oct 2021 14:49:47 +0800
Subject: [PATCH] 保安员证二维码接口新增,修改

---
 src/main/java/org/springblade/modules/system/service/impl/UserServiceImpl.java |  168 +++++++++++++++++++++++++++++++++++++++++++------------
 1 files changed, 131 insertions(+), 37 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 a7e5f52..3a5192f 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
@@ -21,6 +21,7 @@
 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;
@@ -30,6 +31,7 @@
 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;
@@ -49,18 +51,21 @@
 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.*;
 
@@ -81,6 +86,7 @@
 	private final IRoleService roleService;
 	private final BladeTenantProperties tenantProperties;
 	private final IInformationService iInformationService;
+	private final IDeptService deptService;
 
 	@Override
 	@Transactional(rollbackFor = Exception.class)
@@ -386,33 +392,6 @@
 	public void importUser(List<UserExcel> data, Boolean isCovered,String deptId) {
 		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("")) {
@@ -422,14 +401,81 @@
 					}
 				}
 				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()+"]"+"已在["+dept.getDeptName()+"]存在!");
+				}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);
+				}
+			}
 		});
 	}
 
@@ -776,4 +822,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