58 lines
1.6 KiB
Lua
58 lines
1.6 KiB
Lua
-- 导入redis的Lua模块
|
||
local redis = require('resty.redis')
|
||
-- 初始化redis
|
||
local red = redis:new()
|
||
red:set_timeouts(1000, 1000, 1000)
|
||
|
||
-- 关闭redis连接的工具方法，其实是放入连接池
|
||
local function close_redis(red)
|
||
local pool_max_idle_time = 10000 -- 连接的空闲时间，单位是毫秒
|
||
local pool_size = 100 --连接池大小
|
||
local ok, err = red:set_keepalive(pool_max_idle_time, pool_size)
|
||
if not ok then
|
||
ngx.log(ngx.ERR, "放入redis连接池失败: ", err)
|
||
end
|
||
end
|
||
|
||
-- 查询redis的方法 ip和port是redis地址，key是查询的key
|
||
local function read_redis(ip, port, key)
|
||
-- 获取一个连接
|
||
local ok, err = red:connect(ip, port)
|
||
if not ok then
|
||
ngx.log(ngx.ERR, "连接redis失败 : ", err)
|
||
return nil
|
||
end
|
||
-- 查询redis
|
||
local resp, err = red:get(key)
|
||
-- 查询失败处理
|
||
if not resp then
|
||
ngx.log(ngx.ERR, "查询Redis失败: ", err, ", key = " , key)
|
||
end
|
||
--得到的数据为空处理
|
||
if resp == ngx.null then
|
||
resp = nil
|
||
ngx.log(ngx.ERR, "查询Redis数据为空, key = ", key)
|
||
end
|
||
close_redis(red)
|
||
return resp
|
||
end
|
||
|
||
-- 封装函数，发送http请求，并解析响应
|
||
local function read_http(path, params)
|
||
local resp = ngx.location.capture(path,{
|
||
method = ngx.HTTP_GET,
|
||
args = params,
|
||
})
|
||
if not resp then
|
||
-- 记录错误信息，返回404
|
||
ngx.log(ngx.ERR, "http查询失败, path: ", path , ", args: ", args)
|
||
ngx.exit(404)
|
||
end
|
||
return resp.body
|
||
end
|
||
-- 将方法导出
|
||
local _M = {
|
||
read_http = read_http,
|
||
read_redis = read_redis
|
||
}
|
||
return _M |