目的1:自定义RGB 来自定义表头 导出 excel 的表头颜色
目的2:excel 下拉框数据验证
pom 依赖
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>easyexcel</artifactId>
<version>2.2.7</version>
</dependency>
相关示例代码
ExcelUtils 工具类中可以本地测试
/**
* 表头 颜色 定义
*/
public class ComplexHeadStyles {
/**
* 表头横坐标 - 行
*/
private Integer x;
/**
* 表头纵坐标 - 列
*/
private Integer y;
/**
* 内置颜色
*/
private Short indexColor;
private Integer red;
private Integer green;
private Integer blue;
/**
* 字体颜色
*/
private Short fontColor;
public ComplexHeadStyles(Integer x, Integer y, Short indexColor) {
this.x = x;
this.y = y;
this.indexColor = indexColor;
}
public ComplexHeadStyles(Integer x, Integer y, Integer red, Integer green, Integer blue) {
this.x = x;
this.y = y;
this.red = red;
this.green = green;
this.blue = blue;
}
public void setFontColor(Short fontColor) {
this.fontColor = fontColor;
}
public Integer getX() {
return x;
}
public Integer getY() {
return y;
}
public Short getIndexColor() {
return indexColor;
}
public Integer getRed() {
return red;
}
public Integer getGreen() {
return green;
}
public Integer getBlue() {
return blue;
}
public Short getFontColor() {
return fontColor;
}
}
/**
* 自定义表头样式拦截器
*
* @since 2022-07-21 14:26
*/
public class CustomHeadWriteHandler extends AbstractCellStyleStrategy {
/**
* 复杂表头自定义样式队列,先进先出,方便存储
*/
private ArrayBlockingQueue<ComplexHeadStyles> headStylesQueue;
/**
* 需要渲染下拉框
*/
private Map<Integer, String[]> dropdownBox = new HashMap<>();
/**
* CellStyle
*/
private CellStyle cellStyle;
public CustomHeadWriteHandler(ArrayBlockingQueue<ComplexHeadStyles> headStylesQueue) {
this.headStylesQueue = headStylesQueue;
}
public CustomHeadWriteHandler(ArrayBlockingQueue<ComplexHeadStyles> headStylesQueue, Map<Integer, String[]> dropdownBox) {
this.headStylesQueue = headStylesQueue;
this.dropdownBox = dropdownBox;
}
@Override
public void afterCellCreate(WriteSheetHolder writeSheetHolder, WriteTableHolder writeTableHolder, Cell cell, Head head, Integer relativeRowIndex, Boolean isHead) {
Sheet sheet = writeSheetHolder.getSheet();
if (isHead) {
if (headStylesQueue != null && !headStylesQueue.isEmpty()) {
ComplexHeadStyles complexHeadStyle = headStylesQueue.peek();
// 取出队列中的自定义表头信息,与当前坐标比较,判断是否相符
if (cell.getColumnIndex() == complexHeadStyle.getY() && relativeRowIndex.equals(complexHeadStyle.getX())) {
Workbook workbook = sheet.getWorkbook();
CellStyle cellStyle = initCellStyleCustom(workbook, complexHeadStyle.getFontColor());
if (ObjectUtil.isNotEmpty(complexHeadStyle.getIndexColor())) {
//有指定颜色颜色 索引
cellStyle.setFillForegroundColor(complexHeadStyle.getIndexColor());
cellStyle.setFillPattern(FillPatternType.SOLID_FOREGROUND);
cell.setCellStyle(cellStyle);
} else {
//有指定颜色 RGB
if (ObjectUtil.isNotEmpty(complexHeadStyle.getRed()) && ObjectUtil.isNotEmpty(complexHeadStyle.getGreen()) && ObjectUtil.isNotEmpty(complexHeadStyle.getBlue())) {
XSSFCellStyle xssfCellStyle = (XSSFCellStyle)workbook.createCellStyle();
xssfCellStyle.cloneStyleFrom(cellStyle);
xssfCellStyle.setFillForegroundColor(new XSSFColor(new java.awt.Color(complexHeadStyle.getRed(), complexHeadStyle.getGreen(), complexHeadStyle.getBlue())));
xssfCellStyle.setFillPattern(FillPatternType.SOLID_FOREGROUND);
cell.setCellStyle(xssfCellStyle);
}
}
// 样式出队
headStylesQueue.poll();
}
}
} else {
cell.setCellStyle(cellStyle);
}
//设置下拉框选项
DataValidationHelper helper = sheet.getDataValidationHelper();
for (Map.Entry<Integer, String[]> entry : dropdownBox.entrySet()) {
/***起始行、终止行、起始列、终止列**/
CellRangeAddressList addressList = new CellRangeAddressList(1, 9999, entry.getKey(), entry.getKey());
/***设置下拉框数据**/
DataValidationConstraint constraint = helper.createExplicitListConstraint(entry.getValue());
DataValidation dataValidation = helper.createValidation(constraint, addressList);
/***处理Excel兼容性问题**/
if (dataValidation instanceof XSSFDataValidation) {
dataValidation.setSuppressDropDownArrow(true);
dataValidation.setShowErrorBox(true);
} else {
dataValidation.setSuppressDropDownArrow(false);
}
sheet.addValidationData(dataValidation);
}
}
@Override
protected void initCellStyle(Workbook workbook) {
CellStyle style = workbook.createCellStyle();
Font font = workbook.createFont();
font.setFontName("微软雅黑");
font.setFontHeightInPoints((short)11);
font.setColor(IndexedColors.BLACK.getIndex());
style.setFont(font);
DataFormat dataFormat = workbook.createDataFormat();
style.setDataFormat(dataFormat.getFormat("@"));
this.cellStyle = style;
}
@Override
protected void setHeadCellStyle(Cell cell, Head head, Integer relativeRowIndex) {
}
@Override
protected void setContentCellStyle(Cell cell, Head head, Integer relativeRowIndex) {
}
private CellStyle initCellStyleCustom(Workbook workbook, Short fontColor) {
CellStyle style = workbook.createCellStyle();
style.setBorderBottom(BorderStyle.THIN); // 下边框
style.setBorderLeft(BorderStyle.THIN); // 左边框
style.setBorderTop(BorderStyle.THIN); // 上边框
style.setBorderRight(BorderStyle.THIN); // 右边框
Font font = workbook.createFont();
font.setFontName("微软雅黑");
font.setBold(true);
font.setFontHeightInPoints((short)13);
font.setColor(IndexedColors.BLACK.getIndex());
if (ObjectUtil.isNotNull(fontColor)) {
font.setColor(fontColor);
}
style.setFont(font);
style.setAlignment(LEFT);
DataFormat dataFormat = workbook.createDataFormat();
return style;
}
}
/**
* excel 工具类
*/
public class ExcelUtils {
/**
* excel 导出样式 默认
*
* @return
*/
public static WriteHandler setExcelStyle() {
// 头的策略
WriteCellStyle headWriteCellStyle = new WriteCellStyle();
// 背景设置为橙色
headWriteCellStyle.setFillForegroundColor(IndexedColors.ORANGE.getIndex());
WriteFont headWriteFont = new WriteFont();
headWriteFont.setColor(IndexedColors.WHITE.getIndex());
headWriteFont.setFontName("微软雅黑");
headWriteFont.setFontHeightInPoints((short)13);
headWriteCellStyle.setWriteFont(headWriteFont);
// 内容的策略
WriteCellStyle contentWriteCellStyle = new WriteCellStyle();
WriteFont contentWriteFont = new WriteFont();
contentWriteFont.setFontName("微软雅黑");
contentWriteFont.setFontHeightInPoints((short)11);
contentWriteFont.setColor(IndexedColors.BLACK.getIndex());
// 字体大小
contentWriteCellStyle.setWriteFont(contentWriteFont);
// 这个策略是 头是头的样式 内容是内容的样式 其他的策略可以自己实现
return new HorizontalCellStyleStrategy(headWriteCellStyle, contentWriteCellStyle);
}
/**
* 导出文件时为Writer生成OutputStream
*
* @param fileName 文件名
* @param response response
* @return 流
*/
public static OutputStream getOutputStream(String fileName, HttpServletResponse response) throws Exception {
try {
fileName = URLEncoder.encode(fileName, "utf-8");
response.setCharacterEncoding("utf8");
response.setContentType("multipart/form-data");
response.setHeader("Content-Disposition", "attachment;filename*=utf-8'zh_cn'" + fileName + ".xlsx");
return response.getOutputStream();
} catch (IOException e) {
throw new Exception("导出excel表格失败!", e);
}
}
/**
* excel 导出样式 测试
*
* @return
*/
public static WriteHandler setTestTemplateSheet0() {
// 设置表头样式队列【先进先出】
ArrayBlockingQueue<ComplexHeadStyles> complexHeadStylesArrayBlockingQueue = new ArrayBlockingQueue<>(3);
/**
* 颜色
*/
complexHeadStylesArrayBlockingQueue.add(new ComplexHeadStyles(0, 0, 102, 255, 255));
complexHeadStylesArrayBlockingQueue.add(new ComplexHeadStyles(0, 1, IndexedColors.ORANGE.getIndex()));
complexHeadStylesArrayBlockingQueue.add(new ComplexHeadStyles(0, 2, IndexedColors.GREY_25_PERCENT.getIndex()));
/**
* 下拉框
*/
Map<Integer, String[]> dropdownBox = new HashMap<>();
dropdownBox.put(0,new String[]{"选项1","选项2"});
CustomHeadWriteHandler headStyleWriteHandler = new CustomHeadWriteHandler(complexHeadStylesArrayBlockingQueue,dropdownBox);
return headStyleWriteHandler;
}
public static List<List<String>> getTestHeader0(List<List<Object>> list0) {
List<List<String>> list = Lists.newArrayList();
List<String> head0 = Lists.newArrayList("第一列");
List<String> head1 = Lists.newArrayList("第二列");
List<String> head2 = Lists.newArrayList("第三列");
list.add(head0);
list.add(head1);
list.add(head2);
List<Object> row1 = Lists.newArrayList("111", "222", "333");
list0.add(row1);
return list;
}
public static ByteArrayOutputStream getExcelTestOut(HttpServletResponse response) {
ExcelWriter excelWriter = null;
ByteArrayOutputStream out = new ByteArrayOutputStream();
try {
if (ObjectUtil.isNotEmpty(response)) {
excelWriter = EasyExcel.write(getOutputStream("test", response)).registerWriteHandler(setExcelStyle()).build();
} else {
excelWriter = EasyExcel.write(out).registerWriteHandler(setExcelStyle()).build();
}
List<List<Object>> list0 = Lists.newArrayList();
List<List<String>> header0 = getTestHeader0(list0);
//获取sheet对象
WriteSheet sheet0 = EasyExcel.writerSheet(0, "test").head(header0).registerWriteHandler(ExcelUtils.setTestTemplateSheet0()).build();
excelWriter.write(list0, sheet0);
} catch (Exception e) {
e.printStackTrace();
} finally {
excelWriter.finish();
}
return out;
}
public static void main(String[] args) throws IOException {
ByteArrayOutputStream excelTestOut = getExcelTestOut(null);
//建立文件
File file = new File("D:\\test.xlsx");
if (file.exists()) {
System.out.println("文件已存在");
return;
}
file.createNewFile();
FileOutputStream fileOutputStream = new FileOutputStream(file.getPath());
fileOutputStream.write(excelTestOut.toByteArray());
fileOutputStream.close();
}
}