package com.ruoyi.errand.utils;
|
|
import cn.afterturn.easypoi.excel.ExcelExportUtil;
|
import cn.afterturn.easypoi.excel.entity.ExportParams;
|
import cn.afterturn.easypoi.excel.entity.params.ExcelExportEntity;
|
import com.ruoyi.common.exception.GlobalException;
|
import org.apache.poi.ss.usermodel.Workbook;
|
|
import java.io.FileOutputStream;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
|
public class BatchNumberUtils {
|
|
private static final String PREFIX = "第";
|
private static final String SUFFIX = "批";
|
|
/**
|
* 生成批次信息
|
*
|
* @param lastBatchNumber
|
* @return
|
*/
|
public static String getBatchNumber(String lastBatchNumber) {
|
if (lastBatchNumber == null || !lastBatchNumber.startsWith(PREFIX) || !lastBatchNumber.endsWith(SUFFIX)) {
|
return PREFIX + "1" + SUFFIX;
|
}
|
try {
|
// 提取中间的数字
|
String numberPart = lastBatchNumber.substring(PREFIX.length(), lastBatchNumber.length() - SUFFIX.length());
|
int batchNumber = Integer.parseInt(numberPart);
|
|
if (batchNumber < 1 || batchNumber >= 9999) {
|
throw new GlobalException("批次数必须在 1 到 9999 之间");
|
}
|
|
// 计算下一个批次
|
return PREFIX + (batchNumber + 1) + SUFFIX;
|
} catch (NumberFormatException e) {
|
throw new GlobalException("批次中的数字解析失败，请确保格式正确");
|
}
|
}
|
|
public static void main(String[] args) throws Exception {
|
// 1. 创建导出参数
|
ExportParams exportParams = new ExportParams("动态表头示例", "Sheet1");
|
|
// 2. 创建表头
|
List<ExcelExportEntity> columnList = new ArrayList<>();
|
|
// 3. 添加静态列
|
columnList.add(new ExcelExportEntity("编号", "id"));
|
columnList.add(new ExcelExportEntity("补偿资金总额（万元）", "totalCompensation"));
|
columnList.add(new ExcelExportEntity("25%首付款", "firstPayment"));
|
columnList.add(new ExcelExportEntity("过渡补贴", "transitionSubsidy"));
|
|
// 4. 创建父级列 "每季度需支付款项（万元）"，设置 key 为 "dynamic"
|
ExcelExportEntity parentColumn = new ExcelExportEntity("每季度需支付款项（万元）", "dynamic");
|
List<ExcelExportEntity> dynamicColumns = new ArrayList<>();
|
|
// 5. 生成动态列（2025-2027，每年4个季度）
|
List<String> dynamicHeaders = generateDynamicHeaders(2025, 2027);
|
for (String header : dynamicHeaders) {
|
ExcelExportEntity column = new ExcelExportEntity(header, header);
|
column.setFormat("0.00"); // 确保Excel数值格式正确
|
dynamicColumns.add(column);
|
}
|
|
// 6. 关联子列
|
parentColumn.setList(dynamicColumns);
|
columnList.add(parentColumn);
|
|
// 7. 组织数据
|
List<Map<String, Object>> dataList = new ArrayList<>();
|
for (int i = 1; i <= 5; i++) { // 生成5行数据
|
Map<String, Object> row = new HashMap<>();
|
row.put("id", i);
|
row.put("totalCompensation", 500 + i * 10);
|
row.put("firstPayment", 120 + i * 5);
|
row.put("transitionSubsidy", 30 + i * 2);
|
|
// 新建子Map存放动态列数据
|
Map<String, Object> dynamicData = new HashMap<>();
|
for (String header : dynamicHeaders) {
|
dynamicData.put(header, Math.round(Math.random() * 100 * 100.0) / 100.0); // 存储 Double 类型数据
|
}
|
// 将子Map以 key "dynamic" 存入 row 中
|
row.put("dynamic", dynamicData);
|
|
System.out.println("生成数据：" + row); // 调试输出
|
dataList.add(row);
|
}
|
|
// 8. 导出 Excel
|
Workbook workbook = ExcelExportUtil.exportExcel(exportParams, columnList, dataList);
|
|
// 9. 写入文件
|
try (FileOutputStream fos = new FileOutputStream("DynamicTable.xlsx")) {
|
workbook.write(fos);
|
}
|
workbook.close();
|
|
System.out.println("Excel 导出成功！");
|
}
|
|
// 生成动态季度时间列
|
public static List<String> generateDynamicHeaders(int startYear, int endYear) {
|
List<String> headers = new ArrayList<>();
|
String[] quarters = {"1月", "4月", "7月", "10月"};
|
for (int year = startYear; year <= endYear; year++) {
|
for (String quarter : quarters) {
|
headers.add(year + "年" + quarter);
|
}
|
}
|
return headers;
|
}
|
}
|
