支持全局变量

1 files changed, 34 insertions, 0 deletions

diff --git a/server-beta/src/searcher/eachGlobal.lua b/server-beta/src/searcher/eachGlobal.lua
new file mode 100644
index 00000000..21ee2c51
--- /dev/null
+++ b/server-beta/src/searcher/eachGlobal.lua
@@ -0,0 +1,34 @@
+local guide = require 'parser.guide'
+local searcher = require 'searcher.searcher'
+
+local function eachGlobal(source, callback)
+    local root = guide.getRoot(source)
+    local env  = root.locals[1]
+    searcher.eachField(env, callback)
+end
+
+function searcher.eachGlobal(source, callback)
+    local lock <close> = searcher.lock('eachGlobal', source)
+    if not lock then
+        return
+    end
+    local cache = searcher.cache.eachGlobal[source]
+    if cache then
+        for i = 1, #cache do
+            callback(cache[i])
+        end
+        return
+    end
+    cache = {}
+    searcher.cache.eachGlobal[source] = cache
+    local mark = {}
+    eachGlobal(source, function (info)
+        local src = info.source
+        if mark[src] then
+            return
+        end
+        mark[src] = true
+        cache[#cache+1] = info
+        callback(info)
+    end)
+end