diff --git a/pom.xml b/pom.xml
index 25b1f63..fd39473 100644
--- a/pom.xml
+++ b/pom.xml
@@ -137,6 +137,21 @@
true
+
+ org.apache.maven.plugins
+ maven-resources-plugin
+
+
+ xlsx
+ xls
+ zip
+ cer
+ pfx
+ py
+ keystore
+
+
+
diff --git a/web/src/main/java/com/zhehekeji/web/controller/CategoryController.java b/web/src/main/java/com/zhehekeji/web/controller/CategoryController.java
index b115b49..db68c7a 100644
--- a/web/src/main/java/com/zhehekeji/web/controller/CategoryController.java
+++ b/web/src/main/java/com/zhehekeji/web/controller/CategoryController.java
@@ -7,11 +7,16 @@ import com.zhehekeji.web.pojo.category.PageSearch;
import com.zhehekeji.web.service.CategoryService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
+import org.springframework.core.io.ClassPathResource;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import javax.annotation.Resource;
+import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.net.URLEncoder;
import java.util.List;
@Api(value = "category",tags = "分类管理")
@@ -62,4 +67,26 @@ public class CategoryController {
categoryService.del(id);
return Result.success();
}
+
+ @GetMapping("/export/model")
+ @ApiOperation(value = "品规模板下载")
+ public void download(HttpServletResponse response) throws IOException {
+ OutputStream os = response.getOutputStream();
+ byte[] buffer = new byte[4096];
+ int n;
+ response.setContentType("application/vnd.ms-excel");
+ response.setCharacterEncoding("utf-8");
+ // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
+ String fileName = URLEncoder.encode("品规模板", "UTF-8");
+ response.setHeader("Content-disposition", "attachment;filename="+fileName+".xlsx");
+ ClassPathResource classPathResource = new ClassPathResource("品规模板.xlsx");
+ InputStream in = classPathResource.getInputStream();
+ while ((n = in.read(buffer)) > 0) {
+ os.write(buffer, 0, n);
+ }
+ os.flush();
+ os.close();
+ response.flushBuffer();
+
+ }
}
diff --git a/web/src/main/java/com/zhehekeji/web/controller/StockController.java b/web/src/main/java/com/zhehekeji/web/controller/StockController.java
index feff60b..46dd5ec 100644
--- a/web/src/main/java/com/zhehekeji/web/controller/StockController.java
+++ b/web/src/main/java/com/zhehekeji/web/controller/StockController.java
@@ -13,20 +13,21 @@ import com.zhehekeji.web.pojo.stock.StockExportExcel;
import com.zhehekeji.web.service.StockService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
+import org.springframework.core.io.ClassPathResource;
+import org.springframework.util.ResourceUtils;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
-import java.io.IOException;
-import java.io.OutputStream;
+import java.io.*;
import java.net.URLEncoder;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.List;
import java.util.Map;
-@Api(value = "Stock",tags = "库存管理")
+@Api(value = "Stock", tags = "库存管理")
@RestController
@RequestMapping("/stock")
public class StockController {
@@ -38,20 +39,42 @@ public class StockController {
@PostMapping("/import")
@ApiOperation(value = "库存导入")
- public Result upload(MultipartFile file){
+ public Result upload(MultipartFile file) {
stockService.importExcel(file);
return Result.success();
}
+ @GetMapping("/export/model")
+ @ApiOperation(value = "库存模板下载")
+ public void download(HttpServletResponse response) throws IOException {
+ OutputStream os = response.getOutputStream();
+ byte[] buffer = new byte[4096];
+ int n;
+ response.setContentType("application/vnd.ms-excel");
+ response.setCharacterEncoding("utf-8");
+ // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
+ String fileName = URLEncoder.encode("库存模板", "UTF-8");
+ response.setHeader("Content-disposition", "attachment;filename="+fileName+".xlsx");
+ ClassPathResource classPathResource = new ClassPathResource("库存模板.xlsx");
+ InputStream in = classPathResource.getInputStream();
+ while ((n = in.read(buffer)) > 0) {
+ os.write(buffer, 0, n);
+ }
+ os.flush();
+ os.close();
+ response.flushBuffer();
+
+ }
+
@GetMapping("/list")
@ApiOperation(value = "盘点页面 返回map key:货架号 value:库存信息")
- public Result