java使用EasyExcel实现Sheet的复制与填充

@Builder
@Data
@AllArgsConstructor
@NoArgsConstructor
public class CatDTO {
    @ExcelProperty(value = "姓名")
    private String name;
    @ExcelProperty(value = "品种")
    @ColumnWidth(15)
    private String breed;
    @ExcelProperty(value = "年龄")
    private Double age;
    @ExcelProperty(value = "描述")
    @ColumnWidth(30)
    private String desc;
}

@SpringBootTest
class DemoApplicationTests {
    @Test
    void writeTest1() {
        String fileName = "write-" + System.currentTimeMillis() + ".xlsx";
        // 构建数据
        List<CatDTO> catWriteData = List.of(
                CatDTO.builder().name("肥猫").breed("银渐层").age(1.2).desc("每天吃吃睡睡睡睡睡").build(),
                CatDTO.builder().name("虎妞").breed("德文卷毛猫").age(0.6).desc("活泼好动无敌可爱爱粘人").build()
        );
        EasyExcel.write(fileName, CatDTO.class).sheet().doWrite(catWriteData);
    }
}

@Builder
@Data
@AllArgsConstructor
@NoArgsConstructor
public class CatFillDTO {
    private String name;
    private String breed;
    private Double age;
    private String desc;
}

@SpringBootTest
class DemoApplicationTests {
    @Test
    void fillTest1() {
        String fileName = "fill-" + System.currentTimeMillis() + ".xlsx";
        // 构建数据
        List<CatFillDTO> catFillData = List.of(
                CatFillDTO.builder().name("肥猫").breed("银渐层").age(1.2).desc("每天吃吃睡睡睡睡睡").build(),
                CatFillDTO.builder().name("虎妞").breed("德文卷毛猫").age(0.6).desc("活泼好动无敌可爱爱粘人").build()
        );
        FileUtil.touch(fileName);
        // 加载模板
        InputStream resourceAsStream = ResourceUtil.getStreamSafe("temp/cat-template.xlsx");
        // 生成工作簿对象
        ExcelWriter excelWriter = EasyExcel.write(fileName).withTemplate(resourceAsStream).build();
        // 生成工作表对象
        WriteSheet writeSheet = EasyExcel.writerSheet().build();
        // 多组数据填充需要另起一行时 forceNewRow(true)
        FillConfig fillConfig = FillConfig.builder().forceNewRow(true).build();
        // 多组数据填充 List<实体类> 对象
        excelWriter.fill(catFillData, fillConfig, writeSheet);
        // 关闭excelWriter
        excelWriter.finish();
    }
}

@Service
@Slf4j
@RequiredArgsConstructor
public class SpeakerServiceImpl implements ISpeakerService {
   /**
    * 填充模板文件
    *
    * @param sheet1Data sheet1数据
    * @param promptMap  提示地图
    * @return {@code File}
    */
   public File fillTemplateFile(
           CustomSpeakerDemandExcelDTO sheet1Data, Map<String, List<CustomSpeakerPromptDTO>> promptMap) {
      try {
         // 构建模板
         byte[] asInputStream = this.buildTemplateFile(promptMap);
         // 创建临时导出文件
         File temporaryFile = Files.createTempFile("speaker_task_", ".xlsx").toFile();
         // easyExcel API 根据is作为模板填充数据，并写入临时文件temporaryFile
         ExcelWriter excelWriter = EasyExcel.write(temporaryFile)
                 .withTemplate(new ByteArrayInputStream(asInputStream))
                 .build();
         // 填充模板1的sheet
         WriteSheet writeSheet1 = EasyExcel.writerSheet(0).build();
         excelWriter.fill(sheet1Data, writeSheet1);
         // 填充模板2的多个sheet
         int sheetIndex = 1;
         for (String key : promptMap.keySet()) {
            WriteSheet writeSheet = EasyExcel.writerSheet(sheetIndex).build();
            // 写入数据到当前sheet
            List<CustomSpeakerPromptDTO> prompt = promptMap.get(key);
            excelWriter.fill(prompt, writeSheet);
            sheetIndex++;
         }
         excelWriter.finish();
         return temporaryFile;
      } catch (IOException e) {
         throw new RuntimeException(e);
      }
   }
   /**
    * 构建模板文件
    *
    * @param promptMap 提示地图
    * @return {@code byte[]}
    * @throws IOException ioexception
    */
   public byte[] buildTemplateFile(Map<String, List<CustomSpeakerPromptDTO>> promptMap) throws IOException {
      // 加载模板文件
      InputStream templateStream = ResourceUtil.getStreamSafe("temp/speaker-task-template.xlsx");
      Workbook workbook = WorkbookFactory.create(templateStream);
      // 设置模板sheet数量, 需要复制promptMap.size()-1 次工作表2
      boolean first = true;
      for (String key : promptMap.keySet()) {
         if (first) {
            // 第一组数据可以直接使用当前模板，不需要进行复制
            workbook.setSheetName(1, key);
         } else {
            // 复制模板到新工作表,并设置表名
            Sheet newSheet = workbook.cloneSheet(1);
            workbook.setSheetName(workbook.getSheetIndex(newSheet), key);
         }
         first = false;
      }
      ByteArrayOutputStream ops = new ByteArrayOutputStream();
      workbook.write(ops);
      byte[] byteArray = ops.toByteArray();
      // 原文件流后续已不使用，此处关闭
      templateStream.close();
      ops.close();
      return byteArray;
   }
}