feat: 添加接口接收枚举类的通用转换器

Author: cadecode

simple-framework/src/main/java/top/cadecode/sra/framework/config/WebMvcConfig.java

@@ -0,0 +1,21 @@
+package top.cadecode.sra.framework.config;
+
+import org.springframework.context.annotation.Configuration;
+import org.springframework.format.FormatterRegistry;
+import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
+import top.cadecode.sra.framework.convert.EnumConvertorFactory;
+
+/**
+ * @author Cade Li
+ * @date 2022/6/14
+ * @description Spring MVC 配置
+ */
+@Configuration
+public class WebMvcConfig implements WebMvcConfigurer {
+
+    @Override
+    public void addFormatters(FormatterRegistry registry) {
+        // 添加枚举类的转换器工厂
+        registry.addConverterFactory(new EnumConvertorFactory());
+    }
+}

simple-framework/src/main/java/top/cadecode/sra/framework/convert/EnumConvertor.java

@@ -0,0 +1,15 @@
+package top.cadecode.sra.framework.convert;
+
+/**
+ * @author Cade Li
+ * @date 2022/5/28
+ * @description 请求参数枚举转换接口
+ */
+public interface EnumConvertor {
+
+    /**
+     * 返回枚举元素的对应标记
+     */
+    String convertBy();
+
+}

simple-framework/src/main/java/top/cadecode/sra/framework/convert/EnumConvertorFactory.java

@@ -0,0 +1,50 @@
+package top.cadecode.sra.framework.convert;
+
+import org.springframework.core.convert.converter.Converter;
+import org.springframework.core.convert.converter.ConverterFactory;
+
+import java.util.Arrays;
+import java.util.Map;
+import java.util.Objects;
+import java.util.concurrent.ConcurrentHashMap;
+
+import static java.util.stream.Collectors.toMap;
+
+/**
+ * @author Cade Li
+ * @date 2022/6/14
+ * @description 字符串转枚举转换器工厂
+ */
+public class EnumConvertorFactory implements ConverterFactory<String, EnumConvertor> {
+
+    private static final Map<Class<?>, Converter<String, ?>> CONVERTER_MAP = new ConcurrentHashMap<>();
+
+    @SuppressWarnings("unchecked")
+    @Override
+    public <T extends EnumConvertor> Converter<String, T> getConverter(Class<T> targetType) {
+        Converter<String, ?> stringConverter = CONVERTER_MAP.get(targetType);
+        if (Objects.isNull(stringConverter)) {
+            stringConverter = new StringEnumConvertor<>(targetType);
+            CONVERTER_MAP.put(targetType, stringConverter);
+        }
+        return (Converter<String, T>) stringConverter;
+    }
+
+    /**
+     * 字符串转枚举转换器
+     */
+    public static class StringEnumConvertor<T extends EnumConvertor> implements Converter<String, T> {
+
+        private final Map<String, T> enumMap;
+
+        public StringEnumConvertor(Class<T> targetType) {
+            enumMap = Arrays.stream(targetType.getEnumConstants())
+                    .collect(toMap(EnumConvertor::convertBy, o -> o, (p, n) -> n));
+        }
+
+        @Override
+        public T convert(String source) {
+            return enumMap.get(source);
+        }
+    }
+}