导出数据模版增加二级联动功能

最近有个导出表格里边数据进行联动的需求要求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;
  }


}


最后在导出时进行注册即可:

猜你喜欢

转载自blog.csdn.net/u012440725/article/details/131290062
今日推荐