fix actframework#275

greenlaw110 · greenlaw110 · commit efb21570f245 · 2017-07-06T15:07:54.000+10:00
diff --git a/src/main/java/act/inject/genie/GenieProviders.java b/src/main/java/act/inject/genie/GenieProviders.java
@@ -98,18 +98,20 @@ public void visit(ClassNode classNode) throws Osgl.Break {
                     list.add(c);
                 }
             };
-            root.visitTree($.guardedVisitor(new $.Predicate<ClassNode>() {
-                @Override
-                public boolean test(ClassNode classNode) {
-                    if (!loadNonPublic && !classNode.isPublic()) {
-                        return false;
-                    }
-                    if (!loadAbstract && classNode.isAbstract()) {
-                        return false;
+            for (ClassNode node : root.annotatedClasses()) {
+                $.guardedVisitor(new $.Predicate<ClassNode>() {
+                    @Override
+                    public boolean test(ClassNode classNode) {
+                        if (!loadNonPublic && !classNode.isPublic()) {
+                            return false;
+                        }
+                        if (!loadAbstract && classNode.isAbstract()) {
+                            return false;
+                        }
+                        return true;
                     }
-                    return true;
-                }
-            }, visitor));
+                }, visitor).visit(node);
+            }
             return list;
         }
     };
