lua trace

尊重原创：https://github.com/cloudwu/lua-trace

一位牛大哥写的，先贴出来，以后学习

 

trace.lua

local debug = debug
local print = print
local ipairs = ipairs
local pairs = pairs
local string = string
local rawget = rawget

local info = { file = {} }

local trace = {}

local function setname(level)
    info.filename = debug.getinfo(level + 1,"S").short_src
    if info.file[info.filename] == nil then
        info.file[info.filename] = {}
    end
    info.var = info.file[info.filename]
end

local function make_local(index)
    return function()
        local name , value = debug.getlocal(4 , index)
        return name , "local", value
    end
end

local function make_upvalue(func, index)
    return function()
        local name , value = debug.getupvalue(func, index)
        return name , "upvalue", value
    end
end

local function make_global(env, name)
    return function()
        return name , "global" , rawget(env,name)
    end
end

local function gen_var(var_name, level)
    local i = 1
    while true do
        local name,v = debug.getlocal(5,i)
        if name == var_name then
            return make_local(i)
        end
        if name == nil then
            break
        end
        i=i+1
    end
    i = 1
    local f = debug.getinfo(5, "f").func
    while true do
        local name = debug.getupvalue(f,i)

        if name == var_name then
            return make_upvalue(f,i)
        end
        if name == nil then
            break
        end
        i=i+1
    end
    local name,env = debug.getupvalue(f,1)
    if name == '_ENV' then
        return make_global(env, var_name)
    end
end

local function dump_local(level)
    local i = 1
    while true do
        local name,v = debug.getlocal(level,i)
        if name == nil then
            break
        end
        if name == "(*temporary)" then
            print(name,v)
        end
        i=i+1
    end
end

local function gen_vars(var, call)
    local ret = {}
    for _,k in ipairs(var) do
        local f = gen_var(k, call)
        if f then
            table.insert(ret, f)
        end
    end
    return ret
end

local function hookline(var , call, line)
    print(info.filename, ":" , line)
    if info.var[line] == nil then
        info.var[line] = gen_vars(var, call)
    end

    for _,v in ipairs(info.var[line]) do
        local name , type , value = v()
        if info.last[name] ~= value then
            print(name , type, value)
            info.last[name] = value
        end
    end
end

local function hook(var , level)
    local call = 0
    local index = {}
    for w in string.gmatch(var, "%w+") do
        table.insert(index,w)
    end
    local function f (mode, line)
        if mode == 'return' then
            if call <= 0 then
                debug.sethook()
                trace.on = nil
                return
            end
            setname(3)
            call = call - 1
            if call == level then
                debug.sethook(f,'crl')
            end
        elseif mode == 'call' then
            setname(2)
            call = call + 1
            if call > level then
                debug.sethook(f,'cr')
            end
        elseif mode == 'line' then
            hookline(index , call, line)
        end
    end

    return f
end

local function up(level, f)
    local call = 0
    return function(mode)
        if mode == 'return' then
            call = call + 1
            if call == level then
                setname(3)
                debug.sethook(f,'crl')
            end
        elseif mode == 'call' then
            call = call - 1
        end
    end
end

function trace.trace(var , level)
    if trace.on then
        return
    end

    trace.on = true
    info.last = {}
    debug.sethook(up(2 , hook(var or  "" , level or 0)) , 'cr')
end

return trace

main.lua

 

trace = require "trace"

local function factorial(n)
    if n <= 1 then
        return 1
    end
    return factorial(n-1) * n
end

function foo(n)
    trace.trace("n s",n)
    local s =  factorial(100)
    return s
end

function hello()
    print "hello"
end

foo(3)
hello()
foo()