Java中如何用更少量代码实现两个枚举类的双向映射
方案1:利用枚举命名规则自动匹配(代码量最少)
你的两个枚举项命名有明确的前缀对应规则,直接通过枚举名称字符串处理即可,不需要硬编码每个映射关系:
public class EnumMapper { private static final String EXTERNAL_PREFIX = "EXTERNAL_"; public ExternalEnum toExternal(InternalEnum internalEnum) { if (internalEnum == null) return null; try { return ExternalEnum.valueOf(EXTERNAL_PREFIX + internalEnum.name()); } catch (IllegalArgumentException e) { return null; } } public InternalEnum toInternal(ExternalEnum externalEnum) { if (externalEnum == null) return null; try { return InternalEnum.valueOf(externalEnum.name().replaceFirst(EXTERNAL_PREFIX, "")); } catch (IllegalArgumentException e) { return null; } } }
- 优点:不需要维护枚举项的映射关系,新增枚举项时不需要修改转换代码
- 缺点:强依赖命名规则,加异常捕获后可和你原有返回null的逻辑对齐。
方案2:预定义双向映射表(扩展性更强,不依赖命名规则)
如果后续可能出现枚举名不匹配的场景,只需要定义一次映射关系,双向转换都复用,不会重复写映射逻辑:
import java.util.Arrays; import java.util.Map; import java.util.function.Function; import java.util.stream.Collectors; public class EnumMapper { // 只定义一次内部枚举到外部枚举的映射 private static final Map<InternalEnum, ExternalEnum> INTERNAL_TO_EXTERNAL = Arrays.stream(InternalEnum.values()) .collect(Collectors.toMap(Function.identity(), internal -> ExternalEnum.valueOf("EXTERNAL_" + internal.name()))); // 反向映射直接从正向映射生成,不用重复定义 private static final Map<ExternalEnum, InternalEnum> EXTERNAL_TO_INTERNAL = INTERNAL_TO_EXTERNAL.entrySet().stream() .collect(Collectors.toMap(Map.Entry::getValue, Map.Entry::getKey)); public ExternalEnum toExternal(InternalEnum internalEnum) { return INTERNAL_TO_EXTERNAL.get(internalEnum); } public InternalEnum toInternal(ExternalEnum externalEnum) { return EXTERNAL_TO_INTERNAL.get(externalEnum); } }
- 优点:映射关系唯一维护,就算后续枚举命名规则变化,只需要修改
INTERNAL_TO_EXTERNAL的生成逻辑即可,双向自动同步 - 缺点:比方案1多了映射表的初始化代码,适合枚举可能调整的场景
内容的提问来源于stack exchange,提问作者abo
相关产品推荐
相关产品推荐

