From 345d17cf82e1bbc1fbfbeec10193c2bed59cd674 Mon Sep 17 00:00:00 2001
From: Administrator <admin>
Date: Mon, 27 Sep 2021 22:21:21 +0800
Subject: [PATCH] 注册登录,车辆,装备,枪支,招聘,培训考试修改

---
 src/main/java/org/springblade/modules/system/controller/UserController.java |   24 ++++++++++++++++++++----
 1 files changed, 20 insertions(+), 4 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..672d5a9 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;
@@ -71,6 +74,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,8 +366,8 @@
 	@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("操作成功");
 	}
@@ -413,7 +418,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 +454,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