@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";
        // 構(gòu)建數(shù)據(jù)
        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";
        // 構(gòu)建數(shù)據(jù)
        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();
        // 多組數(shù)據(jù)填充需要另起一行時 forceNewRow(true)
        FillConfig fillConfig = FillConfig.builder().forceNewRow(true).build();
        // 多組數(shù)據(jù)填充 List<實體類> 對象
        excelWriter.fill(catFillData, fillConfig, writeSheet);
        // 關(guān)閉excelWriter
        excelWriter.finish();
    }
}

@Service
@Slf4j
@RequiredArgsConstructor
public class SpeakerServiceImpl implements ISpeakerService {
   /**
    * 填充模板文件
    *
    * @param sheet1Data sheet1數(shù)據(jù)
    * @param promptMap  提示地圖
    * @return {@code File}
    */
   public File fillTemplateFile(
           CustomSpeakerDemandExcelDTO sheet1Data, Map<String, List<CustomSpeakerPromptDTO>> promptMap) {
      try {
         // 構(gòu)建模板
         byte[] asInputStream = this.buildTemplateFile(promptMap);
         // 創(chuàng)建臨時導(dǎo)出文件
         File temporaryFile = Files.createTempFile("speaker_task_", ".xlsx").toFile();
         // easyExcel API 根據(jù)is作為模板填充數(shù)據(jù)，并寫入臨時文件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();
            // 寫入數(shù)據(jù)到當(dāng)前sheet
            List<CustomSpeakerPromptDTO> prompt = promptMap.get(key);
            excelWriter.fill(prompt, writeSheet);
            sheetIndex++;
         }
         excelWriter.finish();
         return temporaryFile;
      } catch (IOException e) {
         throw new RuntimeException(e);
      }
   }
   /**
    * 構(gòu)建模板文件
    *
    * @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);
      // 設(shè)置模板sheet數(shù)量, 需要復(fù)制promptMap.size()-1 次工作表2
      boolean first = true;
      for (String key : promptMap.keySet()) {
         if (first) {
            // 第一組數(shù)據(jù)可以直接使用當(dāng)前模板，不需要進(jìn)行復(fù)制
            workbook.setSheetName(1, key);
         } else {
            // 復(fù)制模板到新工作表,并設(shè)置表名
            Sheet newSheet = workbook.cloneSheet(1);
            workbook.setSheetName(workbook.getSheetIndex(newSheet), key);
         }
         first = false;
      }
      ByteArrayOutputStream ops = new ByteArrayOutputStream();
      workbook.write(ops);
      byte[] byteArray = ops.toByteArray();
      // 原文件流后續(xù)已不使用，此處關(guān)閉
      templateStream.close();
      ops.close();
      return byteArray;
   }
}