ContiNew/continew-starter
1
1
Fork 0
mirror of https://github.com/continew-org/continew-starter.git synced 2025-09-12 05:01:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(file/excel): 优化 BaseEnum 转换器 (#10)

Browse Source
This commit is contained in:
Solution-Lin
2025-01-03 11:45:24 +08:00
committed by GitHub
parent a6c9d33024
commit b9779e8944
1 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
continew-starter-file/continew-starter-file-excel/src/main/java/top/continew/starter/file/excel/converter/ExcelBaseEnumConverter.java
View File

@@ -16,7 +16,6 @@
package top.continew.starter.file.excel.converter; package top.continew.starter.file.excel.converter;
import cn.hutool.core.convert.Convert;
import cn.hutool.core.util.ClassUtil; import cn.hutool.core.util.ClassUtil;
import com.alibaba.excel.converters.Converter; import com.alibaba.excel.converters.Converter;
import com.alibaba.excel.enums.CellDataTypeEnum; import com.alibaba.excel.enums.CellDataTypeEnum;
@@ -34,7 +33,7 @@ import top.continew.starter.core.enums.BaseEnum;
* @see BaseEnum * @see BaseEnum
* @since 1.2.0 * @since 1.2.0
*/ */
public class ExcelBaseEnumConverter implements Converter<BaseEnum<Integer>> { public class ExcelBaseEnumConverter implements Converter<BaseEnum<?>> {
@Override @Override
public Class<BaseEnum> supportJavaTypeKey() { public Class<BaseEnum> supportJavaTypeKey() {
@@ -50,17 +49,17 @@ public class ExcelBaseEnumConverter implements Converter<BaseEnum<Integer>> {
* 转换为 Java 数据（读取 Excel） * 转换为 Java 数据（读取 Excel）
*/ */
@Override @Override
public BaseEnum convertToJavaData(ReadCellData<?> cellData, public BaseEnum<?> convertToJavaData(ReadCellData<?> cellData,
ExcelContentProperty contentProperty, ExcelContentProperty contentProperty,
GlobalConfiguration globalConfiguration) { GlobalConfiguration globalConfiguration) {
return this.getEnum(BaseEnum.class, Convert.toStr(cellData.getData())); return this.getEnum(contentProperty.getField().getType(), cellData.getStringValue());
} }
/** /**
* 转换为 Excel 数据（写入 Excel） * 转换为 Excel 数据（写入 Excel）
*/ */
@Override @Override
public WriteCellData<String> convertToExcelData(BaseEnum<Integer> value, public WriteCellData<String> convertToExcelData(BaseEnum<?> value,
ExcelContentProperty contentProperty, ExcelContentProperty contentProperty,
GlobalConfiguration globalConfiguration) { GlobalConfiguration globalConfiguration) {
if (null == value) { if (null == value) {
@@ -76,11 +75,11 @@ public class ExcelBaseEnumConverter implements Converter<BaseEnum<Integer>> {
* @param description 描述 * @param description 描述
* @return 对应枚举 ，获取不到时为 {@code null} * @return 对应枚举 ，获取不到时为 {@code null}
*/ */
private BaseEnum<Integer> getEnum(Class<?> enumType, String description) { private BaseEnum<?> getEnum(Class<?> enumType, String description) {
Object[] enumConstants = enumType.getEnumConstants(); Object[] enumConstants = enumType.getEnumConstants();
for (Object enumConstant : enumConstants) { for (Object enumConstant : enumConstants) {
if (ClassUtil.isAssignable(BaseEnum.class, enumType)) { if (ClassUtil.isAssignable(BaseEnum.class, enumType)) {
BaseEnum<Integer> baseEnum = (BaseEnum<Integer>)enumConstant; BaseEnum<?> baseEnum = (BaseEnum<?>)enumConstant;
if (baseEnum.getDescription().equals(description)) { if (baseEnum.getDescription().equals(description)) {
return baseEnum; return baseEnum;
} }