CRM:code review excel 导出的重构

This commit is contained in:
YunaiV
2024-03-02 00:25:51 +08:00
parent fc7734319b
commit 1f35ef59d1
7 changed files with 26 additions and 14 deletions

View File

@ -3,6 +3,7 @@ package cn.iocoder.yudao.framework.excel.core.handler;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.map.MapUtil;
import cn.hutool.extra.spring.SpringUtil;
import cn.hutool.poi.excel.ExcelUtil;
import cn.iocoder.yudao.framework.common.core.KeyValue;
import cn.iocoder.yudao.framework.excel.core.annotations.ExcelColumnSelect;
import cn.iocoder.yudao.framework.excel.core.service.ExcelColumnSelectDataService;
@ -27,7 +28,9 @@ import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.
*/
public class SelectSheetWriteHandler implements SheetWriteHandler {
// TODO @puhui999注释哈
private static final List<String> ALPHABET = getExcelColumnNameList();
private static final List<ExcelColumnSelectDataService> EXCEL_COLUMN_SELECT_DATA_SERVICES = new ArrayList<>();
/**
@ -43,6 +46,7 @@ public class SelectSheetWriteHandler implements SheetWriteHandler {
private static final String DICT_SHEET_NAME = "字典sheet";
// TODO @puhui999这个 selectMap 可以改成 Map<Integer, List<String>>,然后在 afterSheetCreate 里面进行排序。这样整体的理解成本会更简单
private final List<KeyValue<Integer, List<String>>> selectMap = new ArrayList<>(); // 使用 List + KeyValue 组合方便排序
public SelectSheetWriteHandler(Class<?> head) {
@ -51,11 +55,14 @@ public class SelectSheetWriteHandler implements SheetWriteHandler {
if (MapUtil.isEmpty(beansMap)) {
return;
}
// TODO @puhui999static 是共享。如果并发情况下,会不会存在问题? 其实 EXCEL_COLUMN_SELECT_DATA_SERVICES 不用全局声明,直接 按照 ExcelColumnSelect 去拿下就好了;
if (CollUtil.isEmpty(EXCEL_COLUMN_SELECT_DATA_SERVICES) || EXCEL_COLUMN_SELECT_DATA_SERVICES.size() != beansMap.values().size()) {
EXCEL_COLUMN_SELECT_DATA_SERVICES.clear();
EXCEL_COLUMN_SELECT_DATA_SERVICES.addAll(convertList(beansMap.values(), b -> b));
}
// 解析下拉数据
// TODO @puhui999感觉可以 head 循环 field如果有 ExcelColumnSelect 则进行处理;而 ExcelProperty 可能是非必须的
Map<String, Field> excelPropertyFields = getFieldsWithAnnotation(head, ExcelProperty.class);
Map<String, Field> excelColumnSelectFields = getFieldsWithAnnotation(head, ExcelColumnSelect.class);
int colIndex = 0;
@ -157,6 +164,7 @@ public class SelectSheetWriteHandler implements SheetWriteHandler {
private static List<String> getExcelColumnNameList() {
// TODO @puhui999是不是可以使用 ExcelUtil.indexToColName() 替代
ArrayList<String> strings = new ArrayList<>();
for (int i = 1; i <= 52; i++) { // 生成 52 列名称,需要更多请重写此方法
if (i <= 26) {

View File

@ -9,11 +9,14 @@ import java.util.List;
* Excel 列下拉数据源获取接口
*
* 为什么不直接解析字典还搞个接口?考虑到有的下拉数据不是从字典中获取的所有需要做一个兼容
* TODO @puhui999是不是 @ExcelColumnSelect 可以搞两个属性,一个 dictType一个 functionName如果 dictType则默认走字典否则走 functionName
* 这样的话ExcelColumnSelectDataService 改成 ExcelColumnSelectFunction 用于获取数据。
*
* @author HUIHUI
*/
public interface ExcelColumnSelectDataService {
// TODO @puhui999可以考虑改成 getName
/**
* 获得方法名称
*
@ -21,6 +24,7 @@ public interface ExcelColumnSelectDataService {
*/
String getFunctionName();
// TODO @puhui999可以考虑改成 getOptions因为 select 下面是 option 哈,和前端 html 类似的标签;
/**
* 获得列下拉数据源
*
@ -28,6 +32,7 @@ public interface ExcelColumnSelectDataService {
*/
List<String> getSelectDataList();
// TODO @puhui999这个建议放到 SelectSheetWriteHandler 里
default List<String> handle(String funcName) {
if (StrUtil.isEmptyIfStr(funcName) || !StrUtil.equals(getFunctionName(), funcName)) {
return Collections.emptyList();