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.
cn.renlm.plugins.MyExcel.config.MySheet Maven / Gradle / Ivy
/*
* Copyright (c) 2020 Renlm
* MyExcel is licensed under Mulan PSL v2.
* You can use this software according to the terms and conditions of the Mulan PSL v2.
* You may obtain a copy of Mulan PSL v2 at:
* http://license.coscl.org.cn/MulanPSL2
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
* See the Mulan PSL v2 for more details.
*/
package cn.renlm.plugins.MyExcel.config;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.HorizontalAlignment;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.xssf.streaming.SXSSFSheet;
import org.apache.poi.xssf.streaming.SXSSFWorkbook;
import com.thoughtworks.xstream.annotations.XStreamAsAttribute;
import com.thoughtworks.xstream.annotations.XStreamImplicit;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.map.MapUtil;
import cn.hutool.core.util.CharUtil;
import cn.hutool.core.util.StrUtil;
import cn.renlm.plugins.ConstVal;
import cn.renlm.plugins.MyExcel.config.column.Alias;
import cn.renlm.plugins.MyExcel.config.column.Title;
import cn.renlm.plugins.MyExcel.entity.CellUnit;
import cn.renlm.plugins.MyExcel.entity.CheckResult;
import cn.renlm.plugins.MyExcel.handler.DataReadHandler;
import cn.renlm.plugins.MyExcel.util.MergeUtil;
import cn.renlm.plugins.MyExcel.util.StyleUtil;
import lombok.Data;
/**
* Sheet页配置
*
* @author RenLiMing(任黎明)
*
*/
@Data
public class MySheet implements Serializable {
private static final long serialVersionUID = 1L;
/**
* Sheet页名称
*/
@XStreamAsAttribute
private String name;
/**
* 起始行号（第一行：0）
*/
@XStreamAsAttribute
private int start = 0;
/**
* 冻结列数（左侧）
*/
@XStreamAsAttribute
private int freezes = 0;
/**
* 列配置集
*/
@XStreamImplicit(itemFieldName = "column")
private List columns;
/**
* 标题层级
*
* @return
*/
public int level() {
return this.columns.stream().map(col -> col.getTitle()).mapToInt(title -> {
if (StrUtil.isNotBlank(title.getSplit())) {
return title.getText().split(title.getSplit()).length;
}
return 1;
}).max().getAsInt();
}
/**
* 标题键-字段映射
*
* @return
*/
public Map keyFields() {
Map map = new LinkedHashMap<>();
for (MyColumn col : columns) {
if (col.isIgnore()) {
continue;
}
Title title = col.getTitle();
map.put(title.getText(), col.getField());
if (StrUtil.isNotBlank(title.getSplit())) {
map.put(title.getText().replace(title.getSplit(), ConstVal.NAME), col.getField());
}
if (CollUtil.isEmpty(col.getAliasList())) {
continue;
}
for (Alias alias : col.getAliasList()) {
if (StrUtil.isEmpty(alias.getText())) {
continue;
}
map.put(alias.getText(), col.getField());
if (StrUtil.isNotBlank(title.getSplit())) {
map.put(alias.getText().replace(title.getSplit(), ConstVal.NAME), col.getField());
}
}
}
return map;
}
/**
* 生成[标题|字段-值索引]映射
*
* @param titles
* @param dataReadHandler
* @return
*/
public List generateKeys(List> titles, DataReadHandler dataReadHandler) {
List keys = new ArrayList<>();
Map keyFields = keyFields();
CheckResult checkResult = new CheckResult().setRowIndex(this.start);
int length = titles.stream().mapToInt(it -> it.size()).max().getAsInt();
for (int i = 0; i < length; i++) {
List appends = new ArrayList<>();
for (List it : titles) {
String str = CollUtil.get(it, i);
if (StrUtil.isNotBlank(str)) {
if (appends.size() == 0 || !str.equals(appends.get(appends.size() - 1))) {
appends.add(str);
}
}
}
String key = CollUtil.join(appends, ConstVal.NAME);
if (keyFields.containsKey(key)) {
keys.add(keyFields.get(key));
} else {
keys.add(StrUtil.EMPTY);
}
}
// 检查模板
Map fieldColMap = this.getFieldColMap();
for (Map.Entry entry : fieldColMap.entrySet()) {
String field = entry.getKey();
MyColumn col = entry.getValue();
if (!col.isIgnore() && !col.isOptional() && !keys.contains(field)) {
String message = StrUtil.format("{} / {}，模板错误，缺失列", this.getName(), col.getTitle().getText());
checkResult.getErrors().add(message);
}
}
if (checkResult.isError()) {
dataReadHandler.handle(MapUtil.empty(), checkResult);
}
return keys;
}
/**
* 填充标题（根据模板配置）
*
* @param wb
* @param isTemplate
* @return [ { 字段: 单元格 } ]
*/
public List> fieldTitles(SXSSFWorkbook wb, boolean isTemplate) {
int level = level();
List> titles = new ArrayList<>();
for (int i = 0; i < level; i++) {
Map data = new LinkedHashMap<>();
titles.add(data);
for (MyColumn col : columns) {
// 忽略字段
if (col.isIgnore()) {
continue;
}
// 模板不显示可选字段
if (isTemplate && col.isOptional()) {
continue;
}
// 单元格属性
CellUnit cellUnit = new CellUnit();
cellUnit.setRowIndex(start + i);
cellUnit.setColIndex(i);
cellUnit.setColumn(col);
data.put(col.getField(), cellUnit);
// 单级表头，竖向填充
if (StrUtil.isBlank(col.getTitle().getSplit())) {
short fontSize = level == 1 ? ConstVal.FONT_SIZE : (ConstVal.FONT_SIZE + 1);
CellStyle cellStyle = StyleUtil.createCellStyleWithBorder(wb, ConstVal.FONT, fontSize, true,
col.getAlign());
cellUnit.setText(col.getTitle().getText());
cellUnit.setCellStyle(cellStyle);
}
// 多级表头，由末级竖向填充
else {
String[] arr = col.getTitle().getText().split(col.getTitle().getSplit());
short fontSize = i >= (arr.length - 1) ? ConstVal.FONT_SIZE : (ConstVal.FONT_SIZE + 1);
HorizontalAlignment align = i >= (arr.length - 1) ? col.getAlign() : HorizontalAlignment.CENTER;
CellStyle cellStyle = StyleUtil.createCellStyleWithBorder(wb, ConstVal.FONT, fontSize, true, align);
cellUnit.setText(i >= arr.length ? arr[arr.length - 1] : arr[i]);
cellUnit.setCellStyle(cellStyle);
}
}
}
return titles;
}
/**
* 填充标题（根据读取数据）
*
* @param rowList
* @return
*/
public static final List fillTitle(List rowList) {
String lastTitle = null;
List list = new ArrayList<>();
for (Object obj : rowList) {
if (obj == null) {
list.add(lastTitle);
} else {
String str = obj.toString().chars().filter(c -> !CharUtil.isBlankChar(c))
.mapToObj(c -> CharUtil.toString((char) c)).collect(Collectors.joining());
if (StrUtil.isEmptyIfStr(str)) {
list.add(StrUtil.EMPTY);
} else {
lastTitle = str;
list.add(lastTitle);
}
}
}
return list;
}
/**
* 填充页签表头并返回行列坐标集
*
* @param start
* @param sh
* @param fieldTitles
* @return
*/
public Map> writeSheetTitle(final int start, SXSSFSheet sh,
List> fieldTitles) {
int currentRowIndex = start;
Map> rowColMap = new LinkedHashMap<>();
for (Map data : fieldTitles) {
int currentColIndex = 0;
// 新建行
Row row = sh.createRow(currentRowIndex++);
for (Map.Entry entry : data.entrySet()) {
// 新建列
Cell cell = row.createCell(currentColIndex++);
// 设置单元格值与样式
cell.setCellValue(entry.getValue().getText());
cell.setCellStyle(entry.getValue().getCellStyle());
// 行列坐标映射
MergeUtil.pushTitleRowColMap(rowColMap, entry.getValue().getText(),
new Integer[] { row.getRowNum(), cell.getColumnIndex() });
}
}
return rowColMap;
}
/**
* 数据字段-列配置映射
*/
private Map fieldColMap;
/**
* 获取数据字段-列配置映射
*
* @return
*/
public Map getFieldColMap() {
if (fieldColMap == null) {
fieldColMap = new LinkedHashMap<>();
for (MyColumn col : columns) {
fieldColMap.put(col.getField(), col);
}
}
return fieldColMap;
}
}
