|
@@ -1,6 +1,5 @@
|
|
|
package com.persagy.dmp.rwd.edit.controller;
|
|
|
|
|
|
-import com.persagy.dmp.rwd.edit.model.DownloadModel;
|
|
|
import com.persagy.dmp.rwd.edit.service.DownloadService;
|
|
|
import org.apache.poi.ss.usermodel.Workbook;
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
@@ -10,8 +9,7 @@ import org.springframework.web.bind.annotation.RequestParam;
|
|
|
import org.springframework.web.bind.annotation.RestController;
|
|
|
|
|
|
import javax.servlet.http.HttpServletResponse;
|
|
|
-import java.io.IOException;
|
|
|
-import java.io.OutputStream;
|
|
|
+import java.io.*;
|
|
|
import java.net.URLEncoder;
|
|
|
|
|
|
@RestController
|
|
@@ -21,13 +19,34 @@ public class DownloadController {
|
|
|
@Autowired
|
|
|
private DownloadService downloadService;
|
|
|
|
|
|
+// @GetMapping("/download")
|
|
|
+// private void download(@RequestParam String type, HttpServletResponse response) {
|
|
|
+// DownloadModel result = downloadService.download(type);
|
|
|
+// if ("workbook".equalsIgnoreCase(result.getType())) {
|
|
|
+// String fileName = result.getName();
|
|
|
+// downloadWorkbook(response, fileName, result.getWorkbook());
|
|
|
+// }
|
|
|
+// }
|
|
|
+
|
|
|
@GetMapping("/download")
|
|
|
- private void download(@RequestParam String type, HttpServletResponse response) {
|
|
|
- DownloadModel result = downloadService.download(type);
|
|
|
- if ("workbook".equalsIgnoreCase(result.getType())) {
|
|
|
- String fileName = result.getName();
|
|
|
- downloadWorkbook(response, fileName, result.getWorkbook());
|
|
|
- }
|
|
|
+ private void download(@RequestParam String type, HttpServletResponse response) throws Exception {
|
|
|
+ //读取文件
|
|
|
+ InputStream resourceAsStream = Thread.currentThread().getContextClassLoader().getResourceAsStream("file/批量上传变更记录模板.xlsx");
|
|
|
+ InputStream fis = new BufferedInputStream(resourceAsStream);
|
|
|
+ byte[] buffer = new byte[fis.available()];
|
|
|
+ fis.read(buffer);
|
|
|
+ fis.close();
|
|
|
+ // 清空response
|
|
|
+ response.reset();
|
|
|
+ // 设置response的Header
|
|
|
+ response.setHeader("Content-Disposition", "attachment;filename=" + java.net.URLEncoder.encode("批量导入标准模板.xls", "utf-8"));
|
|
|
+ //response.addHeader("Content-Disposition", "attachment;filename=" + new String(filename.getBytes()));
|
|
|
+ response.addHeader("Content-Length", "" + buffer.length);
|
|
|
+ OutputStream toClient = new BufferedOutputStream(response.getOutputStream());
|
|
|
+ response.setContentType("application/octet-stream");
|
|
|
+ toClient.write(buffer);
|
|
|
+ toClient.flush();
|
|
|
+ toClient.close();
|
|
|
}
|
|
|
|
|
|
private void downloadWorkbook(HttpServletResponse resp, String fileName, Workbook workbook) {
|