下载方法改造

jeecg-boot-base-core/src/main/java/org/jeecg/common/util/ResponseOutputFile.java

@@ -0,0 +1,326 @@
+package org.jeecg.common.util;
+
+import cn.hutool.core.util.StrUtil;
+import lombok.Data;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.poi.ss.usermodel.Workbook;
+import org.jeecg.common.exception.JeecgBootException;
+import org.springframework.http.MediaType;
+import org.springframework.http.MediaTypeFactory;
+
+import javax.servlet.ServletOutputStream;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.io.*;
+import java.net.URLEncoder;
+
+/**
+ * @Author wgx
+ */
+@Slf4j
+public class ResponseOutputFile {
+    public static void output(HttpServletRequest request, HttpServletResponse response, Workbook sheets,
+                                String fileName) {
+        ServletOutputStream outputStream = null;
+        try {
+            //设置下载模式
+            String contentType = getContentType(fileName);
+            response.setContentType(contentType);
+            //设置下载模式
+            response.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode(fileName, "UTF-8") + "");
+            outputStream = response.getOutputStream();
+            sheets.write(outputStream);
+        } catch (IOException e) {
+            e.printStackTrace();
+        } finally {
+
+            if(outputStream != null){
+                try {
+                    outputStream.close();
+                } catch (IOException e) {
+                    throw new RuntimeException(e);
+                }
+            }
+        }
+    }
+    /**
+     * 导出文件
+     * @param request
+     * @param response
+     * @param in
+     * @param fileSize
+     * @param fileName
+     */
+    public static void output(HttpServletRequest request, HttpServletResponse response, InputStream in,
+                              long fileSize, String fileName) {
+        ServletOutputStream outputStream = null;
+        BufferedInputStream bufIn = null;
+        try {
+            //设置下载模式
+            String contentType = getContentType(fileName);
+            response.setContentLengthLong(fileSize);
+            response.setContentType(contentType);
+            // 设置强制下载不打开
+//            response.setContentType("application/force-download");
+            // 在线打开方式 文件名应该编码成UTF-8
+//            response.setHeader("Content-Disposition", "inline; filename=" + URLEncoder.encode(filename, "UTF-8"));
+            //设置下载模式
+            response.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode(fileName, "UTF-8") + "");
+
+            // 读到流中
+            bufIn = new BufferedInputStream(in);
+            byte[] buffer = new byte[2048];
+            int num;
+            outputStream = response.getOutputStream();
+            while ((num = bufIn.read(buffer)) != -1) {
+                outputStream.write(buffer, 0, num);
+            }
+            response.flushBuffer();
+        } catch (IOException e) {
+            e.printStackTrace();
+        }finally {
+            if(bufIn != null){
+                try {
+                    bufIn.close();
+                } catch (IOException e) {
+                    throw new RuntimeException(e);
+                }
+            }
+            if(in != null){
+                try {
+                    in.close();
+                } catch (IOException e) {
+                    throw new RuntimeException(e);
+                }
+            }
+            if(outputStream != null){
+                try {
+                    outputStream.close();
+                } catch (IOException e) {
+                    throw new RuntimeException(e);
+                }
+            }
+
+        }
+    }
+
+
+    //断点续传
+    public static void output(HttpServletRequest request, HttpServletResponse response, File file, boolean del){
+        if(file == null || !file.exists()){
+            throw new JeecgBootException("文件错误");
+        }
+        InputStream inputStream = null;
+        RandomAccessFile in = null;
+        ServletOutputStream outputStream = null;
+        try {
+
+            //设置下载模式
+            String fileName = file.getName();
+//            String type = fileName.substring(fileName.indexOf(".") + 1);
+            String contentType = getContentType(fileName);
+            String range = request.getHeader("Range");
+            long fileSize = file.length();
+            long downloadSize = fileSize;
+            long fromPos = 0, toPos = fileSize - 1;
+            if(range == null){
+                response.setContentLengthLong(downloadSize);
+            }else{// 断点续传
+                // https://blog.csdn.net/liang19890820/article/details/53215087#%E6%96%AD%E7%82%B9%E7%BB%AD%E4%BC%A0%E7%9A%84%E7%94%A8%E9%80%94
+                // https://blog.csdn.net/qq_33868430/article/details/83308315
+                // 若客户端传来Range，说明之前下载了一部分，设置206状态(SC_PARTIAL_CONTENT)
+                //Range 头部的格式有以下几种情况：
+                //Range: bytes=0-499 表示第 0-499 字节范围的内容
+                //Range: bytes=500-999 表示第 500-999 字节范围的内容
+                //Range: bytes=-500 表示最后 500 字节的内容
+                //Range: bytes=500- 表示从第 500 字节开始到文件结束部分的内容
+                //Range: bytes=0-0,-1 表示第一个和最后一个字节
+                //Range: bytes=500-600,601-999 同时指定几个范围
+                response.setStatus(HttpServletResponse.SC_PARTIAL_CONTENT);
+                RangeDTO rangeDTO = executeRangeInfo(range, fileSize);
+                if(rangeDTO == null){
+                    throw new JeecgBootException("range error");
+                }
+                fromPos = rangeDTO.getStartByte();
+                toPos = rangeDTO.getEndByte();
+//                String bytes = range.replaceAll("bytes=", "");
+//                String[] ary = bytes.split("-");
+//                fromPos = Long.parseLong(ary[0]);
+//                if (ary.length == 2) {
+//                    toPos = Long.parseLong(ary[1]);
+//                }
+                int size;
+                if (toPos > fromPos) {
+                    size = (int) (toPos - fromPos + 1);
+                } else {
+                    size = (int) (fileSize - fromPos);
+                }
+                response.setContentLengthLong(size);
+                //Content-Range 说明
+                //Content-Range: bytes 0-499/22400
+                //0－499 是指当前发送的数据的范围，而 22400 则是文件的总大小。
+                response.setHeader("Content-Range", "bytes " + fromPos + "-" + toPos + "/" + fileSize);
+                downloadSize = size;
+            }
+            response.setContentType(contentType);
+            // 设置强制下载不打开
+//            response.setContentType("application/force-download");
+            // 在线打开方式 文件名应该编码成UTF-8
+//            response.setHeader("Content-Disposition", "inline; filename=" + URLEncoder.encode(filename, "UTF-8"));
+            //设置下载模式
+            if(!(contentType.contains("image") || contentType.contains("video"))){
+                //非图片或视频则下载
+                response.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode(file.getName(), "UTF-8") + "");
+            }
+            response.setHeader("Accept-Ranges", "bytes");
+            // 读到流中
+            in = new RandomAccessFile(file, "rw");
+            // 设置下载起始位置
+            if (fromPos > 0) {
+                in.seek(fromPos);
+            }
+            byte[] buffer = new byte[2048];
+            int num;
+            outputStream = response.getOutputStream();
+            while ((num = in.read(buffer)) != -1) {
+                outputStream.write(buffer, 0, num);
+            }
+            response.flushBuffer();
+
+//            inputStream = new BufferedInputStream(new FileInputStream(file));
+//            outputStream = response.getOutputStream();
+//            byte[] buf = new byte[1024];
+//            int len;
+//            while ((len = inputStream.read(buf)) > 0) {
+//                outputStream.write(buf, 0, len);
+//            }
+//            response.flushBuffer();
+        } catch (IOException e) {
+            e.printStackTrace();
+        }finally {
+            if(inputStream != null){
+                try {
+                    inputStream.close();
+                } catch (IOException e) {
+                    throw new RuntimeException(e);
+                }
+            }
+            if(in != null){
+                try {
+                    in.close();
+                } catch (IOException e) {
+                    throw new RuntimeException(e);
+                }
+            }
+            if(outputStream != null){
+                try {
+                    outputStream.close();
+                } catch (IOException e) {
+                    throw new RuntimeException(e);
+                }
+            }
+
+            if(del){
+                try {
+                    file.delete();
+                }catch (Exception e){
+                    log.error(e.getMessage());
+                }
+            }
+        }
+    }
+    private static RangeDTO executeRangeInfo(String range, Long fileSize) {
+
+        if (StrUtil.isEmpty(range) || !range.contains("bytes=") || !range.contains("-")) {
+
+            return null;
+        }
+
+        long startByte = 0;
+        long endByte = fileSize - 1;
+
+        range = range.substring(range.lastIndexOf("=") + 1).trim();
+        String[] ranges = range.split("-");
+
+        if (ranges.length <= 0 || ranges.length > 2) {
+
+            return null;
+        }
+
+        try {
+            if (ranges.length == 1) {
+                if (range.startsWith("-")) {
+
+                    //1. 如：bytes=-1024  从开始字节到第1024个字节的数据
+                    endByte = Long.parseLong(ranges[0]);
+                } else if (range.endsWith("-")) {
+
+                    //2. 如：bytes=1024-  第1024个字节到最后字节的数据
+                    startByte = Long.parseLong(ranges[0]);
+                }
+            } else {
+                //3. 如：bytes=1024-2048  第1024个字节到2048个字节的数据
+                startByte = Long.parseLong(ranges[0]);
+                endByte = Long.parseLong(ranges[1]);
+            }
+        } catch (NumberFormatException e) {
+            startByte = 0;
+            endByte = fileSize - 1;
+        }
+
+        if (startByte >= fileSize) {
+
+            log.error("range error, startByte >= fileSize. " +
+                    "startByte: {}, fileSize: {}", startByte, fileSize);
+            return null;
+        }
+
+        return new RangeDTO(startByte, endByte);
+    }
+
+
+    public static String getContentType(String fileName){
+        return MediaTypeFactory.getMediaType(fileName)
+                .orElse(MediaType.APPLICATION_OCTET_STREAM).toString();
+
+//        //String contentType = "application/force-download"
+//        if(StrUtil.isBlank(fileName)){
+//            return contentType;
+//        }
+//        MediaTypeFactory.getMediaType(fileName);
+//        // fileName 文件名
+//
+//        String type = fileName.substring(fileName.indexOf(".") + 1);
+//        if (Objects.equals(type, "jpg") || Objects.equals(type, "JPG")) {
+//            contentType = "image/jpeg";
+//        }else if(Objects.equals(type, "png") || Objects.equals(type, "PNG")) {
+//            contentType =  "image/png";
+//        }else if(Objects.equals(type, "xlsx") || Objects.equals(type, "xls")){
+//            contentType = "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet";
+//        }else if(Objects.equals(type, "txt")){
+//            contentType = "text/plain; charset=utf-8";
+//        }else if(Objects.equals(type, "pdf")){
+//            contentType = "application/pdf";
+//        }else if(Objects.equals(type, "doc")){
+//            contentType = "application/msword";
+//        }else if(Objects.equals(type, "docx")){
+//            contentType = "application/vnd.openxmlformats-officedocument.wordprocessingml.document";
+//        }else if(Objects.equals(type, "zip")){
+//            contentType = "application/zip";
+//        }
+//        return contentType;
+    }
+
+    @Data
+
+    static class  RangeDTO {
+        private long startByte;
+        private long endByte;
+
+        public RangeDTO(long startByte, long endByte){
+            this.startByte = startByte;
+            this.endByte = endByte;
+        }
+    }
+
+}

