wehub-resource-sync
64 lines
1022 B
Lua
64 lines
1022 B
Lua
local tb_new = require("table.new")
|
|
local tb_nkeys = require("table.nkeys")
|
|
|
|
|
|
local pairs = pairs
|
|
local setmetatable = setmetatable
|
|
|
|
|
|
local _M = {}
|
|
local _MT = { __index = _M, }
|
|
|
|
|
|
function _M.new()
|
|
local self = {
|
|
-- pre-allocate 4 slots
|
|
stats = tb_new(0, 4),
|
|
}
|
|
|
|
return setmetatable(self, _MT)
|
|
end
|
|
|
|
|
|
function _M:_get_stats(name)
|
|
local stats = self.stats
|
|
|
|
if not stats[name] then
|
|
-- keys will be: query/query_last_time/query_fail_nameserver
|
|
-- query_succ/query_fail/stale/runs/...
|
|
-- 6 slots may be a approprate number
|
|
stats[name] = tb_new(0, 6)
|
|
end
|
|
|
|
return stats[name]
|
|
end
|
|
|
|
|
|
function _M:incr(name, key)
|
|
local stats = self:_get_stats(name)
|
|
|
|
stats[key] = (stats[key] or 0) + 1
|
|
end
|
|
|
|
|
|
function _M:set(name, key, value)
|
|
local stats = self:_get_stats(name)
|
|
|
|
stats[key] = value
|
|
end
|
|
|
|
|
|
function _M:emit(fmt)
|
|
local stats = self.stats
|
|
local output = tb_new(0, tb_nkeys(stats))
|
|
|
|
for k, v in pairs(stats) do
|
|
output[fmt(k)] = v
|
|
end
|
|
|
|
return output
|
|
end
|
|
|
|
|
|
return _M
|