1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
local util = require 'utility'
---@class gc
---@field package _list table
local mt = {}
mt.__index = mt
mt.type = 'gc'
mt._removed = false
---@package
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
|
