2. 编写lua脚本
-------dbUtil.lua
dbUtil = {}
function dbUtil.getCon()
if not ngx.ctx.dbCon then
local mysql = require "resty.mysql"
local db, err = mysql:new()
if not db then
ngx.log(ngx.ERR, "failed to instantiate mysql: ", err)
return false, nil
end
db:set_timeout(6000)
local ok, err, errinfo, sqlstate = db:connect{
host = "xxx",
port = xxx,
database = "xxx",
user = "xxx",
password = "xxx",
max_packet_size = 1024 * 1024
}
if not ok then
ngx.log(ngx.ERR, "failed to connect: ", err, ": ", errno, " ", sqlstate)
return false, nil
end
ngx.ctx.dbCon = db
end
return true, ngx.ctx.dbCon
end
function dbUtil.closeCon(dbCon)
if not dbCon then return end
local ok, err = dbCon:set_keepalive(10000, 3000)
if not ok then
ngx.say("failed to set keepalive: ", err)
dbCon:close()
return
end
end
return dbUtil
---------interface.lua
local dbutil = require("dbutil")
local args = ngx.req.get_uri_args()
if args == nil then
ngx.say("{}")
end
local md5 = args["md5"]
if not md5 then
ngx.exit(400)
elseif string.len(tostring(md5)) ~= 32 then
ngx.exit(400)
end
if not cjson then cjson = require "cjson" end
local status, db = dbutil.getCon()
if not status then
ngx.exit(500)
end
local sql = "SELECT xxx FROM xxx WHERE MD5='" .. md5 .. "'"
local res, err, errno, sqlstate = db:query(sql)
if res then
ngx.say(cjson.encode(res))
else
ngx.say(err, errno,sqlstate)
end