假期期间自己在家撸码,刚好用到了导出,导出来之后是多个文件,所以需要打成压缩包并下载来给客户。查阅了一些资料,把这段代码贴在这,相当于有个记录吧。
package com.business.testExcelPort;
import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.List;
import java.util.Random;
import javax.servlet.http.HttpServletResponse;
import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.read.listener.ReadListener;
import com.alibaba.excel.support.ExcelTypeEnum;
import org.apache.log4j.Logger;
import org.apache.tools.zip.ZipEntry;
import org.apache.tools.zip.ZipOutputStream;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
/**
* @ClassName DownLoad
* @Deacription TODO
* @Author SEN
* @Date 2020/2/10 0010 17:49
* @Version 1.0
**/
@Controller
@RequestMapping("/download")
@Slf4j
public class DownLoad {
private Logger Log = Logger.getLogger(DownLoad.class);
// 获取当前系统的临时目录
private static final String FilePath = System.getProperty("java.io.tmpdir") + File.separator;
@RequestMapping(value = "/execute", method=RequestMethod.GET)
public void execute(HttpServletResponse response) {
// 用于存放文件路径
List filePaths = new ArrayList<>();
//生成的ZIP文件名为Demo.zip
String tmpFileName = "Demo.zip";
// zip文件路径
String strZipPath = FilePath + tmpFileName;
filePaths.add(strZipPath);
try {
//创建zip输出流
ZipOutputStream out = new ZipOutputStream(new FileOutputStream(strZipPath));
//声明文件集合用于存放excel文件
List fileList = new ArrayList();
//生成excel文件集合
for (int i = 0; i < 10; i++) {
// 生成随机文件名
String filename = FilePath + generateRandomFilename() + ".xlsx";
// 将文件路径保存
fileList.add(creatFile(filename));
filePaths.add(filename);
List list = new ArrayList<>();
for (int j = 0; j < 10; j++) {
// 造一些表格数据,一般是从数据库查出来的list集合数据
DataOne dataOne = new DataOne();
//。。。
list.add(dataOne);
}
// 使用easyexcel生成excel文件
writeExcel(newFile, DataOne.class, list,ExcelTypeEnum.XLS);
}
byte[] buffer = new byte[1024];
//将excel文件放入zip压缩包
for (int i = 0; i < fileList.size(); i++) {
File file = fileList.get(i);
FileInputStream fis = new FileInputStream(file);
out.putNextEntry(new ZipEntry(file.getName()));
//设置压缩文件内的字符编码,不然会变成乱码
out.setEncoding("GBK");
int len;
// 读入需要下载的文件的内容,打包到zip文件
while ((len = fis.read(buffer)) > 0) {
out.write(buffer, 0, len);
}
out.closeEntry();
fis.close();
}
out.close();
//下载zip文件
this.downFile(response, tmpFileName,filePaths);
} catch (Exception e) {
// 下载失败删除生成的文件
deleteFile(filePaths);
Log.error("文件下载出错", e);
}
}
/**
* 文件下载
* @param response
* @param str
* @param filePaths
*/
private void downFile(HttpServletResponse response, String str, List filePaths) {
try {
String path = FilePath + str;
File file = new File(path);
if (file.exists()) {
InputStream ins = new FileInputStream(path);
BufferedInputStream bins = new BufferedInputStream(ins);// 放到缓冲流里面
OutputStream outs = response.getOutputStream();// 获取文件输出IO流
BufferedOutputStream bouts = new BufferedOutputStream(outs);
response.setContentType("application/x-download");// 设置response内容的类型
response.setHeader(
"Content-disposition",
"attachment;filename="
+ URLEncoder.encode(str, "UTF-8"));// 设置头部信息
int bytesRead = 0;
byte[] buffer = new byte[8192];
// 开始向网络传输文件流
while ((bytesRead = bins.read(buffer, 0, 8192)) != -1) {
bouts.write(buffer, 0, bytesRead);
}
bouts.flush();// 这里一定要调用flush()方法
ins.close();
bins.close();
outs.close();
bouts.close();
deleteFile(filePaths);
}
} catch (IOException e) {
deleteFile(filePaths);
Log.error("文件下载出错", e);
}
}
//创建文件File对象
private File creatFile(String filePath) {
File file = new File(filePath);
return file;
}
//生成随机文件名
public String generateRandomFilename() {
String RandomFilename = "";
Random rand = new Random();//生成随机数
int random = rand.nextInt();
Calendar calCurrent = Calendar.getInstance();
int intDay = calCurrent.get(Calendar.DATE);
int intMonth = calCurrent.get(Calendar.MONTH) + 1;
int intYear = calCurrent.get(Calendar.YEAR);
String now = String.valueOf(intYear) + "_" + String.valueOf(intMonth) + "_" +
String.valueOf(intDay) + "_";
RandomFilename = now + String.valueOf(random > 0 ? random : (-1) * random);
return RandomFilename;
}
//删除文件
public static boolean deleteFile(List filePath){
boolean result = false;
for (String pathname:filePath){
File file = new File(pathname);
if (file.exists()) {
file.delete();
result = true;
}
}
return result;
}
/**
* @Title: writeExcel
* @Description: 写入excel文件到输出流web端
*
*/
private void writeExcel(OutputStream outputStream, Class> clazz, List> datalist,ExcelTypeEnum excelType,String sheetName) throws IOException {
EasyExcel.write(outputStream, clazz).excelType(excelType).sheet(sheetName==null ? "sheet1":sheetName).doWrite(datalist);
outputStream.flush();
}
/**
* @Title: writeExcel
* @Description: 写入excel到本地路径
*/
private void writeExcel(File newFile, Class> clazz, List> datalist,ExcelTypeEnum excelType) {
EasyExcel.write(newFile, clazz).excelType(excelType).sheet("sheet1").doWrite(datalist);
}
/**
* @Title: readExcel
* @Description: 读取excel内容(从输入流)
*/
private List> readExcel(InputStream inputStream, Class> clazz, ReadListener> listener) {
List> list = null;
list = EasyExcel.read(inputStream, clazz, listener).sheet().doReadSync();
return list;
};
}