swagger

src/main/java/com/huiju/estateagents/config/Swagger2Config.java

@@ -1,7 +1,10 @@
 package com.huiju.estateagents.config;
 
+import com.google.common.base.Predicate;
+import io.swagger.annotations.ApiOperation;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
+import springfox.documentation.RequestHandler;
 import springfox.documentation.builders.ApiInfoBuilder;
 import springfox.documentation.builders.PathSelectors;
 import springfox.documentation.builders.RequestHandlerSelectors;
@@ -18,8 +21,19 @@ public class Swagger2Config {
 
     @Bean
     public Docket docket(){
-        return new Docket(DocumentationType.SWAGGER_2).apiInfo(apiInfo()).select()
-                .apis(RequestHandlerSelectors.basePackage("com.huiju.estateagents.controller"))
+        Predicate<RequestHandler> predicate = new Predicate<RequestHandler>() {
+            @Override
+            public boolean apply(RequestHandler input) {
+                //只有添加了ApiOperation注解的method才在API中显示
+                if (input.isAnnotatedWith(ApiOperation.class)) {
+                    return true;
+                } else {
+                    return false;
+                }
+            }
+        };
+        return new Docket(DocumentationType.SWAGGER_2).groupName("docket").apiInfo(apiInfo()).select()
+                .apis(predicate)
                 .paths(PathSelectors.any()).build();
     }
     //构建api文档的详细信息函数