图片下载

src/main/java/com/izouma/awesomeadmin/service/OSSFileService.java

@@ -1,21 +1,26 @@
 package com.izouma.awesomeadmin.service;
 
-import java.io.InputStream;
-import java.io.UnsupportedEncodingException;
-import java.text.SimpleDateFormat;
-import java.util.ArrayList;
-import java.util.Date;
-import java.util.List;
-import java.util.Random;
-
-import com.aliyun.oss.model.*;
+import com.aliyun.oss.OSSClient;
+import com.aliyun.oss.model.OSSObject;
+import com.aliyun.oss.model.ObjectMetadata;
+import com.izouma.awesomeadmin.util.ImagesUtil;
+import com.izouma.awesomeadmin.util.PropertiesFileLoader;
 import org.apache.commons.lang.StringUtils;
 import org.apache.log4j.Logger;
 import org.springframework.stereotype.Service;
 
-import com.aliyun.oss.OSSClient;
-import com.izouma.awesomeadmin.util.ImagesUtil;
-import com.izouma.awesomeadmin.util.PropertiesFileLoader;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.io.*;
+import java.net.URLEncoder;
+import java.text.SimpleDateFormat;
+import java.util.Date;
+import java.util.List;
+import java.util.Random;
+import java.util.zip.Adler32;
+import java.util.zip.CheckedOutputStream;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipOutputStream;
 
 
 @Service
@@ -112,4 +117,158 @@ public class OSSFileService {
     }
 
 
