OpenResty--mysql,redis 項目中的應用

最近剛剛接手同事的OpenResty的項目，發現對mysql，redis的操作沒有用連接池，故對此進行了改造。

 

MYSQL

主要是通過mysql_pool.lua 和 dbutil.lua 來封裝對數據庫的操作

mysql_pool.lua:

module("mysql_pool", package.seeall)

local dbConfig = require"config"
local mysql = require("resty.mysql")

local mysql_pool = {}

--[[
    先從連接池取連接,如果沒有再建立連接.
    返回:
        false,出錯信息.
        true,數據庫連接
--]]
function mysql_pool:get_connect()
    if ngx.ctx[mysql_pool] then
        return true, ngx.ctx[mysql_pool]
    end

    local client, errmsg = mysql:new()
    if not client then
        return false, "mysql.socket_failed: " .. (errmsg or "nil")
    end

    client:set_timeout(10000)  --10秒

    local options = {
        host = dbConfig.DBHOST,
        port = dbConfig.DBPORT,
        user = dbConfig.DBUSER,
        password = dbConfig.DBPASSWORD,
        database = dbConfig.DBNAME
    }

    local result, errmsg, errno, sqlstate = client:connect(options)
    if not result then
        return false, "mysql.cant_connect: " .. (errmsg or "nil") .. ", errno:" .. (errno or "nil") ..
                ", sql_state:" .. (sqlstate or "nil")
    end

    local query = "SET NAMES " .. dbConfig.DEFAULT_CHARSET
    local result, errmsg, errno, sqlstate = client:query(query)
    if not result then
        return false, "mysql.query_failed: " .. (errmsg or "nil") .. ", errno:" .. (errno or "nil") ..
                ", sql_state:" .. (sqlstate or "nil")
    end

    ngx.ctx[mysql_pool] = client
    return true, ngx.ctx[mysql_pool]
end

--[[
    把連接返回到連接池
    用set_keepalive代替close() 將開啟連接池特性,可以為每個nginx工作進程，指定連接最大空閑時間，和連接池最大連接數
 --]]
function mysql_pool:close()
    if ngx.ctx[mysql_pool] then
        ngx.ctx[mysql_pool]:set_keepalive(60000, 1000)
        ngx.ctx[mysql_pool] = nil
    end
end

--[[
    查詢
    有結果數據集時返回結果數據集
    無數據數據集時返回查詢影響
    返回:
        false,出錯信息,sqlstate結構.
        true,結果集,sqlstate結構.
--]]
function mysql_pool:query(sql, flag)
    local ret, client = self:get_connect(flag)
    if not ret then
        return false, client, nil
    end
    
    local result, errmsg, errno, sqlstate = client:query(sql)
    self:close()

    if not result then
        errmsg = concat_db_errmsg("mysql.query_failed:", errno, errmsg, sqlstate)
        return false, errmsg, sqlstate
    end

    return true, result, sqlstate
end

return mysql_pool

 dbutil.lua

module("dbutil", package.seeall)
local mysql_pool = require("mysql_pool")

function query(sql)

    local ret, res, _ = mysql_pool:query(sql)
    if not ret then
        ngx.log(ngx.ERR, "query db error. res: " .. (res or "nil"))
        return nil
    end

    return res
end

function execute(sql)

    local ret, res, sqlstate = mysql_pool:query(sql)
    if not ret then
        ngx.log(ngx.ERR, "mysql.execute_failed. res: " .. (res or 'nil') .. ",sql_state: " .. (sqlstate or 'nil'))
        return -1
    end

    return res.affected_rows
end

 

REDIS

redis_pool.lua:

module("redis_pool", package.seeall)

local redisConfig = require"config"
local redis = require("resty.redis")

local redis_pool = {}

--[[
    先從連接池取連接,如果沒有再建立連接.
    返回:
        false,出錯信息.
        true,redis連接
--]]
function redis_pool:get_connect()
    if ngx.ctx[redis_pool] then
        return true, ngx.ctx[redis_pool]
    end

    local client, errmsg = redis:new()
    if not client then
        return false, "redis.socket_failed: " .. (errmsg or "nil")
    end

    client:set_timeout(10000)  --10秒

    local result, errmsg = client:connect(redisConfig.REDIS_HOST, redisConfig.REDIS_PORT)
    if not result then
        return false, errmsg
    end

    ngx.ctx[redis_pool] = client
    return true, ngx.ctx[redis_pool]
end

function redis_pool:close()
    if ngx.ctx[redis_pool] then
        ngx.ctx[redis_pool]:set_keepalive(60000, 300)
        ngx.ctx[redis_pool] = nil
    end
end

return redis_pool