1. package com.business.testExcelPort;
    3. import java.io.BufferedInputStream;
    4. import java.io.BufferedOutputStream;
    5. import java.io.File;
    6. import java.io.FileInputStream;
    7. import java.io.FileOutputStream;
    8. import java.io.IOException;
    9. import java.io.InputStream;
    10. import java.io.OutputStream;
    11. import java.net.URLEncoder;
    12. import java.util.ArrayList;
    13. import java.util.Calendar;
    14. import java.util.List;
    15. import java.util.Random;
    16. import javax.servlet.http.HttpServletResponse;
    17. import com.alibaba.excel.EasyExcel;
    18. import com.alibaba.excel.read.listener.ReadListener;
    19. import com.alibaba.excel.support.ExcelTypeEnum;
    20. import org.apache.log4j.Logger;
    21. import org.apache.tools.zip.ZipEntry;
    22. import org.apache.tools.zip.ZipOutputStream;
    23. import org.springframework.stereotype.Controller;
    24. import org.springframework.web.bind.annotation.RequestMapping;
    25. import org.springframework.web.bind.annotation.RequestMethod;
    27. /**
    28.  * @ClassName DownLoad
    29.  * @Deacription TODO
    30.  * @Author SEN
    31.  * @Date 2020/2/10 0010 17:49
    32.  * @Version 1.0
    33.  **/
    35. @Controller
    36. @RequestMapping("/download")
    37. @Slf4j
    38. public class DownLoad {
    39.     private Logger Log = Logger.getLogger(DownLoad.class);
    40.     // 获取当前系统的临时目录
    41.     private static final String FilePath = System.getProperty("java.io.tmpdir") + File.separator;
    43.     @RequestMapping(value = "/execute", method=RequestMethod.GET)
    44.     public void execute(HttpServletResponse response) {
    45.         // 用于存放文件路径
    46.         List<String> filePaths = new ArrayList<>();
    47.         //生成的ZIP文件名为Demo.zip
    48.         String tmpFileName = "Demo.zip";
    49.         // zip文件路径
    50.         String strZipPath = FilePath + tmpFileName;
    51.         filePaths.add(strZipPath);
    52.         try {
    53.             //创建zip输出流
    54.             ZipOutputStream out = new ZipOutputStream(new FileOutputStream(strZipPath));
    55.             //声明文件集合用于存放excel文件
    56.             List<File> fileList = new ArrayList<File>();
    57.             //生成excel文件集合
    58.             for (int i = 0; i < 10; i++) {
    59.                 // 生成随机文件名
    60.                 String filename = FilePath + generateRandomFilename() + ".xlsx";
    61.                 // 将文件路径保存
    62.                 fileList.add(creatFile(filename));
    63.                 filePaths.add(filename);
    64.                 List<DataOne> list = new ArrayList<>();
    65.                 for (int j = 0; j < 10; j++) {
    66.                     // 造一些表格数据，一般是从数据库查出来的list集合数据
    67.                     DataOne dataOne = new DataOne();
    68.                     //。。。
    69.                     list.add(dataOne);
    70.                 }
    71.                 // 使用easyexcel生成excel文件
    72.                 writeExcel(newFile, DataOne.class, list,ExcelTypeEnum.XLS);                
    73.             }
    74.             byte[] buffer = new byte[1024];
    75.             //将excel文件放入zip压缩包
    76.             for (int i = 0; i < fileList.size(); i++) {
    77.                 File file = fileList.get(i);
    78.                 FileInputStream fis = new FileInputStream(file);
    79.                 out.putNextEntry(new ZipEntry(file.getName()));
    80.                 //设置压缩文件内的字符编码，不然会变成乱码
    81.                 out.setEncoding("GBK");
    82.                 int len;
    83.                 // 读入需要下载的文件的内容，打包到zip文件
    84.                 while ((len = fis.read(buffer)) > 0) {
    85.                     out.write(buffer, 0, len);
    86.                 }
    87.                 out.closeEntry();
    88.                 fis.close();
    89.             }
    90.             out.close();
    91.             //下载zip文件
    92.             this.downFile(response, tmpFileName,filePaths);
    93.         } catch (Exception e) {
    94.             // 下载失败删除生成的文件
    95.             deleteFile(filePaths);
    96.             Log.error("文件下载出错", e);
    97.         }
    98.     }
    101.     /**
    102.      * 文件下载
    103.      *  @param response
    104.      * @param str
    105.      * @param filePaths
    106.      */
    107.     private void downFile(HttpServletResponse response, String str, List<String> filePaths) {
    108.         try {
    109.             String path = FilePath + str;
    110.             File file = new File(path);
    111.             if (file.exists()) {
    112.                 InputStream ins = new FileInputStream(path);
    113.                 BufferedInputStream bins = new BufferedInputStream(ins);// 放到缓冲流里面
    114.                 OutputStream outs = response.getOutputStream();// 获取文件输出IO流
    115.                 BufferedOutputStream bouts = new BufferedOutputStream(outs);
    116.                 response.setContentType("application/x-download");// 设置response内容的类型
    117.                 response.setHeader(
    118.                         "Content-disposition",
    119.                         "attachment;filename="
    120.                                 + URLEncoder.encode(str, "UTF-8"));// 设置头部信息
    121.                 int bytesRead = 0;
    122.                 byte[] buffer = new byte[8192];
    123.                 // 开始向网络传输文件流
    124.                 while ((bytesRead = bins.read(buffer, 0, 8192)) != -1) {
    125.                     bouts.write(buffer, 0, bytesRead);
    126.                 }
    127.                 bouts.flush();// 这里一定要调用flush()方法
    128.                 ins.close();
    129.                 bins.close();
    130.                 outs.close();
    131.                 bouts.close();
    132.                 deleteFile(filePaths);
    133.             }
    134.         } catch (IOException e) {
    135.             deleteFile(filePaths);
    136.             Log.error("文件下载出错", e);
    137.         }
    138.     }
    139.     //创建文件File对象
    140.     private File creatFile(String filePath) {
    141.         File file = new File(filePath);
    142.         return file;
    143.     }
    144.     //生成随机文件名
    145.     public String generateRandomFilename() {
    146.         String RandomFilename = "";
    147.         Random rand = new Random();//生成随机数
    148.         int random = rand.nextInt();
    149.         Calendar calCurrent = Calendar.getInstance();
    150.         int intDay = calCurrent.get(Calendar.DATE);
    151.         int intMonth = calCurrent.get(Calendar.MONTH) + 1;
    152.         int intYear = calCurrent.get(Calendar.YEAR);
    153.         String now = String.valueOf(intYear) + "_" + String.valueOf(intMonth) + "_" +
    154.                 String.valueOf(intDay) + "_";
    156.         RandomFilename = now + String.valueOf(random > 0 ? random : (-1) * random);
    157.         return RandomFilename;
    158.     }
    159.     //删除文件
    160.     public static boolean deleteFile(List<String> filePath){
    161.         boolean result = false;
    162.         for (String pathname:filePath){
    163.             File file = new File(pathname);
    164.             if (file.exists()) {
    165.                 file.delete();
    166.                 result = true;
    167.             }
    168.         }
    169.         return result;
    170.     }
    171.        /**
    172.      * @Title: writeExcel
    173.      * @Description: 写入excel文件到输出流web端
    174.      *
    175.      */
    176.     private void writeExcel(OutputStream outputStream, Class<?> clazz, List<?> datalist,ExcelTypeEnum excelType,String sheetName) throws IOException {
    177.         EasyExcel.write(outputStream, clazz).excelType(excelType).sheet(sheetName==null ? "sheet1":sheetName).doWrite(datalist);
    178.         outputStream.flush();
    179.     }
    181.     /**
    182.      * @Title: writeExcel
    183.      * @Description: 写入excel到本地路径
    184.      */
    185.     private void writeExcel(File newFile, Class<?> clazz, List<?> datalist,ExcelTypeEnum excelType) {
    186.         EasyExcel.write(newFile, clazz).excelType(excelType).sheet("sheet1").doWrite(datalist);
    187.     }
    189.     /**
    190.      * @Title: readExcel
    191.      * @Description: 读取excel内容（从输入流）
    192.      */
    193.     private List<?> readExcel(InputStream inputStream, Class<?> clazz, ReadListener<?> listener) {
    194.         List<?> list = null;
    195.         list = EasyExcel.read(inputStream, clazz, listener).sheet().doReadSync();
    196.         return list;
    197.     };
    198. }