在我们的控制器中,我们希望使用枚举作为方法参数:
@GetMapping("/feed")
public void feed(@RequestParam myEnum: MyEnum)
这有效,直到我们尝试使用不区分大小写的字符串为止。为此,我发现了一个类似的问题:
但是,这需要为每个枚举设置处理程序。我想为所有枚举实现我的自定义处理程序。像这样:
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,然后在大写版本上尝试,如果是数字,则使用枚举的序数。
我正在寻找一种通用的解决方案,而不是单个枚举的处理程序。
答案 0 :(得分:1)
受StringToEnumConverterFactory的启发。
我将使用Java为您提供示例解决方案。
下面的代码将大写的字符串转换为Enum
,然后调用valueOf
,但是您可以在convert方法中添加任何逻辑。
@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"}