com.whaleal.icefrog.poi.excel.ExcelWriter Maven / Gradle / Ivy
Show all versions of icefrog-poi Show documentation
package com.whaleal.icefrog.poi.excel;
import com.whaleal.icefrog.core.bean.BeanUtil;
import com.whaleal.icefrog.core.collection.CollUtil;
import com.whaleal.icefrog.core.comparator.IndexedComparator;
import com.whaleal.icefrog.core.io.FileUtil;
import com.whaleal.icefrog.core.io.IORuntimeException;
import com.whaleal.icefrog.core.io.IoUtil;
import com.whaleal.icefrog.core.lang.Console;
import com.whaleal.icefrog.core.lang.Precondition;
import com.whaleal.icefrog.core.map.MapUtil;
import com.whaleal.icefrog.core.util.CharsetUtil;
import com.whaleal.icefrog.core.util.IdUtil;
import com.whaleal.icefrog.core.util.StrUtil;
import com.whaleal.icefrog.core.util.URLUtil;
import com.whaleal.icefrog.poi.excel.cell.CellLocation;
import com.whaleal.icefrog.poi.excel.cell.CellUtil;
import com.whaleal.icefrog.poi.excel.style.Align;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.ss.util.CellRangeAddressList;
import org.apache.poi.xssf.usermodel.XSSFDataValidation;
import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.charset.Charset;
import java.util.*;
import java.util.Map.Entry;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;
/**
* Excel 写入器
* 此工具用于通过POI将数据写出到Excel，此对象可完成以下两个功能
*
*
* 1. 编辑已存在的Excel，可写出原Excel文件，也可写出到其它地方（到文件或到流）
* 2. 新建一个空的Excel工作簿，完成数据填充后写出（到文件或到流）
*
*
* @author Looly
* @author wh
* @since 1.0.0
*/
public class ExcelWriter extends ExcelBase {
/**
* 目标文件
*/
protected File destFile;
/**
* 当前行
*/
private AtomicInteger currentRow = new AtomicInteger(0);
/**
* 标题行别名
*/
private Map headerAlias;
/**
* 是否只保留别名对应的字段
*/
private boolean onlyAlias;
/**
* 标题顺序比较器
*/
private Comparator aliasComparator;
/**
* 样式集，定义不同类型数据样式
*/
private StyleSet styleSet;
/**
* 标题项对应列号缓存，每次写标题更新此缓存
*/
private Map headLocationCache;
// -------------------------------------------------------------------------- Constructor start
/**
* 构造，默认生成xls格式的Excel文件
* 此构造不传入写出的Excel文件路径，只能调用{@link #flush(OutputStream)}方法写出到流
* 若写出到文件，还需调用{@link #setDestFile(File)}方法自定义写出的文件，然后调用{@link #flush()}方法写出到文件
*
* @since 1.0.0
*/
public ExcelWriter() {
this(false);
}
/**
* 构造
* 此构造不传入写出的Excel文件路径，只能调用{@link #flush(OutputStream)}方法写出到流
* 若写出到文件，需要调用{@link #flush(File)} 写出到文件
*
* @param isXlsx 是否为xlsx格式
* @since 1.0.0
*/
public ExcelWriter( boolean isXlsx ) {
this(WorkbookUtil.createBook(isXlsx), null);
}
/**
* 构造，默认写出到第一个sheet，第一个sheet名为sheet1
*
* @param destFilePath 目标文件路径，可以不存在
*/
public ExcelWriter( String destFilePath ) {
this(destFilePath, null);
}
/**
* 构造
* 此构造不传入写出的Excel文件路径，只能调用{@link #flush(OutputStream)}方法写出到流
* 若写出到文件，需要调用{@link #flush(File)} 写出到文件
*
* @param isXlsx 是否为xlsx格式
* @param sheetName sheet名，第一个sheet名并写出到此sheet，例如sheet1
* @since 1.0.0
*/
public ExcelWriter( boolean isXlsx, String sheetName ) {
this(WorkbookUtil.createBook(isXlsx), sheetName);
}
/**
* 构造
*
* @param destFilePath 目标文件路径，可以不存在
* @param sheetName sheet名，第一个sheet名并写出到此sheet，例如sheet1
*/
public ExcelWriter( String destFilePath, String sheetName ) {
this(FileUtil.file(destFilePath), sheetName);
}
/**
* 构造，默认写出到第一个sheet，第一个sheet名为sheet1
*
* @param destFile 目标文件，可以不存在
*/
public ExcelWriter( File destFile ) {
this(destFile, null);
}
/**
* 构造
*
* @param destFile 目标文件，可以不存在
* @param sheetName sheet名，做为第一个sheet名并写出到此sheet，例如sheet1
*/
public ExcelWriter( File destFile, String sheetName ) {
this(WorkbookUtil.createBookForWriter(destFile), sheetName);
this.destFile = destFile;
}
/**
* 构造
* 此构造不传入写出的Excel文件路径，只能调用{@link #flush(OutputStream)}方法写出到流
* 若写出到文件，还需调用{@link #setDestFile(File)}方法自定义写出的文件，然后调用{@link #flush()}方法写出到文件
*
* @param workbook {@link Workbook}
* @param sheetName sheet名，做为第一个sheet名并写出到此sheet，例如sheet1
*/
public ExcelWriter( Workbook workbook, String sheetName ) {
this(WorkbookUtil.getOrCreateSheet(workbook, sheetName));
}
/**
* 构造
* 此构造不传入写出的Excel文件路径，只能调用{@link #flush(OutputStream)}方法写出到流
* 若写出到文件，还需调用{@link #setDestFile(File)}方法自定义写出的文件，然后调用{@link #flush()}方法写出到文件
*
* @param sheet {@link Sheet}
* @since 1.0.0
*/
public ExcelWriter( Sheet sheet ) {
super(sheet);
this.styleSet = new StyleSet(workbook);
}
// -------------------------------------------------------------------------- Constructor end
@Override
public ExcelWriter setSheet( int sheetIndex ) {
// 切换到新sheet需要重置开始行
reset();
return super.setSheet(sheetIndex);
}
@Override
public ExcelWriter setSheet( String sheetName ) {
// 切换到新sheet需要重置开始行
reset();
return super.setSheet(sheetName);
}
/**
* 重置Writer，包括：
*
*
* 1. 当前行游标归零
* 2. 清空别名比较器
* 3. 清除标题缓存
*
*
* @return this
*/
public ExcelWriter reset() {
resetRow();
this.headLocationCache = null;
return this;
}
/**
* 重命名当前sheet
*
* @param sheetName 新的sheet名
* @return this
* @since 1.0.0
*/
public ExcelWriter renameSheet( String sheetName ) {
return renameSheet(this.workbook.getSheetIndex(this.sheet), sheetName);
}
/**
* 重命名sheet
*
* @param sheet sheet序号，0表示第一个sheet
* @param sheetName 新的sheet名
* @return this
* @since 1.0.0
*/
public ExcelWriter renameSheet( int sheet, String sheetName ) {
this.workbook.setSheetName(sheet, sheetName);
return this;
}
/**
* 设置所有列为自动宽度，不考虑合并单元格
* 此方法必须在指定列数据完全写出后调用才有效。
* 列数计算是通过第一行计算的
*
* @return this
* @since 1.0.0
*/
public ExcelWriter autoSizeColumnAll() {
final int columnCount = this.getColumnCount();
for (int i = 0; i < columnCount; i++) {
autoSizeColumn(i);
}
return this;
}
/**
* 设置某列为自动宽度，不考虑合并单元格
* 此方法必须在指定列数据完全写出后调用才有效。
*
* @param columnIndex 第几列，从0计数
* @return this
* @since 1.0.0
*/
public ExcelWriter autoSizeColumn( int columnIndex ) {
this.sheet.autoSizeColumn(columnIndex);
return this;
}
/**
* 设置某列为自动宽度
* 此方法必须在指定列数据完全写出后调用才有效。
*
* @param columnIndex 第几列，从0计数
* @param useMergedCells 是否适用于合并单元格
* @return this
* @since 1.0.0
*/
public ExcelWriter autoSizeColumn( int columnIndex, boolean useMergedCells ) {
this.sheet.autoSizeColumn(columnIndex, useMergedCells);
return this;
}
/**
* 禁用默认样式
*
* @return this
* @see #setStyleSet(StyleSet)
* @since 1.0.0
*/
public ExcelWriter disableDefaultStyle() {
return setStyleSet(null);
}
/**
* 获取样式集，样式集可以自定义包括：
*
*
* 1. 头部样式
* 2. 一般单元格样式
* 3. 默认数字样式
* 4. 默认日期样式
*
*
* @return 样式集
* @since 1.0.0
*/
public StyleSet getStyleSet() {
return this.styleSet;
}
/**
* 设置样式集，如果不使用样式，传入{@code null}
*
* @param styleSet 样式集，{@code null}表示无样式
* @return this
* @since 1.0.0
*/
public ExcelWriter setStyleSet( StyleSet styleSet ) {
this.styleSet = styleSet;
return this;
}
/**
* 获取头部样式，获取样式后可自定义样式
*
* @return 头部样式
*/
public CellStyle getHeadCellStyle() {
return this.styleSet.headCellStyle;
}
/**
* 获取单元格样式，获取样式后可自定义样式
*
* @return 单元格样式
*/
public CellStyle getCellStyle() {
if (null == this.styleSet) {
return null;
}
return this.styleSet.cellStyle;
}
/**
* 获得当前行
*
* @return 当前行
*/
public int getCurrentRow() {
return this.currentRow.get();
}
/**
* 设置当前所在行
*
* @param rowIndex 行号
* @return this
*/
public ExcelWriter setCurrentRow( int rowIndex ) {
this.currentRow.set(rowIndex);
return this;
}
/**
* 获取Content-Disposition头对应的值，可以通过调用以下方法快速设置下载Excel的头信息：
*
*
* response.setHeader("Content-Disposition", excelWriter.getDisposition("test.xlsx", CharsetUtil.CHARSET_UTF_8));
*
*
* @param fileName 文件名，如果文件名没有扩展名，会自动按照生成Excel类型补齐扩展名，如果提供空，使用随机UUID
* @param charset 编码，null则使用默认UTF-8编码
* @return Content-Disposition值
*/
public String getDisposition( String fileName, Charset charset ) {
if (null == charset) {
charset = CharsetUtil.CHARSET_UTF_8;
}
if (StrUtil.isBlank(fileName)) {
// 未提供文件名使用随机UUID作为文件名
fileName = IdUtil.fastSimpleUUID();
}
fileName = StrUtil.addSuffixIfNot(URLUtil.encodeAll(fileName, charset), isXlsx() ? ".xlsx" : ".xls");
return StrUtil.format("attachment; filename=\"{}\"; filename*={}''{}", fileName, charset.name(), fileName);
}
/**
* 获取Content-Type头对应的值，可以通过调用以下方法快速设置下载Excel的头信息：
*
*
* response.setContentType(excelWriter.getContentType());
*
*
* @return Content-Type值
* @since 1.0.0
*/
public String getContentType() {
return isXlsx() ? ExcelUtil.XLSX_CONTENT_TYPE : ExcelUtil.XLS_CONTENT_TYPE;
}
/**
* 定位到最后一行的后边，用于追加数据
*
* @return this
* @since 1.0.0
*/
public ExcelWriter setCurrentRowToEnd() {
return setCurrentRow(getRowCount());
}
/**
* 跳过当前行
*
* @return this
*/
public ExcelWriter passCurrentRow() {
this.currentRow.incrementAndGet();
return this;
}
/**
* 跳过指定行数
*
* @param rows 跳过的行数
* @return this
*/
public ExcelWriter passRows( int rows ) {
this.currentRow.addAndGet(rows);
return this;
}
/**
* 重置当前行为0
*
* @return this
*/
public ExcelWriter resetRow() {
this.currentRow.set(0);
return this;
}
/**
* 设置写出的目标文件
*
* @param destFile 目标文件
* @return this
*/
public ExcelWriter setDestFile( File destFile ) {
this.destFile = destFile;
return this;
}
/**
* 设置标题别名，key为Map中的key，value为别名
*
* @param headerAlias 标题别名
* @return this
* @since 1.0.0
*/
public ExcelWriter setHeaderAlias( Map headerAlias ) {
this.headerAlias = headerAlias;
// 新增别名时清除比较器缓存
this.aliasComparator = null;
return this;
}
/**
* 清空标题别名，key为Map中的key，value为别名
*
* @return this
* @since 1.0.0
*/
public ExcelWriter clearHeaderAlias() {
this.headerAlias = null;
// 清空别名时清除比较器缓存
this.aliasComparator = null;
return this;
}
/**
* 设置是否只保留别名中的字段值，如果为true，则不设置alias的字段将不被输出，false表示原样输出
*
* @param isOnlyAlias 是否只保留别名中的字段值
* @return this
* @since 1.0.0
*/
public ExcelWriter setOnlyAlias( boolean isOnlyAlias ) {
this.onlyAlias = isOnlyAlias;
return this;
}
/**
* 增加标题别名
*
* @param name 原标题
* @param alias 别名
* @return this
* @since 1.0.0
*/
public ExcelWriter addHeaderAlias( String name, String alias ) {
Map headerAlias = this.headerAlias;
if (null == headerAlias) {
headerAlias = new LinkedHashMap<>();
}
this.headerAlias = headerAlias;
headerAlias.put(name, alias);
// 新增别名时清除比较器缓存
this.aliasComparator = null;
return this;
}
/**
* 设置窗口冻结，之前冻结的窗口会被覆盖，如果rowSplit为0表示取消冻结
*
* @param rowSplit 冻结的行及行数，2表示前两行
* @return this
* @since 1.0.0
*/
public ExcelWriter setFreezePane( int rowSplit ) {
return setFreezePane(0, rowSplit);
}
/**
* 设置窗口冻结，之前冻结的窗口会被覆盖，如果colSplit和rowSplit为0表示取消冻结
*
* @param colSplit 冻结的列及列数，2表示前两列
* @param rowSplit 冻结的行及行数，2表示前两行
* @return this
* @since 1.0.0
*/
public ExcelWriter setFreezePane( int colSplit, int rowSplit ) {
getSheet().createFreezePane(colSplit, rowSplit);
return this;
}
/**
* 设置列宽（单位为一个字符的宽度，例如传入width为10，表示10个字符的宽度）
*
* @param columnIndex 列号（从0开始计数，-1表示所有列的默认宽度）
* @param width 宽度（单位1~256个字符宽度）
* @return this
* @since 1.0.0
*/
public ExcelWriter setColumnWidth( int columnIndex, int width ) {
if (columnIndex < 0) {
this.sheet.setDefaultColumnWidth(width);
} else {
this.sheet.setColumnWidth(columnIndex, width * 256);
}
return this;
}
/**
* 设置默认行高，值为一个点的高度
*
* @param height 高度
* @return this
* @since 1.0.0
*/
public ExcelWriter setDefaultRowHeight( int height ) {
return setRowHeight(-1, height);
}
/**
* 设置行高，值为一个点的高度
*
* @param rownum 行号（从0开始计数，-1表示所有行的默认高度）
* @param height 高度
* @return this
* @since 1.0.0
*/
public ExcelWriter setRowHeight( int rownum, int height ) {
if (rownum < 0) {
this.sheet.setDefaultRowHeightInPoints(height);
} else {
final Row row = this.sheet.getRow(rownum);
if (null != row) {
row.setHeightInPoints(height);
}
}
return this;
}
/**
* 设置Excel页眉或页脚
*
* @param text 页脚的文本
* @param align 对齐方式枚举 {@link Align}
* @param isFooter 是否为页脚，false表示页眉，true表示页脚
* @return this
* @since 1.0.0
*/
public ExcelWriter setHeaderOrFooter( String text, Align align, boolean isFooter ) {
final HeaderFooter headerFooter = isFooter ? this.sheet.getFooter() : this.sheet.getHeader();
switch (align) {
case LEFT:
headerFooter.setLeft(text);
break;
case RIGHT:
headerFooter.setRight(text);
break;
case CENTER:
headerFooter.setCenter(text);
break;
default:
break;
}
return this;
}
/**
* 增加下拉列表
*
* @param x x坐标，列号，从0开始
* @param y y坐标，行号，从0开始
* @param selectList 下拉列表
* @return this
* @since 1.0.0
*/
public ExcelWriter addSelect( int x, int y, String... selectList ) {
return addSelect(new CellRangeAddressList(y, y, x, x), selectList);
}
/**
* 增加下拉列表
*
* @param regions {@link CellRangeAddressList} 指定下拉列表所占的单元格范围
* @param selectList 下拉列表内容
* @return this
* @since 1.0.0
*/
public ExcelWriter addSelect( CellRangeAddressList regions, String... selectList ) {
final DataValidationHelper validationHelper = this.sheet.getDataValidationHelper();
final DataValidationConstraint constraint = validationHelper.createExplicitListConstraint(selectList);
//设置下拉框数据
final DataValidation dataValidation = validationHelper.createValidation(constraint, regions);
//处理Excel兼容性问题
if (dataValidation instanceof XSSFDataValidation) {
dataValidation.setSuppressDropDownArrow(true);
dataValidation.setShowErrorBox(true);
} else {
dataValidation.setSuppressDropDownArrow(false);
}
return addValidationData(dataValidation);
}
/**
* 增加单元格控制，比如下拉列表、日期验证、数字范围验证等
*
* @param dataValidation {@link DataValidation}
* @return this
* @since 1.0.0
*/
public ExcelWriter addValidationData( DataValidation dataValidation ) {
this.sheet.addValidationData(dataValidation);
return this;
}
/**
* 合并当前行的单元格
* 样式为默认标题样式，可使用{@link #getHeadCellStyle()}方法调用后自定义默认样式
*
* @param lastColumn 合并到的最后一个列号
* @return this
*/
public ExcelWriter merge( int lastColumn ) {
return merge(lastColumn, null);
}
/**
* 合并当前行的单元格，并写入对象到单元格
* 如果写到单元格中的内容非null，行号自动+1，否则当前行号不变
* 样式为默认标题样式，可使用{@link #getHeadCellStyle()}方法调用后自定义默认样式
*
* @param lastColumn 合并到的最后一个列号
* @param content 合并单元格后的内容
* @return this
*/
public ExcelWriter merge( int lastColumn, Object content ) {
return merge(lastColumn, content, true);
}
/**
* 合并某行的单元格，并写入对象到单元格
* 如果写到单元格中的内容非null，行号自动+1，否则当前行号不变
* 样式为默认标题样式，可使用{@link #getHeadCellStyle()}方法调用后自定义默认样式
*
* @param lastColumn 合并到的最后一个列号
* @param content 合并单元格后的内容
* @param isSetHeaderStyle 是否为合并后的单元格设置默认标题样式，只提取边框样式
* @return this
* @since 1.0.0
*/
public ExcelWriter merge( int lastColumn, Object content, boolean isSetHeaderStyle ) {
Precondition.isFalse(this.isClosed, "ExcelWriter has been closed!");
final int rowIndex = this.currentRow.get();
merge(rowIndex, rowIndex, 0, lastColumn, content, isSetHeaderStyle);
// 设置内容后跳到下一行
if (null != content) {
this.currentRow.incrementAndGet();
}
return this;
}
/**
* 合并某行的单元格，并写入对象到单元格
* 如果写到单元格中的内容非null，行号自动+1，否则当前行号不变
* 样式为默认标题样式，可使用{@link #getHeadCellStyle()}方法调用后自定义默认样式
*
* @param firstRow 起始行，0开始
* @param lastRow 结束行，0开始
* @param firstColumn 起始列，0开始
* @param lastColumn 结束列，0开始
* @param content 合并单元格后的内容
* @param isSetHeaderStyle 是否为合并后的单元格设置默认标题样式，只提取边框样式
* @return this
* @since 1.0.0
*/
public ExcelWriter merge( int firstRow, int lastRow, int firstColumn, int lastColumn, Object content, boolean isSetHeaderStyle ) {
Precondition.isFalse(this.isClosed, "ExcelWriter has been closed!");
CellStyle style = null;
if (null != this.styleSet) {
style = (isSetHeaderStyle && null != this.styleSet.headCellStyle) ? this.styleSet.headCellStyle : this.styleSet.cellStyle;
}
return merge(firstRow, lastRow, firstColumn, lastColumn, content, style);
}
/**
* 合并单元格，并写入对象到单元格,使用指定的样式
* 指定样式传入null，则不使用任何样式
*
* @param firstRow 起始行，0开始
* @param lastRow 结束行，0开始
* @param firstColumn 起始列，0开始
* @param lastColumn 结束列，0开始
* @param content 合并单元格后的内容
* @param cellStyle 合并后单元格使用的样式，可以为null
* @return this
* @since 1.0.0
*/
public ExcelWriter merge( int firstRow, int lastRow, int firstColumn, int lastColumn, Object content, CellStyle cellStyle ) {
Precondition.isFalse(this.isClosed, "ExcelWriter has been closed!");
CellUtil.mergingCells(this.getSheet(), firstRow, lastRow, firstColumn, lastColumn, cellStyle);
// 设置内容
if (null != content) {
final Cell cell = getOrCreateCell(firstColumn, firstRow);
CellUtil.setCellValue(cell, content, cellStyle);
Console.log("{} {} {}", firstColumn, firstRow, cell.getStringCellValue());
}
return this;
}
/**
* 写出数据，本方法只是将数据写入Workbook中的Sheet，并不写出到文件
* 写出的起始行为当前行号，可使用{@link #getCurrentRow()}方法调用，根据写出的的行数，当前行号自动增加
* 样式为默认样式，可使用{@link #getCellStyle()}方法调用后自定义默认样式
* 默认的，当当前行号为0时，写出标题（如果为Map或Bean），否则不写标题
*
*
* data中元素支持的类型有：
*
*
* 1. Iterable，即元素为一个集合，元素被当作一行，data表示多行
* 2. Map，即元素为一个Map，第一个Map的keys作为首行，剩下的行为Map的values，data表示多行
* 3. Bean，即元素为一个Bean，第一个Bean的字段名列表会作为首行，剩下的行为Bean的字段值列表，data表示多行
* 4. 其它类型，按照基本类型输出（例如字符串）
*
*
* @param data 数据
* @return this
*/
public ExcelWriter write( Iterable data ) {
return write(data, 0 == getCurrentRow());
}
/**
* 写出数据，本方法只是将数据写入Workbook中的Sheet，并不写出到文件
* 写出的起始行为当前行号，可使用{@link #getCurrentRow()}方法调用，根据写出的的行数，当前行号自动增加
* 样式为默认样式，可使用{@link #getCellStyle()}方法调用后自定义默认样式
*
*
* data中元素支持的类型有：
*
*
* 1. Iterable，即元素为一个集合，元素被当作一行，data表示多行
* 2. Map，即元素为一个Map，第一个Map的keys作为首行，剩下的行为Map的values，data表示多行
* 3. Bean，即元素为一个Bean，第一个Bean的字段名列表会作为首行，剩下的行为Bean的字段值列表，data表示多行
* 4. 其它类型，按照基本类型输出（例如字符串）
*
*
* @param data 数据
* @param isWriteKeyAsHead 是否强制写出标题行（Map或Bean）
* @return this
*/
public ExcelWriter write( Iterable data, boolean isWriteKeyAsHead ) {
Precondition.isFalse(this.isClosed, "ExcelWriter has been closed!");
boolean isFirst = true;
for (Object object : data) {
writeRow(object, isFirst && isWriteKeyAsHead);
if (isFirst) {
isFirst = false;
}
}
return this;
}
/**
* 写出数据，本方法只是将数据写入Workbook中的Sheet，并不写出到文件
* 写出的起始行为当前行号，可使用{@link #getCurrentRow()}方法调用，根据写出的的行数，当前行号自动增加
* 样式为默认样式，可使用{@link #getCellStyle()}方法调用后自定义默认样式
* data中元素支持的类型有：
*
*
* 1. Map，即元素为一个Map，第一个Map的keys作为首行，剩下的行为Map的values，data表示多行
* 2. Bean，即元素为一个Bean，第一个Bean的字段名列表会作为首行，剩下的行为Bean的字段值列表，data表示多行
*
*
* @param data 数据
* @param comparator 比较器，用于字段名的排序
* @return this
* @since 1.0.0
*/
@SuppressWarnings({"rawtypes", "unchecked"})
public ExcelWriter write( Iterable data, Comparator comparator ) {
Precondition.isFalse(this.isClosed, "ExcelWriter has been closed!");
boolean isFirstRow = true;
Map map;
for (Object obj : data) {
if (obj instanceof Map) {
map = new TreeMap<>(comparator);
map.putAll((Map) obj);
} else {
map = BeanUtil.beanToMap(obj, new TreeMap<>(comparator), false, false);
}
writeRow(map, isFirstRow);
if (isFirstRow) {
isFirstRow = false;
}
}
return this;
}
/**
* 写出一行标题数据
* 本方法只是将数据写入Workbook中的Sheet，并不写出到文件
* 写出的起始行为当前行号，可使用{@link #getCurrentRow()}方法调用，根据写出的的行数，当前行号自动+1
* 样式为默认标题样式，可使用{@link #getHeadCellStyle()}方法调用后自定义默认样式
*
* @param rowData 一行的数据
* @return this
*/
public ExcelWriter writeHeadRow( Iterable rowData ) {
Precondition.isFalse(this.isClosed, "ExcelWriter has been closed!");
this.headLocationCache = new ConcurrentHashMap<>();
final Row row = this.sheet.createRow(this.currentRow.getAndIncrement());
int i = 0;
Cell cell;
for (Object value : rowData) {
cell = row.createCell(i);
CellUtil.setCellValue(cell, value, this.styleSet, true);
this.headLocationCache.put(StrUtil.toString(value), i);
i++;
}
return this;
}
/**
* 写出复杂标题的第二行标题数据
* 本方法只是将数据写入Workbook中的Sheet，并不写出到文件
* 写出的起始行为当前行号，可使用{@link #getCurrentRow()}方法调用，根据写出的的行数，当前行号自动+1
* 样式为默认标题样式，可使用{@link #getHeadCellStyle()}方法调用后自定义默认样式
*
*
* 此方法的逻辑是：将一行数据写出到当前行，遇到已存在的单元格跳过，不存在的创建并赋值。
*
*
* @param rowData 一行的数据
* @return this
*/
public ExcelWriter writeSecHeadRow( Iterable rowData ) {
final Row row = RowUtil.getOrCreateRow(this.sheet, this.currentRow.getAndIncrement());
Iterator iterator = rowData.iterator();
//如果获取的row存在单元格，则执行复杂表头逻辑，否则直接调用writeHeadRow(Iterable rowData)
if (row.getLastCellNum() != 0) {
for (int i = 0; i < this.workbook.getSpreadsheetVersion().getMaxColumns(); i++) {
Cell cell = row.getCell(i);
if (cell != null) {
continue;
}
if (iterator.hasNext()) {
cell = row.createCell(i);
CellUtil.setCellValue(cell, iterator.next(), this.styleSet, true);
} else {
break;
}
}
} else {
writeHeadRow(rowData);
}
return this;
}
/**
* 写出一行，根据rowBean数据类型不同，写出情况如下：
*
*
* 1、如果为Iterable，直接写出一行
* 2、如果为Map，isWriteKeyAsHead为true写出两行，Map的keys做为一行，values做为第二行，否则只写出一行values
* 3、如果为Bean，转为Map写出，isWriteKeyAsHead为true写出两行，Map的keys做为一行，values做为第二行，否则只写出一行values
*
*
* @param rowBean 写出的Bean
* @param isWriteKeyAsHead 为true写出两行，Map的keys做为一行，values做为第二行，否则只写出一行values
* @return this
* @see #writeRow(Iterable)
* @see #writeRow(Map, boolean)
* @since 1.0.0
*/
@SuppressWarnings({"rawtypes", "unchecked"})
public ExcelWriter writeRow( Object rowBean, boolean isWriteKeyAsHead ) {
if (rowBean instanceof Iterable) {
return writeRow((Iterable) rowBean);
}
Map rowMap;
if (rowBean instanceof Map) {
if (MapUtil.isNotEmpty(this.headerAlias)) {
rowMap = MapUtil.newTreeMap((Map) rowBean, getCachedAliasComparator());
} else {
rowMap = (Map) rowBean;
}
} else if (BeanUtil.isBean(rowBean.getClass())) {
if (MapUtil.isEmpty(this.headerAlias)) {
rowMap = BeanUtil.beanToMap(rowBean, new LinkedHashMap<>(), false, false);
} else {
// 别名存在情况下按照别名的添加顺序排序Bean数据
rowMap = BeanUtil.beanToMap(rowBean, new TreeMap<>(getCachedAliasComparator()), false, false);
}
} else {
// 其它转为字符串默认输出
return writeRow(CollUtil.newArrayList(rowBean), isWriteKeyAsHead);
}
return writeRow(rowMap, isWriteKeyAsHead);
}
/**
* 将一个Map写入到Excel，isWriteKeyAsHead为true写出两行，Map的keys做为一行，values做为第二行，否则只写出一行values
* 如果rowMap为空（包括null），则写出空行
*
* @param rowMap 写出的Map，为空（包括null），则写出空行
* @param isWriteKeyAsHead 为true写出两行，Map的keys做为一行，values做为第二行，否则只写出一行values
* @return this
*/
public ExcelWriter writeRow( Map rowMap, boolean isWriteKeyAsHead ) {
Precondition.isFalse(this.isClosed, "ExcelWriter has been closed!");
if (MapUtil.isEmpty(rowMap)) {
// 如果写出数据为null或空，跳过当前行
return passCurrentRow();
}
final Map aliasMap = aliasMap(rowMap);
if (isWriteKeyAsHead) {
writeHeadRow(aliasMap.keySet());
}
// 如果已经写出标题行，根据标题行找对应的值写入
if (MapUtil.isNotEmpty(this.headLocationCache)) {
final Row row = RowUtil.getOrCreateRow(this.sheet, this.currentRow.getAndIncrement());
Integer location;
for (Entry entry : aliasMap.entrySet()) {
location = this.headLocationCache.get(StrUtil.toString(entry.getKey()));
if (null != location) {
CellUtil.setCellValue(CellUtil.getOrCreateCell(row, location), entry.getValue(), this.styleSet, false);
}
}
} else {
writeRow(aliasMap.values());
}
return this;
}
/**
* 写出一行数据
* 本方法只是将数据写入Workbook中的Sheet，并不写出到文件
* 写出的起始行为当前行号，可使用{@link #getCurrentRow()}方法调用，根据写出的的行数，当前行号自动+1
* 样式为默认样式，可使用{@link #getCellStyle()}方法调用后自定义默认样式
*
* @param rowData 一行的数据
* @return this
*/
public ExcelWriter writeRow( Iterable rowData ) {
Precondition.isFalse(this.isClosed, "ExcelWriter has been closed!");
RowUtil.writeRow(this.sheet.createRow(this.currentRow.getAndIncrement()), rowData, this.styleSet, false);
return this;
}
/**
* 给指定单元格赋值，使用默认单元格样式
*
* @param locationRef 单元格地址标识符，例如A11，B5
* @param value 值
* @return this
* @since 1.0.0
*/
public ExcelWriter writeCellValue( String locationRef, Object value ) {
final CellLocation cellLocation = ExcelUtil.toLocation(locationRef);
return writeCellValue(cellLocation.getX(), cellLocation.getY(), value);
}
/**
* 给指定单元格赋值，使用默认单元格样式
*
* @param x X坐标，从0计数，即列号
* @param y Y坐标，从0计数，即行号
* @param value 值
* @return this
* @since 1.0.0
*/
public ExcelWriter writeCellValue( int x, int y, Object value ) {
final Cell cell = getOrCreateCell(x, y);
CellUtil.setCellValue(cell, value, this.styleSet, false);
return this;
}
/**
* 设置某个单元格的样式
* 此方法用于多个单元格共享样式的情况
* 可以调用{@link #getOrCreateCellStyle(int, int)} 方法创建或取得一个样式对象。
*
*
* 需要注意的是，共享样式会共享同一个{@link CellStyle}，一个单元格样式改变，全部改变。
*
* @param style 单元格样式
* @param locationRef 单元格地址标识符，例如A11，B5
* @return this
* @since 1.0.0
*/
public ExcelWriter setStyle( CellStyle style, String locationRef ) {
final CellLocation cellLocation = ExcelUtil.toLocation(locationRef);
return setStyle(style, cellLocation.getX(), cellLocation.getY());
}
/**
* 设置某个单元格的样式
* 此方法用于多个单元格共享样式的情况
* 可以调用{@link #getOrCreateCellStyle(int, int)} 方法创建或取得一个样式对象。
*
*
* 需要注意的是，共享样式会共享同一个{@link CellStyle}，一个单元格样式改变，全部改变。
*
* @param style 单元格样式
* @param x X坐标，从0计数，即列号
* @param y Y坐标，从0计数，即行号
* @return this
* @since 1.0.0
*/
public ExcelWriter setStyle( CellStyle style, int x, int y ) {
final Cell cell = getOrCreateCell(x, y);
cell.setCellStyle(style);
return this;
}
/**
* 设置行样式
*
* @param y Y坐标，从0计数，即行号
* @param style 样式
* @return this
* @see Row#setRowStyle(CellStyle)
* @since 1.0.0
*/
public ExcelWriter setRowStyle( int y, CellStyle style ) {
getOrCreateRow(y).setRowStyle(style);
return this;
}
/**
* 对数据行整行加自定义样式 仅对数据单元格设置 write后调用
*
* {@link com.whaleal.icefrog.poi.excel.ExcelWriter#setRowStyle(int, org.apache.poi.ss.usermodel.CellStyle)}
* 这个方法加的样式会使整行没有数据的单元格也有样式
* 特别是加背景色时很不美观 且有数据的单元格样式会被StyleSet中的样式覆盖掉
*
* @param y 行坐标
* @param style 自定义的样式
* @return this
* @since 1.0.0
*/
public ExcelWriter setRowStyleIfHasData( int y, CellStyle style ) {
if (y < 0) {
throw new IllegalArgumentException("Invalid row number (" + y + ")");
}
int columnCount = this.getColumnCount();
for (int i = 0; i < columnCount; i++) {
this.setStyle(style, i, y);
}
return this;
}
/**
* 设置列的默认样式
*
* @param x 列号，从0开始
* @param style 样式
* @return this
* @since 1.0.0
*/
public ExcelWriter setColumnStyle( int x, CellStyle style ) {
this.sheet.setDefaultColumnStyle(x, style);
return this;
}
/**
* 设置整个列的样式 仅对数据单元格设置 write后调用
*
* {@link com.whaleal.icefrog.poi.excel.ExcelWriter#setColumnStyle(int, org.apache.poi.ss.usermodel.CellStyle)}
* 这个方法加的样式会使整列没有数据的单元格也有样式
* 特别是加背景色时很不美观 且有数据的单元格样式会被StyleSet中的样式覆盖掉
*
* @param x 列的索引
* @param y 起始行
* @param style 样式
* @return this
* @since 1.0.0
*/
public ExcelWriter setColumnStyleIfHasData( int x, int y, CellStyle style ) {
if (x < 0) {
throw new IllegalArgumentException("Invalid column number (" + x + ")");
}
if (y < 0) {
throw new IllegalArgumentException("Invalid row number (" + y + ")");
}
int rowCount = this.getRowCount();
for (int i = y; i < rowCount; i++) {
this.setStyle(style, x, i);
}
return this;
}
/**
* 创建字体
*
* @return 字体
* @since 1.0.0
*/
public Font createFont() {
return getWorkbook().createFont();
}
/**
* 将Excel Workbook刷出到预定义的文件
* 如果用户未自定义输出的文件，将抛出{@link NullPointerException}
* 预定义文件可以通过{@link #setDestFile(File)} 方法预定义，或者通过构造定义
*
* @return this
* @throws IORuntimeException IO异常
*/
public ExcelWriter flush() throws IORuntimeException {
return flush(this.destFile);
}
/**
* 将Excel Workbook刷出到文件
* 如果用户未自定义输出的文件，将抛出{@link NullPointerException}
*
* @param destFile 写出到的文件
* @return this
* @throws IORuntimeException IO异常
* @since 1.0.0
*/
public ExcelWriter flush( File destFile ) throws IORuntimeException {
Precondition.notNull(destFile, "[destFile] is null, and you must call setDestFile(File) first or call flush(OutputStream).");
return flush(FileUtil.getOutputStream(destFile), true);
}
/**
* 将Excel Workbook刷出到输出流
*
* @param out 输出流
* @return this
* @throws IORuntimeException IO异常
*/
public ExcelWriter flush( OutputStream out ) throws IORuntimeException {
return flush(out, false);
}
/**
* 将Excel Workbook刷出到输出流
*
* @param out 输出流
* @param isCloseOut 是否关闭输出流
* @return this
* @throws IORuntimeException IO异常
* @since 1.0.0
*/
public ExcelWriter flush( OutputStream out, boolean isCloseOut ) throws IORuntimeException {
Precondition.isFalse(this.isClosed, "ExcelWriter has been closed!");
try {
this.workbook.write(out);
out.flush();
} catch (IOException e) {
throw new IORuntimeException(e);
} finally {
if (isCloseOut) {
IoUtil.close(out);
}
}
return this;
}
/**
* 关闭工作簿
* 如果用户设定了目标文件，先写出目标文件后给关闭工作簿
*/
@Override
public void close() {
if (null != this.destFile) {
flush();
}
closeWithoutFlush();
}
/**
* 关闭工作簿但是不写出
*/
protected void closeWithoutFlush() {
super.close();
// 清空对象
this.currentRow = null;
this.styleSet = null;
}
// -------------------------------------------------------------------------- Private method start
/**
* 为指定的key列表添加标题别名，如果没有定义key的别名，在onlyAlias为false时使用原key
*
* @param rowMap 一行数据
* @return 别名列表
*/
private Map aliasMap( Map rowMap ) {
if (MapUtil.isEmpty(this.headerAlias)) {
return rowMap;
}
final Map filteredMap = MapUtil.newHashMap(rowMap.size(), true);
String aliasName;
for (Entry entry : rowMap.entrySet()) {
aliasName = this.headerAlias.get(StrUtil.toString(entry.getKey()));
if (null != aliasName) {
// 别名键值对加入
filteredMap.put(aliasName, entry.getValue());
} else if (false == this.onlyAlias) {
// 保留无别名设置的键值对
filteredMap.put(entry.getKey(), entry.getValue());
}
}
return filteredMap;
}
/**
* 获取单例的别名比较器，比较器的顺序为别名加入的顺序
*
* @return Comparator
* @since 1.0.0
*/
private Comparator getCachedAliasComparator() {
if (MapUtil.isEmpty(this.headerAlias)) {
return null;
}
Comparator aliasComparator = this.aliasComparator;
if (null == aliasComparator) {
Set keySet = this.headerAlias.keySet();
aliasComparator = new IndexedComparator<>(keySet.toArray(new String[0]));
this.aliasComparator = aliasComparator;
}
return aliasComparator;
}
// -------------------------------------------------------------------------- Private method end
}