tongtong/youlai-boot
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: 用户导入模板下载使用 try-with-resources 释放资源

Browse Source
This commit is contained in:
Ray.Hao
2025-03-31 14:54:26 +08:00
parent 8cfa734238
commit 482bd16f62
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/main/java/com/youlai/boot/system/controller/UserController.java
View File

@@ -136,7 +136,7 @@ public class UserController {
@Operation(summary = "用户导入模板下载") @Operation(summary = "用户导入模板下载")
@GetMapping("/template") @GetMapping("/template")
@Log(value = "用户导入模板下载", module = LogModuleEnum.USER) @Log(value = "用户导入模板下载", module = LogModuleEnum.USER)
public void downloadTemplate(HttpServletResponse response) throws IOException { public void downloadTemplate(HttpServletResponse response) {
String fileName = "用户导入模板.xlsx"; String fileName = "用户导入模板.xlsx";
response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"); response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
response.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode(fileName, StandardCharsets.UTF_8)); response.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode(fileName, StandardCharsets.UTF_8));
@@ -144,10 +144,12 @@ public class UserController {
String fileClassPath = "templates" + File.separator + "excel" + File.separator + fileName; String fileClassPath = "templates" + File.separator + "excel" + File.separator + fileName;
InputStream inputStream = this.getClass().getClassLoader().getResourceAsStream(fileClassPath); InputStream inputStream = this.getClass().getClassLoader().getResourceAsStream(fileClassPath);
ServletOutputStream outputStream = response.getOutputStream(); try (ServletOutputStream outputStream = response.getOutputStream();
ExcelWriter excelWriter = EasyExcel.write(outputStream).withTemplate(inputStream).build(); ExcelWriter excelWriter = EasyExcel.write(outputStream).withTemplate(inputStream).build()) {
excelWriter.finish();
excelWriter.finish(); } catch (IOException e) {
throw new RuntimeException("用户导入模板下载失败", e);
}
} }
@Operation(summary = "导入用户") @Operation(summary = "导入用户")