jeecg-module-system/jeecg-system-biz/src/main/java/org/jeecg/modules/system/controller/CommonController.java

@@ -7,11 +7,8 @@ import org.jeecg.common.api.vo.Result;
 import org.jeecg.common.constant.CommonConstant;
 import org.jeecg.common.constant.SymbolConstant;
 import org.jeecg.common.exception.JeecgBootException;
-import org.jeecg.common.util.CommonUtils;
+import org.jeecg.common.util.*;
-import org.jeecg.common.util.RestUtil;
-import org.jeecg.common.util.TokenUtils;
 import org.jeecg.common.util.filter.FileTypeFilter;
-import org.jeecg.common.util.oConvertUtils;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.http.HttpHeaders;
 import org.springframework.http.HttpMethod;
@@ -205,6 +202,69 @@ public class CommonController {
 //		return result;
 //	}
 
+//    /**
+//     * 预览图片&下载文件
+//     * 请求地址：http://localhost:8080/common/static/{user/20190119/e1fe9925bc315c60addea1b98eb1cb1349547719_1547866868179.jpg}
+//     *
+//     * @param request
+//     * @param response
+//     */
+//    @GetMapping(value = "/static/**")
+//    public void view(HttpServletRequest request, HttpServletResponse response) {
+//        // ISO-8859-1 ==> UTF-8 进行编码转换
+//        String imgPath = extractPathFromPattern(request);
+//        if(oConvertUtils.isEmpty(imgPath) || CommonConstant.STRING_NULL.equals(imgPath)){
+//            return;
+//        }
+//        // 其余处理略
+//        InputStream inputStream = null;
+//        OutputStream outputStream = null;
+//        try {
+//            imgPath = imgPath.replace("..", "").replace("../","");
+//            if (imgPath.endsWith(SymbolConstant.COMMA)) {
+//                imgPath = imgPath.substring(0, imgPath.length() - 1);
+//            }
+//            String filePath = uploadpath + File.separator + imgPath;
+//            File file = new File(filePath);
+//            if(!file.exists()){
+//                response.setStatus(404);
+//                throw new RuntimeException("文件["+imgPath+"]不存在..");
+//            }
+//            // 设置强制下载不打开
+//            response.setContentType("application/force-download");
+//            response.addHeader("Content-Disposition", "attachment;fileName=" + new String(file.getName().getBytes("UTF-8"),"iso-8859-1"));
+//            inputStream = new BufferedInputStream(new FileInputStream(filePath));
+//            outputStream = response.getOutputStream();
+//            byte[] buf = new byte[1024];
+//            int len;
+//            while ((len = inputStream.read(buf)) > 0) {
+//                outputStream.write(buf, 0, len);
+//            }
+//            response.flushBuffer();
+//        } catch (IOException e) {
+//            log.error("预览文件失败" + e.getMessage());
+//            response.setStatus(404);
+//            e.printStackTrace();
+//        } finally {
+//            if (inputStream != null) {
+//                try {
+//                    inputStream.close();
+//                } catch (IOException e) {
+//                    log.error(e.getMessage(), e);
+//                }
+//            }
+//            if (outputStream != null) {
+//                try {
+//                    outputStream.close();
+//                } catch (IOException e) {
+//                    log.error(e.getMessage(), e);
+//                }
+//            }
+//        }
+//
+//    }
+
+
     /**
      * 预览图片&下载文件
      * 请求地址：http://localhost:8080/common/static/{user/20190119/e1fe9925bc315c60addea1b98eb1cb1349547719_1547866868179.jpg}
@@ -216,35 +276,40 @@ public class CommonController {
     public void view(HttpServletRequest request, HttpServletResponse response) {
         // ISO-8859-1 ==> UTF-8 进行编码转换
         String imgPath = extractPathFromPattern(request);
-        if(oConvertUtils.isEmpty(imgPath) || CommonConstant.STRING_NULL.equals(imgPath)){
+        if (oConvertUtils.isEmpty(imgPath) || CommonConstant.STRING_NULL.equals(imgPath)) {
             return;
         }
         // 其余处理略
         InputStream inputStream = null;
         OutputStream outputStream = null;
         try {
-            imgPath = imgPath.replace("..", "").replace("../","");
+            imgPath = imgPath.replace("..", "").replace("../", "");
             if (imgPath.endsWith(SymbolConstant.COMMA)) {
                 imgPath = imgPath.substring(0, imgPath.length() - 1);
             }
             String filePath = uploadpath + File.separator + imgPath;
             File file = new File(filePath);
-            if(!file.exists()){
+            if (!file.exists()) {
                 response.setStatus(404);
-                throw new RuntimeException("文件["+imgPath+"]不存在..");
+                log.info("文件[" + imgPath + "]不存在..");
+                //throw new RuntimeException("文件["+imgPath+"]不存在..");
+                return;
             }
+            ResponseOutputFile.output(request, response, file, false);
             // 设置强制下载不打开
-            response.setContentType("application/force-download");
+//            response.setContentType("application/force-download");
-            response.addHeader("Content-Disposition", "attachment;fileName=" + new String(file.getName().getBytes("UTF-8"),"iso-8859-1"));
+//            response.setContentType(ResponseOutputFile.getContentType(file.getName()));
-            inputStream = new BufferedInputStream(new FileInputStream(filePath));
+//            response.setContentLengthLong(file.length());
-            outputStream = response.getOutputStream();
+//            response.addHeader("Content-Disposition", "attachment;fileName=" + new String(file.getName().getBytes("UTF-8"),"iso-8859-1"));
-            byte[] buf = new byte[1024];
+//            inputStream = new BufferedInputStream(new FileInputStream(filePath));
-            int len;
+//            outputStream = response.getOutputStream();
-            while ((len = inputStream.read(buf)) > 0) {
+//            byte[] buf = new byte[1024];
-                outputStream.write(buf, 0, len);
+//            int len;
-            }
+//            while ((len = inputStream.read(buf)) > 0) {
-            response.flushBuffer();
+//                outputStream.write(buf, 0, len);
-        } catch (IOException e) {
+//            }
+//            response.flushBuffer();
+        } catch (Exception e) {
             log.error("预览文件失败" + e.getMessage());
             response.setStatus(404);
             e.printStackTrace();