优化 getLinks

1 files changed, 16 insertions, 8 deletions

diff --git a/server-beta/src/searcher/getLinks.lua b/server-beta/src/searcher/getLinks.lua
index db1072e3..3e204e1f 100644
--- a/server-beta/src/searcher/getLinks.lua
+++ b/server-beta/src/searcher/getLinks.lua
@@ -3,18 +3,26 @@ local searcher = require 'searcher.searcher'
 
 local function getLinks(root)
     local cache = {}
-    guide.eachSourceType(root, 'call', function (source)
-        local uris = searcher.getLinkUris(source)
-        if uris then
-            for i = 1, #uris do
-                local uri = uris[i]
-                if not cache[uri] then
-                    cache[uri] = {}
+    local ok
+    guide.eachSpecialOf(root, 'require', function (source)
+        local call = source.parent
+        if call.type == 'call' then
+            local uris = searcher.getLinkUris(call)
+            if uris then
+                ok = true
+                for i = 1, #uris do
+                    local uri = uris[i]
+                    if not cache[uri] then
+                        cache[uri] = {}
+                    end
+                    cache[uri][#cache[uri]+1] = call
                 end
-                cache[uri][#cache[uri]+1] = source
             end
         end
     end)
+    if not ok then
+        return nil
+    end
     return cache
 end