+    /**
+     * 批量下载图片
+     *
+     * @param keyList  ossfileName 逗号分隔。
+     * @param zipName  压缩包名称
+     * @param request
+     * @param response
+     * @return
+     */
+    public HttpServletResponse zipFilesDown(List<String> keyList, String zipName, HttpServletRequest request, HttpServletResponse response) {
+
+        String accessKeyId = PropertiesFileLoader.getDefaultProperties("aliossid", "");
+        String accessKeySecret = PropertiesFileLoader.getDefaultProperties("aliosskey", "");
+        String endpoint = PropertiesFileLoader.getDefaultProperties("aliossendpoit", "");
+        String bucketName = PropertiesFileLoader.getDefaultProperties("alibucketname", "");
+
+        // endpoint以杭州为例，其它region请按实际情况填写
+        // 云账号AccessKey有所有API访问权限，建议遵循阿里云安全最佳实践，创建并使用RAM子账号进行API访问或日常运维，请登录 https://ram.console.aliyun.com 创建
+        try {
+            // 初始化
+            OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
+            String fileName = zipName + ".zip";
+            // 创建临时文件
+            File zipFile = File.createTempFile(zipName, ".zip");
+            FileOutputStream f = new FileOutputStream(zipFile);
+            /**
+             * 作用是为任何OutputStream产生校验和
+             * 第一个参数是制定产生校验和的输出流，第二个参数是指定Checksum的类型 （Adler32（较快）和CRC32两种）
+             */
+            CheckedOutputStream csum = new CheckedOutputStream(f, new Adler32());
+            // 用于将数据压缩成Zip文件格式
+            ZipOutputStream zos = new ZipOutputStream(csum);
+
+            for (String ossfile : keyList) {
+                // 获取Object，返回结果为OSSObject对象
+                OSSObject ossObject = ossClient.getObject(bucketName, ossfile);
+                // 读去Object内容  返回
+                InputStream inputStream = ossObject.getObjectContent();
+                // 对于每一个要被存放到压缩包的文件，都必须调用ZipOutputStream对象的putNextEntry()方法，确保压缩包里面文件不同名
+
+                zos.putNextEntry(new ZipEntry(ossfile.split("/")[ossfile.split("/").length - 1]));
+                int bytesRead = 0;
+                // 向压缩文件中输出数据
+                while ((bytesRead = inputStream.read()) != -1) {
+                    zos.write(bytesRead);
+                }
+                inputStream.close();
+                zos.closeEntry(); // 当前文件写完，定位为写入下一条项目
+            }
+            zos.close();
+            String header = request.getHeader("User-Agent").toUpperCase();
+            if (header.contains("MSIE") || header.contains("TRIDENT") || header.contains("EDGE")) {
+                fileName = URLEncoder.encode(fileName, "utf-8");
+                fileName = fileName.replace("+", "%20");    //IE下载文件名空格变+号问题
+            } else {
+                fileName = new String(fileName.getBytes(), "ISO8859-1");
+            }
+            response.reset();
+            response.setContentType("text/plain");
+            response.setContentType("application/octet-stream; charset=utf-8");
+            response.setHeader("Location", fileName);
+            response.setHeader("Cache-Control", "max-age=0");
+            response.setHeader("Content-Disposition", "attachment; filename=" + fileName);
+
+            FileInputStream fis = new FileInputStream(zipFile);
+            BufferedInputStream buff = new BufferedInputStream(fis);
+            BufferedOutputStream out = new BufferedOutputStream(response.getOutputStream());
+            byte[] car = new byte[1024];
+            int l = 0;
+            while (l < zipFile.length()) {
+                int j = buff.read(car, 0, 1024);
+                l += j;
+                out.write(car, 0, j);
+            }
+            // 关闭流
+            fis.close();
+            buff.close();
+            out.close();
+
+            ossClient.shutdown();
+            // 删除临时文件
+            zipFile.delete();
+        } catch (Exception e) {
+            logger.error(e);
+        }
+        return response;
+    }
+
+    /**
+     * 单张图片下载
+     *
+     * @param key
+     * @param request
+     * @param response
+     * @return
+     */
+    public HttpServletResponse singleFileDown(String key, HttpServletRequest request, HttpServletResponse response) {
+
+        String accessKeyId = PropertiesFileLoader.getDefaultProperties("aliossid", "");
+        String accessKeySecret = PropertiesFileLoader.getDefaultProperties("aliosskey", "");
+        String endpoint = PropertiesFileLoader.getDefaultProperties("aliossendpoit", "");
+        String bucketName = PropertiesFileLoader.getDefaultProperties("alibucketname", "");
+
+        // endpoint以杭州为例，其它region请按实际情况填写
+        // 云账号AccessKey有所有API访问权限，建议遵循阿里云安全最佳实践，创建并使用RAM子账号进行API访问或日常运维，请登录 https://ram.console.aliyun.com 创建
+        try {
+            // 初始化
+            OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
+
+            // 获取Object，返回结果为OSSObject对象
+            OSSObject ossObject = ossClient.getObject(bucketName, key);
+            // 读去Object内容  返回
+            InputStream inputStream = ossObject.getObjectContent();
+            // 对于每一个要被存放到压缩包的文件，都必须调用ZipOutputStream对象的putNextEntry()方法，确保压缩包里面文件不同名
+            String fileName = key.split("/")[key.split("/").length - 1];
+
+            String header = request.getHeader("User-Agent").toUpperCase();
+            if (header.contains("MSIE") || header.contains("TRIDENT") || header.contains("EDGE")) {
+                fileName = URLEncoder.encode(fileName, "utf-8");
+                fileName = fileName.replace("+", "%20");    //IE下载文件名空格变+号问题
+            } else {
+                fileName = new String(fileName.getBytes(), "ISO8859-1");
+            }
+            response.reset();
+            response.setContentType("text/plain");
+            response.setContentType("application/octet-stream; charset=utf-8");
+            response.setHeader("Location", fileName);
+            response.setHeader("Cache-Control", "max-age=0");
+            response.setHeader("Content-Disposition", "attachment; filename=" + fileName);
+
+            BufferedInputStream buff = new BufferedInputStream(inputStream);
+            BufferedOutputStream out = new BufferedOutputStream(response.getOutputStream());
+            byte[] car = new byte[1024];
+            int l = 0;
+
+            while ((l = inputStream.read(car)) != -1) {
+                if (car.length != 0) {
+                    out.write(car, 0, l);
+                }
+            }
+
+            // 关闭流
+            inputStream.close();
+            buff.close();
+            out.close();
+
+            ossClient.shutdown();
+        } catch (Exception e) {
+            logger.error(e);
+        }
+        return response;
+    }
+
+
 }

src/main/java/com/izouma/awesomeadmin/web/OrderImageController.java

@@ -5,12 +5,15 @@ import com.izouma.awesomeadmin.dto.Page;
 import com.izouma.awesomeadmin.dto.Result;
 import com.izouma.awesomeadmin.model.ImageUploadLog;
 import com.izouma.awesomeadmin.model.OrderImage;
+import com.izouma.awesomeadmin.model.UserOrder;
 import com.izouma.awesomeadmin.service.ImageUploadLogService;
 import com.izouma.awesomeadmin.service.OSSFileService;
 import com.izouma.awesomeadmin.service.OrderImageService;
+import com.izouma.awesomeadmin.service.UserOrderService;
 import com.izouma.awesomeadmin.util.ExportExcelUtil;
 import com.izouma.awesomeadmin.util.PropertiesFileLoader;
 import org.apache.commons.lang.StringUtils;
