package org.dromara.common.excel.handler;
|
|
import cn.hutool.core.collection.CollUtil;
|
import com.alibaba.excel.metadata.data.DataFormatData;
|
import com.alibaba.excel.metadata.data.WriteCellData;
|
import com.alibaba.excel.util.StyleUtil;
|
import com.alibaba.excel.write.handler.CellWriteHandler;
|
import com.alibaba.excel.write.handler.SheetWriteHandler;
|
import com.alibaba.excel.write.handler.context.CellWriteHandlerContext;
|
import com.alibaba.excel.write.metadata.holder.WriteSheetHolder;
|
import com.alibaba.excel.write.metadata.style.WriteCellStyle;
|
import com.alibaba.excel.write.metadata.style.WriteFont;
|
import org.apache.poi.ss.usermodel.*;
|
import org.apache.poi.xssf.usermodel.XSSFClientAnchor;
|
import org.apache.poi.xssf.usermodel.XSSFRichTextString;
|
import org.dromara.common.core.utils.reflect.ReflectUtils;
|
import org.dromara.common.excel.annotation.ExcelNotation;
|
import org.dromara.common.excel.annotation.ExcelRequired;
|
|
import java.lang.reflect.Field;
|
import java.util.HashMap;
|
import java.util.Map;
|
|
/**
|
* 批注、必填
|
*
|
* @author guzhouyanyu
|
*/
|
public class DataWriteHandler implements SheetWriteHandler, CellWriteHandler {
|
|
/**
|
* 批注
|
*/
|
private final Map<Integer, String> notationMap;
|
|
/**
|
* 头列字体颜色
|
*/
|
private final Map<Integer, Short> headColumnMap;
|
|
|
public DataWriteHandler(Class<?> clazz) {
|
notationMap = getNotationMap(clazz);
|
headColumnMap = getRequiredMap(clazz);
|
}
|
|
@Override
|
public void afterCellDispose(CellWriteHandlerContext context) {
|
if (CollUtil.isEmpty(notationMap) && CollUtil.isEmpty(headColumnMap)) {
|
return;
|
}
|
WriteCellData<?> cellData = context.getFirstCellData();
|
WriteCellStyle writeCellStyle = cellData.getOrCreateStyle();
|
|
DataFormatData dataFormatData = new DataFormatData();
|
// 单元格设置为文本格式
|
dataFormatData.setIndex((short) 49);
|
writeCellStyle.setDataFormatData(dataFormatData);
|
|
if (context.getHead()) {
|
Cell cell = context.getCell();
|
WriteSheetHolder writeSheetHolder = context.getWriteSheetHolder();
|
Sheet sheet = writeSheetHolder.getSheet();
|
Workbook workbook = writeSheetHolder.getSheet().getWorkbook();
|
Drawing<?> drawing = sheet.createDrawingPatriarch();
|
// 设置标题字体样式
|
WriteFont headWriteFont = new WriteFont();
|
// 加粗
|
headWriteFont.setBold(true);
|
if (CollUtil.isNotEmpty(headColumnMap) && headColumnMap.containsKey(cell.getColumnIndex())) {
|
// 设置字体颜色
|
headWriteFont.setColor(headColumnMap.get(cell.getColumnIndex()));
|
}
|
writeCellStyle.setWriteFont(headWriteFont);
|
CellStyle cellStyle = StyleUtil.buildCellStyle(workbook, null, writeCellStyle);
|
cell.setCellStyle(cellStyle);
|
|
if (CollUtil.isNotEmpty(notationMap) && notationMap.containsKey(cell.getColumnIndex())) {
|
// 批注内容
|
String notationContext = notationMap.get(cell.getColumnIndex());
|
// 创建绘图对象
|
Comment comment = drawing.createCellComment(new XSSFClientAnchor(0, 0, 0, 0, (short) cell.getColumnIndex(), 0, (short) 5, 5));
|
comment.setString(new XSSFRichTextString(notationContext));
|
cell.setCellComment(comment);
|
}
|
}
|
}
|
|
/**
|
* 获取必填列
|
*/
|
private static Map<Integer, Short> getRequiredMap(Class<?> clazz) {
|
Map<Integer, Short> requiredMap = new HashMap<>();
|
Field[] fields = clazz.getDeclaredFields();
|
// 检查 fields 数组是否为空
|
if (fields.length == 0) {
|
return requiredMap;
|
}
|
Field[] filteredFields = ReflectUtils.getFields(clazz, field -> !"serialVersionUID".equals(field.getName()));
|
|
for (int i = 0; i < filteredFields.length; i++) {
|
Field field = filteredFields[i];
|
if (!field.isAnnotationPresent(ExcelRequired.class)) {
|
continue;
|
}
|
ExcelRequired excelRequired = field.getAnnotation(ExcelRequired.class);
|
int columnIndex = excelRequired.index() == -1 ? i : excelRequired.index();
|
requiredMap.put(columnIndex, excelRequired.fontColor().getIndex());
|
}
|
return requiredMap;
|
}
|
|
/**
|
* 获取批注
|
*/
|
private static Map<Integer, String> getNotationMap(Class<?> clazz) {
|
Map<Integer, String> notationMap = new HashMap<>();
|
Field[] fields = clazz.getDeclaredFields();
|
// 检查 fields 数组是否为空
|
if (fields.length == 0) {
|
return notationMap;
|
}
|
Field[] filteredFields = ReflectUtils.getFields(clazz, field -> !"serialVersionUID".equals(field.getName()));
|
for (int i = 0; i < filteredFields.length; i++) {
|
Field field = filteredFields[i];
|
if (!field.isAnnotationPresent(ExcelNotation.class)) {
|
continue;
|
}
|
ExcelNotation excelNotation = field.getAnnotation(ExcelNotation.class);
|
int columnIndex = excelNotation.index() == -1 ? i : excelNotation.index();
|
notationMap.put(columnIndex, excelNotation.value());
|
}
|
return notationMap;
|
}
|
}
|
