do full gc after scope diag

#1745

2 files changed, 12 insertions, 2 deletions

diff --git a/script/provider/diagnostic.lua b/script/provider/diagnostic.lua
index fd78b9c2..90c5a81c 100644
--- a/script/provider/diagnostic.lua
+++ b/script/provider/diagnostic.lua
@@ -508,11 +508,15 @@ local function askForDisable(uri)
     end
 end
 
-local function clearMemory()
+local function clearMemory(finished)
     if m.scopeDiagCount > 0 then
         return
     end
     vm.clearNodeCache()
+    if finished then
+        collectgarbage()
+        collectgarbage()
+    end
 end
 
 ---@async
@@ -524,10 +528,11 @@ function m.awaitDiagnosticsScope(suri, callback)
     while loading.count() > 0 do
         await.sleep(1.0)
     end
+    local finished
     m.scopeDiagCount = m.scopeDiagCount + 1
     local scopeDiag <close> = util.defer(function ()
         m.scopeDiagCount = m.scopeDiagCount - 1
-        clearMemory()
+        clearMemory(finished)
     end)
     local clock = os.clock()
     local bar <close> = progress.create(suri, lang.script.WORKSPACE_DIAGNOSTIC, 1)
@@ -567,6 +572,7 @@ function m.awaitDiagnosticsScope(suri, callback)
     end
     bar:remove()
     log.info(('Diagnostics scope [%s] finished, takes [%.3f] sec.'):format(scp:getName(), os.clock() - clock))
+    finished = true
 end
 
 function m.diagnosticsScope(uri, force)
diff --git a/script/provider/provider.lua b/script/provider/provider.lua
index d7c5c655..319d00a9 100644
--- a/script/provider/provider.lua
+++ b/script/provider/provider.lua
@@ -1159,6 +1159,7 @@ m.register '$/status/click' {
             titleDiagnostic,
             DEVELOP and 'Restart Server' or nil,
             DEVELOP and 'Clear Node Cache' or nil,
+            DEVELOP and 'GC' or nil,
         })
         if not result then
             return
@@ -1177,6 +1178,9 @@ m.register '$/status/click' {
             vm.clearNodeCache()
             collectgarbage()
             collectgarbage()
+        elseif result == 'GC' then
+            collectgarbage()
+            collectgarbage()
         end
     end
 }