package com.ycl.common.utils.excel;
|
|
import cn.hutool.core.collection.CollUtil;
|
import cn.hutool.core.util.IdUtil;
|
import cn.hutool.core.util.ObjectUtil;
|
import cn.hutool.core.util.ZipUtil;
|
import com.alibaba.excel.EasyExcel;
|
import com.alibaba.excel.annotation.ExcelProperty;
|
import com.alibaba.excel.write.builder.ExcelWriterSheetBuilder;
|
import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy;
|
import com.ycl.common.utils.excel.convert.ExcelBigNumberConvert;
|
import com.ycl.common.utils.excel.core.CellMergeStrategy;
|
import com.ycl.common.utils.excel.core.DropDownOptions;
|
import com.ycl.common.utils.excel.core.ExcelDownHandler;
|
import com.ycl.common.utils.file.FileUtils;
|
import org.apache.commons.codec.Charsets;
|
|
import javax.servlet.ServletOutputStream;
|
import javax.servlet.http.HttpServletResponse;
|
import javax.validation.constraints.NotNull;
|
import java.io.*;
|
import java.lang.reflect.Field;
|
import java.net.URLEncoder;
|
import java.nio.file.Files;
|
import java.util.Arrays;
|
import java.util.HashSet;
|
import java.util.List;
|
import java.util.Set;
|
|
/**
|
* @Author: ljx
|
* @CreateTime: 2024-10-18 10:13
|
*/
|
|
public class OutputExcelUtils {
|
|
/**
|
* 忽略部分导出字段方法
|
* @param response
|
* @param fileName 文件名称
|
* @param sheetName sheet名称
|
* @param dataList 需要导出的数据
|
* @param clazz 类
|
* @param
|
* @param <T>
|
* @throws IOException
|
*/
|
public static <T> void export(HttpServletResponse response, String fileName, String sheetName, List<T> dataList, Class<T> clazz, List<String> fieldNames) throws IOException {
|
response.setContentType("application/zip");
|
response.setCharacterEncoding(Charsets.UTF_8.name());
|
fileName = URLEncoder.encode(fileName, Charsets.UTF_8);
|
response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".zip");
|
Set<Integer> selectedIndexes = getSelectFields(fieldNames, clazz);
|
|
//下载模板
|
downloadTemplate(response, sheetName, dataList, clazz, selectedIndexes);
|
|
|
// 临时向资源文件夹写 名为template的文件夹
|
// ClassPathResource classPathResource = new ClassPathResource("/template/test.xls");
|
// File file = classPathResource.getFile();
|
// try(FileOutputStream fileOutputStream = new FileOutputStream(file,false)) {
|
// EasyExcel.write(fileOutputStream, clazz).sheet(sheetName).doWrite(dataList);
|
// }
|
//
|
// ClassPathResource classPathResource1 = new ClassPathResource("/template");
|
// File zip = ZipUtil.zip(classPathResource1.getFile());
|
// byte[] bytes;
|
// try (FileInputStream fileInputStream = new FileInputStream(zip)) {
|
// bytes = fileInputStream.readAllBytes();
|
// }
|
// response.getOutputStream().write(bytes);
|
|
// if (selectedIndexes.size() > 0) {
|
// EasyExcel.write(response.getOutputStream(), clazz).excludeColumnIndexes(selectedIndexes).sheet(sheetName).doWrite(dataList);
|
// } else {
|
// EasyExcel.write(response.getOutputStream(), clazz).sheet(sheetName).doWrite(dataList);
|
// }
|
}
|
|
private synchronized static <T> void downloadTemplate(HttpServletResponse response, String sheetName, List<T> dataList, Class<T> clazz, Set<Integer> columnIndex) throws IOException {
|
|
File tempDir = null;
|
try {
|
// 创建临时目录
|
tempDir = Files.createTempDirectory("temp").toFile();
|
|
File templateDir = new File(tempDir, "template");
|
if (!templateDir.exists()) {
|
templateDir.mkdirs();
|
}
|
|
// 创建 Excel 文件
|
File excelFile = new File(templateDir, "excel.xlsx");
|
if (!excelFile.exists()) {
|
excelFile.createNewFile();
|
}
|
|
// 写入 Excel 模板数据
|
try (FileOutputStream fileOutputStream = new FileOutputStream(excelFile, false)) {
|
EasyExcel.write(fileOutputStream, clazz).includeColumnIndexes(columnIndex).sheet(sheetName).doWrite(dataList);
|
}
|
|
// 创建附件目录
|
File attachmentDir = new File(templateDir, "attachment");
|
if (!attachmentDir.exists()) {
|
attachmentDir.mkdirs();
|
}
|
|
// 打包 ZIP 文件
|
File zipFile = ZipUtil.zip(templateDir);
|
byte[] zipBytes = Files.readAllBytes(zipFile.toPath());
|
|
// 将 ZIP 文件写入响应
|
try(ServletOutputStream outputStream = response.getOutputStream()) {
|
outputStream.write(zipBytes);
|
}
|
} finally {
|
deleteDirectoryOrFile(tempDir);
|
}
|
}
|
|
private static void deleteDirectoryOrFile(File file) {
|
if (ObjectUtil.isNull(file)) {
|
return;
|
}
|
|
if (file.isDirectory()) {
|
File[] files = file.listFiles();
|
if (files != null) {
|
for (File f : files) {
|
deleteDirectoryOrFile(f);
|
}
|
}
|
}
|
file.delete();
|
|
}
|
|
/**
|
* 导出模板
|
* @param response
|
* @param fileName
|
* @param sheetName
|
* @param dataList
|
* @param clazz
|
* @param fieldNames
|
*/
|
public static <T> void exportTemplate(HttpServletResponse response,String fileName, String sheetName, List<T> dataList, Class<T> clazz, List<String> fieldNames) throws IOException {
|
Set<Integer> selectedIndexes = getSelectFields(fieldNames, clazz);
|
|
resetResponse(fileName, response);
|
|
exportExcel(dataList, sheetName, clazz, false, response.getOutputStream(), null, selectedIndexes);
|
}
|
|
public static <T> @NotNull Set<Integer> getSelectFields(List<String> fieldNames, Class<T> clazz) {
|
Set<Integer> selectedIndexes = new HashSet<>();
|
if (CollUtil.isNotEmpty(fieldNames)) {
|
// 反射获取字段属性,这里只获取easyExcel注解的字段
|
Field[] declaredFields = Arrays.stream(clazz.getDeclaredFields())
|
.filter(field -> field.isAnnotationPresent(ExcelProperty.class))
|
.toArray(Field[]::new);
|
// // 匹配需要导入的字段
|
for (int i = 0; i < declaredFields.length; i++) {
|
if (fieldNames.contains(declaredFields[i].getName())) {
|
// 获取需要导入的字段下标
|
selectedIndexes.add(i);
|
}
|
}
|
}
|
return selectedIndexes;
|
|
}
|
|
/**
|
* 导出excel
|
*
|
* @param list 导出数据集合
|
* @param sheetName 工作表的名称
|
* @param clazz 实体类
|
* @param merge 是否合并单元格
|
* @param os 输出流
|
*/
|
public static <T> void exportExcel(List<T> list, String sheetName, Class<T> clazz, boolean merge,
|
OutputStream os, List<DropDownOptions> options, Set<Integer> selectedIndexes) {
|
ExcelWriterSheetBuilder builder = EasyExcel.write(os, clazz)
|
.autoCloseStream(false)
|
.includeColumnIndexes(selectedIndexes)
|
// 自动适配
|
.registerWriteHandler(new LongestMatchColumnWidthStyleStrategy())
|
// 大数值自动转换 防止失真
|
.registerConverter(new ExcelBigNumberConvert())
|
.sheet(sheetName);
|
if (merge) {
|
// 合并处理器
|
builder.registerWriteHandler(new CellMergeStrategy(list, true));
|
}
|
// 添加下拉框操作
|
builder.registerWriteHandler(new ExcelDownHandler(options));
|
builder.doWrite(list);
|
}
|
|
/**
|
* 重置响应体
|
*/
|
private static void resetResponse(String sheetName, HttpServletResponse response) throws UnsupportedEncodingException {
|
String filename = encodingFilename(sheetName);
|
FileUtils.setAttachmentResponseHeader(response, filename);
|
response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=UTF-8");
|
}
|
|
/**
|
* 编码文件名
|
*/
|
public static String encodingFilename(String filename) {
|
return IdUtil.fastSimpleUUID() + "_" + filename + ".xlsx";
|
}
|
|
|
}
|
