springboot下载模板

Author: deserveeeeee

pom.xml

@@ -50,7 +50,28 @@
 				<groupId>org.springframework.boot</groupId>
 				<artifactId>spring-boot-maven-plugin</artifactId>
 			</plugin>
+			<plugin>
+				<groupId>org.apache.maven.plugins</groupId>
+				<artifactId>maven-resources-plugin</artifactId>
+				<configuration>
+					<!-- 过滤不编译文件类型，解决maven打包时会编译特定文件导致文件不可用的问题 -->
+					<nonFilteredFileExtensions>
+						<nonFilteredFileExtension>xlsx</nonFilteredFileExtension>
+						<nonFilteredFileExtension>xls</nonFilteredFileExtension>
+					</nonFilteredFileExtensions>
+				</configuration>
+			</plugin>
 		</plugins>
+		<resources>
+			<resource>
+				<directory>src/main/resources</directory>
+				<filtering>true</filtering>
+				<includes>
+					<!-- 将 resources 目录下的所有文件都一起打包 -->
+					<include>**/*.*</include>
+				</includes>
+			</resource>
+		</resources>
 	</build>
 
 </project>

src/main/java/com/ramostear/application/controller/FileController.java

@@ -2,8 +2,11 @@
 
 import com.ramostear.application.model.FileInfo;
 import com.ramostear.application.util.FileUtil;
+import org.apache.tomcat.util.http.fileupload.IOUtils;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.core.io.InputStreamResource;
+import org.springframework.core.io.Resource;
+import org.springframework.core.io.ResourceLoader;
 import org.springframework.http.HttpHeaders;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.MediaType;
@@ -14,10 +17,14 @@
 import org.springframework.web.multipart.MultipartFile;
 
 import javax.annotation.PostConstruct;
+import javax.servlet.ServletOutputStream;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
 import java.io.*;
 import java.util.Collection;
 import java.util.HashMap;
 import java.util.Map;
+import java.util.Objects;
 
 /**
  * @author : ramostear
@@ -39,6 +46,9 @@ public class FileController {
 
     private static Map<String,FileInfo> fileRepository = new HashMap<>();
 
+//    @Resource
+    private ResourceLoader resourceLoader;
+
     @PostConstruct
     public void initFileRepository(){
         FileInfo file1 = new FileInfo ().setFileName ( "bg1.jpg" );
@@ -90,24 +100,35 @@ public String fileUpload(@RequestParam("file") MultipartFile file) throws IOExce
 
     @GetMapping("/download/{fileName}")
     @ResponseBody
-    public ResponseEntity<Object> downloadFile(@PathVariable(name = "fileName") String fileName) throws FileNotFoundException {
-
-        File file = new File ( fileUploadRootDir+fileName);
-        InputStreamResource resource = new InputStreamResource ( new FileInputStream ( file ) );
-
-        HttpHeaders headers = new HttpHeaders();
-        headers.add ( "Content-Disposition",String.format("attachment;filename=\"%s",fileName));
-        headers.add ( "Cache-Control","no-cache,no-store,must-revalidate" );
-        headers.add ( "Pragma","no-cache" );
-        headers.add ( "Expires","0" );
-
-        ResponseEntity<Object> responseEntity = ResponseEntity.ok()
-                .headers ( headers )
-                .contentLength ( file.length ())
-                .contentType(MediaType.parseMediaType ( "application/txt" ))
-                .body(resource);
-
-        return responseEntity;
+    public void downloadFile(@PathVariable(name = "fileName") String fileName,
+                             HttpServletResponse response){
+        try {
+            //获取要下载的模板名称
+            //设置要下载的文件的名称
+            response.setHeader("Content-Disposition", "attachment;filename=" + fileName);
+            //通知客服文件的MIME类型
+            response.setContentType("application/vnd.ms-excel;charset=UTF-8");
+            //获取文件的路径
+//            String filePath = getClass().getResource("/"+fileName).getPath();
+            InputStream input = getClass().getClassLoader().getResourceAsStream(fileName);
+//            ByteArrayOutputStream out = new ByteArrayOutputStream();
+//            System.out.println(filePath);
+//            FileInputStream input = new FileInputStream(filePath);
+            OutputStream out = response.getOutputStream();
+
+            byte[] b = new byte[2048];
+            int len;
+            while ((len = input.read(b)) != -1) {
+                out.write(b, 0, len);
+            }
+            out.flush();
+            out.close();
+            input.close();
+        } catch (Exception ex) {
+//            log.error("getApplicationTemplate :", ex);
+            System.out.println(ex);
+            //return Response.ok("应用导入模板下载失败！");
+        }
     }