工具类
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.io.IoUtil;
import cn.hutool.core.lang.UUID;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.poi.excel.ExcelWriter;
import lombok.extern.slf4j.Slf4j;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicReference;
import java.util.function.BiFunction;
import java.util.function.Supplier;
@Slf4j
public class ExcelUtil {
private static final Integer PAGE_SIZE = 100000;
private static final Integer SHEET_MAX_SIZE = 100000;
public static <T> void executeTask(HttpServletResponse response, Supplier<Long> countSupplier
, BiFunction<Integer, Integer, List<T>> queryFunc
, LinkedHashMap<String, String> header
, Integer width
, String excelName) {
ExcelWriter writer;
ServletOutputStream out = null;
Long totalCount = countSupplier.get();
log.info("总条数;{}", totalCount);
int writeCount = 0;
long pageTotal = (totalCount % PAGE_SIZE) > 0 ? (totalCount / PAGE_SIZE) + 1 : totalCount / PAGE_SIZE;
writer = cn.hutool.poi.excel.ExcelUtil.getBigWriter(UUID.fastUUID()+".xlsx", "sheet1");
int sheetNo = writeCount / SHEET_MAX_SIZE;
for (int pageNo = 1; pageNo <= pageTotal; pageNo++) {
writer.setSheet("sheet" + pageNo);
List<T> list = queryFunc.apply(pageNo, PAGE_SIZE);
log.info("pageNo:{},resultSize:{}",pageNo,list.size());
writeCount += list.size();
AtomicReference<Integer> index = new AtomicReference<>(0);
header.forEach((k, v) -> {
writer.addHeaderAlias(k, v);
writer.setColumnWidth(index.get(), width);
index.getAndSet(index.get() + 1);
});
// writer.autoSizeColumnAll();
writer.setOnlyAlias(true);
writer.write(list, true);
}
try {
response.setContentType("application/vnd.ms-excel;charset=utf-8");
response.setHeader("Content-disposition", "attachment; filename=" + URLEncoder.encode(excelName + "-" + DateUtil.today() + ".xls", "utf-8"));
out = response.getOutputStream();
writer.flush(out, true);
} catch (IOException e) {
log.error("导出异常错误!!");
throw new RuntimeException("导出异常");
} finally {
if (ObjectUtil.isNotEmpty(writer)) {
writer.close();
}
IoUtil.close(out);
}
}
/**
* @param response
* @param maps 导出数据
* @param header 头部
* @param width 表格宽度
* @param excelName 导出文件名字
* @throws IOException
*/
public static void exportExcel(HttpServletResponse response, List<Map<String, Object>> maps, LinkedHashMap<String, String> header, Integer width, String excelName) {
ExcelWriter writer = null;
ServletOutputStream out = null;
try {
writer = cn.hutool.poi.excel.ExcelUtil.getBigWriter();
AtomicReference<Integer> index = new AtomicReference<>(0);
ExcelWriter finalWriter = writer;
header.forEach((k, v) -> {
finalWriter.addHeaderAlias(k, v);
finalWriter.setColumnWidth(index.get(), width);
index.getAndSet(index.get() + 1);
});
finalWriter.setOnlyAlias(true);
writer.write(maps, true);
response.setContentType("application/vnd.ms-excel;charset=utf-8");
response.setHeader("Content-disposition", "attachment; filename=" + URLEncoder.encode(excelName + "-" + DateUtil.today() + ".xls", "utf-8"));
out = response.getOutputStream();
writer.flush(out, true);
} catch (IOException e) {
log.error("导出异常错误!!");
throw new RuntimeException("导出异常");
} finally {
if (ObjectUtil.isNotEmpty(writer)) {
writer.close();
}
IoUtil.close(out);
}
}
/**
* @param response
* @param maps 导出数据
* @param header 头部
* @param width 表格宽度
* @param excelName 导出文件名字
* @param doAction 接口函数
* @throws IOException
*/
public static void exportExcel(HttpServletResponse response, List<Map<String, Object>> maps, LinkedHashMap<String, String> header, Integer width, String excelName, Action doAction) {
ExcelWriter writer = null;
ServletOutputStream out = null;
try {
writer = cn.hutool.poi.excel.ExcelUtil.getBigWriter();
AtomicReference<Integer> index = new AtomicReference<>(0);
ExcelWriter finalWriter = writer;
doAction.doAction(writer);
header.forEach((k, v) -> {
finalWriter.addHeaderAlias(k, v);
finalWriter.setColumnWidth(index.get(), width);
index.getAndSet(index.get() + 1);
});
finalWriter.setOnlyAlias(true);
writer.write(maps, true);
response.setContentType("application/vnd.ms-excel;charset=utf-8");
response.setHeader("Content-disposition", "attachment; filename=" + URLEncoder.encode(excelName + "-" + DateUtil.today() + ".xls", "utf-8"));
out = response.getOutputStream();
writer.flush(out, true);
} catch (IOException e) {
log.error("导出异常错误!!");
throw new RuntimeException("导出异常");
} finally {
writer.close();
IoUtil.close(out);
}
}
public interface Action {
void doAction(ExcelWriter writer);
}
}
调用类
ExcelUtil.executeTask(response, () -> baseMapper.getCount(tabInsBase), (no, size) -> {
List<TabInsBase> records = this.getPage(new Page<>(no, size), tabInsBase).getRecords();
return records;
}, linkedHashMap, 50, "导出名称");