🔧 build: 调整 库 目录
xiaojin
5 years ago
parent
dd6ab6d6a7
commit
8dc9ed8a49
@ -0,0 +1,41 @@
|
||||
local skynet = require "skynet"
|
||||
local mc = require "skynet.multicast"
|
||||
local st = require "skynet.sharetable"
|
||||
|
||||
local shared
|
||||
local channel
|
||||
|
||||
local M
|
||||
|
||||
local function init()
|
||||
assert(not shared)
|
||||
assert(not channel)
|
||||
shared = skynet.uniqueservice "shared"
|
||||
local name = skynet.call(shared, "lua", "channel")
|
||||
channel = mc.new {
|
||||
channel = name,
|
||||
dispatch = function(_, _, filenames)
|
||||
for _, filename in pairs(filenames) do
|
||||
if M[filename] then
|
||||
M[filename] = nil
|
||||
end
|
||||
end
|
||||
end,
|
||||
}
|
||||
end
|
||||
|
||||
M = setmetatable({}, {
|
||||
__index = function(self, filename)
|
||||
if not shared then
|
||||
init()
|
||||
end
|
||||
|
||||
local obj = st.query(filename)
|
||||
if obj then
|
||||
self[filename] = obj
|
||||
end
|
||||
return obj
|
||||
end,
|
||||
})
|
||||
|
||||
return M
|
||||
@ -0,0 +1,31 @@
|
||||
local skynet = require "skynet"
|
||||
local redlock = {}
|
||||
|
||||
local addr
|
||||
local make_uuid
|
||||
|
||||
local function default_cb()
|
||||
end
|
||||
|
||||
local function _lock(lockname, func, cb, timeout, ...)
|
||||
cb = cb or default_cb
|
||||
local uuid = make_uuid()
|
||||
local ok, data = skynet.call(addr, "lua", "lock", lockname, uuid, timeout)
|
||||
if not ok then
|
||||
return cb(ok, data)
|
||||
end
|
||||
cb(pcall(func, ...))
|
||||
skynet.call(addr, "lua", "unlock", lockname, uuid)
|
||||
end
|
||||
|
||||
function redlock.lock(lockname, func, cb, timeout, ...)
|
||||
skynet.fork(_lock, lockname, func, cb, timeout, ...)
|
||||
end
|
||||
|
||||
skynet.init(function()
|
||||
local conf = require "redlock_conf"
|
||||
make_uuid = conf.make_uuid
|
||||
addr = skynet.uniqueservice("redlockd")
|
||||
end)
|
||||
|
||||
return redlock
|
||||
@ -0,0 +1,78 @@
|
||||
local lfs = require "lfs"
|
||||
--------------------------------------------------------------------------------
|
||||
--------------------------------------------------------------------------------
|
||||
local Share = class("Share")
|
||||
|
||||
function Share:initialize(root)
|
||||
self._root = root
|
||||
end
|
||||
|
||||
function Share:get_paths(rootpath, paths)
|
||||
paths = paths or {}
|
||||
for entry in lfs.dir(rootpath) do
|
||||
if entry ~= "." and entry ~= ".." then
|
||||
local path = string.format("%s/%s", rootpath, entry)
|
||||
|
||||
local attr = lfs.attributes(path)
|
||||
assert(type(attr) == "table")
|
||||
|
||||
if attr.mode == "directory" then
|
||||
self:get_paths(path, paths)
|
||||
else
|
||||
-- 获取文件
|
||||
entry = entry:match("(.+)%.lua$")
|
||||
if entry then
|
||||
paths[entry] = path
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
return paths
|
||||
end
|
||||
|
||||
function Share:read_file(fileName)
|
||||
local f, errMsg = io.open(fileName, "r")
|
||||
if not f then
|
||||
return false, errMsg
|
||||
end
|
||||
|
||||
local source = f:read("a")
|
||||
io.close(f)
|
||||
return true, source
|
||||
end
|
||||
|
||||
function Share:load_file(fileName, ...)
|
||||
local ok, source = self:read_file(fileName)
|
||||
if ok then
|
||||
local chunk, err = load(source)
|
||||
if "function" == type(chunk) then
|
||||
return pcall(chunk, ...)
|
||||
else
|
||||
return false, err
|
||||
end
|
||||
else
|
||||
local err = source
|
||||
return false, err
|
||||
end
|
||||
end
|
||||
|
||||
function Share:get_share_data()
|
||||
local ret = {}
|
||||
if not self._root or "table" ~= type(self._root) then
|
||||
return ret
|
||||
end
|
||||
|
||||
local files = {}
|
||||
for _, rootPath in pairs(self._root) do
|
||||
self:get_paths(rootPath, files)
|
||||
end
|
||||
for fileName, filePath in pairs(files) do
|
||||
local ok, data = self:load_file(filePath)
|
||||
assert(ok, string.format("%s err: %s", filePath, data))
|
||||
ret[fileName] = data
|
||||
end
|
||||
return ret
|
||||
end
|
||||
|
||||
return Share
|
||||
|
||||
@ -0,0 +1,99 @@
|
||||
local single_entity = require "single_entity"
|
||||
local multi_entity = require "multi_entity"
|
||||
local util = require "store_util"
|
||||
|
||||
local traceback = debug.traceback
|
||||
|
||||
local mt = {}
|
||||
mt.__index = mt
|
||||
|
||||
function mt:ctor(opts)
|
||||
self.conf = util.config(opts)
|
||||
self.entities = {}
|
||||
end
|
||||
|
||||
function mt:load(id)
|
||||
local conf = self.conf
|
||||
local entity
|
||||
if conf.multi then
|
||||
entity = multi_entity.new(id, conf)
|
||||
else
|
||||
entity = single_entity.new(id, conf)
|
||||
end
|
||||
assert(entity)
|
||||
self.entities[id] = entity
|
||||
return entity
|
||||
end
|
||||
|
||||
function mt:unload(id)
|
||||
local entity = self.entities[id]
|
||||
if not entity then
|
||||
return
|
||||
end
|
||||
entity:flush()
|
||||
self.entities[id] = nil
|
||||
end
|
||||
|
||||
function mt:unload_all()
|
||||
self:flush()
|
||||
self.entities = {}
|
||||
end
|
||||
|
||||
function mt:entity(id)
|
||||
return self.entities[id]
|
||||
end
|
||||
|
||||
function mt:get(id, ...)
|
||||
local entity = self.entities[id]
|
||||
assert(entity, id)
|
||||
if self.conf.multi then
|
||||
local subid = ...
|
||||
return entity:get(subid)
|
||||
else
|
||||
return entity:get()
|
||||
end
|
||||
end
|
||||
|
||||
function mt:set(id, ...)
|
||||
local data1, data2 = ...
|
||||
local entity = self.entities[id]
|
||||
assert(entity, id)
|
||||
if self.conf.multi then
|
||||
entity:set(data1, data2)
|
||||
else
|
||||
entity:set(data1)
|
||||
end
|
||||
end
|
||||
|
||||
function mt:flush()
|
||||
local rediscmds = {}
|
||||
local sqls = {}
|
||||
|
||||
for _, entity in pairs(self.entities) do
|
||||
entity:collect_dirty(rediscmds, sqls)
|
||||
end
|
||||
|
||||
for addr, cmds in pairs(rediscmds) do
|
||||
xpcall(util.redis_execute, traceback, addr, cmds)
|
||||
end
|
||||
|
||||
for addr, sqls in pairs(sqls) do
|
||||
xpcall(util.mysql_execute, traceback, addr, sqls)
|
||||
end
|
||||
end
|
||||
|
||||
function mt:dump()
|
||||
local obj = {}
|
||||
for key, entity in pairs(self.entities) do
|
||||
obj[key] = entity.rs
|
||||
end
|
||||
return util.encode(obj)
|
||||
end
|
||||
|
||||
return {
|
||||
new = function(opts)
|
||||
local obj = setmetatable({}, mt)
|
||||
obj:ctor(opts)
|
||||
return obj
|
||||
end,
|
||||
}
|
||||
@ -0,0 +1,6 @@
|
||||
local skynet = require "skynet"
|
||||
|
||||
return function(dbconf)
|
||||
local addr = skynet.uniqueservice("dbmgr")
|
||||
return skynet.call(addr, "lua", "init", dbconf)
|
||||
end
|
||||
@ -0,0 +1,177 @@
|
||||
local util = require "store_util"
|
||||
|
||||
local strformat = string.format
|
||||
local tinsert = table.insert
|
||||
local mysql_execute = util.mysql_execute
|
||||
local redis_execute = util.redis_execute
|
||||
local redis_key_field = util.redis_execute
|
||||
local quote_sql_str = util.quote_sql_str
|
||||
local decode = util.decode
|
||||
local encode = util.encode
|
||||
local guid = util.guid
|
||||
local traceback = debug.traceback
|
||||
|
||||
local mt = {}
|
||||
mt.__index = mt
|
||||
|
||||
function mt:ctor(id, conf)
|
||||
assert(conf)
|
||||
assert(type(id) == "number")
|
||||
self.conf = conf
|
||||
self.id = id
|
||||
|
||||
local redisd_addr = conf.redisd_addr
|
||||
local mysqld_addr = conf.mysqld_addr
|
||||
|
||||
-- load from redis
|
||||
if redisd_addr then
|
||||
local key, _ = redis_key_field(conf, 0)
|
||||
local res = redis_execute(redisd_addr, "hgetall", key)
|
||||
local sz = #res
|
||||
if sz > 0 then
|
||||
local rs = {}
|
||||
for i = 1, sz, 2 do
|
||||
local id = tonumber(res[i])
|
||||
rs[id] = decode(res[i + 1])
|
||||
end
|
||||
self.rs = rs
|
||||
end
|
||||
end
|
||||
|
||||
-- load from mysql
|
||||
if not self.rs and mysqld_addr then
|
||||
local mysql_idname = conf.mysql_idname
|
||||
local mysql_valname = conf.mysql_valname
|
||||
local sql = strformat([[SELECT `%s`, `%s` FROM `%s` where `%s` = %d]], mysql_idname, mysql_valname, conf.tbl,
|
||||
conf.mysql_owner_idname, self.id)
|
||||
local res = mysql_execute(mysqld_addr, sql)
|
||||
local sz = #res
|
||||
if sz > 0 then
|
||||
local rs = {}
|
||||
for i = 1, sz do
|
||||
local data = res[i]
|
||||
local id = data[mysql_idname]
|
||||
assert(type(id) == "number", id)
|
||||
rs[id] = data[mysql_valname]
|
||||
end
|
||||
self.rs = rs
|
||||
end
|
||||
end
|
||||
|
||||
if not self.rs then
|
||||
self.rs = {}
|
||||
end
|
||||
|
||||
-- dirty id
|
||||
self.dirty = {}
|
||||
end
|
||||
|
||||
function mt:guid()
|
||||
return guid()
|
||||
end
|
||||
|
||||
function mt:get(id)
|
||||
return self.rs[id]
|
||||
end
|
||||
|
||||
function mt:set(id, val)
|
||||
local conf = self.conf
|
||||
if self.rs[id] then
|
||||
if val then
|
||||
-- update
|
||||
self.dirty[id] = true
|
||||
else
|
||||
-- delete
|
||||
self.dirty[id] = nil
|
||||
if conf.mysqld_addr then
|
||||
local sql = strformat([[delete from `%s` where `%s` = %d and `%s` = %d limit 1]], conf.tbl,
|
||||
conf.mysql_idname, id, conf.mysql_owner_idname, self.id)
|
||||
mysql_execute(conf.mysqld_addr, sql)
|
||||
end
|
||||
if conf.redisd_addr then
|
||||
local key, field = redis_key_field(conf, id)
|
||||
redis_execute(conf.redisd_addr, "hdel", key, field)
|
||||
end
|
||||
end
|
||||
elseif val then
|
||||
-- insert
|
||||
local sval = encode(val)
|
||||
assert(sval)
|
||||
if conf.mysqld_addr then
|
||||
local sql = strformat([[insert into `%s`(`%s`,`%s`,`%s`) values (%d,%d,%s)]], conf.tbl, conf.mysql_idname,
|
||||
conf.mysql_owner_idname, conf.mysql_valname, id, self.id, quote_sql_str(sval))
|
||||
mysql_execute(conf.mysqld_addr, sql)
|
||||
end
|
||||
if conf.redisd_addr then
|
||||
local key, field = redis_key_field(conf, id)
|
||||
redis_execute(conf.redisd_addr, "hset", key, field, sval)
|
||||
end
|
||||
end
|
||||
|
||||
self.rs[id] = val
|
||||
end
|
||||
|
||||
function mt:_collect_dirty(id, rediscmds, sqls)
|
||||
local val = self.rs[id]
|
||||
val = encode(val)
|
||||
assert(val)
|
||||
local conf = self.conf
|
||||
local redisd_addr = conf.redid_addr
|
||||
local mysqld_addr = conf.mysqld_addr
|
||||
|
||||
if redisd_addr then
|
||||
local key, field = redis_key_field(conf, id)
|
||||
local list = rediscmds[redisd_addr]
|
||||
if not list then
|
||||
list = {}
|
||||
rediscmds[redisd_addr] = list
|
||||
end
|
||||
tinsert(list, {"HSET", key, field, val})
|
||||
end
|
||||
|
||||
if mysqld_addr then
|
||||
local sql = strformat([[UPDATE `%s` SET `%s` = %s where `%s` = %d and `%s` = %d limit 1]], conf.tbl,
|
||||
conf.mysql_valname, quote_sql_str(val), conf.mysql_idname, id, conf.mysql_owner_idname, self.id)
|
||||
|
||||
local list = sqls[mysqld_addr]
|
||||
if not list then
|
||||
list = {}
|
||||
sqls[mysqld_addr] = list
|
||||
end
|
||||
tinsert(list, sql)
|
||||
end
|
||||
end
|
||||
|
||||
function mt:collect_dirty(rediscmds, sqls)
|
||||
if not next(self.dirty) then
|
||||
return
|
||||
end
|
||||
for id, _ in pairs(self.dirty) do
|
||||
xpcall(self._collect_dirty, traceback, self, id, rediscmds, sqls)
|
||||
end
|
||||
self.dirty = {}
|
||||
end
|
||||
|
||||
function mt:flush()
|
||||
local conf = self.conf
|
||||
local rediscmds = {}
|
||||
local sqls = {}
|
||||
|
||||
self:collect_dirty(rediscmds, sqls)
|
||||
|
||||
if #rediscmds > 0 then
|
||||
xpcall(redis_execute, traceback, conf.redisd_addr, rediscmds)
|
||||
end
|
||||
|
||||
if #sqls > 0 then
|
||||
mysql_execute(conf.mysqld_addr, sqls)
|
||||
end
|
||||
end
|
||||
|
||||
return {
|
||||
new = function(...)
|
||||
local obj = setmetatable({}, mt)
|
||||
obj:ctor(...)
|
||||
return obj
|
||||
end,
|
||||
}
|
||||
@ -0,0 +1,133 @@
|
||||
local util = require "store_util"
|
||||
|
||||
local strformat = string.format
|
||||
local tinsert = table.insert
|
||||
local mysql_execute = util.mysql_execute
|
||||
local redis_execute = util.redis_execute
|
||||
local redis_key_field = util.redis_execute
|
||||
local quote_sql_str = util.quote_sql_str
|
||||
local decode = util.decode
|
||||
local encode = util.encode
|
||||
local traceback = debug.traceback
|
||||
|
||||
local mt = {}
|
||||
mt.__index = mt
|
||||
|
||||
function mt:ctor(id, conf)
|
||||
assert(conf)
|
||||
assert(type(id) == "number")
|
||||
self.conf = conf
|
||||
self.id = id
|
||||
-- load record
|
||||
if conf.redisd_addr then
|
||||
local key, field = redis_key_field(conf)
|
||||
local ret = redis_execute(conf.redisd_addr, "hget", key, field)
|
||||
self.rs = decode(ret)
|
||||
end
|
||||
|
||||
if not self.rs and conf.mysqld_addr then
|
||||
local sql = strformat("select `%s` from `%s` where `%s` = %d limit 1", conf.mysql_valname, conf.tbl,
|
||||
conf.mysql_idname, self.id)
|
||||
|
||||
local ret = mysql_execute(conf.mysqld_addr, sql)
|
||||
if ret then
|
||||
self.rs = decode(ret[1][conf.mysql_valname])
|
||||
end
|
||||
end
|
||||
|
||||
if not self.rs and not conf.readonly then
|
||||
self:set({})
|
||||
end
|
||||
|
||||
self.dirty = false
|
||||
end
|
||||
|
||||
function mt:get()
|
||||
return self.rs
|
||||
end
|
||||
|
||||
function mt:set(val)
|
||||
local conf = self.conf
|
||||
assert(not conf.readonly, conf.tbl)
|
||||
|
||||
local sval = encode(val)
|
||||
if not sval then
|
||||
assert(false, ("[tbl] %s [id] %s"):format(conf.tbl, self.id))
|
||||
return
|
||||
end
|
||||
|
||||
if not self.rs then
|
||||
-- insert
|
||||
if conf.mysqld_addr then
|
||||
local sql = strformat([[insert into `%s`(`%s`, `%s`) values (%d, %s)]], conf.tbl, conf.mysql_idname,
|
||||
conf.mysql_valname, self.id, quote_sql_str(sval))
|
||||
|
||||
mysql_execute(conf.mysqld_addr, sql)
|
||||
end
|
||||
|
||||
if conf.redisd_addr then
|
||||
local key, field = redis_key_field(conf)
|
||||
redis_execute(conf.redisd_addr, "hset", key, field, sval)
|
||||
end
|
||||
else
|
||||
self.dirty = true
|
||||
end
|
||||
self.rs = val
|
||||
end
|
||||
|
||||
function mt:collect_dirty(rediscmds, sqls)
|
||||
local conf = self.conf
|
||||
if not self.dirty then
|
||||
return
|
||||
end
|
||||
local val = encode(self.rs)
|
||||
local redisd_addr = conf.redisd_addr
|
||||
local mysqld_addr = conf.mysqld_addr
|
||||
|
||||
if redisd_addr then
|
||||
local key, field = redis_key_field(conf)
|
||||
local list = rediscmds[redisd_addr]
|
||||
if not list then
|
||||
list = {}
|
||||
rediscmds[redisd_addr] = list
|
||||
end
|
||||
tinsert(list, {"hset", key, field, val})
|
||||
end
|
||||
|
||||
if mysqld_addr then
|
||||
local sql = strformat([[update `%s` set `%s` = %s where `%s` = %d limit 1]], conf.tbl, conf.mysql_valname,
|
||||
quote_sql_str(val), conf.mysql_idname, self.id)
|
||||
local list = sqls[mysqld_addr]
|
||||
if not list then
|
||||
list = {}
|
||||
sqls[mysqld_addr] = list
|
||||
end
|
||||
tinsert(list, sql)
|
||||
end
|
||||
|
||||
self.dirty = false
|
||||
end
|
||||
|
||||
function mt:flush()
|
||||
local conf = self.conf
|
||||
local rediscmds = {}
|
||||
local sqls = {}
|
||||
|
||||
self:collect_dirty(rediscmds, sqls)
|
||||
|
||||
if #rediscmds > 0 then
|
||||
xpcall(redis_execute, traceback, conf.redisd_addr, rediscmds)
|
||||
end
|
||||
|
||||
if #sqls > 0 then
|
||||
mysql_execute(conf.mysqld_addr, sqls)
|
||||
end
|
||||
end
|
||||
|
||||
return {
|
||||
new = function(...)
|
||||
local obj = setmetatable({}, mt)
|
||||
obj:ctor(...)
|
||||
return obj
|
||||
end,
|
||||
}
|
||||
@ -0,0 +1,40 @@
|
||||
local skynet = require "skynet"
|
||||
local entity_pool = requier "entity_pool"
|
||||
local _store = {}
|
||||
|
||||
local inited = false
|
||||
|
||||
local function add_pool(name, opts)
|
||||
if _store[name] then
|
||||
error(("entity pool [%d] has exist"):format(name))
|
||||
end
|
||||
_store[name] = entity_pool.new(opts)
|
||||
end
|
||||
|
||||
local function open(flush_interval)
|
||||
if flush_interval then
|
||||
skynet.fork(function()
|
||||
while inited do
|
||||
for _, pool in pairs(_store) do
|
||||
pool:flush()
|
||||
skynet.yield()
|
||||
end
|
||||
skynet.sleep(flush_interval*100)
|
||||
end
|
||||
end)
|
||||
end
|
||||
end
|
||||
|
||||
local function close()
|
||||
for _, pool in pairs(_store) do
|
||||
pool:unload_all()
|
||||
end
|
||||
_store = {}
|
||||
inited = false
|
||||
end
|
||||
|
||||
local store = setmetatable(_store, {__index = function(self, name)
|
||||
error(("can not find entity pool [%s]"):format(name))
|
||||
end})
|
||||
|
||||
return store, add_pool, open, close
|
||||
@ -0,0 +1,149 @@
|
||||
local skynet = require "skynet"
|
||||
local json = require "json"
|
||||
|
||||
local strformat = string.format
|
||||
local strgsub = string.gsub
|
||||
local random = math.random
|
||||
local traceback = debug.traceback
|
||||
|
||||
local mysqld_service_num = {}
|
||||
local redisd_service_num = {}
|
||||
local guid_generator_addrs
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.config(opts)
|
||||
local schema = opts.schema
|
||||
assert(type(schema.tbl) == "string", schema.tbl)
|
||||
|
||||
local conf = {}
|
||||
conf.mysqld_addr = opts.mysqld_addr
|
||||
conf.redisd_addr = opts.redisd_addr
|
||||
conf.tbl = schema.tbl
|
||||
conf.readonly = schema.readonly
|
||||
conf.multi = false
|
||||
if schema.multi then
|
||||
conf.multi = true
|
||||
end
|
||||
if conf.mysqld then
|
||||
local field = schema.field or {}
|
||||
conf.mysql_valname = field.data or "data"
|
||||
conf.mysql_idname = field.id or "id"
|
||||
if conf.multi then
|
||||
conf.mysql_owner_idname = field.owner_id or "owner_id"
|
||||
end
|
||||
end
|
||||
return conf
|
||||
end
|
||||
|
||||
function M.mysql_sname(dbkey, index)
|
||||
return ".mysqld." .. dbkey .. "." .. index
|
||||
end
|
||||
|
||||
function M.redis_sname(dbkey, index)
|
||||
return ".redisd." .. dbkey .. "." .. index
|
||||
end
|
||||
|
||||
function M.balance_mysqld_addr(dbkey, id)
|
||||
local num = mysqld_service_num[dbkey]
|
||||
if not num then
|
||||
num = skynet.call(".dbmgr", "lua", "mysql_service_num", dbkey)
|
||||
mysqld_service_num[dbkey] = num
|
||||
end
|
||||
assert(num)
|
||||
local sname = M.mysqld_sname(dbkey, (id % num) + 1)
|
||||
return skynet.localname(sname)
|
||||
end
|
||||
|
||||
function M.balance_redisd_addr(dbkey, id)
|
||||
local num = redisd_service_num[dbkey]
|
||||
if not num then
|
||||
num = skynet.call(".dbmgr", "lua", "redis_service_num", dbkey)
|
||||
redisd_service_num[dbkey] = num
|
||||
end
|
||||
assert(num)
|
||||
local sname = M.redis_sname(dbkey, (id % num) + 1)
|
||||
return skynet.localname(sname)
|
||||
end
|
||||
|
||||
function M.guid()
|
||||
if not guid_generator_addrs then
|
||||
guid_generator_addrs = skynet.call(".dbmgr", "lua", "guid_generators")
|
||||
end
|
||||
assert(guid_generator_addrs, ".dbmgr must be init")
|
||||
local num = #guid_generator_addrs
|
||||
return skynet.call(guid_generator_addrs[random(1, num)], "lua")
|
||||
end
|
||||
|
||||
function M.encode(val)
|
||||
if not val then
|
||||
return
|
||||
end
|
||||
local ok, ret = xpcall(json.encode, traceback, val)
|
||||
if not ok then
|
||||
assert(false, strformat("encode error. val=[%s] ret=[%s]", tostring(val), tostring(ret)))
|
||||
return
|
||||
end
|
||||
return ret
|
||||
end
|
||||
|
||||
function M.decode(val)
|
||||
if not val then
|
||||
return
|
||||
end
|
||||
local ok, ret = xpcall(json.decode, traceback, val)
|
||||
if not ok then
|
||||
assert(false, strformat("decode error. val=[%s] ret=[%s]", tostring(val), tostring(ret)))
|
||||
return
|
||||
end
|
||||
return ret
|
||||
end
|
||||
|
||||
local escape_map = {
|
||||
['\0'] = "\\0",
|
||||
['\b'] = "\\b",
|
||||
['\n'] = "\\n",
|
||||
['\r'] = "\\r",
|
||||
['\t'] = "\\t",
|
||||
['\26'] = "\\Z",
|
||||
['\\'] = "\\\\",
|
||||
["'"] = "\\'",
|
||||
['"'] = '\\"',
|
||||
}
|
||||
|
||||
function M.quote_sql_str(str)
|
||||
return strformat("'%s'", strgsub(str, "[\0\b\n\r\t\26\\\'\"]", escape_map))
|
||||
end
|
||||
|
||||
function M.redis_key_field(conf, id)
|
||||
if not conf.redisd_addr then
|
||||
return
|
||||
end
|
||||
local key, field
|
||||
if id then
|
||||
key = conf.tbl .. ":" .. conf.id
|
||||
field = tostring(id)
|
||||
else
|
||||
key = conf.tbl
|
||||
field = conf.id
|
||||
end
|
||||
return key, field
|
||||
end
|
||||
|
||||
function M.mysql_execute(addr, sqls)
|
||||
if type(sqls) == "table" then
|
||||
return skynet.call(addr, "lua", "exec", sqls)
|
||||
elseif type(sqls) == "string" then
|
||||
return skynet.call(addr, "lua", "exec_one", sqls)
|
||||
end
|
||||
end
|
||||
|
||||
function M.redis_execute(addr, cmd, ...)
|
||||
if type(cmd) == "table" then
|
||||
return skynet.call(addr, "lua", "exec", cmd)
|
||||
elseif type(cmd) == "string" then
|
||||
return skynet.call(addr, "lua", "exec_one", cmd, ...)
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,42 @@
|
||||
local conf = {
|
||||
|
||||
-- redis服务器信息, redis服务器数为奇数
|
||||
-- 最好将奇数台reids部署在不同机器, 达到容错、高可用目的
|
||||
servers = {{
|
||||
host = "127.0.0.1",
|
||||
port = 6379,
|
||||
db = 0,
|
||||
auth = "123456",
|
||||
}},
|
||||
|
||||
-- uuid生成器, 请替换成真uuid(可使用uuid库uuid_generate)
|
||||
make_uuid = function()
|
||||
local fn = function(x)
|
||||
local r = math.random(16) - 1
|
||||
r = (x == "x") and (r + 1) or (r % 4) + 9
|
||||
return ("0123456789abcdef"):sub(r, r)
|
||||
end
|
||||
return (("xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx"):gsub("[xy]", fn))
|
||||
end,
|
||||
|
||||
-- 默认锁过期时间
|
||||
timeout = 3000,
|
||||
|
||||
-- 用计算服务器漂移时间 (http://redis.io/topics/distlock)
|
||||
drift_factor = 0.01,
|
||||
|
||||
-- 重试获得锁最大次数
|
||||
retry_count = 10,
|
||||
|
||||
-- 重试获得锁间隔(毫秒)
|
||||
retry_delay = 200,
|
||||
|
||||
-- 重试获得锁抖动时间(毫秒) https://www.awsarchitectureblog.com/2015/03/backoff.html
|
||||
retry_jitter = 100,
|
||||
|
||||
-- 请求redis超时时间(毫秒)
|
||||
request_timeout = 500,
|
||||
}
|
||||
|
||||
return conf
|
||||
|
||||
@ -0,0 +1,16 @@
|
||||
local M = {}
|
||||
|
||||
function M.new(sep)
|
||||
sep = sep or ' '
|
||||
|
||||
return function(...)
|
||||
local argc, argv = select('#', ...), {...}
|
||||
for i = 1, argc do
|
||||
argv[i] = tostring(argv[i])
|
||||
end
|
||||
return (table.concat(argv, sep))
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
|
||||
@ -0,0 +1,9 @@
|
||||
local M = {}
|
||||
|
||||
function M.new()
|
||||
return function(msg)
|
||||
return msg
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,9 @@
|
||||
local string = require "string"
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.new()
|
||||
return string.format
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,24 @@
|
||||
---
|
||||
-- compatiable with lualogging
|
||||
--
|
||||
local M = {}
|
||||
|
||||
function M.new(default)
|
||||
if not default then
|
||||
default = require"log.formatter.format".new()
|
||||
end
|
||||
|
||||
return function(...)
|
||||
if type((...)) == 'function' then
|
||||
return (...)(select(2, ...))
|
||||
end
|
||||
|
||||
if select('#', ...) < 2 then
|
||||
return tostring((...))
|
||||
end
|
||||
|
||||
return default(...)
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,93 @@
|
||||
local lpeg = require "lpeg"
|
||||
local table = require "table"
|
||||
local string = require "string"
|
||||
|
||||
local unpack = unpack or table.unpack
|
||||
|
||||
local HAS_A_FORMAT = pcall(string.format, '%a', 10)
|
||||
|
||||
local P, C, Cs, Ct, Cp, S, R = lpeg.P, lpeg.C, lpeg.Cs, lpeg.Ct, lpeg.Cp, lpeg.S, lpeg.R
|
||||
|
||||
local any = P(1)
|
||||
local empty = P(0)
|
||||
|
||||
local esc = P '%%'
|
||||
local flags = S '-+ #0'
|
||||
local digit = R '09'
|
||||
|
||||
local fsym = S('cdiouxXeEfgGqs' .. (HAS_A_FORMAT and 'aA' or ''))
|
||||
local width = digit * digit + digit
|
||||
local precision = P '.' * (digit * digit + digit)
|
||||
local format = (flags + empty) * (width + empty) * (precision + empty) * (P '.' + empty)
|
||||
local valid_format = P '%' * format * fsym
|
||||
local valid_format_capture = Cs(valid_format)
|
||||
|
||||
local any_fsym = any - (flags + digit + P '.')
|
||||
local any_format = P '%' * (flags + digit + P '.') ^ 0 * any_fsym
|
||||
|
||||
local types = {
|
||||
c = 'number',
|
||||
d = 'number',
|
||||
i = 'number',
|
||||
o = 'number',
|
||||
u = 'number',
|
||||
x = 'number',
|
||||
X = 'number',
|
||||
a = 'number',
|
||||
A = 'number',
|
||||
e = 'number',
|
||||
E = 'number',
|
||||
f = 'number',
|
||||
g = 'number',
|
||||
G = 'number',
|
||||
q = 'string',
|
||||
s = 'string',
|
||||
}
|
||||
|
||||
local function safe_format(protect_only_args, fmt, ...)
|
||||
local n, args = 0, {...}
|
||||
|
||||
local function fix_fmt(f)
|
||||
local fmt = valid_format_capture:match(f)
|
||||
|
||||
if not fmt then
|
||||
if protect_only_args then
|
||||
return
|
||||
end
|
||||
return '%' .. f
|
||||
end
|
||||
|
||||
local typ = string.sub(fmt, -1)
|
||||
|
||||
n = n + 1
|
||||
|
||||
if types[typ] ~= type(args[n]) then
|
||||
args[n], fmt = tostring(args[n]), '%s'
|
||||
end
|
||||
|
||||
return fmt
|
||||
end
|
||||
|
||||
local pattern = Cs((esc + any_format / fix_fmt + any) ^ 0)
|
||||
fmt = pattern:match(fmt)
|
||||
|
||||
return string.format(fmt, unpack(args, 1, n))
|
||||
end
|
||||
|
||||
local function buld_formatter(protect_only_args, no_warning)
|
||||
return function(...)
|
||||
local ok, msg = pcall(string.format, ...)
|
||||
if not ok then
|
||||
local err = msg
|
||||
msg = safe_format(protect_only_args, ...)
|
||||
if not no_warning then
|
||||
msg = msg .. ' - ' .. 'WARNING: Error formatting log message: ' .. err
|
||||
end
|
||||
end
|
||||
return msg
|
||||
end
|
||||
end
|
||||
|
||||
return {
|
||||
new = buld_formatter,
|
||||
}
|
||||
@ -0,0 +1,223 @@
|
||||
---
|
||||
-- @module log
|
||||
--
|
||||
local _COPYRIGHT = "Copyright (C) 2013-2016 Alexey Melnichuk";
|
||||
local _VERSION = "0.1.7-dev"
|
||||
|
||||
local table = require "table"
|
||||
local date = require "date"
|
||||
|
||||
local destroy_list = {}
|
||||
local loggers_list = setmetatable({}, {
|
||||
__mode = 'k',
|
||||
})
|
||||
local emptyfn = function()
|
||||
end
|
||||
|
||||
local LOG_LVL = {
|
||||
EMERG = 1,
|
||||
ALERT = 2,
|
||||
FATAL = 3,
|
||||
ERROR = 4,
|
||||
WARNING = 5,
|
||||
NOTICE = 6,
|
||||
INFO = 7,
|
||||
DEBUG = 8,
|
||||
TRACE = 9,
|
||||
}
|
||||
|
||||
local writer_names = {}
|
||||
local LOG_LVL_NAMES = {}
|
||||
for k, v in pairs(LOG_LVL) do
|
||||
LOG_LVL_NAMES[v] = k
|
||||
writer_names[v] = k:lower()
|
||||
end
|
||||
|
||||
local LOG_LVL_COUNT = #LOG_LVL_NAMES
|
||||
|
||||
local function lvl2number(lvl)
|
||||
if type(lvl) == 'number' then
|
||||
return lvl
|
||||
end
|
||||
if type(lvl) == 'string' then
|
||||
lvl = lvl:upper()
|
||||
local n
|
||||
if lvl == 'NONE' then
|
||||
n = 0
|
||||
else
|
||||
n = LOG_LVL[lvl]
|
||||
end
|
||||
if not n then
|
||||
return nil, "unknown log level: '" .. lvl .. "'"
|
||||
end
|
||||
return n
|
||||
end
|
||||
|
||||
return nil, 'unsupported log leve type: ' .. type(lvl)
|
||||
end
|
||||
|
||||
local Log = {}
|
||||
Log._COPYRIGHT = _COPYRIGHT
|
||||
Log._NAME = "log"
|
||||
Log._VERSION = _VERSION
|
||||
Log._LICENSE = "MIT"
|
||||
|
||||
Log.LVL = LOG_LVL
|
||||
Log.LVL_NAMES = LOG_LVL_NAMES
|
||||
|
||||
Log.lvl2number = lvl2number
|
||||
|
||||
function Log.new(max_lvl, writer, formatter, logformat)
|
||||
if max_lvl and type(max_lvl) ~= 'number' and type(max_lvl) ~= 'string' then
|
||||
max_lvl, writer, formatter, logformat = nil, max_lvl, writer, formatter
|
||||
end
|
||||
|
||||
max_lvl = assert(lvl2number(max_lvl or LOG_LVL.INFO))
|
||||
|
||||
if not writer then
|
||||
writer = require"log.writer.stdout".new()
|
||||
end
|
||||
|
||||
if not formatter then
|
||||
formatter = require"log.formatter.default".new()
|
||||
end
|
||||
|
||||
if not logformat then
|
||||
logformat = require"log.logformat.default".new()
|
||||
end
|
||||
|
||||
local write = function(lvl, ...)
|
||||
local now = date()
|
||||
writer(logformat, formatter(...), lvl, now)
|
||||
end;
|
||||
|
||||
local dump = function(lvl, fn, ...)
|
||||
local now = date()
|
||||
writer(logformat, (fn(...) or ''), lvl, now)
|
||||
end
|
||||
|
||||
local logger = {}
|
||||
|
||||
function logger.writer()
|
||||
return writer
|
||||
end
|
||||
|
||||
function logger.formatter()
|
||||
return formatter
|
||||
end
|
||||
|
||||
function logger.format()
|
||||
return logformat
|
||||
end
|
||||
|
||||
function logger.lvl()
|
||||
return max_lvl
|
||||
end
|
||||
|
||||
function logger.set_writer(value)
|
||||
assert(value)
|
||||
writer, value = value, writer
|
||||
return value
|
||||
end
|
||||
|
||||
function logger.set_formatter(value)
|
||||
assert(value)
|
||||
formatter, value = value, formatter
|
||||
return value
|
||||
end
|
||||
|
||||
function logger.set_format(value)
|
||||
assert(value)
|
||||
logformat, value = value, logformat
|
||||
return value
|
||||
end
|
||||
|
||||
function logger.log(lvl, ...)
|
||||
local err
|
||||
lvl, err = lvl2number(lvl)
|
||||
if not lvl then
|
||||
return nil, err
|
||||
end
|
||||
return write(lvl, ...)
|
||||
end
|
||||
|
||||
function logger.dump(lvl, ...)
|
||||
local err
|
||||
lvl, err = lvl2number(lvl)
|
||||
if not lvl then
|
||||
return nil, err
|
||||
end
|
||||
return dump(lvl, ...)
|
||||
end
|
||||
|
||||
function logger.set_lvl(lvl)
|
||||
local err
|
||||
lvl, err = lvl2number(lvl)
|
||||
if not lvl then
|
||||
return nil, err
|
||||
end
|
||||
max_lvl = lvl
|
||||
for i = 1, max_lvl do
|
||||
logger[writer_names[i]] = function(...)
|
||||
write(i, ...)
|
||||
end
|
||||
end
|
||||
for i = 1, max_lvl do
|
||||
logger[writer_names[i] .. '_dump'] = function(...)
|
||||
dump(i, ...)
|
||||
end
|
||||
end
|
||||
for i = max_lvl + 1, LOG_LVL_COUNT do
|
||||
logger[writer_names[i]] = emptyfn
|
||||
end
|
||||
for i = max_lvl + 1, LOG_LVL_COUNT do
|
||||
logger[writer_names[i] .. '_dump'] = emptyfn
|
||||
end
|
||||
return true
|
||||
end
|
||||
|
||||
assert(logger.set_lvl(max_lvl))
|
||||
|
||||
loggers_list[logger] = true;
|
||||
|
||||
return logger
|
||||
end
|
||||
|
||||
function Log.add_cleanup(fn)
|
||||
assert(type(fn) == 'function')
|
||||
for k, v in ipairs(destroy_list) do
|
||||
if v == fn then
|
||||
return
|
||||
end
|
||||
end
|
||||
table.insert(destroy_list, 1, fn)
|
||||
return fn
|
||||
end
|
||||
|
||||
function Log.remove_cleanup(fn)
|
||||
for k, v in ipairs(destroy_list) do
|
||||
if v == fn then
|
||||
table.remove(destroy_list, k)
|
||||
break
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
function Log.close()
|
||||
for _, fn in ipairs(destroy_list) do
|
||||
pcall(fn)
|
||||
end
|
||||
for logger in pairs(loggers_list) do
|
||||
logger.fotal = emptyfn;
|
||||
logger.error = emptyfn;
|
||||
logger.warning = emptyfn;
|
||||
logger.info = emptyfn;
|
||||
logger.notice = emptyfn;
|
||||
logger.debug = emptyfn;
|
||||
logger.closed = true;
|
||||
loggers_list[logger] = nil
|
||||
end
|
||||
destroy_list = {}
|
||||
end
|
||||
|
||||
return Log
|
||||
@ -0,0 +1,18 @@
|
||||
local string = require "string"
|
||||
local Log = require "log"
|
||||
|
||||
local sformat = string.format
|
||||
local function date_fmt(now)
|
||||
local Y, M, D = now:getdate()
|
||||
return sformat("%.4d-%.2d-%.2d %.2d:%.2d:%.2d", Y, M, D, now:gettime())
|
||||
end
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.new()
|
||||
return function(msg, lvl, now)
|
||||
return date_fmt(now) .. ' [' .. Log.LVL_NAMES[lvl] .. '] ' .. msg
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,11 @@
|
||||
local packer = require "log.logformat.proxy.pack"
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.new()
|
||||
return function(now, lvl, msg)
|
||||
return packer.pack(now, lvl, msg)
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,36 @@
|
||||
local Log = require "log"
|
||||
local date = require "date"
|
||||
local string = require "string"
|
||||
local schar = string.char
|
||||
local sbyte = string.byte
|
||||
local sformat = string.format
|
||||
local ssub = string.sub
|
||||
local tn = tonumber
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.pack(msg, lvl, now)
|
||||
local Y, M, D = now:getdate()
|
||||
local h, m, s = now:gettime()
|
||||
local now_s = sformat("%.4d-%.2d-%.2d %.2d:%.2d:%.2d", Y, M, D, h, m, s)
|
||||
|
||||
return schar(lvl) .. now_s .. msg
|
||||
end
|
||||
|
||||
function M.unpack(str)
|
||||
local lvl = sbyte(ssub(str, 1, 1))
|
||||
if not Log.LVL_NAMES[lvl] then
|
||||
return
|
||||
end
|
||||
local now_s = ssub(str, 2, 20)
|
||||
local Y, M, D = ssub(str, 2, 5), ssub(str, 7, 8), ssub(str, 10, 11)
|
||||
local h, m, s = ssub(str, 13, 14), ssub(str, 16, 17), ssub(str, 19, 20)
|
||||
Y, M, D, h, m, s = tn(Y), tn(M), tn(D), tn(h), tn(m), tn(s)
|
||||
if not (Y and M and D and h and m and s) then
|
||||
return
|
||||
end
|
||||
|
||||
return ssub(str, 21), lvl, date(Y, M, D, h, m, s)
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,148 @@
|
||||
local string = require "string"
|
||||
local math = require "math"
|
||||
local Log = require "log"
|
||||
|
||||
local mod, floor, ceil, abs, pow = math.fmod, math.floor, math.ceil, math.abs, math.pow
|
||||
local fmt = string.format
|
||||
|
||||
-- removes the decimal part of a number
|
||||
local function fix(n)
|
||||
n = tonumber(n)
|
||||
return n and ((n > 0 and floor or ceil)(n))
|
||||
end
|
||||
|
||||
local function lshift(a, b)
|
||||
if math.pow then
|
||||
return a * math.pow(2, b)
|
||||
else
|
||||
return a << b
|
||||
end
|
||||
end
|
||||
|
||||
--[[ RFC5424
|
||||
0 Emergency: system is unusable
|
||||
1 Alert: action must be taken immediately
|
||||
2 Critical: critical conditions
|
||||
3 Error: error conditions
|
||||
4 Warning: warning conditions
|
||||
5 Notice: normal but significant condition
|
||||
6 Informational: informational messages
|
||||
7 Debug: debug-level messages
|
||||
--]]
|
||||
local SEVERITY = {
|
||||
EMERG = 0,
|
||||
ALERT = 1,
|
||||
CRIT = 2,
|
||||
ERR = 3,
|
||||
WARNING = 4,
|
||||
NOTICE = 5,
|
||||
INFO = 6,
|
||||
DEBUG = 7,
|
||||
}
|
||||
|
||||
--[[ RFC5424
|
||||
0 kernel messages
|
||||
1 user-level messages
|
||||
2 mail system
|
||||
3 system daemons
|
||||
4 security/authorization messages
|
||||
5 messages generated internally by syslogd
|
||||
6 line printer subsystem
|
||||
7 network news subsystem
|
||||
8 UUCP subsystem
|
||||
9 clock daemon
|
||||
10 security/authorization messages
|
||||
11 FTP daemon
|
||||
12 NTP subsystem
|
||||
13 log audit
|
||||
14 log alert
|
||||
15 clock daemon (note 2)
|
||||
16 local use 0 (local0)
|
||||
17 local use 1 (local1)
|
||||
18 local use 2 (local2)
|
||||
19 local use 3 (local3)
|
||||
20 local use 4 (local4)
|
||||
21 local use 5 (local5)
|
||||
22 local use 6 (local6)
|
||||
23 local use 7 (local7)
|
||||
--]]
|
||||
local FACILITY = {
|
||||
KERN = lshift(0, 3),
|
||||
USER = lshift(1, 3),
|
||||
MAIL = lshift(2, 3),
|
||||
DAEMON = lshift(3, 3),
|
||||
AUTH = lshift(4, 3),
|
||||
SYSLOG = lshift(5, 3),
|
||||
LPR = lshift(6, 3),
|
||||
NEWS = lshift(7, 3),
|
||||
UUCP = lshift(8, 3),
|
||||
CRON = lshift(9, 3),
|
||||
CLKD = lshift(9, 3),
|
||||
AUTHPRIV = lshift(10, 3),
|
||||
FTP = lshift(11, 3),
|
||||
NTP = lshift(12, 3),
|
||||
SECURITY = lshift(13, 3),
|
||||
AUDIT = lshift(13, 3),
|
||||
CONSOLE = lshift(14, 3),
|
||||
ALERT = lshift(14, 3),
|
||||
CLKD2 = lshift(15, 3),
|
||||
LOCAL0 = lshift(16, 3),
|
||||
LOCAL1 = lshift(17, 3),
|
||||
LOCAL2 = lshift(18, 3),
|
||||
LOCAL3 = lshift(19, 3),
|
||||
LOCAL4 = lshift(20, 3),
|
||||
LOCAL5 = lshift(21, 3),
|
||||
LOCAL6 = lshift(22, 3),
|
||||
LOCAL7 = lshift(23, 3),
|
||||
}
|
||||
|
||||
local LVL2SYSLOG = {
|
||||
[Log.LVL.EMERG] = SEVERITY.EMERG,
|
||||
[Log.LVL.ALERT] = SEVERITY.ALERT,
|
||||
[Log.LVL.FATAL] = SEVERITY.CRIT,
|
||||
[Log.LVL.ERROR] = SEVERITY.ERR,
|
||||
[Log.LVL.WARNING] = SEVERITY.WARNING,
|
||||
[Log.LVL.NOTICE] = SEVERITY.NOTICE,
|
||||
[Log.LVL.INFO] = SEVERITY.INFO,
|
||||
[Log.LVL.DEBUG] = SEVERITY.DEBUG,
|
||||
[Log.LVL.TRACE] = SEVERITY.DEBUG,
|
||||
}
|
||||
|
||||
local function Date2SysLog(now)
|
||||
local Y, M, D = now:getdate()
|
||||
local h, m, s = now:gettime()
|
||||
|
||||
local b = -now:getbias();
|
||||
local x = abs(b);
|
||||
|
||||
return fmt("%.4d-%.2d-%.2dT%.2d:%.2d:%.2d%s%.2d:%.2d", Y, M, D, h, m, s, b < 0 and "-" or "+", fix(x / 60),
|
||||
floor(mod(x, 60)))
|
||||
end
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.new(facility, host_name, app_name, procid, msgid)
|
||||
if not facility then
|
||||
facility = FACILITY.USER
|
||||
else
|
||||
facility = FACILITY[facility:upper()]
|
||||
end
|
||||
host_name = host_name or '-'
|
||||
app_name = app_name or '-'
|
||||
procid = procid or '-'
|
||||
msgid = msgid or '-'
|
||||
|
||||
-- HOSTNAME APP-NAME PROCID MSGID
|
||||
local header = host_name .. ' ' .. app_name .. ' ' .. procid .. ' ' .. msgid
|
||||
|
||||
return function(msg, lvl, now)
|
||||
local slvl = assert(LVL2SYSLOG[lvl])
|
||||
return -- HEADER
|
||||
-- PRI VERSION TIMESTAMP
|
||||
'<' .. slvl + facility .. '>1 ' .. Date2SysLog(now) .. ' ' .. -- HEADER STRUCTURED-DATA MSG
|
||||
header .. ' - ' .. msg
|
||||
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,101 @@
|
||||
local function prequire(...)
|
||||
local ok, mod = pcall(require, ...)
|
||||
return ok and mod, mod or nil
|
||||
end
|
||||
|
||||
local llthreads = prequire "llthreads2"
|
||||
if not llthreads then
|
||||
llthreads = require "llthreads"
|
||||
end
|
||||
|
||||
local runstring = function(code, ...)
|
||||
code = [[do
|
||||
local string = require "string"
|
||||
local os = require "os"
|
||||
local loadstring = loadstring or load
|
||||
local lua_init = os.getenv("lua_init")
|
||||
if lua_init and #lua_init > 0 then
|
||||
if lua_init:sub(1,1) == '@' then dofile((lua_init:sub(2)))
|
||||
else assert(loadstring(lua_init))() end
|
||||
end
|
||||
end;]] .. code
|
||||
return llthreads.new(code, ...)
|
||||
end
|
||||
|
||||
local sleep
|
||||
do
|
||||
repeat
|
||||
local socket = prequire "socket"
|
||||
if socket then
|
||||
sleep = function(ms)
|
||||
socket.sleep(ms / 1000)
|
||||
end
|
||||
break
|
||||
end
|
||||
|
||||
local ztimer = prequire "lzmq.timer"
|
||||
if ztimer then
|
||||
sleep = ztimer.sleep
|
||||
break
|
||||
end
|
||||
|
||||
-- @todo find another way (use os.execute("sleep " .. ms)) on *nix
|
||||
|
||||
sleep = function()
|
||||
end
|
||||
break
|
||||
until true
|
||||
end
|
||||
|
||||
local Worker = [=[
|
||||
(function(server, maker, logformat, ...)
|
||||
local Log = require "log"
|
||||
local logformat = require(logformat).new()
|
||||
|
||||
local loadstring = loadstring or load
|
||||
local writer = assert(loadstring(maker))()
|
||||
|
||||
require(server).run(writer, Log.close, logformat, ...)
|
||||
end)(...)
|
||||
]=]
|
||||
|
||||
local function run_server(server, maker, logformat, ...)
|
||||
if type(maker) == 'function' then
|
||||
maker = string.dump(maker)
|
||||
end
|
||||
|
||||
assert(type(server) == 'string')
|
||||
assert(type(maker) == 'string')
|
||||
assert(type(logformat) == 'string')
|
||||
|
||||
local child_thread = assert(runstring(Worker, server, maker, logformat, ...))
|
||||
child_thread:start(true, true)
|
||||
sleep(500)
|
||||
return
|
||||
end
|
||||
|
||||
local Z
|
||||
local function run_zserver(server, maker, logformat, ctx, ...)
|
||||
Z = Z or require "log.writer.net.zmq._private.compat"
|
||||
|
||||
if type(maker) == 'function' then
|
||||
maker = string.dump(maker)
|
||||
end
|
||||
|
||||
assert(type(server) == 'string')
|
||||
assert(type(maker) == 'string')
|
||||
assert(type(logformat) == 'string')
|
||||
assert(Z.is_ctx(ctx))
|
||||
|
||||
local zthreads = assert(Z.threads)
|
||||
local child_thread = assert((zthreads.run or zthreads.runstring)(ctx, Worker, server, maker, logformat, ...))
|
||||
child_thread:start(true)
|
||||
return
|
||||
end
|
||||
|
||||
local M = {}
|
||||
|
||||
M.run = run_server
|
||||
M.zrun = run_zserver
|
||||
|
||||
return M
|
||||
@ -0,0 +1,22 @@
|
||||
local server = require "log.writer.async.server.lane"
|
||||
local packer = require "log.logformat.proxy.pack"
|
||||
|
||||
local function create_writer(channel, maker)
|
||||
if maker then
|
||||
server.run(channel, maker, "log.logformat.default")
|
||||
end
|
||||
|
||||
local queue = server.channel()
|
||||
local pack = packer.pack
|
||||
|
||||
return function(fmt, ...)
|
||||
local msg = pack(...)
|
||||
queue:send(channel, msg)
|
||||
end
|
||||
end
|
||||
|
||||
local M = {}
|
||||
|
||||
M.new = create_writer
|
||||
|
||||
return M
|
||||
@ -0,0 +1,71 @@
|
||||
local on_lane_create = function()
|
||||
local loadstring = loadstring or load
|
||||
local lua_init = os.getenv("lua_init")
|
||||
if lua_init and #lua_init > 0 then
|
||||
if lua_init:sub(1, 1) == '@' then
|
||||
dofile(lua_init:sub(2))
|
||||
else
|
||||
assert(loadstring(lua_init))()
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
local LOG = require "log"
|
||||
local lanes = require"lanes".configure {
|
||||
with_timers = false,
|
||||
on_state_create = on_lane_create,
|
||||
}
|
||||
local pack = require"log.logformat.proxy.pack".pack
|
||||
|
||||
local queue
|
||||
|
||||
local function context()
|
||||
queue = queue or assert(lanes.linda())
|
||||
return queue
|
||||
end
|
||||
|
||||
local function log_thread_fn(maker, logformat, channel)
|
||||
local log_packer = require "log.logformat.proxy.pack"
|
||||
local logformat = require(logformat).new()
|
||||
local unpack = log_packer.unpack
|
||||
|
||||
local loadstring = loadstring or load
|
||||
local writer = assert(assert(loadstring(maker))())
|
||||
while (true) do
|
||||
local key, val = queue:receive(1.0, channel)
|
||||
-- print(maker, channel, key, val)
|
||||
if not (key and val) then
|
||||
key, val = nil, 'timeout'
|
||||
end
|
||||
if key then
|
||||
local msg, lvl, now = unpack(val)
|
||||
if msg and lvl and now then
|
||||
writer(logformat, msg, lvl, now)
|
||||
end
|
||||
else
|
||||
if val ~= 'timeout' then
|
||||
io.stderror:write('lane_logger: ', val)
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
local function start_log_thread(...)
|
||||
local fn = assert(lanes.gen("*", log_thread_fn))
|
||||
return assert(fn(...))
|
||||
end
|
||||
|
||||
local M = {}
|
||||
|
||||
M.run = function(channel, maker, logformat)
|
||||
logformat = logformat or "log.logformat.default"
|
||||
context() -- init context
|
||||
local child_thread = start_log_thread(maker, logformat, channel)
|
||||
LOG.add_cleanup(function()
|
||||
child_thread:cancel(60)
|
||||
end)
|
||||
end
|
||||
|
||||
M.channel = context
|
||||
|
||||
return M
|
||||
@ -0,0 +1,10 @@
|
||||
local server = require "log.writer.async._private.server"
|
||||
|
||||
local M = {}
|
||||
|
||||
M.run = function(host, port, maker, logformat)
|
||||
logformat = logformat or "log.logformat.default"
|
||||
server.run("log.writer.net.server.udp", maker, logformat, host, port)
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,42 @@
|
||||
local Z = require "log.writer.net.zmq._private.compat"
|
||||
local IMPL = require "log.writer.net.zmq._private.impl"
|
||||
local server = require "log.writer.async._private.server"
|
||||
|
||||
local zmq, ETERM, zstrerror, zassert, zrecv = Z.zmq, Z.ETERM, Z.strerror, Z.assert, Z.recv
|
||||
|
||||
local function rand_str(n)
|
||||
math.randomseed(os.time())
|
||||
local str = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'
|
||||
local res = ''
|
||||
for i = 1, n do
|
||||
local n = math.random(1, #str)
|
||||
res = res .. str:sub(n, n)
|
||||
end
|
||||
return res
|
||||
end
|
||||
|
||||
local function create_server(ctx, addr, maker, logformat)
|
||||
if ctx and not Z.is_ctx(ctx) then
|
||||
ctx, addr, maker, logformat = nil, ctx, addr, maker
|
||||
end
|
||||
logformat = logformat or "log.logformat.default"
|
||||
|
||||
ctx = IMPL.context(ctx)
|
||||
|
||||
if maker then
|
||||
local addr_sync = 'inproc://' .. rand_str(15)
|
||||
local skt_sync = zassert(ctx:socket(zmq.PAIR))
|
||||
zassert(skt_sync:bind(addr_sync))
|
||||
server.zrun("log.writer.net.server.zmq", maker, logformat, ctx, false, 'PULL', addr, addr_sync)
|
||||
zassert(skt_sync:recv())
|
||||
skt_sync:close()
|
||||
end
|
||||
end
|
||||
|
||||
local M = {}
|
||||
|
||||
M.run = create_server
|
||||
|
||||
M.context = IMPL.context
|
||||
|
||||
return M
|
||||
@ -0,0 +1,18 @@
|
||||
local server = require "log.writer.async.server.udp"
|
||||
|
||||
local function create_writer(host, port, maker)
|
||||
if maker then
|
||||
server.run(host, port, maker, "log.logformat.default")
|
||||
end
|
||||
|
||||
local writer = require"log.writer.format".new(require"log.logformat.proxy".new(),
|
||||
require"log.writer.net.udp".new(host, port))
|
||||
|
||||
return writer
|
||||
end
|
||||
|
||||
local M = {}
|
||||
|
||||
M.new = create_writer
|
||||
|
||||
return M
|
||||
@ -0,0 +1,21 @@
|
||||
local Z = require "log.writer.net.zmq._private.compat"
|
||||
local server = require "log.writer.async.server.zmq"
|
||||
|
||||
local function create_writer(ctx, addr, maker)
|
||||
if ctx and not Z.is_ctx(ctx) then
|
||||
ctx, addr, maker = nil, ctx, addr
|
||||
end
|
||||
|
||||
if maker then
|
||||
server.run(ctx, addr, maker, "log.logformat.default")
|
||||
end
|
||||
|
||||
return require"log.writer.format".new(require"log.logformat.proxy".new(),
|
||||
require"log.writer.net.zmq.push".new(ctx, addr))
|
||||
end
|
||||
|
||||
local M = {}
|
||||
|
||||
M.new = create_writer
|
||||
|
||||
return M
|
||||
@ -0,0 +1 @@
|
||||
return require "log.writer.stdout"
|
||||
@ -0,0 +1,334 @@
|
||||
local io = require "io"
|
||||
local os = require "os"
|
||||
local Log = require "log"
|
||||
|
||||
local make_attr, color_writeln, COLORS
|
||||
|
||||
if not COLORS then -- afx cio
|
||||
local ok, cio = pcall(require, "cio")
|
||||
if ok then
|
||||
COLORS = {
|
||||
BLACK = 'n',
|
||||
BLUE = 'b',
|
||||
GREEN = 'g',
|
||||
CYAN = 'c',
|
||||
RED = 'r',
|
||||
MAGENTA = 'm',
|
||||
BROWN = 'y',
|
||||
LIGHTGRAY = 'w',
|
||||
DARKGRAY = 'n+',
|
||||
LIGHTBLUE = 'b+',
|
||||
LIGHTGREEN = 'g+',
|
||||
LIGHTCYAN = 'c+',
|
||||
LIGHTRED = 'r+',
|
||||
LIGHTMAGENTA = 'm+',
|
||||
YELLOW = 'y+',
|
||||
WHITE = 'w+',
|
||||
}
|
||||
make_attr = function(F, B)
|
||||
return F .. '/' .. B
|
||||
end
|
||||
color_writeln = function(attr, text)
|
||||
cio.textattr(attr)
|
||||
cio.writeln(text or "")
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
if not COLORS then -- conio
|
||||
local ok, conio = pcall(require, "conio")
|
||||
if ok then
|
||||
COLORS = {
|
||||
BLACK = conio.COLOR_BLACK,
|
||||
BLUE = conio.COLOR_BLUE,
|
||||
GREEN = conio.COLOR_GREEN,
|
||||
CYAN = conio.COLOR_CYAN,
|
||||
RED = conio.COLOR_RED,
|
||||
MAGENTA = conio.COLOR_MAGENTA,
|
||||
BROWN = conio.COLOR_BROWN,
|
||||
LIGHTGRAY = conio.COLOR_LIGHTGRAY,
|
||||
DARKGRAY = conio.COLOR_DARKGRAY,
|
||||
LIGHTBLUE = conio.COLOR_LIGHTBLUE,
|
||||
LIGHTGREEN = conio.COLOR_LIGHTGREEN,
|
||||
LIGHTCYAN = conio.COLOR_LIGHTCYAN,
|
||||
LIGHTRED = conio.COLOR_LIGHTRED,
|
||||
LIGHTMAGENTA = conio.COLOR_LIGHTMAGENTA,
|
||||
YELLOW = conio.COLOR_YELLOW,
|
||||
WHITE = conio.COLOR_WHITE,
|
||||
}
|
||||
make_attr = function(F, B)
|
||||
return {F, B}
|
||||
end
|
||||
color_writeln = function(attr, text)
|
||||
conio.textcolor(attr[1])
|
||||
conio.textbackground(attr[2])
|
||||
conio.cputs((text or "") .. '\n')
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
if not COLORS then -- ansicolors
|
||||
local IS_WINDOWS = (package.config:sub(1, 1) == '\\')
|
||||
if IS_WINDOWS and os.getenv("ANSICON") then
|
||||
IS_WINDOWS = nil
|
||||
end
|
||||
local ok, c
|
||||
if not IS_WINDOWS then
|
||||
ok, c = pcall(require, "ansicolors")
|
||||
end
|
||||
if ok and (type(c) == 'table') then
|
||||
if c.black then -- version above 1.0.2
|
||||
COLORS = {
|
||||
BLACK = 1,
|
||||
BLUE = 2,
|
||||
GREEN = 3,
|
||||
CYAN = 4,
|
||||
RED = 5,
|
||||
MAGENTA = 6,
|
||||
BROWN = 7,
|
||||
LIGHTGRAY = 8,
|
||||
DARKGRAY = 9,
|
||||
LIGHTBLUE = 10,
|
||||
LIGHTGREEN = 11,
|
||||
LIGHTCYAN = 12,
|
||||
LIGHTRED = 13,
|
||||
LIGHTMAGENTA = 14,
|
||||
YELLOW = 15,
|
||||
WHITE = 16,
|
||||
}
|
||||
local reset = tostring(c.reset)
|
||||
local fore = {
|
||||
[COLORS.BLACK] = c.black,
|
||||
[COLORS.BLUE] = c.blue,
|
||||
[COLORS.GREEN] = c.green,
|
||||
[COLORS.CYAN] = c.cyan,
|
||||
[COLORS.RED] = c.red,
|
||||
[COLORS.MAGENTA] = c.magenta,
|
||||
[COLORS.BROWN] = c.yellow,
|
||||
[COLORS.LIGHTGRAY] = c.white,
|
||||
[COLORS.DARKGRAY] = c.black .. c.bright,
|
||||
[COLORS.LIGHTBLUE] = c.blue .. c.bright,
|
||||
[COLORS.LIGHTGREEN] = c.green .. c.bright,
|
||||
[COLORS.LIGHTCYAN] = c.cyan .. c.bright,
|
||||
[COLORS.LIGHTRED] = c.red .. c.bright,
|
||||
[COLORS.LIGHTMAGENTA] = c.magenta .. c.bright,
|
||||
[COLORS.YELLOW] = c.yellow .. c.bright,
|
||||
[COLORS.WHITE] = c.white .. c.bright,
|
||||
}
|
||||
|
||||
local back = {
|
||||
[COLORS.BLACK] = c.onblack,
|
||||
[COLORS.BLUE] = c.onblue,
|
||||
[COLORS.GREEN] = c.ongreen,
|
||||
[COLORS.CYAN] = c.oncyan,
|
||||
[COLORS.RED] = c.onred,
|
||||
[COLORS.MAGENTA] = c.onmagenta,
|
||||
[COLORS.BROWN] = c.onyellow,
|
||||
[COLORS.LIGHTGRAY] = c.onwhite,
|
||||
[COLORS.DARKGRAY] = c.onblack .. c.bright,
|
||||
[COLORS.LIGHTBLUE] = c.onblue .. c.bright,
|
||||
[COLORS.LIGHTGREEN] = c.ongreen .. c.bright,
|
||||
[COLORS.LIGHTCYAN] = c.oncyan .. c.bright,
|
||||
[COLORS.LIGHTRED] = c.onred .. c.bright,
|
||||
[COLORS.LIGHTMAGENTA] = c.onmagenta .. c.bright,
|
||||
[COLORS.YELLOW] = c.onyellow .. c.bright,
|
||||
[COLORS.WHITE] = c.onwhite .. c.bright,
|
||||
}
|
||||
|
||||
make_attr = function(F, B)
|
||||
return fore[F] .. back[B]
|
||||
end
|
||||
color_writeln = function(attr, text)
|
||||
io.write(attr, text, reset, '\n')
|
||||
end
|
||||
elseif c.noReset then -- 1.0.2
|
||||
COLORS = {
|
||||
BLACK = 'black',
|
||||
BLUE = 'blue',
|
||||
GREEN = 'green',
|
||||
CYAN = 'cyan',
|
||||
RED = 'red',
|
||||
MAGENTA = 'magenta',
|
||||
BROWN = 'yellow',
|
||||
LIGHTGRAY = 'white',
|
||||
DARKGRAY = 'bright black',
|
||||
LIGHTBLUE = 'bright blue',
|
||||
LIGHTGREEN = 'bright green',
|
||||
LIGHTCYAN = 'bright cyan',
|
||||
LIGHTRED = 'bright red',
|
||||
LIGHTMAGENTA = 'bright magenta',
|
||||
YELLOW = 'bright yellow',
|
||||
WHITE = 'bright white',
|
||||
}
|
||||
|
||||
make_attr = function(F, B)
|
||||
return c.noReset("%{" .. F .. " " .. B .. "bg}")
|
||||
end
|
||||
local RESET = c.noReset("%{reset}")
|
||||
color_writeln = function(attr, text)
|
||||
io.write(attr, (text), RESET, '\n')
|
||||
end
|
||||
end
|
||||
end
|
||||
if ok and (type(c) == 'function') then
|
||||
COLORS = {
|
||||
BLACK = 1,
|
||||
BLUE = 2,
|
||||
GREEN = 3,
|
||||
CYAN = 4,
|
||||
RED = 5,
|
||||
MAGENTA = 6,
|
||||
BROWN = 7,
|
||||
LIGHTGRAY = 8,
|
||||
DARKGRAY = 9,
|
||||
LIGHTBLUE = 10,
|
||||
LIGHTGREEN = 11,
|
||||
LIGHTCYAN = 12,
|
||||
LIGHTRED = 13,
|
||||
LIGHTMAGENTA = 14,
|
||||
YELLOW = 15,
|
||||
WHITE = 16,
|
||||
}
|
||||
local fore = {
|
||||
[COLORS.BLACK] = "%{black}",
|
||||
[COLORS.BLUE] = "%{blue}",
|
||||
[COLORS.GREEN] = "%{green}",
|
||||
[COLORS.CYAN] = "%{cyan}",
|
||||
[COLORS.RED] = "%{red}",
|
||||
[COLORS.MAGENTA] = "%{magenta}",
|
||||
[COLORS.BROWN] = "%{yellow}",
|
||||
[COLORS.LIGHTGRAY] = "%{white}",
|
||||
[COLORS.DARKGRAY] = "%{black}" .. "%{bright}",
|
||||
[COLORS.LIGHTBLUE] = "%{blue}" .. "%{bright}",
|
||||
[COLORS.LIGHTGREEN] = "%{green}" .. "%{bright}",
|
||||
[COLORS.LIGHTCYAN] = "%{cyan}" .. "%{bright}",
|
||||
[COLORS.LIGHTRED] = "%{red}" .. "%{bright}",
|
||||
[COLORS.LIGHTMAGENTA] = "%{magenta}" .. "%{bright}",
|
||||
[COLORS.YELLOW] = "%{yellow}" .. "%{bright}",
|
||||
[COLORS.WHITE] = "%{white}" .. "%{bright}",
|
||||
}
|
||||
local back = {
|
||||
[COLORS.BLACK] = "%{blackbg}",
|
||||
[COLORS.BLUE] = "%{bluebg}",
|
||||
[COLORS.GREEN] = "%{greenbg}",
|
||||
[COLORS.CYAN] = "%{cyanbg}",
|
||||
[COLORS.RED] = "%{redbg}",
|
||||
[COLORS.MAGENTA] = "%{magentabg}",
|
||||
[COLORS.BROWN] = "%{yellowbg}",
|
||||
[COLORS.LIGHTGRAY] = "%{whitebg}",
|
||||
[COLORS.DARKGRAY] = "%{blackbg}" .. "%{bright}",
|
||||
[COLORS.LIGHTBLUE] = "%{bluebg}" .. "%{bright}",
|
||||
[COLORS.LIGHTGREEN] = "%{greenbg}" .. "%{bright}",
|
||||
[COLORS.LIGHTCYAN] = "%{cyanbg}" .. "%{bright}",
|
||||
[COLORS.LIGHTRED] = "%{redbg}" .. "%{bright}",
|
||||
[COLORS.LIGHTMAGENTA] = "%{magentabg}" .. "%{bright}",
|
||||
[COLORS.YELLOW] = "%{yellowbg}" .. "%{bright}",
|
||||
[COLORS.WHITE] = "%{whitebg}" .. "%{bright}",
|
||||
}
|
||||
make_attr = function(F, B)
|
||||
return fore[F] .. back[B]
|
||||
end
|
||||
color_writeln = function(attr, text)
|
||||
io.write(c(attr .. text), '\n')
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
if not COLORS then
|
||||
local ok, term = pcall(require, "term")
|
||||
if ok then
|
||||
COLORS = {
|
||||
BLACK = 1,
|
||||
BLUE = 2,
|
||||
GREEN = 3,
|
||||
CYAN = 4,
|
||||
RED = 5,
|
||||
MAGENTA = 6,
|
||||
BROWN = 7,
|
||||
LIGHTGRAY = 8,
|
||||
DARKGRAY = 9,
|
||||
LIGHTBLUE = 10,
|
||||
LIGHTGREEN = 11,
|
||||
LIGHTCYAN = 12,
|
||||
LIGHTRED = 13,
|
||||
LIGHTMAGENTA = 14,
|
||||
YELLOW = 15,
|
||||
WHITE = 16,
|
||||
}
|
||||
|
||||
local c = term.colors
|
||||
local reset = c.reset
|
||||
local fore = {
|
||||
[COLORS.BLACK] = c.black,
|
||||
[COLORS.BLUE] = c.blue,
|
||||
[COLORS.GREEN] = c.green,
|
||||
[COLORS.CYAN] = c.cyan,
|
||||
[COLORS.RED] = c.red,
|
||||
[COLORS.MAGENTA] = c.magenta,
|
||||
[COLORS.BROWN] = c.yellow,
|
||||
[COLORS.LIGHTGRAY] = c.white,
|
||||
[COLORS.DARKGRAY] = c.black .. c.bright,
|
||||
[COLORS.LIGHTBLUE] = c.blue .. c.bright,
|
||||
[COLORS.LIGHTGREEN] = c.green .. c.bright,
|
||||
[COLORS.LIGHTCYAN] = c.cyan .. c.bright,
|
||||
[COLORS.LIGHTRED] = c.red .. c.bright,
|
||||
[COLORS.LIGHTMAGENTA] = c.magenta .. c.bright,
|
||||
[COLORS.YELLOW] = c.yellow .. c.bright,
|
||||
[COLORS.WHITE] = c.white .. c.bright,
|
||||
}
|
||||
|
||||
local back = {
|
||||
[COLORS.BLACK] = c.onblack,
|
||||
[COLORS.BLUE] = c.onblue,
|
||||
[COLORS.GREEN] = c.ongreen,
|
||||
[COLORS.CYAN] = c.oncyan,
|
||||
[COLORS.RED] = c.onred,
|
||||
[COLORS.MAGENTA] = c.onmagenta,
|
||||
[COLORS.BROWN] = c.onyellow,
|
||||
[COLORS.LIGHTGRAY] = c.onwhite,
|
||||
[COLORS.DARKGRAY] = c.onblack .. c.bright,
|
||||
[COLORS.LIGHTBLUE] = c.onblue .. c.bright,
|
||||
[COLORS.LIGHTGREEN] = c.ongreen .. c.bright,
|
||||
[COLORS.LIGHTCYAN] = c.oncyan .. c.bright,
|
||||
[COLORS.LIGHTRED] = c.onred .. c.bright,
|
||||
[COLORS.LIGHTMAGENTA] = c.onmagenta .. c.bright,
|
||||
[COLORS.YELLOW] = c.onyellow .. c.bright,
|
||||
[COLORS.WHITE] = c.onwhite .. c.bright,
|
||||
}
|
||||
|
||||
make_attr = function(F, B)
|
||||
return fore[F] .. back[B]
|
||||
end
|
||||
color_writeln = function(attr, text)
|
||||
io.write(attr .. text .. reset, '\n')
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
if not COLORS then -- fallback to console
|
||||
return require "log.writer.console"
|
||||
end
|
||||
|
||||
local colors = {
|
||||
[Log.LVL.EMERG] = make_attr(COLORS.WHITE, COLORS.LIGHTRED),
|
||||
[Log.LVL.ALERT] = make_attr(COLORS.BLUE, COLORS.LIGHTRED),
|
||||
[Log.LVL.FATAL] = make_attr(COLORS.BLACK, COLORS.LIGHTRED),
|
||||
[Log.LVL.ERROR] = make_attr(COLORS.LIGHTRED, COLORS.BLACK),
|
||||
[Log.LVL.WARNING] = make_attr(COLORS.LIGHTMAGENTA, COLORS.BLACK),
|
||||
[Log.LVL.NOTICE] = make_attr(COLORS.LIGHTCYAN, COLORS.BLACK),
|
||||
[Log.LVL.INFO] = make_attr(COLORS.WHITE, COLORS.BLACK),
|
||||
[Log.LVL.DEBUG] = make_attr(COLORS.YELLOW, COLORS.BLACK),
|
||||
[Log.LVL.TRACE] = make_attr(COLORS.LIGHTGREEN, COLORS.BLACK),
|
||||
}
|
||||
|
||||
local function console_writer(fmt, msg, lvl, now)
|
||||
color_writeln(colors[lvl], fmt(msg, lvl, now))
|
||||
end
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.new()
|
||||
return console_writer
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,17 @@
|
||||
local LOG = require "log"
|
||||
local file_logger = require "log.writer.file.private.impl"
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.new(opt)
|
||||
local logger = file_logger:new(opt)
|
||||
LOG.add_cleanup(function()
|
||||
logger:close()
|
||||
end)
|
||||
|
||||
return function(fmt, msg, lvl, now)
|
||||
logger:write((fmt(msg, lvl, now)))
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,17 @@
|
||||
local file = require "log.writer.file"
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.new(log_dir, log_name, max_rows)
|
||||
return file.new {
|
||||
log_dir = log_dir,
|
||||
log_name = log_name,
|
||||
max_rows = max_rows,
|
||||
by_day = true,
|
||||
close_file = false,
|
||||
flush_interval = 1,
|
||||
}
|
||||
end
|
||||
|
||||
return M
|
||||
|
||||
@ -0,0 +1,484 @@
|
||||
local Log = require "log"
|
||||
local io = require "io"
|
||||
local os = require "os"
|
||||
local string = require "string"
|
||||
local date = require "date"
|
||||
local lfs = require "lfs"
|
||||
|
||||
local DIR_SEP = package.config:sub(1, 1)
|
||||
local IS_WINDOWS = DIR_SEP == '\\'
|
||||
|
||||
local function remove_dir_end(str)
|
||||
return (string.gsub(str, '[\\/]+$', ''))
|
||||
end
|
||||
|
||||
local function ensure_dir_end(str)
|
||||
return remove_dir_end(str) .. DIR_SEP
|
||||
end
|
||||
|
||||
local function path_normolize_sep(P)
|
||||
return (string.gsub(P, '\\', DIR_SEP):gsub('/', DIR_SEP))
|
||||
end
|
||||
|
||||
local function path_fullpath(P)
|
||||
P = path_normolize_sep(P)
|
||||
local ch1, ch2 = P:sub(1, 1), P:sub(2, 2)
|
||||
if IS_WINDOWS then
|
||||
if ch1 == DIR_SEP then -- \temp => c:\temp
|
||||
local cwd = lfs.currentdir()
|
||||
local disk = cwd:sub(1, 2)
|
||||
P = disk .. P
|
||||
elseif ch1 == '~' then -- ~\temp
|
||||
local base = os.getenv('USERPROFILE') or (os.getenv('HOMEDRIVE') .. os.getenv('HOMEPATH'))
|
||||
P = ((ch2 == DIR_SEP) and remove_dir_end(base) or ensure_dir_end(base)) .. string.sub(P, 2)
|
||||
elseif ch2 ~= ':' then
|
||||
P = ensure_dir_end(lfs.currentdir()) .. P
|
||||
end
|
||||
else
|
||||
if ch1 == '~' then -- ~/temp
|
||||
local base = os.getenv('HOME')
|
||||
P = ((ch2 == DIR_SEP) and remove_dir_end(base) or ensure_dir_end(base)) .. string.sub(P, 2)
|
||||
else
|
||||
if P:sub(1, 1) ~= '/' then
|
||||
P = ensure_dir_end(lfs.currentdir()) .. P
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
P = string.gsub(P, DIR_SEP .. '%.' .. DIR_SEP, DIR_SEP):gsub(DIR_SEP .. DIR_SEP, DIR_SEP)
|
||||
while true do
|
||||
local first, last = string.find(P, DIR_SEP .. "[^" .. DIR_SEP .. "]+" .. DIR_SEP .. '%.%.' .. DIR_SEP)
|
||||
if not first then
|
||||
break
|
||||
end
|
||||
P = string.sub(P, 1, first) .. string.sub(P, last + 1)
|
||||
end
|
||||
|
||||
return P
|
||||
end
|
||||
|
||||
local function attrib(P, ...)
|
||||
if IS_WINDOWS then
|
||||
if #P < 4 and P:sub(2, 2) == ':' then
|
||||
P = ensure_dir_end(P) -- c: => c:\
|
||||
else
|
||||
P = remove_dir_end(P) -- c:\temp\ => c:\temp
|
||||
end
|
||||
end
|
||||
return lfs.attributes(P, ...)
|
||||
end
|
||||
|
||||
local function path_exists(P)
|
||||
return attrib(P, 'mode') ~= nil and P
|
||||
end
|
||||
|
||||
local function path_isdir(P)
|
||||
return attrib(P, 'mode') == 'directory' and P
|
||||
end
|
||||
|
||||
local function path_mkdir(P)
|
||||
local P = path_fullpath(P)
|
||||
local p = ''
|
||||
|
||||
for str in string.gmatch(ensure_dir_end(P), '.-' .. DIR_SEP) do
|
||||
p = p .. str
|
||||
if path_exists(p) then
|
||||
if not path_isdir(p) then
|
||||
return nil, 'can not create ' .. p
|
||||
end
|
||||
else
|
||||
local ok, err = lfs.mkdir(remove_dir_end(p))
|
||||
if not ok then
|
||||
return nil, err .. ' ' .. p
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
return true
|
||||
end
|
||||
|
||||
local function path_getctime(P)
|
||||
return attrib(P, 'change')
|
||||
end
|
||||
|
||||
local function path_getmtime(P)
|
||||
return attrib(P, 'modification')
|
||||
end
|
||||
|
||||
local function path_getatime(P)
|
||||
return attrib(P, 'access')
|
||||
end
|
||||
|
||||
local function path_getsize(P)
|
||||
return attrib(P, 'size')
|
||||
end
|
||||
|
||||
local function path_getrows(P)
|
||||
local f, err = io.open(P, "r")
|
||||
if not f then
|
||||
return 0
|
||||
end
|
||||
local count = 0
|
||||
for _ in f:lines() do
|
||||
count = count + 1
|
||||
end
|
||||
f:close()
|
||||
return count
|
||||
end
|
||||
|
||||
local function path_remove(P)
|
||||
return os.remove(P)
|
||||
end
|
||||
|
||||
local function path_rename(from, to)
|
||||
path_remove(to)
|
||||
return os.rename(from, to)
|
||||
end
|
||||
|
||||
local function reset_out(FileName, rewrite)
|
||||
local END_OF_LINE = '\n'
|
||||
local FILE_APPEND = 'a'
|
||||
|
||||
if rewrite then
|
||||
local FILE_REWRITE = 'w+'
|
||||
local f, err = io.open(FileName, FILE_REWRITE);
|
||||
if not f then
|
||||
return nil, err
|
||||
end
|
||||
f:close();
|
||||
end
|
||||
|
||||
return function(msg)
|
||||
local f, err = io.open(FileName, FILE_APPEND)
|
||||
if not f then
|
||||
return nil, err
|
||||
end
|
||||
f:write(msg, END_OF_LINE)
|
||||
f:close()
|
||||
end
|
||||
end
|
||||
|
||||
local function make_no_close_reset(flush_interval)
|
||||
return function(FileName, rewrite)
|
||||
local END_OF_LINE = '\n'
|
||||
local FILE_APPEND = 'a'
|
||||
|
||||
if rewrite then
|
||||
local FILE_REWRITE = 'w+'
|
||||
local f, err = io.open(FileName, FILE_REWRITE);
|
||||
if not f then
|
||||
return nil, err
|
||||
end
|
||||
f:close()
|
||||
end
|
||||
|
||||
local f, err = io.open(FileName, FILE_APPEND);
|
||||
if not f then
|
||||
return nil, err
|
||||
end
|
||||
|
||||
local writer
|
||||
if flush_interval then
|
||||
local flush_interval, counter = flush_interval, 0
|
||||
writer = function(msg)
|
||||
f:write(msg, END_OF_LINE)
|
||||
counter = counter + 1
|
||||
if counter >= flush_interval then
|
||||
f:flush()
|
||||
counter = 0
|
||||
end
|
||||
end
|
||||
else
|
||||
writer = function(msg)
|
||||
f:write(msg, END_OF_LINE)
|
||||
end
|
||||
end
|
||||
return writer, function()
|
||||
f:close()
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
local function split_ext(fname)
|
||||
local s1, s2 = string.match(fname, '([^\\/]*)([.][^.\\/]*)$')
|
||||
if s1 then
|
||||
return s1, s2
|
||||
end
|
||||
s1 = string.match(fname, '([^\\/]+)$')
|
||||
if s1 then
|
||||
return s1, ''
|
||||
end
|
||||
end
|
||||
|
||||
local function assert_2(f1, f2, v1, v2)
|
||||
assert(f1 == v1, string.format("Expected '%s' got '%s'", tostring(f1), tostring(v1)))
|
||||
assert(f2 == v2, string.format("Expected '%s' got '%s'", tostring(f2), tostring(v2)))
|
||||
end
|
||||
|
||||
assert_2("events", ".log", split_ext("events.log"))
|
||||
assert_2("events", '', split_ext("events"))
|
||||
assert_2(nil, nil, split_ext("events\\"))
|
||||
assert_2('', '.log', split_ext("events\\.log"))
|
||||
assert_2('log', '', split_ext("events\\log"))
|
||||
|
||||
local file_logger = {}
|
||||
|
||||
local FILE_LOG_DATE_FMT = "%Y%m%d"
|
||||
local EOL_SIZE = IS_WINDOWS and 2 or 1
|
||||
|
||||
local function get_file_date(fname)
|
||||
local mdate = path_getmtime(fname)
|
||||
if mdate then
|
||||
mdate = date(mdate):tolocal()
|
||||
else
|
||||
mdate = date()
|
||||
end
|
||||
return mdate:fmt(FILE_LOG_DATE_FMT)
|
||||
end
|
||||
|
||||
function file_logger:close()
|
||||
if self.private_.logger and self.private_.logger_close then
|
||||
self.private_.logger_close()
|
||||
end
|
||||
self.private_.logger = nil
|
||||
self.private_.logger_close = nil
|
||||
end
|
||||
|
||||
function file_logger:open()
|
||||
local full_name = self:current_name()
|
||||
|
||||
local logger, err = self.private_.reset_out(full_name)
|
||||
if not logger then
|
||||
return nil, string.format("can not create logger for file '%s':", full_name, err)
|
||||
end
|
||||
|
||||
self.private_.logger = logger
|
||||
self.private_.logger_close = err
|
||||
self.private_.log_date = os.date(FILE_LOG_DATE_FMT)
|
||||
self.private_.log_rows = 0
|
||||
self.private_.log_size = 0
|
||||
|
||||
return true
|
||||
end
|
||||
|
||||
function file_logger:current_name()
|
||||
return self.private_.log_dir .. self.private_.log_name
|
||||
end
|
||||
|
||||
function file_logger:archive_roll_name(i)
|
||||
return self.private_.log_dir .. string.format("%s.%.5d.log", self.private_.arc_pfx, i)
|
||||
end
|
||||
|
||||
function file_logger:archive_date_name(d, i)
|
||||
return self.private_.log_dir .. string.format("%s.%s.%.5d.log", self.private_.arc_pfx, d, i)
|
||||
end
|
||||
|
||||
function file_logger:reset_log_by_roll()
|
||||
self:close()
|
||||
|
||||
local full_name = self:current_name()
|
||||
local first_name = self:archive_roll_name(1)
|
||||
|
||||
-- we must "free" space for current file
|
||||
if path_exists(first_name) then
|
||||
for i = self.private_.roll_count - 1, 1, -1 do
|
||||
local fname1 = self:archive_roll_name(i)
|
||||
local fname2 = self:archive_roll_name(i + 1)
|
||||
path_rename(fname1, fname2)
|
||||
end
|
||||
end
|
||||
|
||||
if path_exists(full_name) then
|
||||
local ok, err = path_rename(full_name, first_name)
|
||||
if not ok then
|
||||
return nil, string.format("can not rename '%s' to '%s' : %s", full_name, first_name, err or '')
|
||||
end
|
||||
end
|
||||
|
||||
return self:open()
|
||||
end
|
||||
|
||||
function file_logger:next_date_name(log_date)
|
||||
local id = self.private_.id
|
||||
|
||||
local fname = self:archive_date_name(log_date, id)
|
||||
while path_exists(fname) do
|
||||
id = id + 1
|
||||
fname = self:archive_date_name(log_date, id)
|
||||
end
|
||||
|
||||
self.private_.id = id
|
||||
return fname
|
||||
end
|
||||
|
||||
function file_logger:reset_log_by_date(log_date)
|
||||
self:close()
|
||||
|
||||
local full_name = self:current_name()
|
||||
if path_exists(full_name) then -- previews file
|
||||
log_date = log_date or get_file_date(full_name)
|
||||
local next_fname = self:next_date_name(log_date)
|
||||
local ok, err = path_rename(full_name, next_fname)
|
||||
if not ok then
|
||||
return nil, string.format("can not rename '%s' to '%s' : ", full_name, next_fname, err or '')
|
||||
end
|
||||
end
|
||||
|
||||
return self:open()
|
||||
end
|
||||
|
||||
function file_logger:reset_log(...)
|
||||
if self.private_.roll_count then
|
||||
return self:reset_log_by_roll(...)
|
||||
end
|
||||
return self:reset_log_by_date(...)
|
||||
end
|
||||
|
||||
function file_logger:check()
|
||||
if self.private_.by_day then
|
||||
local now = os.date(FILE_LOG_DATE_FMT)
|
||||
if self.private_.log_date ~= now then
|
||||
local ok, err = self:reset_log_by_date(self.private_.log_date)
|
||||
self.private_.id = 1
|
||||
return ok, err
|
||||
end
|
||||
end
|
||||
|
||||
if self.private_.max_rows and (self.private_.log_rows >= self.private_.max_rows) then
|
||||
return self:reset_log()
|
||||
end
|
||||
|
||||
if self.private_.max_size and (self.private_.log_size >= self.private_.max_size) then
|
||||
return self:reset_log()
|
||||
end
|
||||
|
||||
return true
|
||||
end
|
||||
|
||||
function file_logger:write(msg)
|
||||
local ok, err = self:check()
|
||||
if not ok then
|
||||
io.stderr:write("logger error: ", err, '\n')
|
||||
return
|
||||
end
|
||||
|
||||
self.private_.logger(msg)
|
||||
self.private_.log_rows = self.private_.log_rows + 1
|
||||
self.private_.log_size = self.private_.log_size + #msg + EOL_SIZE
|
||||
end
|
||||
|
||||
function file_logger:init(opt)
|
||||
|
||||
if (opt.by_day or opt.roll_count) then
|
||||
assert(not (opt.by_day and opt.roll_count), "Can not set 'by_day' and 'roll_count' fields at the same time!")
|
||||
end
|
||||
assert(opt.log_name, 'field log_name is required')
|
||||
|
||||
local log_dir = path_fullpath(opt.log_dir or '.')
|
||||
|
||||
if path_exists(log_dir) then
|
||||
assert(path_isdir(log_dir))
|
||||
else
|
||||
assert(path_mkdir(log_dir))
|
||||
end
|
||||
|
||||
local log_name, log_ext = string.match(opt.log_name, '([^\\/]+)([.][^.\\/]+)$')
|
||||
assert(log_name and log_ext)
|
||||
|
||||
log_dir = ensure_dir_end(log_dir)
|
||||
local full_name = log_dir .. log_name .. log_ext
|
||||
local current_size = path_getsize(full_name)
|
||||
if 0 == current_size then
|
||||
-- prevent rename zero size logfile
|
||||
path_remove(full_name)
|
||||
end
|
||||
|
||||
local flush_interval = opt.flush_interval and
|
||||
assert(tonumber(opt.flush_interval), 'flush_interval must be a number') or 1
|
||||
self.private_ = {
|
||||
-- options
|
||||
log_dir = log_dir,
|
||||
log_name = log_name .. log_ext,
|
||||
max_rows = opt.max_rows or math.huge,
|
||||
max_size = opt.max_size or math.huge,
|
||||
reset_out = opt.close_file and reset_out or make_no_close_reset(flush_interval),
|
||||
arc_pfx = opt.archive_prefix or log_name,
|
||||
roll_count = opt.roll_count and assert(tonumber(opt.roll_count), 'roll_count must be a number'),
|
||||
by_day = not not opt.by_day,
|
||||
|
||||
-- state
|
||||
-- log_date = ; -- date when current log file was create
|
||||
-- log_rows = 0; -- how many lines in current log file
|
||||
-- log_size = 0;
|
||||
id = 1, -- numbers of file in current log_date
|
||||
}
|
||||
if self.private_.roll_count then
|
||||
assert(self.private_.roll_count > 0)
|
||||
end
|
||||
|
||||
local reuse_log = opt.reuse
|
||||
|
||||
if reuse_log and current_size and (current_size > 0) then
|
||||
self.private_.log_date = get_file_date(full_name)
|
||||
|
||||
if opt.max_rows then
|
||||
self.private_.log_rows = path_getrows(full_name) or 0
|
||||
else
|
||||
self.private_.log_rows = 0
|
||||
end
|
||||
|
||||
if opt.max_size then
|
||||
self.private_.log_size = path_getsize(full_name) or 0
|
||||
else
|
||||
self.private_.log_size = 0
|
||||
end
|
||||
|
||||
local logger, err = self.private_.reset_out(full_name)
|
||||
if not logger then
|
||||
error(string.format("can not create logger for file '%s':", full_name, err))
|
||||
end
|
||||
|
||||
self.private_.logger = logger
|
||||
self.private_.logger_close = err
|
||||
|
||||
else
|
||||
assert(self:reset_log())
|
||||
end
|
||||
|
||||
return self
|
||||
end
|
||||
|
||||
function file_logger:new(...)
|
||||
local o = setmetatable({}, {
|
||||
__index = self,
|
||||
}):init(...)
|
||||
Log.add_cleanup(function()
|
||||
o:close()
|
||||
end)
|
||||
return o
|
||||
end
|
||||
|
||||
local function do_profile()
|
||||
require"profiler".start()
|
||||
|
||||
local logger = file_logger:new{
|
||||
log_dir = './logs',
|
||||
log_name = "events.log",
|
||||
max_rows = 1000,
|
||||
max_size = 70,
|
||||
roll_count = 11,
|
||||
-- by_day = true;
|
||||
close_file = false,
|
||||
flush_interval = 1,
|
||||
reuse = true,
|
||||
}
|
||||
|
||||
for i = 1, 10000 do
|
||||
local msg = string.format("%5d", i)
|
||||
logger:write(msg)
|
||||
end
|
||||
|
||||
logger:close()
|
||||
end
|
||||
|
||||
return file_logger
|
||||
@ -0,0 +1,18 @@
|
||||
local file = require "log.writer.file"
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.new(log_dir, log_name, roll_count, max_size)
|
||||
return file.new {
|
||||
log_dir = log_dir,
|
||||
log_name = log_name,
|
||||
max_size = max_size or 10 * 1024 * 1024,
|
||||
roll_count = assert(roll_count),
|
||||
close_file = false,
|
||||
flush_interval = 1,
|
||||
reuse = true,
|
||||
}
|
||||
end
|
||||
|
||||
return M
|
||||
|
||||
@ -0,0 +1 @@
|
||||
return require "log.writer.filter.lvl.le"
|
||||
@ -0,0 +1,14 @@
|
||||
local Log = require "log"
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.new(max_lvl, writer)
|
||||
max_lvl = assert(Log.lvl2number(max_lvl))
|
||||
return function(fmt, msg, lvl, now)
|
||||
if lvl == max_lvl then
|
||||
writer(fmt, msg, lvl, now)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,14 @@
|
||||
local Log = require "log"
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.new(max_lvl, writer)
|
||||
max_lvl = assert(Log.lvl2number(max_lvl))
|
||||
return function(fmt, msg, lvl, now)
|
||||
if lvl <= max_lvl then
|
||||
writer(fmt, msg, lvl, now)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,9 @@
|
||||
local M = {}
|
||||
|
||||
function M.new(newfmt, writer)
|
||||
return function(oldfmt, msg, lvl, now)
|
||||
writer(newfmt, msg, lvl, now)
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,42 @@
|
||||
local log = require "log"
|
||||
|
||||
local freeswitch = assert(freeswitch)
|
||||
|
||||
local FS_LVL = {
|
||||
console = 0,
|
||||
alert = 1,
|
||||
crit = 2,
|
||||
err = 3,
|
||||
warning = 4,
|
||||
notice = 5,
|
||||
info = 6,
|
||||
debug = 7,
|
||||
}
|
||||
|
||||
local LOG2FS = {
|
||||
[log.LVL.EMERG] = FS_LVL.alert,
|
||||
[log.LVL.ALERT] = FS_LVL.alert,
|
||||
[log.LVL.FATAL] = FS_LVL.crit,
|
||||
[log.LVL.ERROR] = FS_LVL.err,
|
||||
[log.LVL.WARNING] = FS_LVL.warning,
|
||||
[log.LVL.NOTICE] = FS_LVL.notice,
|
||||
[log.LVL.INFO] = FS_LVL.info,
|
||||
[log.LVL.DEBUG] = FS_LVL.debug,
|
||||
[log.LVL.TRACE] = FS_LVL.debug,
|
||||
}
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.new(session)
|
||||
if session then
|
||||
return function(fmt, msg, lvl, now)
|
||||
session:consoleLog(LOG2FS[lvl], msg .. '\n')
|
||||
end
|
||||
else
|
||||
return function(fmt, msg, lvl, now)
|
||||
freeswitch.consoleLog(LOG2FS[lvl], msg .. '\n')
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,13 @@
|
||||
local M = {}
|
||||
|
||||
function M.new(...)
|
||||
local writers = {...}
|
||||
return function(...)
|
||||
for i = 1, #writers do
|
||||
writers[i](...)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
|
||||
@ -0,0 +1,29 @@
|
||||
local socket = require "socket"
|
||||
local log_packer = require "log.logformat.proxy.pack"
|
||||
|
||||
local _M = {}
|
||||
|
||||
function _M.run(writer, final, logformat, host, port)
|
||||
local uskt = assert(socket.udp())
|
||||
assert(uskt:setsockname(host, port))
|
||||
local unpack = log_packer.unpack
|
||||
|
||||
while true do
|
||||
local msg, err = uskt:receivefrom()
|
||||
if msg then
|
||||
local msg, lvl, now = unpack(msg)
|
||||
if msg and lvl and now then
|
||||
writer(logformat, msg, lvl, now)
|
||||
end
|
||||
else
|
||||
if err ~= 'timeout' then
|
||||
io.stderr:write('log.writer.net.udp.server: ', err, '\n')
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
-- @todo
|
||||
-- if final then final() end
|
||||
end
|
||||
|
||||
return _M
|
||||
@ -0,0 +1,55 @@
|
||||
local Z = require "log.writer.net.zmq._private.compat"
|
||||
local IMPL = require "log.writer.net.zmq._private.impl"
|
||||
|
||||
local zmq, ETERM, zstrerror, zassert, zrecv = Z.zmq, Z.ETERM, Z.strerror, Z.assert, Z.recv
|
||||
local zerrcode = Z.errcode
|
||||
|
||||
local log_packer = require "log.logformat.proxy.pack"
|
||||
|
||||
local _M = {}
|
||||
|
||||
function _M.run(writer, final, logformat, ctx, stype, address, addr_sync)
|
||||
-- print(writer, logformat, ctx, stype, address, addr_sync)
|
||||
local stypes = {
|
||||
SUB = zmq.SUB,
|
||||
PULL = zmq.PULL,
|
||||
}
|
||||
stype = assert(stypes[stype], 'Unsupported socket type')
|
||||
|
||||
ctx = IMPL.context(ctx)
|
||||
|
||||
local skt = zassert(ctx:socket(stype))
|
||||
zassert(skt:bind(address))
|
||||
|
||||
if addr_sync then
|
||||
local skt_sync = zassert(ctx:socket(zmq.PAIR))
|
||||
zassert(skt_sync:connect(addr_sync))
|
||||
skt_sync:send("")
|
||||
skt_sync:close()
|
||||
end
|
||||
|
||||
local unpack = log_packer.unpack
|
||||
|
||||
while (true) do
|
||||
local msg, err = zrecv(skt)
|
||||
if msg then
|
||||
local msg, lvl, now = unpack(msg)
|
||||
if msg and lvl and now then
|
||||
writer(logformat, msg, lvl, now)
|
||||
end
|
||||
else
|
||||
if zerrcode(err) == ETERM then
|
||||
break
|
||||
end
|
||||
io.stderr:write('log.writer.net.zmq.server: ', tostring(err), zstrerror(err), '\n')
|
||||
end
|
||||
end
|
||||
|
||||
if final then
|
||||
final()
|
||||
end
|
||||
|
||||
skt:close()
|
||||
end
|
||||
|
||||
return _M
|
||||
@ -0,0 +1,26 @@
|
||||
local Log = require "log"
|
||||
local sendmail = require "sendmail"
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.new(from, to, server, subject)
|
||||
assert(to, "'to' parameter is required")
|
||||
assert(from, "'from' parameter is required")
|
||||
assert(server, "'server' parameter is required")
|
||||
|
||||
subject = subject or ''
|
||||
|
||||
return function(fmt, msg, lvl, now)
|
||||
msg = fmt(msg, lvl, now)
|
||||
sendmail(from, to, server, {
|
||||
subject = now:fmt("%F %T") .. ' [' .. Log.LVL_NAMES[lvl] .. '] ' .. subject,
|
||||
file = {
|
||||
name = 'message.txt',
|
||||
data = msg,
|
||||
},
|
||||
text = msg,
|
||||
})
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,19 @@
|
||||
local socket = require("socket")
|
||||
|
||||
local function create_socket(host, port, timeout)
|
||||
local skt = assert(socket.udp())
|
||||
assert(skt:settimeout(timeout or 0.1))
|
||||
assert(skt:setpeername(host, port))
|
||||
return skt
|
||||
end
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.new(host, port, timeout)
|
||||
local skt = create_socket(host, port, timeout)
|
||||
return function(fmt, ...)
|
||||
skt:send((fmt(...)))
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1 @@
|
||||
return require "log.writer.net.zmq.pub"
|
||||
@ -0,0 +1,150 @@
|
||||
local function prequire(...)
|
||||
local ok, mod = pcall(require, ...)
|
||||
return ok and mod, mod or nil
|
||||
end
|
||||
|
||||
local zmq, zthreads, zpoller
|
||||
local zstrerror, zassert, ETERM
|
||||
local zconnect, zbind
|
||||
local zrecv_all, zrecv
|
||||
local zerrcode
|
||||
|
||||
local function has_member(t, key)
|
||||
local ok, has
|
||||
if type(key) == "table" then
|
||||
ok, has = pcall(function()
|
||||
for _, k in ipairs(key) do
|
||||
if nil == t[k] then
|
||||
return false
|
||||
end
|
||||
end
|
||||
return true
|
||||
end)
|
||||
else
|
||||
ok, has = pcall(function()
|
||||
return nil ~= t[key]
|
||||
end)
|
||||
end
|
||||
return ok and has
|
||||
end
|
||||
|
||||
local function is_ctx(ctx)
|
||||
local tname = type(ctx)
|
||||
if (tname ~= 'table') and (tname ~= 'userdata') then
|
||||
return false
|
||||
end
|
||||
return has_member(ctx, {'socket', 'term'})
|
||||
end
|
||||
|
||||
zmq = prequire "lzmq"
|
||||
if zmq then
|
||||
zpoller = prequire "lzmq.poller"
|
||||
zthreads = prequire "lzmq.threads"
|
||||
ETERM = zmq.errors.ETERM
|
||||
zstrerror = function(err)
|
||||
if type(err) == "number" then
|
||||
return zmq.strerror(err)
|
||||
end
|
||||
if type(err) == "string" then
|
||||
return err
|
||||
end
|
||||
return err:msg()
|
||||
end
|
||||
zerrcode = function(err)
|
||||
if type(err) == "number" then
|
||||
return err
|
||||
end
|
||||
if type(err) == "string" then
|
||||
return err
|
||||
end -- @todo extract no from string
|
||||
return err:no()
|
||||
end
|
||||
zassert = zmq.assert
|
||||
zrecv_all = function(skt)
|
||||
return skt:recv_all()
|
||||
end
|
||||
zconnect = function(skt, addr)
|
||||
return skt:connect(addr)
|
||||
end
|
||||
zbind = function(skt, addr)
|
||||
return skt:bind(addr)
|
||||
end
|
||||
else
|
||||
zmq = require "zmq"
|
||||
zpoller = require "zmq.poller"
|
||||
zthreads = prequire "zmq.threads"
|
||||
ETERM = 'closed'
|
||||
zstrerror = function(err)
|
||||
return err
|
||||
end
|
||||
zerrcode = function(err)
|
||||
return err
|
||||
end
|
||||
zassert = assert
|
||||
zrecv_all = function(skt)
|
||||
local t = {}
|
||||
local r, err = skt:recv()
|
||||
if not r then
|
||||
return nil, err
|
||||
end
|
||||
table.insert(t, r)
|
||||
while skt:rcvmore() == 1 do
|
||||
r, err = skt:recv()
|
||||
if not r then
|
||||
return nil, err, t
|
||||
end
|
||||
table.insert(t, r)
|
||||
end
|
||||
return t
|
||||
end
|
||||
zconnect = function(skt, addr)
|
||||
if type(addr) == 'table' then
|
||||
for i, a in ipairs(addr) do
|
||||
local ok, err = skt:connect(a)
|
||||
if not ok then
|
||||
return nil, err, i
|
||||
end
|
||||
end
|
||||
return true
|
||||
end
|
||||
return skt:connect(addr)
|
||||
end
|
||||
zbind = function(skt, addr)
|
||||
if type(addr) == 'table' then
|
||||
for i, a in ipairs(addr) do
|
||||
local ok, err = skt:bind(a)
|
||||
if not ok then
|
||||
return nil, err, i
|
||||
end
|
||||
end
|
||||
return true
|
||||
end
|
||||
return skt:bind(addr)
|
||||
end
|
||||
end
|
||||
|
||||
zrecv = function(skt)
|
||||
local ok, err, t = zrecv_all(skt)
|
||||
if not ok then
|
||||
if t and t[1] then
|
||||
return t[1]
|
||||
end
|
||||
return nil, err
|
||||
end
|
||||
return ok[1]
|
||||
end
|
||||
|
||||
return {
|
||||
zmq = zmq,
|
||||
threads = zthreads,
|
||||
poller = zpoller,
|
||||
connect = zconnect,
|
||||
bind = zbind,
|
||||
recv_all = zrecv_all,
|
||||
recv = zrecv,
|
||||
strerror = zstrerror,
|
||||
errcode = zerrcode,
|
||||
assert = zassert,
|
||||
ETERM = ETERM,
|
||||
is_ctx = is_ctx,
|
||||
}
|
||||
@ -0,0 +1,74 @@
|
||||
local Log = require "log"
|
||||
local Z = require "log.writer.net.zmq._private.compat"
|
||||
|
||||
local zmq, zthreads = Z.zmq, Z.threads
|
||||
local zstrerror, zassert = Z.strerror, Z.assert
|
||||
local ETERM = Z.ETERM
|
||||
local zconnect, zbind = Z.connect, Z.bind
|
||||
|
||||
local log_ctx
|
||||
|
||||
local function context(ctx)
|
||||
-- we have to use same context for all writers
|
||||
if ctx and log_ctx then
|
||||
assert(ctx == log_ctx)
|
||||
end
|
||||
|
||||
if log_ctx then
|
||||
return log_ctx
|
||||
end
|
||||
|
||||
log_ctx = ctx or (zthreads and zthreads.get_parent_ctx()) or zassert(zmq.init(1))
|
||||
|
||||
return log_ctx
|
||||
end
|
||||
|
||||
local function socket(ctx, stype, is_srv, addr, timeout)
|
||||
local stypes = {
|
||||
PUSH = zmq.PUSH,
|
||||
PUB = zmq.PUB,
|
||||
}
|
||||
stype = assert(stypes[stype], 'Unsupported socket type')
|
||||
timeout = timeout or 100
|
||||
ctx = context(ctx)
|
||||
|
||||
local skt = ctx:socket(stype)
|
||||
if ctx.autoclose then
|
||||
ctx:autoclose(skt)
|
||||
end
|
||||
skt:set_sndtimeo(timeout)
|
||||
skt:set_linger(timeout)
|
||||
if is_srv then
|
||||
zassert(zbind(skt, addr))
|
||||
else
|
||||
zassert(zconnect(skt, addr))
|
||||
end
|
||||
if not ctx.autoclose then
|
||||
Log.add_cleanup(function()
|
||||
skt:close()
|
||||
end)
|
||||
end
|
||||
return skt
|
||||
end
|
||||
|
||||
local function init(stype, is_srv)
|
||||
local M = {}
|
||||
|
||||
function M.new(ctx, addr, timeout)
|
||||
if ctx and not Z.is_ctx(ctx) then
|
||||
ctx, addr, timeout = nil, ctx, addr
|
||||
end
|
||||
|
||||
local skt = socket(ctx, stype, is_srv, addr, timeout)
|
||||
return function(fmt, ...)
|
||||
skt:send((fmt(...)))
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
end
|
||||
|
||||
return {
|
||||
init = init,
|
||||
context = context,
|
||||
}
|
||||
@ -0,0 +1 @@
|
||||
return require"log.writer.net.zmq._private.impl".init('PUB', false)
|
||||
@ -0,0 +1 @@
|
||||
return require"log.writer.net.zmq._private.impl".init('PUSH', false)
|
||||
@ -0,0 +1 @@
|
||||
return require"log.writer.net.zmq._private.impl".init('PUB', true)
|
||||
@ -0,0 +1,9 @@
|
||||
local M = {}
|
||||
|
||||
function M.new(prefix, writer)
|
||||
return function(fmt, msg, lvl, now)
|
||||
writer(fmt, prefix .. msg, lvl, now)
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,11 @@
|
||||
local io = require "io"
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.new()
|
||||
return function(fmt, ...)
|
||||
io.stderr:write(fmt(...), '\n')
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,11 @@
|
||||
local io = require "io"
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.new()
|
||||
return function(fmt, ...)
|
||||
io.stdout:write(fmt(...), '\n')
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
@ -0,0 +1,78 @@
|
||||
local skynet = require "skynet"
|
||||
local queue = require "skynet.queue"
|
||||
local util = require "store_util"
|
||||
require "skynet.manager"
|
||||
|
||||
local dbconf
|
||||
local lock = queue()
|
||||
local guid_generator_addrs = {}
|
||||
|
||||
local CMD = {}
|
||||
|
||||
function CMD.init(conf)
|
||||
assert(not dbconf, "dbmgr has been initialized.")
|
||||
dbconf = conf -- init is allowed only once
|
||||
|
||||
local uidconf = dbconf.guid_generator
|
||||
assert(uidconf)
|
||||
for _, worker_id in pairs(uidconf.worker_ids) do
|
||||
local addr = skynet.newservice("guid_generator", worker_id)
|
||||
table.insert(guid_generator_addrs, addr)
|
||||
end
|
||||
|
||||
local redisconf = dbconf.redis
|
||||
for dbkey, conf in pairs(redisconf) do
|
||||
for index = 1, conf.service_num do
|
||||
local addr = skynet.newservice("redisd", dbkey, index)
|
||||
local ok = skynet.call(addr, "lua", "init", conf)
|
||||
if not ok then
|
||||
assert(false, ("redisd init failed. [dbkey] %s [id] %d"):format(dbkey, index))
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
local mysqlconf = dbconf.mysql
|
||||
for dbkey, conf in pairs(mysqlconf) do
|
||||
for index = 1, conf.service_num do
|
||||
local addr = skynet.newservice("mysqld", dbkey, index)
|
||||
local ok = skynet.call(addr, "lua", "init", conf)
|
||||
if not ok then
|
||||
assert(false, ("mysqld init failed. [dbkey] %s [id] %d"):format(dbkey, index))
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
return true
|
||||
end
|
||||
|
||||
function CMD.mysql_service_num(dbkey)
|
||||
if not dbconf then return end
|
||||
local mysqlconf = dbconf.mysql
|
||||
if not mysqlconf then return end
|
||||
local conf = mysqlconf[dbkey]
|
||||
if not conf then return end
|
||||
return conf.service_num
|
||||
end
|
||||
|
||||
function CMD.redis_service_num(dbkey)
|
||||
if not dbconf then return end
|
||||
local redisconf = dbconf.redis
|
||||
if not redisconf then return end
|
||||
local conf = redisconf[dbkey]
|
||||
if not conf then return end
|
||||
return conf.service_num
|
||||
end
|
||||
|
||||
function CMD.guid_generators()
|
||||
return guid_generator_addrs
|
||||
end
|
||||
|
||||
skynet.start(function()
|
||||
skynet.dispatch("lua", function(_, _, cmd, ...)
|
||||
local f = CMD[cmd]
|
||||
assert(f, cmd)
|
||||
skynet.retpack(f(...))
|
||||
end)
|
||||
end)
|
||||
|
||||
skynet.register(".dbmgr")
|
||||
@ -0,0 +1,34 @@
|
||||
local skynet = require "skynet"
|
||||
local st = require "skynet.sharetable"
|
||||
local mc = require "skynet.multicast"
|
||||
|
||||
local channel
|
||||
|
||||
local CMD = {}
|
||||
|
||||
function CMD.query(filename)
|
||||
assert(channel)
|
||||
return st.query(filename)
|
||||
end
|
||||
|
||||
function CMD.loadfiles(filenames)
|
||||
assert(type(filenames) == "table")
|
||||
for _, filename in pairs(filenames) do
|
||||
st.loadfile(filename)
|
||||
end
|
||||
channel:publish(filenames)
|
||||
end
|
||||
|
||||
function CMD.channel()
|
||||
return channel.channel
|
||||
end
|
||||
|
||||
skynet.start(function()
|
||||
channel = mc.new()
|
||||
skynet.dispatch("lua", function(_, _, cmd, ...)
|
||||
local f = CMD[cmd]
|
||||
assert(f, cmd)
|
||||
skynet.retpack(f(...))
|
||||
end)
|
||||
end)
|
||||
|
||||
@ -0,0 +1,52 @@
|
||||
local skynet = require "skynet"
|
||||
local time = skynet.time
|
||||
local floor = math.floor
|
||||
local ceil = math.ceil
|
||||
|
||||
--[[
|
||||
sign | delta seconds | worker id | sequence
|
||||
1bit | 30bits | 20bits | 13bits
|
||||
|
||||
sign - 0为正数 1为负数
|
||||
delta seconds - 2020/01/01到当前时间时间差, 30bits可以用到2054-01-09
|
||||
worker id - 0~1048575 每个服务worker id都应该不一样,才能保证唯一性, worker_id需使用者自行规划
|
||||
sequence - 0~8191 当前时刻的id序列
|
||||
]]
|
||||
|
||||
local DELTA_SECS_BITS = 30
|
||||
local WORKER_ID_BITS = 20
|
||||
local SEQ_BITS = 13
|
||||
local TIME_LEFT_SHIFT_BITS = WORKER_ID_BITS + SEQ_BITS
|
||||
|
||||
local START_TIME = 1577808000 -- 2020/01/01
|
||||
local MAX_WORKER_ID = (1<<WORKER_ID_BITS) - 1
|
||||
local MAX_SEQ = (1<<SEQ_BITS) - 1
|
||||
|
||||
local worker_id = tonumber(...)
|
||||
assert(worker_id >= 0 and worker_id <= MAX_WORKER_ID)
|
||||
|
||||
local seq = 0
|
||||
local last_delta_time = 0
|
||||
local gen_guid
|
||||
|
||||
gen_guid = function()
|
||||
local now = time()
|
||||
local floor_now = floor(now)
|
||||
local delta_time = floor_now - START_TIME
|
||||
if last_delta_time ~= delta_time then
|
||||
last_delta_time = delta_time
|
||||
seq = 0
|
||||
elseif seq < MAX_SEQ then
|
||||
seq = seq + 1
|
||||
else
|
||||
skynet.sleep(100 - ceil(now - floor_now)*100)
|
||||
return gen_guid()
|
||||
end
|
||||
return (delta_time<<TIME_LEFT_SHIFT_BITS) + (worker_id<<SEQ_BITS) + seq
|
||||
end
|
||||
|
||||
skynet.start(function()
|
||||
skynet.dispatch("lua", function()
|
||||
skynet.retpack(gen_guid())
|
||||
end)
|
||||
end)
|
||||
@ -0,0 +1,50 @@
|
||||
local skynet = require "skynet"
|
||||
local mysql = require "skynet.db.mysql"
|
||||
local util = require "store_util"
|
||||
require "skynet.manager"
|
||||
|
||||
local traceback = debug.traceback
|
||||
|
||||
local dbkey, index = ...
|
||||
local db
|
||||
|
||||
local CMD = {}
|
||||
|
||||
local function success(ret)
|
||||
if not ret or ret.err or ret.badresult then
|
||||
return false
|
||||
end
|
||||
return true
|
||||
end
|
||||
|
||||
function CMD.init(conf)
|
||||
db = mysql.connect(conf)
|
||||
db:query("set names utf8mb4")
|
||||
return true
|
||||
end
|
||||
|
||||
function CMD.exec_one(sql)
|
||||
local ok, ret = xpcall(db.query, traceback, db, sql)
|
||||
if not ok or not success(ret) then
|
||||
assert(false, ("sql=[%s] ret=[%s]"):format(sql, util.encode(ret)))
|
||||
return
|
||||
end
|
||||
return ret
|
||||
end
|
||||
|
||||
function CMD.exec(sqls)
|
||||
for i = 1, #sqls do
|
||||
local sql = sqls[i]
|
||||
CMD.exec_one(sql)
|
||||
end
|
||||
end
|
||||
|
||||
skynet.start(function()
|
||||
skynet.dispatch("lua", function(_, _, cmd, ...)
|
||||
local f = CMD[cmd]
|
||||
assert(f, cmd)
|
||||
skynet.retpack(f(...))
|
||||
end)
|
||||
end)
|
||||
|
||||
skynet.register(util.mysql_sname(dbkey, index))
|
||||
@ -0,0 +1,42 @@
|
||||
local skynet = require "skynet"
|
||||
local redis = require "skynet.db.redis"
|
||||
local util = require "store_util"
|
||||
require "skynet.manager"
|
||||
|
||||
local traceback = debug.traceback
|
||||
local tunpack = table.unpack
|
||||
local tconcat = table.concat
|
||||
local dbkey, index = ...
|
||||
local db
|
||||
|
||||
local CMD = {}
|
||||
|
||||
function CMD.init(conf)
|
||||
db = redis.connect(conf)
|
||||
return true
|
||||
end
|
||||
|
||||
function CMD.exec_one(cmd, ...)
|
||||
local ok, ret = xpcall(db[cmd], traceback, db, ...)
|
||||
if not ok then
|
||||
assert(false, ("cmd=[%s %s] ret=[%s]"):format(cmd, tconcat({...}, " "), ret))
|
||||
return
|
||||
end
|
||||
return ret
|
||||
end
|
||||
|
||||
function CMD.exec(cmds)
|
||||
for _, cmd in pairs(cmds) do
|
||||
xpcall(CMD.exec_one, traceback, tunpack(cmd))
|
||||
end
|
||||
end
|
||||
|
||||
skynet.start(function()
|
||||
skynet.dispatch("lua", function(_, _, cmd, ...)
|
||||
local f = CMD[cmd]
|
||||
assert(f, cmd)
|
||||
skynet.retpack(f(...))
|
||||
end)
|
||||
end)
|
||||
|
||||
skynet.register(util.redis_sname(dbkey, index))
|
||||
@ -0,0 +1,198 @@
|
||||
local skynet = require "skynet"
|
||||
local redis = require "skynet.db.redis"
|
||||
local crypt = require "skynet.crypt"
|
||||
|
||||
local math_floor = math.floor
|
||||
local math_ceil = math.ceil
|
||||
local math_random = math.random
|
||||
|
||||
local function hash(script)
|
||||
local key = crypt.sha1(script)
|
||||
return crypt.hexencode(key)
|
||||
end
|
||||
|
||||
local QUORUM
|
||||
|
||||
local SCRIPT = {
|
||||
LOCK = [[
|
||||
local key = KEYS[1]
|
||||
if redis.call("exists", key) == 1 then
|
||||
return 0
|
||||
end
|
||||
redis.call("set", key, ARGV[1], "PX", ARGV[2])
|
||||
return 1
|
||||
]],
|
||||
UNLOCK = [[
|
||||
local key = KEYS[1]
|
||||
if redis.call("get", key) == ARGV[1] then
|
||||
redis.pcall("del", key)
|
||||
return 1
|
||||
end
|
||||
return 0
|
||||
]],
|
||||
EXTEND = [[
|
||||
local key = KEYS[1]
|
||||
if redis.call("get", key) ~= ARGV[1] then
|
||||
return 0
|
||||
end
|
||||
redis.call("set", key, ARGV[1], "PX", ARGV[2])
|
||||
return 1
|
||||
]],
|
||||
}
|
||||
|
||||
local SCRIPT_HASH = {
|
||||
LOCK = hash(SCRIPT.LOCK),
|
||||
UNLOCK = hash(SCRIPT.UNLOCK),
|
||||
EXTEND = hash(SCRIPT.EXTEND),
|
||||
}
|
||||
|
||||
local conf
|
||||
local dbs = {}
|
||||
local sessions = {}
|
||||
|
||||
local function execute_script(db, type, s)
|
||||
local ok, ret = pcall(db["evalsha"], db, SCRIPT_HASH[type], 1, s.lockname, s.uuid, s.timeout)
|
||||
if not ok and ret:find("NOSCRIPT") then
|
||||
ok, ret = pcall(db["eval"], db, SCRIPT[type], 1, s.lockname, s.uuid, s.timeout)
|
||||
end
|
||||
if not ok then
|
||||
skynet.error("redis execute_script err.", ret, s.lockname, s.uuid, s.timeout)
|
||||
return false
|
||||
end
|
||||
if ret == 1 then
|
||||
return true
|
||||
end
|
||||
return false
|
||||
end
|
||||
|
||||
local function execute_script_timeout(db, type, s)
|
||||
local co = coroutine.running()
|
||||
local ok, ret = false, "timeout"
|
||||
|
||||
skynet.fork(function()
|
||||
ok, ret = execute_script(db, type, s)
|
||||
if co then
|
||||
skynet.wakeup(co)
|
||||
co = nil
|
||||
end
|
||||
end)
|
||||
|
||||
skynet.sleep(conf.request_timeout / 10)
|
||||
if co then
|
||||
co = nil
|
||||
end
|
||||
return ok, ret
|
||||
end
|
||||
|
||||
local function calc_time(s)
|
||||
local now = skynet.now() * 10
|
||||
local drift = math_floor(conf.drift_factor * s.timeout) + 2
|
||||
s.starttime = now
|
||||
s.endtime = now + s.timeout - drift
|
||||
end
|
||||
|
||||
local function make_session(lockname, uuid, timeout)
|
||||
local s = {
|
||||
lockname = lockname,
|
||||
uuid = uuid,
|
||||
timeout = timeout,
|
||||
attempts = 0,
|
||||
starttime = 0,
|
||||
endtime = 0,
|
||||
}
|
||||
calc_time(s)
|
||||
return s
|
||||
end
|
||||
|
||||
local function unlock(s)
|
||||
s.endtime = 0
|
||||
for _, db in pairs(dbs) do
|
||||
execute_script(db, "UNLOCK", s)
|
||||
end
|
||||
end
|
||||
|
||||
local function attempt(s, is_extend)
|
||||
s.attempts = s.attempts + 1
|
||||
local votes = 0
|
||||
for _, db in pairs(dbs) do
|
||||
local ok
|
||||
if is_extend then
|
||||
ok = execute_script_timeout(db, "EXTEND", s)
|
||||
else
|
||||
ok = execute_script_timeout(db, "LOCK", s)
|
||||
end
|
||||
if ok then
|
||||
votes = votes + 1
|
||||
end
|
||||
end
|
||||
|
||||
local now = skynet.now() * 10
|
||||
if votes >= QUORUM and s.endtime > now then
|
||||
local ti = s.timeout / 3 - (now - s.starttime)
|
||||
ti = math_floor(ti / 10)
|
||||
if ti < 0 then
|
||||
ti = 0
|
||||
end
|
||||
skynet.timeout(ti, function()
|
||||
if s.endtime == 0 then
|
||||
return
|
||||
end
|
||||
s.attempts = 0
|
||||
calc_time(s)
|
||||
attempt(s, true)
|
||||
end)
|
||||
return true
|
||||
else
|
||||
unlock(s)
|
||||
-- retry
|
||||
if conf.retry_count == -1 or s.attempts <= conf.retry_count then
|
||||
local t = conf.retry_delay + math_floor((math_random() * 2 - 1) * conf.retry_jitter)
|
||||
skynet.sleep(math_ceil(t / 10))
|
||||
calc_time(s)
|
||||
return attempt(s)
|
||||
end
|
||||
-- failed
|
||||
sessions[s.uuid] = nil
|
||||
return false, "timeout"
|
||||
end
|
||||
end
|
||||
|
||||
local CMD = {}
|
||||
|
||||
function CMD.lock(lockname, uuid, timeout)
|
||||
timeout = timeout or conf.timeout
|
||||
local s = sessions[uuid]
|
||||
if s then
|
||||
return false, "session exist"
|
||||
end
|
||||
s = make_session(lockname, uuid, timeout)
|
||||
sessions[uuid] = s
|
||||
|
||||
return attempt(s)
|
||||
end
|
||||
|
||||
function CMD.unlock(_, uuid)
|
||||
local s = sessions[uuid]
|
||||
if not s then
|
||||
return false, "session not exist."
|
||||
end
|
||||
sessions[uuid] = nil
|
||||
return unlock(s)
|
||||
end
|
||||
|
||||
skynet.init(function()
|
||||
conf = require "redlock_conf"
|
||||
for _, client in ipairs(conf.servers) do
|
||||
table.insert(dbs, redis.connect(client))
|
||||
end
|
||||
QUORUM = math_floor(#conf.servers / 2) + 1
|
||||
end)
|
||||
|
||||
skynet.start(function()
|
||||
skynet.dispatch("lua", function(_, _, cmd, ...)
|
||||
local f = CMD[cmd]
|
||||
assert(f, cmd)
|
||||
skynet.retpack(f(...))
|
||||
end)
|
||||
end)
|
||||
|
||||
@ -1 +1,60 @@
|
||||
-- 统一的服务基础逻辑
|
||||
local _service_name, _service_path = ...
|
||||
assert(_service_name, "_service_name nil")
|
||||
assert(_service_path, "_service_path nil")
|
||||
|
||||
SERVICE_NAME = _service_name
|
||||
local _arguments = table.pack(...)
|
||||
|
||||
local skynet = require "skynet"
|
||||
require "skynet.manager"
|
||||
|
||||
require "sharetable"
|
||||
require "cluster"
|
||||
require "multicast"
|
||||
require "datacenter"
|
||||
|
||||
local function _exception(e)
|
||||
skynet.error(e)
|
||||
return e
|
||||
end
|
||||
|
||||
local _xpcall = function(f, ...)
|
||||
return xpcall(f, _exception, ...)
|
||||
end
|
||||
|
||||
local single -- 服务逻辑单例对象
|
||||
|
||||
local _info = function()
|
||||
end
|
||||
|
||||
local _dispatch = function(session, source, cmd, ...)
|
||||
local func = single:get_cmd_func(cmd)
|
||||
if not func then
|
||||
skynet.retpack({
|
||||
ok = false,
|
||||
msg = string.format("[%s]UNKNOWN_CMD[%s]FROM[%s]", _service_name, cmd, tostring(source)),
|
||||
})
|
||||
return
|
||||
end
|
||||
|
||||
if session > 0 then
|
||||
skynet.retpack(_xpcall(func, single, ...))
|
||||
else
|
||||
_xpcall(func, single, ...)
|
||||
end
|
||||
end
|
||||
|
||||
skynet.start(function()
|
||||
single = require(_service_path):instance(table.unpack(_arguments, 2, _arguments.n))
|
||||
assert(single.dispatch, "zeus service dispatch function nil")
|
||||
|
||||
skynet.dispatch("lua", _dispatch)
|
||||
skynet.info_func(_info)
|
||||
|
||||
local fsm = single:get_fsm()
|
||||
fsm.initial()
|
||||
fsm.running()
|
||||
|
||||
single:add_timer(_info, 3600)
|
||||
end)
|
||||
|
||||
Loading…
Reference in New Issue