package com.panzhihua.common.excel;
|
|
import com.alibaba.excel.write.handler.SheetWriteHandler;
|
import com.alibaba.excel.write.metadata.holder.WriteSheetHolder;
|
import com.alibaba.excel.write.metadata.holder.WriteWorkbookHolder;
|
import org.apache.poi.ss.usermodel.DataValidation;
|
import org.apache.poi.ss.usermodel.DataValidationConstraint;
|
import org.apache.poi.ss.usermodel.DataValidationHelper;
|
import org.apache.poi.ss.util.CellRangeAddressList;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
|
/**
|
* 自定义拦截器.对第一列第一行和第二行的数据新增下拉框，显示 测试1 测试2
|
*
|
* @author Jiaju Zhuang
|
*/
|
public class CustomSheetWriteHandler implements SheetWriteHandler {
|
|
private static final Logger LOGGER = LoggerFactory.getLogger(CustomSheetWriteHandler.class);
|
|
@Override
|
public void beforeSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) {
|
|
}
|
|
@Override
|
public void afterSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) {
|
LOGGER.info("第{}个Sheet写入成功。", writeSheetHolder.getSheetNo());
|
|
// 区间设置 第一列第一行和第二行的数据。由于第一行是头，所以第一、二行的数据实际上是第二三行
|
CellRangeAddressList cellRangeAddressList = new CellRangeAddressList(1, 2000, 1, 1);
|
CellRangeAddressList cellRangeAddressList1 = new CellRangeAddressList(1, 2000, 3 ,3);
|
CellRangeAddressList cellRangeAddressList2 = new CellRangeAddressList(1, 2000, 10 ,10);
|
DataValidationHelper helper = writeSheetHolder.getSheet().getDataValidationHelper();
|
DataValidationConstraint constraint = helper.createExplicitListConstraint(new String[] {"是", "否"});
|
DataValidationConstraint constraint1 = helper.createExplicitListConstraint(new String[] {"中共党员","中共预备党员","共青团员","民革党员","民盟盟员","民建会员","农工党党员","致公党党员","九三学社社员","台盟盟员","无党派人士","群众"});
|
DataValidationConstraint constraint2 = helper.createExplicitListConstraint(new String[] {"男", "女","未知"});
|
DataValidation dataValidation = helper.createValidation(constraint2, cellRangeAddressList);
|
DataValidation dataValidation1 = helper.createValidation(constraint, cellRangeAddressList1);
|
DataValidation dataValidation2= helper.createValidation(constraint1, cellRangeAddressList2);
|
|
writeSheetHolder.getSheet().addValidationData(dataValidation);
|
writeSheetHolder.getSheet().addValidationData(dataValidation1);
|
writeSheetHolder.getSheet().addValidationData(dataValidation2);
|
}
|
}
|