+import org.apache.log4j.Logger;
 import org.apache.shiro.authz.annotation.RequiresAuthentication;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Controller;
@@ -35,9 +38,19 @@ import java.util.*;
 @RequestMapping("/orderImage")
 public class OrderImageController {
 
+
+    /**
+     * 日志对象
+     */
+    private static Logger logger = Logger.getLogger(OrderImageController.class);
+
+
     @Autowired
     private OrderImageService orderImageService;
 
+    @Autowired
+    private UserOrderService userOrderService;
+
     @Autowired
     private OSSFileService OSSFileService;
 
@@ -230,7 +243,7 @@ public class OrderImageController {
 
                         paths.add(imageUploadLog);
                     } catch (IOException e) {
-                        e.printStackTrace();
+                        logger.error("上传订单图片失败", e);
                         return new Result(false, "上传失败");
                     }
                 }
@@ -313,5 +326,104 @@ public class OrderImageController {
         return new Result(true, data);
     }
 
+
+    /**
+     * 下载压缩包
+     *
+     * @param orderId  订单ID
+     * @param typeFlag 类型  original / finished
+     * @param request
+     * @param response
+     * @return
+     */
+    @RequestMapping(value = "/zipFilesDown", method = RequestMethod.GET)
+    @ResponseBody
+    public void zipFilesDown(@RequestParam(required = true, value = "orderId") String orderId,
+                             @RequestParam(required = true, value = "typeFlag") String typeFlag,
+                             HttpServletRequest request, HttpServletResponse response) {
+
+        try {
+            OrderImage record = new OrderImage();
+            record.setOrderId(orderId);
+            List<OrderImage> orderImageList = orderImageService.getOrderImageList(record);
+            if (orderImageList.size() > 0) {
+                UserOrder userOrder = userOrderService.getUserOrderById(orderId);
+                if (userOrder != null) {
+
+                    List<String> keyList = new ArrayList<>();
+                    String zipName = userOrder.getOrderCode() + "_" + typeFlag;
+
+                    for (OrderImage orderImage : orderImageList) {
+
+
+                        if ("original".equals(typeFlag)) {
+                            if (StringUtils.isNotEmpty(orderImage.getOriginalImage())) {
+                                String imageName = orderImage.getOriginalImage().replace(PropertiesFileLoader.getProperties("aliImageSever") + "/", "");
+                                keyList.add(imageName);
+                            }
+                        } else if ("finished".equals(typeFlag)) {
+                            if (StringUtils.isNotEmpty(orderImage.getFinishedImage())) {
+                                String imageName = orderImage.getFinishedImage().replace(PropertiesFileLoader.getProperties("aliImageSever") + "/", "");
+                                keyList.add(imageName);
+                            }
+                        }
+                    }
+
+                    if (keyList.size() > 0) {
+                        OSSFileService.zipFilesDown(keyList, zipName, request, response);
+
+                    }
+
+
+                }
+
+            }
+
+
+        } catch (Exception e) {
+            logger.error("zipFilesDown,批量下载异常", e);
+        }
+
+    }
+
+    /**
+     * 单张下载
+     *
+     * @param id       orderImage id
+     * @param typeFlag 类型  original / finished
+     * @param request
+     * @param response
+     */
+    @RequestMapping(value = "/singleFileDown", method = RequestMethod.GET)
+    @ResponseBody
+    public void singleFileDown(@RequestParam(required = true, value = "id") String id,
+                               @RequestParam(required = true, value = "typeFlag") String typeFlag,
+                               HttpServletRequest request, HttpServletResponse response) {
+
+        try {
+            OrderImage orderImage = orderImageService.getOrderImageById(id);
+
+            if (orderImage != null) {
+                if ("original".equals(typeFlag)) {
+                    if (StringUtils.isNotEmpty(orderImage.getOriginalImage())) {
+                        String imageName = orderImage.getOriginalImage().replace(PropertiesFileLoader.getProperties("aliImageSever") + "/", "");
+                        OSSFileService.singleFileDown(imageName, request, response);
+                    }
+                } else if ("finished".equals(typeFlag)) {
+                    if (StringUtils.isNotEmpty(orderImage.getFinishedImage())) {
+                        String imageName = orderImage.getFinishedImage().replace(PropertiesFileLoader.getProperties("aliImageSever") + "/", "");
+                        OSSFileService.singleFileDown(imageName, request, response);
+                    }
+                }
+            }
+
+
+        } catch (Exception e) {
+            logger.error("singleFileDown,单张下载异常", e);
+        }
+
+    }
+
+
 }