wehub-resource-sync
39 lines
617 B
Lua
39 lines
617 B
Lua
local _M = {}
|
|
|
|
local timing
|
|
|
|
|
|
local function before()
|
|
timing.enter_context("redis")
|
|
end
|
|
|
|
|
|
local function after()
|
|
timing.leave_context() -- leave redis
|
|
end
|
|
|
|
|
|
function _M.register_hooks(timing_module)
|
|
local req_dyn_hook = require("kong.dynamic_hook")
|
|
|
|
local redis = require("resty.redis")
|
|
for method_name, _ in pairs(redis) do
|
|
if type(redis[method_name]) ~= "function" then
|
|
goto continue
|
|
end
|
|
|
|
req_dyn_hook.hook_function("timing", redis, method_name, "varargs", {
|
|
befores = { before },
|
|
afters = { after },
|
|
})
|
|
|
|
::continue::
|
|
end
|
|
|
|
timing = timing_module
|
|
end
|
|
|
|
|
|
|
|
return _M
|