improve performance

2 files changed, 21 insertions, 0 deletions

diff --git a/script/core/completion/completion.lua b/script/core/completion/completion.lua
index 428c8e96..75f75999 100644
--- a/script/core/completion/completion.lua
+++ b/script/core/completion/completion.lua
@@ -1936,6 +1936,8 @@ local function completion(uri, position, triggerCharacter)
         return nil
     end
     clearStack()
+    vm.lockCache()
+    local _ <close> = vm.unlockCache
     local results = {}
     tracy.ZoneBeginN 'completion #2'
     tryCompletions(state, position, triggerCharacter, results)
diff --git a/script/vm/node.lua b/script/vm/node.lua
index ea6a5874..21f95a49 100644
--- a/script/vm/node.lua
+++ b/script/vm/node.lua
@@ -232,7 +232,26 @@ function vm.removeNode(source)
     vm.nodeCache[source] = nil
 end
 
+local lockCount = 0
+local needClearCache = false
+function vm.lockCache()
+    lockCount = lockCount + 1
+end
+
+function vm.unlockCache()
+    lockCount = lockCount - 1
+    if needClearCache then
+        needClearCache = false
+        vm.clearNodeCache()
+    end
+end
+
 function vm.clearNodeCache()
+    if lockCount > 0 then
+        needClearCache = true
+        return
+    end
+    log.debug('clearNodeCache')
     vm.nodeCache = {}
 end