121 lines
2.9 KiB
Lua
121 lines
2.9 KiB
Lua
|
local util = require 'utility'
|
|||
|
|
|
|||
|
|
---@class gc
|
|||
|
|
---@field _list table
|
|||
|
|
local mt = {}
|
|||
|
|
mt.__index = mt
|
|||
|
|
mt.type = 'gc'
|
|||
|
|
mt._removed = false
|
|||
|
|
|
|||
|
|
mt._max = 10
|
|||
|
|
|
|||
|
|
local function destroyGCObject(obj)
|
|||
|
|
local tp = type(obj)
|
|||
|
|
if tp == 'function' then
|
|||
|
|
xpcall(obj, log.error)
|
|||
|
|
elseif tp == 'table' then
|
|||
|
|
local remove = obj.remove
|
|||
|
|
if type(remove) == 'function' then
|
|||
|
|
xpcall(remove, log.error, obj)
|
|||
|
|
end
|
|||
|
|
end
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
local function isRemoved(obj)
|
|||
|
|
local tp = type(obj)
|
|||
|
|
if tp == 'function' then
|
|||
|
|
for i = 1, 1000 do
|
|||
|
|
local n, v = debug.getupvalue(obj, i)
|
|||
|
|
if not n then
|
|||
|
|
if i > 1 then
|
|||
|
|
log.warn('函数式析构器没有 removed 上值！', util.dump(debug.getinfo(obj)))
|
|||
|
|
end
|
|||
|
|
break
|
|||
|
|
end
|
|||
|
|
if n == 'removed' then
|
|||
|
|
if v then
|
|||
|
|
return true
|
|||
|
|
end
|
|||
|
|
break
|
|||
|
|
end
|
|||
|
|
end
|
|||
|
|
elseif tp == 'table' then
|
|||
|
|
if obj._removed then
|
|||
|
|
return true
|
|||
|
|
end
|
|||
|
|
end
|
|||
|
|
return false
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
local function zip(self)
|
|||
|
|
local list = self._list
|
|||
|
|
local index = 1
|
|||
|
|
for i = 1, #list do
|
|||
|
|
local obj = list[index]
|
|||
|
|
if not obj then
|
|||
|
|
break
|
|||
|
|
end
|
|||
|
|
if isRemoved(obj) then
|
|||
|
|
if index == #list then
|
|||
|
|
list[#list] = nil
|
|||
|
|
break
|
|||
|
|
end
|
|||
|
|
list[index] = list[#list]
|
|||
|
|
else
|
|||
|
|
index = index + 1
|
|||
|
|
end
|
|||
|
|
end
|
|||
|
|
self._max = #list * 1.5
|
|||
|
|
if self._max < 10 then
|
|||
|
|
self._max = 10
|
|||
|
|
end
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
function mt:remove()
|
|||
|
|
if self._removed then
|
|||
|
|
return
|
|||
|
|
end
|
|||
|
|
self._removed = true
|
|||
|
|
local list = self._list
|
|||
|
|
for i = 1, #list do
|
|||
|
|
destroyGCObject(list[i])
|
|||
|
|
end
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
--- 标记`obj`在buff移除时自动移除。如果`obj`是个`function`,
|
|||
|
|
--- 则直接调用；如果`obj`是个`table`，则调用内部的`remove`方法。
|
|||
|
|
--- 其他情况不做处理
|
|||
|
|
---@param obj any
|
|||
|
|
---@return any
|
|||
|
|
function mt:add(obj)
|
|||
|
|
if self._removed then
|
|||
|
|
destroyGCObject(obj)
|
|||
|
|
return nil
|
|||
|
|
end
|
|||
|
|
self._list[#self._list+1] = obj
|
|||
|
|
if #self._list > self._max then
|
|||
|
|
zip(self)
|
|||
|
|
end
|
|||
|
|
return obj
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
--- 创建一个gc容器，使用 `gc:add(obj)` 将析构器放入gc容器。
|
|||
|
|
---
|
|||
|
|
--- 当gc容器被销毁时，会调用内部的析构器（不保证调用顺序）
|
|||
|
|
---
|
|||
|
|
--- 析构器必须是以下格式中的一种：
|
|||
|
|
--- 1. 一个对象，使用 `obj:remove()` 方法来析构，使用 `obj._removed` 属性来标记已被析构。
|
|||
|
|
--- 2. 一个析构函数，使用上值 `removed` 来标记已被析构。
|
|||
|
|
---
|
|||
|
|
--- ```lua
|
|||
|
|
--- local gc = ac.gc() -- 创建gc容器
|
|||
|
|
--- gc:add(obj1) -- 将obj1放入gc容器
|
|||
|
|
--- gc:add(obj2) -- 将obj2放入gc容器
|
|||
|
|
--- gc:remove() -- 移除gc容器，同时也会移除obj1与obj2
|
|||
|
|
--- ```
|
|||
|
|
return function ()
|
|||
|
|
return setmetatable({
|
|||
|
|
_list = {},
|
|||
|
|
}, mt)
|
|||
|
|
end
|