Spring枚举识别

Question

在控制器中，我们希望使用枚举作为方法参数：

@GetMapping("/feed")
public void feed(@RequestParam myEnum: MyEnum)

这是可行的，直到我们尝试使用字符串不匹配的情况下。为此，我发现了一个类似的问题：

• Spring3.0MVC绑定Enum区分大小写

但是，这需要为每个枚举设置处理程序。我宁愿为所有枚举实现我的自定义处理程序。类似于：

val ordinal = param.toString().toIntOrNull()
// Reflection on built-in Kotlin types is not yet fully supported... so ...
return if (null == ordinal)
    try {
        type.jvmErasure.java.methods.find { it.name == "valueOf" }!!.invoke(null, param.toString())
    } catch (e: Exception) {
        type.jvmErasure.java.methods.find { it.name == "valueOf" }!!.invoke(null, param.toString().toUpperCase())
    }
else
    type.jvmErasure.java.enumConstants[ordinal]

对不起，这是用Kotlin写的，但应该不难理解。它基本上在原始字符串上尝试valueOf，然后在大写版本上尝试，如果它是数字，它使用枚举的序号值。

我正在寻找一个通用的解决方案，而不是单个枚举的处理程序。

Answer 1

受StringToEnumConverterFactory的启发。

我将使用Java为您提供一个示例解决方案。

下面的代码大写字符串将被转换为Enum，然后调用valueOf，但是您可以将任何逻辑放入转换方法。

@SuppressWarnings({"unchecked", "rawtypes"})
public class CaseInsensitiveStringToEnumConverterFactory implements ConverterFactory<String, Enum> {
  @Override
  public <T extends Enum> Converter<String, T> getConverter(Class<T> targetType) {

    return new CaseInsensitiveStringToEnumConverter(getEnumType(targetType));
  }

  // ConversionUtils are package private. The method is copy-pasted
  private static Class<?> getEnumType(Class<?> targetType) {
    Class<?> enumType = targetType;
    while (enumType != null && !enumType.isEnum()) {
      enumType = enumType.getSuperclass();
    }
    Assert.notNull(enumType, () -> "The target type " + targetType.getName() + " does not refer to an enum");
    return enumType;
  }

  @SuppressWarnings("unchecked")
  private class CaseInsensitiveStringToEnumConverter<T extends Enum> implements Converter<String, T> {
    private final Class<T> enumClass;

    CaseInsensitiveStringToEnumConverter(Class<T> enumClass) {
      this.enumClass = enumClass;
    }

    @Override
    public T convert(String source) {
      if (source.isEmpty()) {
        return null;
      }
      source = source.toUpperCase().trim();
      return (T)Enum.valueOf(enumClass, source);
    }
  }
}

MvcConfig：

@Configuration
public class MvcConfig implements WebMvcConfigurer {

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

样品枚举：

public enum Color {
  RED,
  BLUE,
  GREEN
}

SampleContoller：

@RestController
public class MyController {
  @RequestMapping("/")
  public Map<String, Object> testMethod(@RequestParam("color") Color color) {
    Map<String, Object> map = new HashMap<>();
    map.put("color", color);
    return map;
  }
}

测试：

$ curl 'localhost:8080/?color=red'

输出：

{"color":"RED"}