最近有个导出表格里边数据进行联动的需求要求GH列表进行联动,G列有2个选择,文档库和数据库,当选择文档库时,H列选择项目有ftp和sftp2个,当选择数据库时H列有mysql
,oracle
,postgersql,kingbase,sqlserver2017,
dm,
greenplum
,db2,gbase,sybase,mongoDB,gbase,hive,redis,sqlserver其他等如下:
使用EasyExcel如下获取数据列表需要依据自己的实际情况进行调整:
public class CustomDataAssetSheetWriteHandler implements SheetWriteHandler {
private DmssDictionaryService dmssDictionaryService;
public CustomDataAssetSheetWriteHandler(DmssDictionaryService dmssDictionaryService) {
this.dmssDictionaryService = dmssDictionaryService;
}
@Override
public void beforeSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) {
}
@Override
public void afterSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) {
Long start = System.currentTimeMillis();
//是否集群
List<DmssDictionaryPo> lists = dmssDictionaryService.findByType("BOOLEAN_CONSTANT");
//区间设置 第一列第一行和第二行的数据。由于第一行是头,所以第一、二行的数据实际上是第二三行
CellRangeAddressList cellRangeAddressList3 = new CellRangeAddressList(1, 20000, 8, 8);
CustomHandlerHelper.addValidation(writeSheetHolder, lists, cellRangeAddressList3);
//是否敏感资产
CellRangeAddressList cellRangeAddressList4 = new CellRangeAddressList(1, 20000, 13, 13);
CustomHandlerHelper.addValidation(writeSheetHolder, lists, cellRangeAddressList4);
//导出数据下拉二级联动
// 获取数据类型、数据库类型和文件类型列表
List<DmssDictionaryPo> dataTypes = dmssDictionaryService.findByType("DATA_TYPE");
List<DmssDictionaryPo> dbTypes = dmssDictionaryService.findByType("DB_TYPE");
List<DmssDictionaryPo> fileTypes = dmssDictionaryService.findByType("FILE_TYPE");
// 列表值
List<String> typeLists = dataTypes.stream().map(DmssDictionaryPo::getMessage).collect(Collectors.toList());
ArrayList<String> firstTypeList = (ArrayList<String>) dbTypes.stream().map(DmssDictionaryPo::getMessage).collect(Collectors.toList());
ArrayList<String> secondTypeList = (ArrayList<String>) fileTypes.stream().map(DmssDictionaryPo::getMessage).collect(Collectors.toList());
HashMap<String, ArrayList<String>> firstKeyMap = new HashMap<>();
firstKeyMap.put("数据库", firstTypeList);
firstKeyMap.put("文档库", secondTypeList);
// 获取工作簿和工作表对象
Workbook book = writeWorkbookHolder.getWorkbook();
Sheet sheet = writeSheetHolder.getSheet();
Sheet hideProvinceCityDistrictSheet = book.createSheet("typeDistrict");
book.setSheetHidden(book.getSheetIndex(hideProvinceCityDistrictSheet), true);
int rowId = 0;
// 设置第一行,存列表
Row firstRow = hideProvinceCityDistrictSheet.createRow(rowId++);
firstRow.createCell(0).setCellValue("类型");
for (int i = 0; i < typeLists.size(); i++) {
Cell cell = firstRow.createCell(i + 1);
cell.setCellValue(typeLists.get(i));
}
// 添加名称管理器
String provinceRange = DataValidationUtil.getRange(1, rowId, 2);
Name firstListName = book.createName();
//key不可重复
firstListName.setNameName("资产类型");
String provinceFormula = "typeDistrict!" + provinceRange;
firstListName.setRefersToFormula(provinceFormula);
// firstListName,行开头为父级区域,后面是子区域。
String[] firstListNameArr = typeLists.toArray(new String[typeLists.size()]);
DataValidationHelper dvHelper = sheet.getDataValidationHelper();
//类型
DataValidationConstraint typeConstraint = dvHelper.createExplicitListConstraint(firstListNameArr);
CellRangeAddressList provRangeAddressList = new CellRangeAddressList(1, 20000, 6, 6);
DataValidation typeDataValidation = dvHelper.createValidation(typeConstraint, provRangeAddressList);
//验证
typeDataValidation.createErrorBox("error", "请选择正确的数据");
typeDataValidation.setShowErrorBox(true);
typeDataValidation.setSuppressDropDownArrow(true);
sheet.addValidationData(typeDataValidation);
// 创建线程池,线程数根据需要设定
ExecutorService executorService = Executors.newFixedThreadPool(4);
// 对前2w行设置有效性,每个任务处理1000行
for (int i = 1; i < 20000; i += 1000) {
// 提交任务到线程池
int finalI = i;
executorService.submit(() -> {
for (int j = finalI; j < finalI + 1000 && j < 20000; j++) {
setDataValidation("G", sheet, j, 8, dvHelper);
}
});
}
// 关闭线程池
executorService.shutdown();
//首列的下拉列表
Set<Map.Entry<String, ArrayList<String>>> provinceCityEntries = firstKeyMap.entrySet();
for (Map.Entry<String, ArrayList<String>> provinceCityEntry : provinceCityEntries) {
String provinceKey = provinceCityEntry.getKey();
ArrayList<String> cityList = provinceCityEntry.getValue();
Row row = hideProvinceCityDistrictSheet.createRow(rowId++);
row.createCell(0).setCellValue(provinceKey);
for (int j = 0; j < cityList.size(); j++) {
Cell cell = row.createCell(j + 1);
cell.setCellValue(cityList.get(j));
}
// 添加名称管理器
String range = DataValidationUtil.getRange(1, rowId, cityList.size());
Name name = book.createName();
//key不可重复
name.setNameName(provinceKey);
String formula = "typeDistrict!" + range;
name.setRefersToFormula(formula);
}
}
/**
* 设置有效性
*
* @param offset 主影响单元格所在列,即此单元格由哪个单元格影响联动
* @param sheet 工作簿
* @param rowNum 行数
* @param colNum 列数
* @param dvHelper 校验器
*/
public static void setDataValidation(String offset, Sheet sheet, int rowNum, int colNum, DataValidationHelper dvHelper) {
DataValidation data_validation_list = getDataValidationByFormula("INDIRECT($" + offset + (rowNum) + ")", rowNum, colNum, dvHelper);
sheet.addValidationData(data_validation_list);
}
/**
* 加载下拉列表内容
*
* @param formulaString 函数
* @param naturalRowIndex 行数
* @param naturalColumnIndex 列数
* @param dvHelper
* @return
*/
private static DataValidation getDataValidationByFormula(String formulaString, int naturalRowIndex, int naturalColumnIndex, DataValidationHelper dvHelper) {
// 加载下拉列表内容
// 举例:若formulaString = "INDIRECT($A$2)" 表示规则数据会从名称管理器中获取key与单元格 A2 值相同的数据,
XSSFDataValidationConstraint dvConstraint = (XSSFDataValidationConstraint) dvHelper.createFormulaListConstraint(formulaString);
// 设置数据有效性加载在哪个单元格上。
// 四个参数分别是:起始行、终止行、起始列、终止列
int firstRow = naturalRowIndex - 1;
int lastRow = naturalRowIndex - 1;
int firstCol = naturalColumnIndex - 1;
int lastCol = naturalColumnIndex - 1;
CellRangeAddressList regions = new CellRangeAddressList(firstRow, lastRow, firstCol, lastCol);
// 数据有效性对象
// 绑定
XSSFDataValidation data_validation_list = (XSSFDataValidation) dvHelper.createValidation(dvConstraint, regions);
data_validation_list.setEmptyCellAllowed(false);
if (data_validation_list instanceof XSSFDataValidation) {
data_validation_list.setSuppressDropDownArrow(true);
data_validation_list.setShowErrorBox(true);
} else {
data_validation_list.setSuppressDropDownArrow(false);
}
// 设置输入信息提示信息
data_validation_list.createPromptBox("下拉选择提示", "请使用下拉方式选择合适的值!");
// 设置输入错误提示信息
data_validation_list.createErrorBox("选择错误提示", "你输入的值未在备选列表中,请下拉选择合适的值!");
return data_validation_list;
}
}
最后在导出时进行注册即可: