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()  --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(, )
         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 -
     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()  --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(, )
         ngx.ctx[redis_pool] = nil
     end
 end

 return redis_pool