-
Notifications
You must be signed in to change notification settings - Fork 30
/
make_pooled.lua
47 lines (40 loc) · 1011 Bytes
/
make_pooled.lua
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
--[[
add pooling functionality to a class
adds a handful of class and instance methods to do with pooling
todo: automatically use the pool by replacing __call, so you really just need to :release()
]]
return function(class, limit)
--shared pooled storage
local _pool = {}
--size limit for tuning memory upper bound
local _pool_limit = limit or 128
--flush the entire pool
function class:flush_pool()
if #_pool > 0 then
_pool = {}
end
end
--drain one element from the pool, if it exists
function class:drain_pool()
if #_pool > 0 then
return table.remove(_pool)
end
return nil
end
--get a pooled object
--(re-initialised with new, or freshly constructed if the pool was empty)
function class:pooled(...)
local instance = class:drain_pool()
if not instance then
return class(...)
end
instance:new(...)
return instance
end
--release an object back to the pool
function class:release()
if #_pool < _pool_limit then
table.insert(_pool, self)
end
end
end