首页 > 解决方案 > 如何在不使用@JsonCreator的情况下将请求参数中的整数转换为枚举类型字段的枚举?

问题描述

在我们的项目中,我们定义如下枚举

public enum GenderEnum implements IEnum{
  MALE(1),
  FEMALE(2);

  private Integer code;

  GenderEnum(int code) {
    this.code = code;
  }

  public static GenderEnum getByCode(int code) {
    for (GenderEnum genderEnum : values()) {
        if (genderEnum.code == code) {
            return genderEnum;
        }
    }
    return null;
  }
}

我们想在用于参数的类中使用这些枚举,例如

public class Param{
  String id;
  GenderEnum gender;
}

我们有像这样的控制器

@RestController
class MyController{
  @PostMapping('/alink')
  ResponseBean doSomething(@RequestBody Param param){
      ...
  }
}

对于上述情况,当请求正文为

{"id":"L323", "gender":1}

我们希望1可以转换成枚举MALE

一种解决方案是添加@JsonCreatorover getByCode(int code)。但是在我们现有的项目中,存储库层将 enum 转换为其 String 值(MALEFEMALE)而不是codevalue,然后再将它们插入数据库。因此,如果我们添加@JsonCreatorover getByCode(int code),我们会在尝试从数据库中检索枚举时遇到错误。

我们如何使 mvc/controller 将输入的整数值转换为枚举而不影响存储库层?

我看到一个解决方案如下:

    @Configuration
    public class WebMvcConfig implements WebMvcConfigurer {

        @Override
        public void addFormatters(FormatterRegistry registry) {  
            registry.addConverterFactory(new EnumConverterFactory());
        }
    }



static class EnumConverterFactory implements ConverterFactory<Integer, IEnum> {
    @Override
    public <T extends IEnum> Converter<Integer, T> getConverter(Class<T> targetType) {
        if (!targetType.isEnum()) {
            throw new UnsupportedOperationException(targetType + "is not enum");
        }
        return new EnumConverter(targetType);
    }

    private static class EnumConverter<T extends IEnum> implements Converter<Integer, T> {
        private final Class<T> enumType;

        public EnumConverter(Class<T> enumType) {
            this.enumType = enumType;
        }

        @Override
        public T convert(Integer code) {
            for (T t : enumType.getEnumConstants()) {
                if (code.equals(t.getCode())) {
                    return t;
                }
            }
            return null;
        }
    }
}

但这个解决方案不起作用。调试时,我注意到参数的解析取决于杰克逊的序列化(它根据序数而不是代码值将整数转换为枚举),并且根本不涉及转换器。

那么,有什么解决方法吗?

标签: javaspringspring-bootspring-mvcenums

解决方案


您可以在控制器中声明用于特定类型参数的自定义映射的方法:

@InitBinder
public void initBinder(WebDataBinder dataBinder) {

        dataBinder.registerCustomEditor(GenderEnum.class,
            new PropertyEditorSupport() {
                @Override
                public void setAsText(String value){                    
                    setValue(GenderEnum.getByCode(Integer.parseInt(value));
                }               
            });     
    }

推荐阅读