All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.dromara.hutool.poi.excel.ExcelBase Maven / Gradle / Ivy
/*
* Copyright (c) 2013-2024 Hutool Team and hutool.cn
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.dromara.hutool.poi.excel;
import org.apache.poi.common.usermodel.HyperlinkType;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.ss.util.CellReference;
import org.apache.poi.xssf.streaming.SXSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.dromara.hutool.core.io.IoUtil;
import org.dromara.hutool.core.lang.Assert;
import org.dromara.hutool.poi.excel.cell.CellUtil;
import org.dromara.hutool.poi.excel.style.StyleUtil;
import java.io.Closeable;
import java.io.File;
import java.util.ArrayList;
import java.util.List;
/**
* Excel基础类，用于抽象ExcelWriter和ExcelReader中共用部分的对象和方法
*
* @param 子类类型，用于返回this
* @param ExcelConfig类型
* @author looly
* @since 4.1.4
*/
public class ExcelBase, C extends ExcelConfig> implements Closeable {
/**
* Excel配置，此项不为空
*/
protected C config;
/**
* 是否被关闭
*/
protected boolean isClosed;
/**
* 目标文件，如果用户读取为流或自行创建的Workbook或Sheet,此参数为{@code null}
*/
protected File targetFile;
/**
* 工作簿
*/
protected Workbook workbook;
/**
* Excel中对应的Sheet
*/
protected Sheet sheet;
/**
* 构造
*
* @param config config
* @param sheet Excel中的sheet
*/
public ExcelBase(final C config, final Sheet sheet) {
this.config = Assert.notNull(config);
this.sheet = Assert.notNull(sheet, "No Sheet provided.");
this.workbook = sheet.getWorkbook();
}
/**
* 设置Excel配置
*
* @param config Excel配置
* @return this
*/
@SuppressWarnings("unchecked")
public T setConfig(final C config) {
this.config = config;
return (T) this;
}
/**
* 获取Excel配置
*
* @return Excel配置
*/
public C getConfig() {
return this.config;
}
/**
* 获取Workbook
*
* @return Workbook
*/
public Workbook getWorkbook() {
return this.workbook;
}
// region ----- sheet ops
/**
* 返回工作簿表格数
*
* @return 工作簿表格数
* @since 4.0.10
*/
public int getSheetCount() {
return this.workbook.getNumberOfSheets();
}
/**
* 获取此工作簿所有Sheet表
*
* @return sheet表列表
* @since 4.0.3
*/
public List getSheets() {
final int totalSheet = getSheetCount();
final List result = new ArrayList<>(totalSheet);
for (int i = 0; i < totalSheet; i++) {
result.add(this.workbook.getSheetAt(i));
}
return result;
}
/**
* 获取表名列表
*
* @return 表名列表
* @since 4.0.3
*/
public List getSheetNames() {
final int totalSheet = workbook.getNumberOfSheets();
final List result = new ArrayList<>(totalSheet);
for (int i = 0; i < totalSheet; i++) {
result.add(this.workbook.getSheetAt(i).getSheetName());
}
return result;
}
/**
* 获取当前Sheet
*
* @return {@link Sheet}
*/
public Sheet getSheet() {
return this.sheet;
}
/**
* 重命名当前sheet
*
* @param newName 新名字
* @return this
* @see Workbook#setSheetName(int, String)
* @since 5.7.10
*/
@SuppressWarnings("unchecked")
public T renameSheet(final String newName) {
this.workbook.setSheetName(this.workbook.getSheetIndex(this.sheet), newName);
return (T) this;
}
/**
* 自定义需要读取或写出的Sheet，如果给定的sheet不存在，创建之。
* 最后一行序号 + 1
*
*
* @return 行数
* @since 4.5.4
*/
public int getRowCount() {
return this.sheet.getLastRowNum() + 1;
}
/**
* 获取有记录的行数，计算方法为：
*
*
* 最后一行序号 - 第一行序号 + 1
*
*
* @return 行数
* @since 4.5.4
*/
public int getPhysicalRowCount() {
return this.sheet.getPhysicalNumberOfRows();
}
// endregion
// region ----- style ops
/**
* 为指定单元格获取或者创建样式，返回样式后可以设置样式内容
*
* @param locationRef 单元格地址标识符，例如A11，B5
* @return {@link CellStyle}
* @since 5.1.4
*/
public CellStyle getOrCreateCellStyle(final String locationRef) {
final CellReference cellReference = new CellReference(locationRef);
return getOrCreateCellStyle(cellReference.getCol(), cellReference.getRow());
}
/**
* 为指定单元格获取或者创建样式，返回样式后可以设置样式内容
*
* @param x X坐标，从0计数，即列号
* @param y Y坐标，从0计数，即行号
* @return {@link CellStyle}
* @since 4.1.4
*/
public CellStyle getOrCreateCellStyle(final int x, final int y) {
final CellStyle cellStyle = getOrCreateCell(x, y).getCellStyle();
return StyleUtil.isNullOrDefaultStyle(this.workbook, cellStyle) ? createCellStyle(x, y) : cellStyle;
}
/**
* 为指定单元格创建样式，返回样式后可以设置样式内容
*
* @param locationRef 单元格地址标识符，例如A11，B5
* @return {@link CellStyle}
* @since 5.1.4
*/
public CellStyle createCellStyle(final String locationRef) {
final CellReference cellReference = new CellReference(locationRef);
return createCellStyle(cellReference.getCol(), cellReference.getRow());
}
/**
* 为指定单元格创建样式，返回样式后可以设置样式内容
*
* @param x X坐标，从0计数，即列号
* @param y Y坐标，从0计数，即行号
* @return {@link CellStyle}
* @since 4.6.3
*/
public CellStyle createCellStyle(final int x, final int y) {
final Cell cell = getOrCreateCell(x, y);
final CellStyle cellStyle = this.workbook.createCellStyle();
cell.setCellStyle(cellStyle);
return cellStyle;
}
/**
* 创建单元格样式
*
* @return {@link CellStyle}
* @see Workbook#createCellStyle()
* @since 5.4.0
*/
public CellStyle createCellStyle() {
return StyleUtil.createCellStyle(this.workbook);
}
/**
* 获取或创建某一行的样式，返回样式后可以设置样式内容
* 最后一列序号 + 1
*
*
* @return 列数
*/
public int getColumnCount() {
return getColumnCount(0);
}
/**
* 获取总列数，计算方法为：
*
*
* 最后一列序号 + 1
*
*
* @param rowNum 行号
* @return 列数，-1表示获取失败
*/
public int getColumnCount(final int rowNum) {
final Row row = this.sheet.getRow(rowNum);
if (null != row) {
// getLastCellNum方法返回序号+1的值
return row.getLastCellNum();
}
return -1;
}
/**
* 判断是否为xlsx格式的Excel表（Excel07格式）
*
* @return 是否为xlsx格式的Excel表（Excel07格式）
* @since 4.6.2
*/
public boolean isXlsx() {
return this.sheet instanceof XSSFSheet || this.sheet instanceof SXSSFSheet;
}
/**
* 获取Content-Type头对应的值，可以通过调用以下方法快速设置下载Excel的头信息：
*
*
* response.setContentType(excelWriter.getContentType());
*
*
* @return Content-Type值
* @since 5.6.7
*/
public String getContentType() {
return isXlsx() ? ExcelUtil.XLSX_CONTENT_TYPE : ExcelUtil.XLS_CONTENT_TYPE;
}
/**
* 关闭工作簿
