lua实现详细日志记录，lua日志记录

lua实现详细日志记录，lua日志记录

find .|xargs grep -ri "IBM"

--[[
logger
--]]


local _M = {}
local mt = { __index = _M }


local random = require("resty.random")
local str = require("resty.string")
local time = require("time")
local cjson = require("cjson")
 
local startTime = os.date("%H:%M:%S")


local function trim(s)
return string.gsub(s, "^%s*(.-)%s*$", "%1")
end


--[[
用分隔符切分字符串


@param pString 要切分的字符串
@param pPattern 分隔符

@return 返回一个table 里面存有切分后的字符串
--]]


local function split(str, delimiter)


    local fields = {}
    str:gsub(string.format("([^%s]*)%s?", delimiter, delimiter), function(c) table.insert(fields, c) end)


if fields[#fields] == "" then
fields[#fields] = nil
end


    return fields
end


--[[
将一个lua变量转换为字符串
可以让你更加清楚的看到这个变量里面到底存了什么内容


@param value 要转换成字符串的变量


@return 变量内容的字符串形式
--]]


local function to_str(value)
local str = ''


if type(value) ~= 'table' then
if type(value) == 'string' then
str = string.format("%q", value)
else
str = tostring(value)
end
else
local auxTable = {}


for key in pairs(value) do
if tonumber(key) ~= key then
table.insert(auxTable, key)
else
table.insert(auxTable, to_str(key))
end
end


table.sort(auxTable)


str = str .. '{'


local separator = ""
local entry = ""


for _, fieldName in ipairs(auxTable) do
if tonumber(fieldName) and tonumber(fieldName) > 0 then
entry = to_str(value[tonumber(fieldName)])
else
entry = fieldName .. " = " .. to_str(value[fieldName])
end
str = str .. separator .. entry


separator = ", "
end


str = str .. '}'


end


return str
end


--- log ID 
-- </br> 当前交易的log标识号，查询log的时候可以用来定位一笔交易。
-- @class table
-- @name id
-- @field id 用于定义当前交易log的8位随机16进制字符。


_M.id = str.to_hex(random.bytes(4, true))


_M.DAY  = 1
_M.HOUR = 2
_M.rotateType = _M.DAY




--- 设置记录日志文件的类型。
-- @param rotateType 要设定的类型。 
-- @return 没有返回。
-- @usage logger.setRotateType(logger.HOUR)


function _M.setRotateType(rotateType)


_M.rotateType = rotateType


end


--- 设置记录日志的目录.
-- @param path 记录日志的目录。
-- @return 没有返回
-- @usage logger.setPath("../logs/quickpay/")


function _M.setPath(path)


_M.path = path


local f, err = io.open(path, "r")
if not f then
os.execute(string.format("mkdir %s -p", path))
return
end 
f:close()


local f1, err = io.open(path .. "/tradelog", "r")
    if not f1 then
        os.execute(string.format("mkdir %s -p", path .. "/tradelog"))
        return
    end
    f1:close()


local f2, err = io.open(path .. "/monitor", "r")
    if not f2 then
        os.execute(string.format("mkdir %s -p", path .. "/monitor"))
        return
    end
    f2:close()


local f3, err = io.open(path .. "monitor.json", "r")
    if not f3 then
        os.execute(string.format("mkdir %s -p", path .. "/monitor.json"))
        return
    end
    f3:close()


end


--- 记录日志信息到文件traceYYYYMMDD.log中.
-- </br>文件被记录由setPath指定的目录中。
-- 格式 ： [时间] [日志id] [文件, 行] 日志信息
-- @param str 记录的日志信息，如果str不是string类型，将会被转换为字符串。
-- @return 没有返回
-- @usage logger.log("retcode = " .. retcode)


function _M.log(uid, str, level)


-- 1. 如果不是string 会将变量转换为字符串。
if type(str) ~= "string" then
str = to_str(str)
end


local file


if _M.rotateType == _M.DAY then
file = string.format("%s/tradelog/trace%s.log", _M.path, os.date("%Y%m%d"))
else
file = string.format("%s/tradelog/trace%s.log", _M.path, os.date("%Y%m%d.%H"))
end


local f, err = io.open(file, "a")


-- 2. 如果打开文件失败直接返回，如果发现没有日志文件可以检查目录是否存在,还有权限。
if not f then return end


-- 3. 获取打日志时候的代码位置
level = tonumber(level) or 2


if level < 2 then 
level = 2
end


local debug_info = debug.getinfo(level, "Snl")


local i, j = debug_info.short_src:find("[^/]+$")
local filename = string.sub(debug_info.short_src, i, j)
local position = string.format("%s, %d", filename, debug_info.currentline)




    local tv = time.gettimeofday()
    local t = os.date("%H:%M:%S", tv.sec) .. ":" .. string.format("%03d", math.floor(tv.usec/1000))


-- 4. 每一行都要按照日志格式输出到日志文件中。
local lines = split(str, "\n")


for i, v in ipairs(lines) do
local line = string.format("[%s] [%s] [%s] [%s] %s\n", t, _M.id, uid, position, v)
f:write(line)
end

f:close()
end




--- 记录日志信息到文件traceYYYYMMDD.log中.
-- </br>文件被记录由setPath指定的目录中。
-- 格式 ： [时间] [日志id] [文件, 行] 日志信息
-- @param str 记录的日志信息，如果str不是string类型，将会被转换为字符串。
-- @return 没有返回
-- @usage logger.debug("retcode = " .. retcode)


function _M.debug(str, level)


if not _M.debug then
return 
end


-- 1. 如果不是string 会将变量转换为字符串。
if type(str) ~= "string" then
str = to_str(str)
end


local file


if _M.rotateType == _M.DAY then
file = string.format("%s/trace%s.log", _M.path, os.date("%Y%m%d"))
else
file = string.format("%s/trace%s.log", _M.path, os.date("%Y%m%d.%H"))
end


local f, err = io.open(file, "a")


-- 2. 如果打开文件失败直接返回，如果发现没有日志文件可以检查目录是否存在,还有权限。
if not f then return end


-- 3. 获取打日志时候的代码位置


level = tonumber(level) or 2


if level < 2 then
level = 2
end


local debug_info = debug.getinfo(level, "Snl")


local i, j = debug_info.short_src:find("[^/]+$")
local filename = string.sub(debug_info.short_src, i, j)
local position = string.format("%s, %d", filename, debug_info.currentline)


-- 4. 每一行都要按照日志格式输出到日志文件中。
local lines = split(str, "\n")
lines = { str }


local tv = time.gettimeofday()
local t = os.date("%H:%M:%S", tv.sec) .. ":" .. tostring(math.floor(tv.usec/1000))


for i, v in ipairs(lines) do
local line = string.format("[%s] [%s] [%s] %s\n", t, _M.id, position, v)
f:write(line)
end

f:close()
end


local monitorData   = {}
local monitorDefine = {}
local monitorFields = {}


local placeholder   = "_"




--- 设置monitor的格式定义。
-- @param define monitor格式定义。
-- @return 没有返回
-- @usage logger.setMonitorDefine(require("paymentMonDefine"))


function _M.setMonitorDefine(define)


monitorDefine = define


for i, v in ipairs(monitorDefine) do
monitorFields[v.name] = placeholder
end


end




--- 设置monitor相关域的值。
-- @param key 要设置的域。
-- @param value 要设置的相关域对应的值。
-- @return 没有返回。
-- @usage logger.monitorSet("mcssn", mcssn)


function _M.monitorSet(key, value)


monitorFields[key] = value or placeholder


end




--- 记录monitor日志.
-- @return 没有返回
-- @usage logger.monitor()


function _M.monitor()


-- 1. 获取文件名
local file = string.format("%s/monitor/monitor%s.log", _M.path, os.date("%Y%m%d"))


local f, err = io.open(file, "a")


-- 2. 如果打开文件失败直接返回，如果发现没有日志文件可以检查目录是否存在,还有权限。
if not f then return end


local endTime = os.date("%H:%M:%S")


local line = string.format("%s %s %s", startTime, endTime, _M.id)


for i, v in ipairs(monitorDefine) do
local s = monitorFields[v.name]
s = trim(s)
if #s == 0 then s = placeholder end
s = s:sub(-tonumber(v.maxlen))
line = line .. string.format(" %-" .. v.maxlen .. "s", s)
end


line = line .. "\n"


f:write(line)
f:close()
end




---JSON log
local monitorDefine4Json = {}
local monitorFields4Json = {}


local placeholder4Json   = ""




function _M.setMonitorDefine4Json(define)


monitorFields4Json = define


for i, v in ipairs(monitorFields4Json) do
monitorFields4Json[v.name] = placeholder
end


end




--- 设置monitor相关域的值。
-- @param key 要设置的域。
-- @param value 要设置的相关域对应的值。
-- @return 没有返回。
-- @usage logger.monitorSet("mcssn", mcssn)


function _M.monitorSet4Json(key, value)


monitorFields4Json[key] = value or placeholder4Json


end




function _M.monitor4json()
  
-- 1. 获取文件名
local file = string.format("%s/monitor.json/Nposp_monitor_json.%s", _M.path, os.date("%Y%m%d"))


local f, err = io.open(file, "a")


-- 2. 如果打开文件失败直接返回，如果发现没有日志文件可以检查目录是否存在,还有权限。
if not f then return end


local endTime = os.date("%H:%M:%S")
  monitorFields4Json["logssn"] = _M.id
  monitorFields4Json["trans_start_time"] = startTime
  monitorFields4Json["trans_end_time"] = endTime
    
    
local line = cjson.encode(monitorFields4Json)


line = line .. "\n"


f:write(line)
f:close()
end


return _M