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/controller/UserController.java |   41 ++++++++++++++++++++++++++++++++++-------
 1 files changed, 34 insertions(+), 7 deletions(-)

diff --git a/src/main/java/org/springblade/modules/system/controller/UserController.java b/src/main/java/org/springblade/modules/system/controller/UserController.java
index 48e5973..b34520c 100644
--- a/src/main/java/org/springblade/modules/system/controller/UserController.java
+++ b/src/main/java/org/springblade/modules/system/controller/UserController.java
@@ -17,6 +17,7 @@
 package org.springblade.modules.system.controller;
 
 
+import com.alibaba.excel.EasyExcel;
 import com.aliyun.oss.ServiceException;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
@@ -28,9 +29,11 @@
 import io.swagger.annotations.ApiParam;
 import com.alibaba.fastjson.JSON;
 import lombok.AllArgsConstructor;
+import org.apache.commons.codec.Charsets;
 import org.springblade.common.cache.DictCache;
 import org.springblade.common.config.FtpConfig;
 import org.springblade.common.enums.DictEnum;
+import org.springblade.common.excel.RowWriteHandler;
 import org.springblade.common.utils.arg;
 import org.springblade.core.cache.utils.CacheUtil;
 import org.springblade.core.excel.util.ExcelUtil;
@@ -54,9 +57,7 @@
 import org.springblade.modules.system.entity.Dept;
 import org.springblade.modules.system.entity.Role;
 import org.springblade.modules.system.entity.User;
-import org.springblade.modules.system.excel.SecurityImporter;
-import org.springblade.modules.system.excel.UserExcel;
-import org.springblade.modules.system.excel.UserImporter;
+import org.springblade.modules.system.excel.*;
 import org.springblade.modules.system.node.TreeNode;
 import org.springblade.modules.system.service.IDeptService;
 import org.springblade.modules.system.service.IRoleService;
@@ -71,6 +72,8 @@
 
 import javax.servlet.http.HttpServletResponse;
 import javax.validation.Valid;
+import java.io.IOException;
+import java.net.URLEncoder;
 import java.text.SimpleDateFormat;
 import java.util.*;
 
@@ -361,9 +364,22 @@
 	@PostMapping("import-user")
 	@ApiOperationSupport(order = 12)
 	@ApiOperation(value = "导入用户", notes = "传入excel")
-	public R importUser(MultipartFile file, Integer isCovered) {
-		UserImporter userImporter = new UserImporter(userService, false);
+	public R importUser(MultipartFile file, Integer isCovered,String deptId) {
+		UserImporter userImporter = new UserImporter(userService, false,deptId);
 		ExcelUtil.save(file, userImporter, UserExcel.class);
+		return R.success("操作成功");
+	}
+
+
+	/**
+	 * 保安员证二维码批量生成
+	 */
+	@PostMapping("import-qrCode")
+	@ApiOperationSupport(order = 12)
+	@ApiOperation(value = "导入用户", notes = "传入excel")
+	public R importQrCode(@RequestParam("file") MultipartFile file, Integer isCovered,String deptId) {
+		QrCodeImporter qrCodeImporter = new QrCodeImporter(userService, false,deptId);
+		ExcelUtil.save(file, qrCodeImporter, QrCodeExcel.class);
 		return R.success("操作成功");
 	}
 
@@ -413,7 +429,7 @@
 	@GetMapping("export-template-security")
 	@ApiOperationSupport(order = 14)
 	@ApiOperation(value = "导出模板")
-	public void exportSecurity(HttpServletResponse response) {
+	public void exportSecurity(HttpServletResponse response) throws IOException {
 		List<UserExcel> list = new ArrayList<>();
 		UserExcel userExcel = new UserExcel();
 		userExcel.setDeptId("xxx保安公司");
@@ -449,7 +465,18 @@
 		userExcel1.setHold("否");
 		userExcel1.setSecuritynumber("");
 		list.add(userExcel1);
-		ExcelUtil.export(response, "保安员导入数据模板", "保安员数据表", list, UserExcel.class);
+		String fileName = null;
+		try {
+			response.setContentType("application/vnd.ms-excel");
+			response.setCharacterEncoding(org.apache.commons.codec.Charsets.UTF_8.name());
+			fileName = URLEncoder.encode("保安员导入数据模板", Charsets.UTF_8.name());
+			response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");
+			//修改单元格格式为文本格式
+			EasyExcel.write(response.getOutputStream(), UserExcel.class).sheet("保安员数据表").registerWriteHandler(new RowWriteHandler()).doWrite(list);
+		} catch (Throwable var6) {
+			throw var6;
+		}
+//		ExcelUtil.export(response, "保安员导入数据模板", "保安员数据表", list, UserExcel.class);
 	}
 
 

--
Gitblit v1.9.3