初始化提交
This commit is contained in:
USER-20221017CE\Administrator
110
doc/nginx/lua/MySQLOps.lua
Normal file
110
doc/nginx/lua/MySQLOps.lua
Normal file
@@ -0,0 +1,110 @@
|
||||
---
|
||||
--- Created by mark.
|
||||
--- Desc: 演示对OpenResty中使用Lua对MySQL操作
|
||||
--- Note：本Lua脚本借鉴了网络，未经测试，仅供参考，也不提供任何技术支持
|
||||
---
|
||||
local function close_db(db)
|
||||
if not db then
|
||||
return
|
||||
end
|
||||
db:close()
|
||||
end
|
||||
|
||||
local mysql = require("resty.mysql")
|
||||
|
||||
local db, err = mysql:new()
|
||||
if not db then
|
||||
ngx.say("new mysql error : ", err)
|
||||
return
|
||||
end
|
||||
|
||||
db:set_timeout(1000)
|
||||
|
||||
local props = {
|
||||
host = "127.0.0.1",
|
||||
port = 3306,
|
||||
database = "mysql",
|
||||
user = "root",
|
||||
password = "123"
|
||||
}
|
||||
|
||||
local res, err, errno, sqlstate = db:connect(props)
|
||||
|
||||
if not res then
|
||||
ngx.say("connect to mysql error : ", err, " , errno : ", errno, " , sqlstate : ", sqlstate)
|
||||
return close_db(db)
|
||||
end
|
||||
|
||||
---------------------------------------
|
||||
-- 执行SQL语句范例
|
||||
|
||||
local create_table_sql = "create table test(id int primary key auto_increment, ch varchar(100))"
|
||||
res, err, errno, sqlstate = db:query(create_table_sql)
|
||||
if not res then
|
||||
ngx.say("create table error : ", err, " , errno : ", errno, " , sqlstate : ", sqlstate)
|
||||
return close_db(db)
|
||||
end
|
||||
|
||||
local drop_table_sql = "drop table if exists test"
|
||||
res, err, errno, sqlstate = db:query(drop_table_sql)
|
||||
if not res then
|
||||
ngx.say("drop table error : ", err, " , errno : ", errno, " , sqlstate : ", sqlstate)
|
||||
return close_db(db)
|
||||
end
|
||||
|
||||
local insert_sql = "insert into test (ch) values('hello')"
|
||||
res, err, errno, sqlstate = db:query(insert_sql)
|
||||
if not res then
|
||||
ngx.say("insert error : ", err, " , errno : ", errno, " , sqlstate : ", sqlstate)
|
||||
return close_db(db)
|
||||
end
|
||||
res, err, errno, sqlstate = db:query(insert_sql)
|
||||
ngx.say("insert rows : ", res.affected_rows, " , id : ", res.insert_id, "<br/>")
|
||||
|
||||
|
||||
local update_sql = "update test set ch = 'hello2' where id =" .. res.insert_id
|
||||
res, err, errno, sqlstate = db:query(update_sql)
|
||||
if not res then
|
||||
ngx.say("update error : ", err, " , errno : ", errno, " , sqlstate : ", sqlstate)
|
||||
return close_db(db)
|
||||
end
|
||||
ngx.say("update rows : ", res.affected_rows, "<br/>")
|
||||
|
||||
|
||||
local select_sql = "select id, ch from test"
|
||||
res, err, errno, sqlstate = db:query(select_sql)
|
||||
if not res then
|
||||
ngx.say("select error : ", err, " , errno : ", errno, " , sqlstate : ", sqlstate)
|
||||
return close_db(db)
|
||||
end
|
||||
for i, row in ipairs(res) do
|
||||
for name, value in pairs(row) do
|
||||
ngx.say("select row ", i, " : ", name, " = ", value, "<br/>")
|
||||
end
|
||||
end
|
||||
ngx.say("<br/>")
|
||||
|
||||
|
||||
local ch_param = ngx.req.get_uri_args()["ch"] or ''
|
||||
local query_sql = "select id, ch from test where ch = " .. ngx.quote_sql_str(ch_param)
|
||||
res, err, errno, sqlstate = db:query(query_sql)
|
||||
if not res then
|
||||
ngx.say("select error : ", err, " , errno : ", errno, " , sqlstate : ", sqlstate)
|
||||
return close_db(db)
|
||||
end
|
||||
for i, row in ipairs(res) do
|
||||
for name, value in pairs(row) do
|
||||
ngx.say("select row ", i, " : ", name, " = ", value, "<br/>")
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
local delete_sql = "delete from test"
|
||||
res, err, errno, sqlstate = db:query(delete_sql)
|
||||
if not res then
|
||||
ngx.say("delete error : ", err, " , errno : ", errno, " , sqlstate : ", sqlstate)
|
||||
return close_db(db)
|
||||
end
|
||||
ngx.say("delete rows : ", res.affected_rows, "<br/>")
|
||||
|
||||
close_db(db)
|
||||
320
doc/nginx/lua/RedisExtOps.lua
Normal file
320
doc/nginx/lua/RedisExtOps.lua
Normal file
@@ -0,0 +1,320 @@
|
||||
---
|
||||
--- Desc: 对OpenResty中使用Lua对Redis操作的封装库，支持订阅、管道等功能
|
||||
--- Note：本Lua脚本借鉴了网络，未经测试，仅供参考，也不提供任何技术支持
|
||||
---
|
||||
local redis_c = require "resty.redis"
|
||||
|
||||
local ok, new_tab = pcall(require, "table.new")
|
||||
if not ok or type(new_tab) ~= "function" then
|
||||
new_tab = function (narr, nrec) return {} end
|
||||
end
|
||||
|
||||
local _M = new_tab(0, 155)
|
||||
_M._VERSION = '0.01'
|
||||
|
||||
local commands = {
|
||||
"append", "auth", "bgrewriteaof",
|
||||
"bgsave", "bitcount", "bitop",
|
||||
"blpop", "brpop",
|
||||
"brpoplpush", "client", "config",
|
||||
"dbsize",
|
||||
"debug", "decr", "decrby",
|
||||
"del", "discard", "dump",
|
||||
"echo",
|
||||
"eval", "exec", "exists",
|
||||
"expire", "expireat", "flushall",
|
||||
"flushdb", "get", "getbit",
|
||||
"getrange", "getset", "hdel",
|
||||
"hexists", "hget", "hgetall",
|
||||
"hincrby", "hincrbyfloat", "hkeys",
|
||||
"hlen",
|
||||
"hmget", "hmset", "hscan",
|
||||
"hset",
|
||||
"hsetnx", "hvals", "incr",
|
||||
"incrby", "incrbyfloat", "info",
|
||||
"keys",
|
||||
"lastsave", "lindex", "linsert",
|
||||
"llen", "lpop", "lpush",
|
||||
"lpushx", "lrange", "lrem",
|
||||
"lset", "ltrim", "mget",
|
||||
"migrate",
|
||||
"monitor", "move", "mset",
|
||||
"msetnx", "multi", "object",
|
||||
"persist", "pexpire", "pexpireat",
|
||||
"ping", "psetex", "psubscribe",
|
||||
"pttl",
|
||||
"publish", --[[ "punsubscribe", ]] "pubsub",
|
||||
"quit",
|
||||
"randomkey", "rename", "renamenx",
|
||||
"restore",
|
||||
"rpop", "rpoplpush", "rpush",
|
||||
"rpushx", "sadd", "save",
|
||||
"scan", "scard", "script",
|
||||
"sdiff", "sdiffstore",
|
||||
"select", "set", "setbit",
|
||||
"setex", "setnx", "setrange",
|
||||
"shutdown", "sinter", "sinterstore",
|
||||
"sismember", "slaveof", "slowlog",
|
||||
"smembers", "smove", "sort",
|
||||
"spop", "srandmember", "srem",
|
||||
"sscan",
|
||||
"strlen", --[[ "subscribe", ]] "sunion",
|
||||
"sunionstore", "sync", "time",
|
||||
"ttl",
|
||||
"type", --[[ "unsubscribe", ]] "unwatch",
|
||||
"watch", "zadd", "zcard",
|
||||
"zcount", "zincrby", "zinterstore",
|
||||
"zrange", "zrangebyscore", "zrank",
|
||||
"zrem", "zremrangebyrank", "zremrangebyscore",
|
||||
"zrevrange", "zrevrangebyscore", "zrevrank",
|
||||
"zscan",
|
||||
"zscore", "zunionstore", "evalsha"
|
||||
}
|
||||
|
||||
local mt = { __index = _M }
|
||||
|
||||
local function is_redis_null( res )
|
||||
if type(res) == "table" then
|
||||
for k,v in pairs(res) do
|
||||
if v ~= ngx.null then
|
||||
return false
|
||||
end
|
||||
end
|
||||
return true
|
||||
elseif res == ngx.null then
|
||||
return true
|
||||
elseif res == nil then
|
||||
return true
|
||||
end
|
||||
|
||||
return false
|
||||
end
|
||||
|
||||
function _M.close_redis(self, redis)
|
||||
if not redis then
|
||||
return
|
||||
end
|
||||
--释放连接(连接池实现)
|
||||
local pool_max_idle_time = self.pool_max_idle_time --最大空闲时间 毫秒
|
||||
local pool_size = self.pool_size --连接池大小
|
||||
|
||||
local ok, err = redis:set_keepalive(pool_max_idle_time, pool_size)
|
||||
if not ok then
|
||||
ngx.say("set keepalive error : ", err)
|
||||
end
|
||||
end
|
||||
|
||||
-- change connect address as you need
|
||||
function _M.connect_mod( self, redis )
|
||||
redis:set_timeout(self.timeout)
|
||||
|
||||
local ok, err = redis:connect(self.ip, self.port)
|
||||
if not ok then
|
||||
ngx.say("connect to redis error : ", err)
|
||||
return self:close_redis(redis)
|
||||
end
|
||||
|
||||
if self.password then ----密码认证
|
||||
local count, err = redis:get_reused_times()
|
||||
if 0 == count then ----新建连接，需要认证密码
|
||||
ok, err = redis:auth(self.password)
|
||||
if not ok then
|
||||
ngx.say("failed to auth: ", err)
|
||||
return
|
||||
end
|
||||
elseif err then ----从连接池中获取连接，无需再次认证密码
|
||||
ngx.say("failed to get reused times: ", err)
|
||||
return
|
||||
end
|
||||
end
|
||||
|
||||
return ok,err;
|
||||
end
|
||||
|
||||
function _M.init_pipeline( self )
|
||||
self._reqs = {}
|
||||
end
|
||||
|
||||
function _M.commit_pipeline( self )
|
||||
local reqs = self._reqs
|
||||
|
||||
if nil == reqs or 0 == #reqs then
|
||||
return {}, "no pipeline"
|
||||
else
|
||||
self._reqs = nil
|
||||
end
|
||||
|
||||
local redis, err = redis_c:new()
|
||||
if not redis then
|
||||
return nil, err
|
||||
end
|
||||
|
||||
local ok, err = self:connect_mod(redis)
|
||||
if not ok then
|
||||
return {}, err
|
||||
end
|
||||
|
||||
redis:init_pipeline()
|
||||
for _, vals in ipairs(reqs) do
|
||||
local fun = redis[vals[1]]
|
||||
table.remove(vals , 1)
|
||||
|
||||
fun(redis, unpack(vals))
|
||||
end
|
||||
|
||||
local results, err = redis:commit_pipeline()
|
||||
if not results or err then
|
||||
return {}, err
|
||||
end
|
||||
|
||||
if is_redis_null(results) then
|
||||
results = {}
|
||||
ngx.log(ngx.WARN, "is null")
|
||||
end
|
||||
-- table.remove (results , 1)
|
||||
|
||||
--self.set_keepalive_mod(redis)
|
||||
self:close_redis(redis)
|
||||
|
||||
for i,value in ipairs(results) do
|
||||
if is_redis_null(value) then
|
||||
results[i] = nil
|
||||
end
|
||||
end
|
||||
|
||||
return results, err
|
||||
end
|
||||
|
||||
|
||||
local function do_command(self, cmd, ... )
|
||||
if self._reqs then
|
||||
table.insert(self._reqs, {cmd, ...})
|
||||
return
|
||||
end
|
||||
|
||||
local redis, err = redis_c:new()
|
||||
if not redis then
|
||||
return nil, err
|
||||
end
|
||||
|
||||
local ok, err = self:connect_mod(redis)
|
||||
if not ok or err then
|
||||
return nil, err
|
||||
end
|
||||
|
||||
redis:select(self.db_index)
|
||||
|
||||
local fun = redis[cmd]
|
||||
local result, err = fun(redis, ...)
|
||||
if not result or err then
|
||||
-- ngx.log(ngx.ERR, "pipeline result:", result, " err:", err)
|
||||
return nil, err
|
||||
end
|
||||
|
||||
if is_redis_null(result) then
|
||||
result = nil
|
||||
end
|
||||
|
||||
--self.set_keepalive_mod(redis)
|
||||
self:close_redis(redis)
|
||||
|
||||
return result, err
|
||||
end
|
||||
|
||||
for i = 1, #commands do
|
||||
local cmd = commands[i]
|
||||
_M[cmd] =
|
||||
function (self, ...)
|
||||
return do_command(self, cmd, ...)
|
||||
end
|
||||
end
|
||||
|
||||
function _M.new(self, opts)
|
||||
opts = opts or {}
|
||||
local timeout = (opts.timeout and opts.timeout * 1000) or 1000
|
||||
local db_index= opts.db_index or 0
|
||||
local ip = opts.ip or '127.0.0.1'
|
||||
local port = opts.port or 6379
|
||||
local password = opts.password
|
||||
local pool_max_idle_time = opts.pool_max_idle_time or 60000
|
||||
local pool_size = opts.pool_size or 100
|
||||
|
||||
return setmetatable({
|
||||
timeout = timeout,
|
||||
db_index = db_index,
|
||||
ip = ip,
|
||||
port = port,
|
||||
password = password,
|
||||
pool_max_idle_time = pool_max_idle_time,
|
||||
pool_size = pool_size,
|
||||
_reqs = nil }, mt)
|
||||
end
|
||||
|
||||
function _M.subscribe( self, channel )
|
||||
local redis, err = redis_c:new()
|
||||
if not redis then
|
||||
return nil, err
|
||||
end
|
||||
|
||||
local ok, err = self:connect_mod(redis)
|
||||
if not ok or err then
|
||||
return nil, err
|
||||
end
|
||||
|
||||
local res, err = redis:subscribe(channel)
|
||||
if not res then
|
||||
return nil, err
|
||||
end
|
||||
|
||||
local function do_read_func ( do_read )
|
||||
if do_read == nil or do_read == true then
|
||||
res, err = redis:read_reply()
|
||||
if not res then
|
||||
return nil, err
|
||||
end
|
||||
return res
|
||||
end
|
||||
|
||||
redis:unsubscribe(channel)
|
||||
self.set_keepalive_mod(redis)
|
||||
return
|
||||
end
|
||||
|
||||
return do_read_func
|
||||
end
|
||||
|
||||
return _M
|
||||
|
||||
---------------------------------------
|
||||
-- 调用案例
|
||||
|
||||
local redis = require "RedisExtOps"
|
||||
local opts = {
|
||||
ip = "10.11.0.215",
|
||||
port = "6379",
|
||||
password = "redis123",
|
||||
db_index = 1
|
||||
}
|
||||
local red = redis:new(opts)
|
||||
local ok, err = red:set("dog", "an animal")
|
||||
if not ok then
|
||||
ngx.say("failed to set dog: ", err)
|
||||
return
|
||||
end
|
||||
ngx.say("set result: ", ok)
|
||||
|
||||
---------------------------------------
|
||||
-- 管道
|
||||
red:init_pipeline()
|
||||
red:set("cat", "Marry")
|
||||
red:set("horse", "Bob")
|
||||
red:get("cat")
|
||||
red:get("horse")
|
||||
local results, err = red:commit_pipeline()
|
||||
if not results then
|
||||
ngx.say("failed to commit the pipelined requests: ", err)
|
||||
return
|
||||
end
|
||||
for i, res in ipairs(results) do
|
||||
ngx.say(res,"<br/>");
|
||||
end
|
||||
14
doc/nginx/lua/product.lua
Normal file
14
doc/nginx/lua/product.lua
Normal file
@@ -0,0 +1,14 @@
|
||||
-- 导入lua-resty-template函数库
|
||||
local template = require('resty.template')
|
||||
local flashPromotionId = ngx.var.arg_flashPromotionId
|
||||
ngx.log(ngx.ERR, "秒杀活动ID: ", flashPromotionId)
|
||||
local promotionProductId = ngx.var.arg_promotionProductId
|
||||
ngx.log(ngx.ERR, "秒杀产品ID: ", promotionProductId)
|
||||
local templateName = "seckill_"..flashPromotionId.."_"..promotionProductId..".html"
|
||||
local context = {
|
||||
memberId = ngx.var.arg_memberId,
|
||||
productId = promotionProductId,
|
||||
flashPromotionId = flashPromotionId
|
||||
}
|
||||
ngx.log(ngx.ERR, "渲染页面输出，获得当前用户ID: ", context.memberId)
|
||||
template.render(templateName, context)
|
||||
58
doc/nginx/lua/redisOps.lua
Normal file
58
doc/nginx/lua/redisOps.lua
Normal file
@@ -0,0 +1,58 @@
|
||||
-- 导入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
|
||||
42
doc/nginx/lua/stock.lua
Normal file
42
doc/nginx/lua/stock.lua
Normal file
@@ -0,0 +1,42 @@
|
||||
-- 导入redisOps函数库
|
||||
local redisOps = require('redisOps')
|
||||
local read_redis = redisOps.read_redis
|
||||
-- 导入cjson库
|
||||
local cjson = require('cjson')
|
||||
-- 导入共享词典，本地缓存
|
||||
-- 本地缓存的主要目的为库存检查，当商品的库存<=0时，提前终止秒杀
|
||||
-- 这里从业务上来说，同样需要解决退单等引发的库存增加允许重新秒杀的情况，
|
||||
-- 解决思路：同样可以订阅对应的Redis的channel，本次不做具体实现，Lua订阅Redis的Channel的参考代码写在RedisExtOps.lua中
|
||||
local item_cache = ngx.shared.stock_cache
|
||||
|
||||
-- 封装查询函数
|
||||
function read_data(key, expire)
|
||||
-- 查询本地缓存
|
||||
local val = item_cache:get(key)
|
||||
if not val then
|
||||
ngx.log(ngx.ERR, "本地缓存查询失败，尝试查询Redis， key: ", key)
|
||||
-- 查询redis
|
||||
val = read_redis("127.0.0.1", 6379, key)
|
||||
-- 判断查询结果
|
||||
if not val then
|
||||
ngx.log(ngx.ERR, "redis查询失败，key: ", key)
|
||||
-- redis查询失败，给一个缺省值
|
||||
val = 0
|
||||
end
|
||||
end
|
||||
-- 查询成功，把数据写入本地缓存,expire秒后过期
|
||||
if tonumber(val) <= 0 then
|
||||
item_cache:set(key, val, expire)
|
||||
end
|
||||
-- 返回数据
|
||||
return val
|
||||
end
|
||||
|
||||
-- 获取请求参数中的productId，也可以使用ngx.req.get_uri_args["productId"]，req.get_uri_args在productId有多个时，会返回一个table
|
||||
local product_id = ngx.var.arg_productId
|
||||
|
||||
-- 查询库存信息
|
||||
local stock = read_data("miaosha:stock:cache:"..product_id, 3600)
|
||||
|
||||
-- 返回结果
|
||||
ngx.say(cjson.encode(stock))
|
||||
Reference in New Issue
Block a user