枚举类转换风险，一不小心炸项目

后端

2022-11-21 01:16:50

枚举转换：规避陷阱，保障后端接口畅通

一、枚举转换的常见陷阱

在前端和后端交互中，枚举类型扮演着至关重要的角色，用于表示性别、状态或其他分类信息。然而，在进行枚举转换时，一个常见的陷阱可能会给开发带来麻烦。

当前端将枚举值传递给后端时，如果后端枚举类型中不存在该枚举值，反序列化过程就会失败并抛出异常。这会导致程序崩溃，影响系统稳定性和用户体验。

二、枚举转换的处理方法

为了规避枚举转换陷阱，有几种不同的处理方法：

1. 将枚举序列化为int或其他类型

此方法简单易行，只需重写枚举类的 toString() 方法，将其转换为int或其他类型。后端反序列化时再将其转换回枚举值即可。

public enum SexType {
    M(1, "男"),
    F(2, "女");

    private Integer code;
    private String name;

    ...

    @Override
    public String toString() {
        return String.valueOf(getCode());
    }

    ...
}

2. 使用枚举值和名称映射

此方法需要维护一个枚举值和名称的映射表。反序列化时，根据枚举值从映射表中查找对应的名称，再将名称反序列化为枚举值。

Map<Integer, String> sexTypeMap = new HashMap<>();
for (SexType value : SexType.values()) {
    sexTypeMap.put(value.getCode(), value.getName());
}

...

SexType sexType = SexType.fromCode(code);
if (sexType == null) {
    // 处理不存在的枚举值情况
}

3. 使用枚举转换器

此方法利用枚举转换器，实现 Converter 接口来完成枚举转换。它可以将枚举值序列化为其他类型，也可以将其他类型反序列化为枚举值。

@Converter
public class SexTypeConverter implements AttributeConverter<SexType, Integer> {

    @Override
    public Integer convertToDatabaseColumn(SexType sexType) {
        return sexType.getCode();
    }

    @Override
    public SexType convertToEntityAttribute(Integer code) {
        return SexType.fromCode(code);
    }
}