use async loadFile

Author: sumneko

script/brave/work.lua

@@ -24,3 +24,7 @@ brave.on('timer', function (time)
         brave.push('wakeup')
     end
 end)
+
+brave.on('loadFile', function (path)
+    return util.loadFile(path)
+end)

script/pub/report.lua

@@ -1,5 +1,6 @@
 local pub   = require 'pub.pub'
 local await = require 'await'
+local util  = require 'utility'
 
 pub.on('log', function (params, brave)
     log.raw(brave.id, params.level, params.msg, params.src, params.line, params.clock)

script/workspace/loading.lua

@@ -6,6 +6,7 @@ local config   = require 'config.config'
 local client   = require 'client'
 local util     = require 'utility'
 local furi     = require 'file-uri'
+local pub      = require 'pub'
 
 ---@class workspace.loading
 ---@field scp scope
@@ -75,14 +76,18 @@ function mt:loadFile(uri, libraryUri)
         end
         self.max = self.max + 1
         self:update()
+        ---@async
         self._stash[#self._stash+1] = function ()
-            self.read = self.read + 1
             if files.getFile(uri) then
+                self.read = self.read + 1
+                self:update()
+                self._cache[uri] = true
                 files.addRef(uri)
                 log.info(('Skip loaded file: %s'):format(uri))
                 return
             end
-            local content = util.loadFile(furi.decode(uri))
+            local content = pub.awaitTask('loadFile', furi.decode(uri))
+            self.read = self.read + 1
             self:update()
             if not content then
                 return
@@ -102,14 +107,18 @@ function mt:loadFile(uri, libraryUri)
     elseif files.isDll(uri) then
         self.max = self.max + 1
         self:update()
+        ---@async
         self._stash[#self._stash+1] = function ()
-            self.read = self.read + 1
             if files.getFile(uri) then
+                self.read = self.read + 1
+                self:update()
+                self._cache[uri] = true
                 files.addRef(uri)
                 log.info(('Skip loaded file: %s'):format(uri))
                 return
             end
-            local content = util.loadFile(furi.decode(uri))
+            local content = pub.awaitTask('loadFile', furi.decode(uri))
+            self.read = self.read + 1
             self:update()
             if not content then
                 return
@@ -129,28 +138,23 @@ function mt:loadFile(uri, libraryUri)
     await.delay()
 end
 
----@async
 function mt:loadStashed(max)
     for _ = 1, max do
         local loader = table.remove(self._stash)
         if not loader then
-            return false
+            return
         end
-        loader()
-        await.delay()
+        await.call(loader)
     end
-    return true
 end
 
 ---@async
 function mt:loadAll()
-    while true do
+    while self.read < self.max do
         log.info(('Loaded %d/%d files'):format(self.read, self.max))
-        local suc = self:loadStashed(10)
+        self:loadStashed(100)
         self:update()
-        if not suc then
-            break
-        end
+        await.sleep(0.1)
     end
     log.info('Loaded finish.')
 end

test/tclient/lclient.lua

@@ -3,6 +3,7 @@ local util    = require 'utility'
 local proto   = require 'proto'
 local await   = require 'await'
 local timer   = require 'timer'
+local pub     = require 'pub'
 
 require 'provider'
 
@@ -36,6 +37,23 @@ function mt:_flushServer()
     files.reset()
 end
 
+function mt:_localLoadFile()
+    local awaitTask = pub.awaitTask
+    ---@async
+    ---@param name   string
+    ---@param params any
+    pub.awaitTask = function (name, params)
+        if name == 'loadFile' then
+            local path = params
+            return util.loadFile(path)
+        end
+        return awaitTask(name, params)
+    end
+    self:gc(function ()
+        pub.awaitTask = awaitTask
+    end)
+end
+
 ---@async
 function mt:initialize(params)
     self:awaitRequest('initialize', params or {})
@@ -60,6 +78,7 @@ end
 function mt:start(callback)
     self:_fakeProto()
     self:_flushServer()
+    self:_localLoadFile()
 
     local finished = false