Springboot操作Poi进行Excel导入

以上就是我所要导入的Excel模板。

需求重点
经过这次需求的实现,发现这个需求最难的点 就在 模板模板模板！
模板中数据的关系，理清楚思路
模板中的数据，到底是多sheet还是说有合并单元进行导入

实现感悟
不管是多个sheet、还是说合并单元格、最重要的思想都是说将Excel中的数据进行拼接为一条一条的List记录进行新增数据。

 /**
 * 用于检测是否是表头行,此方法就是说代码再执行的时候找到一级这个字段的名称,然后知道获取字段从此开始
 *
 * @param row
 * @return
 */
 private boolean isHeaderRow(Row row) {
 Cell firstCell = row.getCell(1);
 return firstCell != null && "一级".equals(firstCell.getStringCellValue().trim());
 }

 /**
 * 
 * 上传文件，进行解析
 * @param file 文件
 * @return
 */
 public Result<?> importExcels(MultipartFile file) {
 Map<Integer, List<Object>> sheetDataMap = new HashMap<>();
 try {
 // 获取工作簿，解析整个Excel文件
 Workbook workbook = new XSSFWorkbook(file.getInputStream());
 // 循环遍历每个Sheet
 for (int i = 0; i < workbook.getNumberOfSheets(); i++) {
 Sheet sheet = workbook.getSheetAt(i);
 // 解析每个Sheet的数据
 List<Object> sheetData = parseSheet(sheet);
 // 将Sheet的数据存储到Map中，以Sheet名称为Key
 sheetDataMap.put(i, sheetData);
 }
 } catch (Exception e) {
 e.printStackTrace();
 }
 // 处理数据
 processingData(sheetDataMap);
 return Result.ok(sheetDataMap);
 }

 /**
 * 解析给定的Sheet，并将数据存储到List中
 *
 * @param sheet 要解析的Sheet
 * @return 包含每行数据的List
 */
 private List<Object> parseSheet(Sheet sheet) {
 List<Object> data = new ArrayList<>();
 boolean isHeaderFound = false;
 // 循环遍历每一行数据
 for (int rowIndex = 1; rowIndex <= sheet.getLastRowNum(); rowIndex++) {
 Row row = sheet.getRow(rowIndex);
 if (row == null) continue; // 跳过空行
 // 检查是否为标题行
 if (!isHeaderFound) {
 if (isHeaderRow(row)) {
 isHeaderFound = true; // 找到标题行
 }
 continue; // 跳过标题行的处理
 }
 // 收集每列的数据
 List<Object> rowData = new ArrayList<>();
 for (int cellIndex = 1; cellIndex <= row.getLastCellNum(); cellIndex++) {
 Cell cell = row.getCell(cellIndex);
 if (cell != null && isMergedCell(sheet, row, cell)) {
 rowData.add(getMergedCellValue(sheet, row, cell));
 } else if (cell != null && !cell.toString().trim().isEmpty()) {
 rowData.add(cell.toString()); // 仅当单元格不为空且不为空字符串时才添加
 }
 }
 if (!rowData.isEmpty()) {
 data.add(rowData); // 仅在行数据非空时才添加到数据列表中
 }
 }
 return data;
 }

 /**
 * 获取合并单元格的值
 */
 private String getMergedCellValue(Sheet sheet, Row row, Cell cell) {
 int rowIndex = row.getRowNum();
 int colIndex = cell.getColumnIndex();
 for (int i = 1; i < sheet.getNumMergedRegions(); i++) {
 CellRangeAddress range = sheet.getMergedRegion(i);
 if (range.isInRange(rowIndex, colIndex)) {
 Row firstRow = sheet.getRow(range.getFirstRow());
 Cell firstCell = firstRow.getCell(range.getFirstColumn());
 return firstCell.toString().replace("\n", " ");
 }
 }
 return null;
 }

 /**
 * 判断指定单元格是否是合并单元格
 */
 private boolean isMergedCell(Sheet sheet, Row row, Cell cell) {
 int rowIndex = row.getRowNum();
 int colIndex = cell.getColumnIndex();
 for (int i = 0; i < sheet.getNumMergedRegions(); i++) {
 CellRangeAddress range = sheet.getMergedRegion(i);
 if (range.isInRange(rowIndex, colIndex)) {
 return true;
 }
 }
 return false;
 }

 /**
 * 对于读取的Excel数据进行处理。
 *
 * @param data
 */
 private boolean processingData(Map<Integer, List<Object>> data) {
 boolean flag = false;
 try {
 // 遍历每个层级的数据，从0开始，每个level对应一个sheet的数据
 for (int level = 0; level < data.size(); level++) {
 List<Object> levelList = data.get(level);
 if (levelList == null) {
 continue; // 如果当前层级没有数据，跳过
 }
 // 遍历当前层级中的每一行数据
 for (Object o : levelList) {
 List<String> rowData = (List<String>) o;
 // 获取父类名称（如果是第一级，则父类名称为null）
 String parentName = level == 0 ? null : rowData.get(level - 1);
 // 获取当前的名称
 String currentName = rowData.get(level);
 try {
 String parentId = "0"; // 默认父类ID为"0"，表示顶级目录
 if (level > 0 && parentName != null) {
 // 如果不是第一级，通过父类名称查询父类ID
 Zbk parentZbk = zbkMapper.selectOne(new LambdaQueryWrapper<Zbk>()
 .eq(Zbk::getName, parentName.replace("\n", "")) // 去除换行符后匹配名称
 .eq(Zbk::getType, 1) // 类型为目录
 .eq(Zbk::getDeleted, 0)); // 未被删除
 if (parentZbk != null) {
 parentId = parentZbk.getId(); // 如果找到父类，获取父类ID
 } else {
 log.warn("父类不存在: " + parentName);
 continue;
 }
 }
 // 查询是否已经存在当前名称的指标/目录
 Zbk existingZbk = zbkMapper.selectOne(new LambdaQueryWrapper<Zbk>()
 .eq(Zbk::getName, currentName.replace("\n", "")) // 去除换行符后匹配名称
 .eq(Zbk::getType, level == data.size() - 1 ? 0 : 1) // 如果是最后一级，类型为指标，否则为目录
 .eq(Zbk::getDeleted, 0) // 未被删除
 .eq(Zbk::getPid, parentId)); // 父类ID匹配
 // 如果当前名称的指标/目录不存在，插入新的记录
 if (existingZbk == null) {
 Zbk zbk = new Zbk();
 zbk.setName(currentName);
 zbk.setDeleted(0); // 逻辑删除标志设为未删除
 zbk.setType(level == data.size() - 1 ? 0 : 1); // 设置类型，0为指标，1为目录
 zbk.setEvaluationObject("1382586667761336322");
 zbk.setZbSourceType("build");
 zbk.setDisable(false); // 未禁用
 zbk.setPid(parentId); // 设置父类ID
 zbkMapper.insert(zbk); // 插入新记录
 // 如果是最后一级（观测点），处理观测点的额外逻辑
 if (level == data.size() - 1) {
 flag = true;
 // 获取指标类型
 String indicatorType = rowData.get(rowData.size() - 2);
 // 获取指标分值
 String indicatorScore = rowData.get(rowData.size() - 1);
 if (!StringUtils.isEmpty(indicatorType) && !StringUtils.isEmpty(indicatorScore)) {
 // 处理指标数据
 processingIndicators(indicatorType, indicatorScore, zbk.getId());
 }
 }
 }
 } catch (Exception e) {
 log.error("处理数据时发生异常: ", e);
 continue; // 如果发生异常，跳过当前循环
 }
 }
 }
 } catch (Exception e) {
 log.error("处理数据时发生异常: ", e);
 e.printStackTrace();
 }
 return flag; // 返回处理结果的标志
 }