style(微服务化改造): Swagger依赖升级改造

src/main/java/com/gitlink/softbot/SoftBotApplication.java

@@ -1,5 +1,7 @@
 package com.gitlink.softbot;
 
+
+import com.gitlink.softbot.annotation.EnableCustomSwagger2;
 import org.mybatis.spring.annotation.MapperScan;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
@@ -10,6 +12,7 @@ import org.springframework.scheduling.annotation.EnableAsync;
 @MapperScan("com.gitlink.softbot.dao")
 @EnableAsync
 @EnableRetry
+@EnableCustomSwagger2
 public class SoftBotApplication {
 
     public static void main(String[] args) {

src/main/java/com/gitlink/softbot/annotation/EnableCustomSwagger2.java

@@ -0,0 +1,17 @@
+package com.gitlink.softbot.annotation;
+
+import com.gitlink.softbot.config.SwaggerConfig;
+import org.springframework.context.annotation.Import;
+
+import java.lang.annotation.*;
+
+@Target({ ElementType.TYPE })
+@Retention(RetentionPolicy.RUNTIME)
+@Documented
+@Inherited
+@Import({ SwaggerConfig.class })
+public @interface EnableCustomSwagger2
+{
+
+}
+

src/main/java/com/gitlink/softbot/config/SwaggerBeanPostProcessor.java

@@ -0,0 +1,50 @@
+package com.gitlink.softbot.config;
+
+import org.springframework.beans.BeansException;
+import org.springframework.beans.factory.config.BeanPostProcessor;
+import org.springframework.stereotype.Component;
+import org.springframework.util.ReflectionUtils;
+import org.springframework.web.servlet.mvc.method.RequestMappingInfoHandlerMapping;
+import springfox.documentation.spring.web.plugins.WebFluxRequestHandlerProvider;
+import springfox.documentation.spring.web.plugins.WebMvcRequestHandlerProvider;
+
+import java.lang.reflect.Field;
+import java.util.List;
+import java.util.stream.Collectors;
+
+@Component
+public class SwaggerBeanPostProcessor implements BeanPostProcessor
+{
+    @Override
+    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException
+    {
+        if (bean instanceof WebMvcRequestHandlerProvider || bean instanceof WebFluxRequestHandlerProvider)
+        {
+            customizeSpringfoxHandlerMappings(getHandlerMappings(bean));
+        }
+        return bean;
+    }
+
+    private <T extends RequestMappingInfoHandlerMapping> void customizeSpringfoxHandlerMappings(List<T> mappings)
+    {
+        List<T> copy = mappings.stream().filter(mapping -> mapping.getPatternParser() == null)
+                .collect(Collectors.toList());
+        mappings.clear();
+        mappings.addAll(copy);
+    }
+
+    @SuppressWarnings("unchecked")
+    private List<RequestMappingInfoHandlerMapping> getHandlerMappings(Object bean)
+    {
+        try
+        {
+            Field field = ReflectionUtils.findField(bean.getClass(), "handlerMappings");
+            field.setAccessible(true);
+            return (List<RequestMappingInfoHandlerMapping>) field.get(bean);
+        }
+        catch (IllegalArgumentException | IllegalAccessException e)
+        {
+            throw new IllegalStateException(e);
+        }
+    }
+}

src/main/java/com/gitlink/softbot/config/SwaggerConfig.java

@@ -8,25 +8,27 @@ import springfox.documentation.builders.PathSelectors;
 import springfox.documentation.builders.RequestHandlerSelectors;
 import springfox.documentation.service.ApiInfo;
 import springfox.documentation.spi.DocumentationType;
+import springfox.documentation.spring.web.plugins.ApiSelectorBuilder;
 import springfox.documentation.spring.web.plugins.Docket;
 import springfox.documentation.swagger2.annotations.EnableSwagger2;
 
+import java.util.ArrayList;
+import java.util.List;
+import java.util.function.Predicate;
+
 
 @Configuration
 @EnableSwagger2
 public class SwaggerConfig {
     @Bean
-    public Docket createRestApi() {
-        return new Docket(DocumentationType.SWAGGER_2)
-                .apiInfo(apiInfo())
-                //是否开启 (true 开启  false隐藏。生产环境建议隐藏)
-                //.enable(false)
-                .select()
-                //扫描的路径包,设置basePackage会将包下的所有被@Api标记类的所有方法作为api
+    public Docket createRestApi()
+    {
+        ApiSelectorBuilder builder = new Docket(DocumentationType.SWAGGER_2)
+                .apiInfo(apiInfo()).select()
                 .apis(RequestHandlerSelectors.basePackage("com.gitlink.softbot.controller"))
-                //指定路径处理PathSelectors.any()代表所有的路径
-                .paths(PathSelectors.any())
-                .build();
+                .paths(PathSelectors.any());
+
+        return builder.build();
     }
 
     private ApiInfo apiInfo() {