整理代码

1 files changed, 27 insertions, 0 deletions

diff --git a/script-beta/vm/eachDef.lua b/script-beta/vm/eachDef.lua
new file mode 100644
index 00000000..75c29526
--- /dev/null
+++ b/script-beta/vm/eachDef.lua
@@ -0,0 +1,27 @@
+local vm    = require 'vm.vm'
+local guide = require 'parser.guide'
+
+local function eachDef(source)
+    local results = guide.requestDefinition(source)
+    return results
+end
+
+function vm.eachDef(source, callback)
+    local cache = vm.cache.eachDef[source]
+    if cache ~= nil then
+        for i = 1, #cache do
+            callback(cache[i])
+        end
+        return
+    end
+    local unlock = vm.lock('eachDef', source)
+    if not unlock then
+        return
+    end
+    cache = eachDef(source) or false
+    vm.cache.eachDef[source] = cache
+    unlock()
+    for i = 1, #cache do
+        callback(cache[i])
+    end
+end