批量导出接口

Author: zxcvbnmzsedr

apidoc-core/src/main/java/com/ztianzeng/apidoc/Reader.java

@@ -84,7 +84,6 @@ public OpenAPI read(Class<?> cls) {
         BeanDescription beanDesc = mapper.getSerializationConfig().introspect(targetType);
 
 
-
         // 处理方法
         for (JavaMethod method : classByName.getMethods()) {
             boolean isRquestMapping = false;
@@ -139,6 +138,39 @@ public OpenAPI read(Class<?> cls) {
     }
 
 
+    public OpenAPI read(Set<Class<?>> classes) {
+        Set<Class<?>> sortedClasses = new TreeSet<>(new Comparator<Class<?>>() {
+            @Override
+            public int compare(Class<?> class1, Class<?> class2) {
+                if (class1.equals(class2)) {
+                    return 0;
+                } else if (class1.isAssignableFrom(class2)) {
+                    return -1;
+                } else if (class2.isAssignableFrom(class1)) {
+                    return 1;
+                }
+                return class1.getName().compareTo(class2.getName());
+            }
+        });
+        sortedClasses.addAll(classes);
+
+
+        for (Class<?> aClass : sortedClasses) {
+            OpenAPI read = read(aClass);
+
+            paths.putAll(read.getPaths());
+            if (components.getSchemas() == null) {
+                components.setSchemas(new HashMap<>(20));
+            }
+            components.getSchemas().putAll(read.getComponents().getSchemas());
+
+        }
+
+
+        return openAPI;
+    }
+
+
     protected String getOperationId(String operationId) {
         boolean operationIdUsed = existOperationId(operationId);
         String operationIdToFind = null;

apidoc-core/src/main/java/com/ztianzeng/apidoc/SourceBuilder.java

@@ -1,25 +1,17 @@
 package com.ztianzeng.apidoc;
 
-import com.fasterxml.jackson.annotation.JsonInclude;
-import com.fasterxml.jackson.databind.DeserializationFeature;
-import com.fasterxml.jackson.databind.ObjectMapper;
-import com.fasterxml.jackson.databind.SerializationFeature;
+import com.fasterxml.jackson.databind.type.TypeFactory;
 import com.thoughtworks.qdox.JavaProjectBuilder;
-import com.thoughtworks.qdox.model.*;
-import com.ztianzeng.apidoc.constants.RequestMethod;
+import com.thoughtworks.qdox.model.JavaAnnotation;
+import com.thoughtworks.qdox.model.JavaClass;
 import com.ztianzeng.apidoc.utils.DocUtils;
-import org.apache.commons.lang3.StringUtils;
 
 import java.io.File;
-import java.util.Collection;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.lang.reflect.Type;
+import java.util.*;
 
-import static com.ztianzeng.apidoc.constants.GlobalConstants.IGNORE_TAG;
-import static com.ztianzeng.apidoc.constants.SpringMvcConstants.*;
-import static com.ztianzeng.apidoc.utils.DocUtils.getRequestMappingMethod;
-import static com.ztianzeng.apidoc.utils.DocUtils.isRequestMapping;
+import static com.ztianzeng.apidoc.constants.SpringMvcConstants.CONTROLLER_FULLY;
+import static com.ztianzeng.apidoc.constants.SpringMvcConstants.REST_CONTROLLER_FULLY;
 
 /**
  * 核心处理器
@@ -57,7 +49,34 @@ private void loadJavaFiles(String uri) {
         }
     }
 
+    public Set<Class<?>> getControllerData() throws ClassNotFoundException {
+        Set<Class<?>> apiMethodDocs = new HashSet<>();
+        for (JavaClass javaClass : javaClasses) {
+            if (isController(javaClass)) {
+                apiMethodDocs.add(TypeFactory.defaultInstance().findClass(javaClass.getBinaryName()));
+            }
+        }
+        return apiMethodDocs;
+    }
 
+    /**
+     * 检测controller上的注解
+     *
+     * @param cls
+     * @return
+     */
+    private boolean isController(JavaClass cls) {
+        List<JavaAnnotation> classAnnotations = cls.getAnnotations();
+        for (JavaAnnotation annotation : classAnnotations) {
+            String annotationName = annotation.getType().getName();
+            if ("Controller".equals(annotationName) || "RestController".equals(annotationName)
+                    || REST_CONTROLLER_FULLY.equals(annotationName)
+                    || CONTROLLER_FULLY.equals(annotationName)) {
+                return true;
+            }
+        }
+        return false;
+    }
 
     public JavaProjectBuilder getBuilder() {
         return builder;

pom.xml

@@ -58,5 +58,16 @@
             <scope>test</scope>
         </dependency>
     </dependencies>
-
+    <distributionManagement>
+        <repository>
+            <id>nexus-releases</id>
+            <name>CNABC Release Maven Repository Group</name>
+            <url>http://nexus.cnabc-inc.com/content/repositories/share_releases</url>
+        </repository>
+        <snapshotRepository>
+            <id>nexus-snapshots</id>
+            <name>CNABC Snapshot Maven Repository Group</name>
+            <url>http://nexus.cnabc-inc.com/content/repositories/share_snapshots/</url>
+        </snapshotRepository>
+    </distributionManagement>
 </project>