整理代码

1 files changed, 3 insertions, 3 deletions

diff --git a/script-beta/core/reference.lua b/script-beta/core/reference.lua
index 5c18de1b..e61e2940 100644
--- a/script-beta/core/reference.lua
+++ b/script-beta/core/reference.lua
@@ -1,5 +1,6 @@
 local guide      = require 'parser.guide'
 local files      = require 'files'
+local vm         = require 'vm'
 local findSource = require 'core.find-source'
 
 local function isValidFunction(source, offset)
@@ -56,8 +57,7 @@ return function (uri, offset)
     end
 
     local results = {}
-    local refs = guide.requestReference(source)
-    for _, src in ipairs(refs) do
+    vm.eachRef(source, function (src)
         local root = guide.getRoot(src)
         if     src.type == 'setfield'
         or     src.type == 'getfield'
@@ -75,7 +75,7 @@ return function (uri, offset)
             target = src,
             uri    = files.getOriginUri(root.uri),
         }
-    end
+    end)
 
     if #results == 0 then
         return nil