Merge pull request #9 from wangpeng2020/hotfix

添加对Rest类解析，兼容有些采用Rest作为Controller类结尾的项目
liuhuagui · Nov 19, 2019 · 5fc4e9d · 5fc4e9d
2 parents 3a7b951 + be86436
commit 5fc4e9d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/smalldoc-core/src/main/java/com/github/liuhuagui/smalldoc/core/DefaultSmallDocletImpl.java b/smalldoc-core/src/main/java/com/github/liuhuagui/smalldoc/core/DefaultSmallDocletImpl.java
@@ -46,7 +46,7 @@ protected boolean process(RootDoc root) {
     private void handleClassDocs(RootDoc root) {
         ClassDoc[] classes = root.classes();
         for (ClassDoc classDoc : classes) {
-            if (!classDoc.name().endsWith(Constants.CONTROLLER))//只解析*Controller类
+            if (!classDoc.name().endsWith(Constants.CONTROLLER) && !classDoc.name().endsWith(Constants.REST))//解析*Controller类和Rest类
                 continue;
             handleClassDoc(classDoc);
         }

diff --git a/smalldoc-core/src/main/java/com/github/liuhuagui/smalldoc/core/constant/Constants.java b/smalldoc-core/src/main/java/com/github/liuhuagui/smalldoc/core/constant/Constants.java
@@ -11,4 +11,5 @@ public class Constants {
 
     public static final String CONTROLLER = "Controller";
 
+    public static final String REST = "Rest";
 }