From 6da2b175e20ed3c03b0dfcfc9046de1e0e5d4444 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9C=80=E8=90=8C=E5=B0=8F=E6=B1=90?= <sumneko@hotmail.com>
Date: Sat, 23 Nov 2019 00:05:30 +0800
Subject: =?UTF-8?q?=E6=AD=A3=E8=B7=AF=E7=9B=AE=E5=BD=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 script/vm/chain.lua | 65 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 script/vm/chain.lua

(limited to 'script/vm/chain.lua')

diff --git a/script/vm/chain.lua b/script/vm/chain.lua
new file mode 100644
index 00000000..6e7c6ac7
--- /dev/null
+++ b/script/vm/chain.lua
@@ -0,0 +1,65 @@
+local valueMgr = require 'vm.value'
+local sourceMgr = require 'vm.source'
+
+local mt = {}
+mt.__index = mt
+mt.type = 'chain'
+
+mt.min = 100
+mt.max = 100
+mt.count = 0
+
+function mt:clearCache()
+    if self.count <= self.max then
+        return
+    end
+    local clock = os.clock()
+    local n = 0
+    for uri, value in pairs(self.cache) do
+        local ok = value:eachInfo(function ()
+            return true
+        end)
+        if ok then
+            n = n + 1
+        else
+            value:getSource():kill()
+            self.cache[uri] = nil
+        end
+    end
+    self.count = n
+    self.max = self.count * 1.1 + 10
+    if self.max < self.min then
+        self.max = self.min
+    end
+    local passed = os.clock() - clock
+    if passed > 0.1 then
+        log.warn(('chain:clearCache takes: [%.3f]sec, self.count: %d'):format(passed, self.count))
+    end
+end
+
+function mt:get(uri)
+    if not self.cache[uri] then
+        self.count = self.count + 1
+        self:clearCache()
+        self.cache[uri] = valueMgr.create('any', sourceMgr.dummy())
+        self.cache[uri]:markGlobal()
+        self.cache[uri].uri = uri
+    end
+    return self.cache[uri]
+end
+
+function mt:remove()
+    if self.removed then
+        return
+    end
+    self.removed = true
+    for _, value in pairs(self.cache) do
+        value:getSource():kill()
+    end
+end
+
+return function ()
+    return setmetatable({
+        cache = {},
+    }, mt)
+end
-- 
cgit v1.2.3