Controller

  1. /**
  2.      * @description: 试题篮子里的音频打包下载
  3.      * @author: zcq
  4.      * @date: 2020/11/26 6:28 下午
  5.      */
  6.     @RequestMapping("/downloadAudioZip")
  7.     public void downloadAudioZip(@RequestBody List<AudioDownloadDto> audioDownloadDto, HttpServletResponse response) {
  8.         try {
  9.             String zipName = DateUtils.format(Field.DEFAULT_LONG_FORMAT, new Date());
  10.             FileUtil.downloadAllFilesZip(zipName, audioDownloadDto, response);
  11.         } catch (Exception e) {
  12.             Result result = Result.buildFailResult(e.getMessage());
  13.             AlertMessageUtil.packageErrorResult(result);
  14.             log.error("试题篮子里的音频打包下载失败", e);
  15.         }
  16.     }

FileUtils

  1. import com.hwl.tool.FileUtils;
  2. import com.hwl.tool.ListUtils;
  3. import com.tal.search.business.model.dto.AudioDownloadDto;
  4. import com.tal.search.core.common.contants.Constants;
  5. import com.tal.search.core.exceptions.ServiceException;
  6. import lombok.extern.slf4j.Slf4j;
  7. import org.apache.commons.lang.StringUtils;
  9. import javax.servlet.http.HttpServletResponse;
  10. import java.io.IOException;
  11. import java.io.InputStream;
  12. import java.net.URLEncoder;
  13. import java.util.HashSet;
  14. import java.util.List;
  15. import java.util.Set;
  16. import java.util.UUID;
  17. import java.util.zip.ZipEntry;
  18. import java.util.zip.ZipOutputStream;
  20. /**
  21.  * @description: 文件相关工具类
  22.  * @author: zcq
  23.  * @date: 2020/11/24 08:34 下午
  24.  */
  25. @Slf4j
  26. public class FileUtil {
  28.     public static final String ZIP_FILE = "/Users/zhangchuanqiang/Desktop/file.zip";
  30.     public static void downloadAllFilesZip(String zipName, List<AudioDownloadDto> downloads, HttpServletResponse response) throws ServiceException {
  31.         long beginTime = System.currentTimeMillis();
  32.         if (ListUtils.isEmpty(downloads)) {
  33.             return;
  34.         }
  35.         dealRepeatFileName(downloads);
  36.         ZipOutputStream zos = null;
  37.         try {
  38.             String downloadFilename = zipName + ".zip";
  39.             downloadFilename = URLEncoder.encode(downloadFilename, "UTF-8");
  40.             // 指明response的返回对象是文件流
  41.             response.setContentType("application/octet-stream");
  42.             // 设置在下载框默认显示的文件名
  43.             response.setHeader("Content-Disposition", "attachment;filename=" + downloadFilename);
  44.             zos = new ZipOutputStream(response.getOutputStream());
  45. //            zos = new ZipOutputStream(new FileOutputStream(new File(ZIP_FILE)));
  46.             for (AudioDownloadDto download : downloads) {
  47.                 String fileUrl = download.getUrl();
  48.                 ZipEntry zipEntry = new ZipEntry(getFileName(download.getQueId(), download.getUrl()));
  49.                 zos.putNextEntry(zipEntry);
  50.                 InputStream fis = FileUtils.getInputStreamByFileUrl(fileUrl);
  51.                 if (fis == null) {
  52.                     continue;
  53.                 }
  54.                 byte[] buffer = new byte[2048];
  55.                 int r;
  56.                 while ((r = fis.read(buffer)) != -1) {
  57.                     zos.write(buffer, 0, r);
  58.                 }
  59.                 fis.close();
  60.                 //  注意写完一个文件，需要关闭这个文件，再写下一个
  61.                 zos.closeEntry();
  62.                 zos.flush();
  63.             }
  64.             zos.finish();
  65.             printInfo(beginTime);
  66.         } catch (Exception e) {
  67.             log.error("下载音频压缩包失败 ：{}", e.getMessage());
  68.             throw new ServiceException("下载音频压缩包失败");
  69.         } finally {
  70.             try {
  71.                 if (zos != null) {
  72.                     zos.close();
  73.                 }
  74.             } catch (IOException e) {
  75.                 log.error("ZipOutputStream close fail ：{}", e.getMessage());
  76.             }
  77.         }
  78.     }
  80.     /**
  81.      * @description: 文件名重复问题
  82.      * @author: zcq
  83.      * @date: 2020/11/27 4:44 下午
  84.      */
  85.     private static void dealRepeatFileName(List<AudioDownloadDto> downloads) {
  86.         Set<String> queIdSet = new HashSet();
  87.         for (AudioDownloadDto download : downloads) {
  88.             if (!queIdSet.add(download.getQueId())) {
  89.                 // 试题重复，通过增加时间戳来去重
  90.                 download.setQueId(download.getQueId() + "-" + UUID.randomUUID().toString().replace("-", ""));
  91.             }
  92.         }
  93.     }
  95.     /**
  96.      * @description: 根据文件路径获取文件的扩展名
  97.      * @author: zcq
  98.      * @date: 2020/11/26 6:18 下午
  99.      */
  100.     private static String getFileName(String queId, String url) {
  101.         return queId + Constants.Decollator.POINT_DECOLLATOR + StringUtils.substringAfterLast(url, Constants.Decollator.POINT_DECOLLATOR);
  102.     }
  104.     private static void printInfo(long beginTime) {
  105.         log.info("file zip execute time → " + (System.currentTimeMillis() - beginTime));
  106.     }
  107. }

资料