kong--kong
wehub-resource-sync
59 行
1.1 KiB
Lua
59 行
1.1 KiB
Lua
local _M = {}
|
|
local cjson = require("cjson")
|
|
local snappy = require("resty.snappy")
|
|
local lrucache = require("resty.lrucache")
|
|
|
|
|
|
local string_sub = string.sub
|
|
local assert = assert
|
|
local cjson_encode = cjson.encode
|
|
local cjson_decode = cjson.decode
|
|
local rfind = require("pl.stringx").rfind
|
|
local snappy_compress = snappy.compress
|
|
local snappy_uncompress = snappy.uncompress
|
|
|
|
|
|
local cap_names = lrucache.new(100)
|
|
|
|
|
|
function _M.parse_method_name(method)
|
|
local cap = cap_names:get(method)
|
|
|
|
if cap then
|
|
return cap[1], cap[2]
|
|
end
|
|
|
|
local pos = rfind(method, ".")
|
|
if not pos then
|
|
return nil, "not a valid method name"
|
|
end
|
|
|
|
local cap = { method:sub(1, pos - 1), method:sub(pos + 1) }
|
|
|
|
cap_names:set(method, cap)
|
|
|
|
return cap[1], cap[2]
|
|
end
|
|
|
|
|
|
function _M.is_timeout(err)
|
|
return err and (err == "timeout" or string_sub(err, -7) == "timeout")
|
|
end
|
|
|
|
|
|
function _M.compress_payload(payload)
|
|
local json = cjson_encode(payload)
|
|
local data = assert(snappy_compress(json))
|
|
return data
|
|
end
|
|
|
|
|
|
function _M.decompress_payload(compressed)
|
|
local json = assert(snappy_uncompress(compressed))
|
|
local data = cjson_decode(json)
|
|
return data
|
|
end
|
|
|
|
|
|
return _M
|