mirrors/snacks.nvim
1
0
Fork 0
mirror of https://github.com/folke/snacks.nvim synced 2025-08-04 10:49:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

refactor: move all dashboard tracing to debug

Browse source
This commit is contained in:
Folke Lemaitre 2024-11-19 14:55:04 +01:00
parent 9c0d2e2e93
commit 143c244346
No known key found for this signature in database
GPG key ID: 41F8B1FBACAE2040
2 changed files with 79 additions and 55 deletions
Download patch file Download diff file Expand all files Collapse all files

80
lua/snacks/debug.lua
View file

@ -94,9 +94,10 @@ function M.log(...)
end
---@alias snacks.debug.Trace {name: string, time: number, [number]:snacks.debug.Trace}
---@alias snacks.debug.Stat {name:string, time:number, count?:number, depth?:number}
---@type snacks.debug.Trace[]
M._traces = { { name = "snacks" } }
M._traces = { { name = "__TOP__", time = 0 } }
---@param name string?
function M.trace(name)
@ -112,25 +113,70 @@ function M.trace(name)
end
end
---@param opts? {min?: number}
function M.stats(opts)
opts = opts or {}
local lines = {}
---@param stat snacks.debug.Trace
---@param level number
local function collect(stat, level)
for _, entry in ipairs(stat) do
local ms = math.floor(entry.time / 1e4) / 1e2
if ms >= (opts.min or 0) then
table.insert(lines, ("%s- `%s`: **%.2f**ms"):format((" "):rep(level), entry.name, entry.time / 1e6))
if entry[1] then
collect(entry, level + 1)
end
---@param modname string
---@param mod? table
---@param suffix? string
function M.tracemod(modname, mod, suffix)
mod = mod or require(modname)
suffix = suffix or "."
for k, v in pairs(mod) do
if type(v) == "function" and k ~= "trace" then
mod[k] = function(...)
M.trace(modname .. suffix .. k)
local ok, ret = pcall(v, ...)
M.trace()
return ok == false and error(ret) or ret
end
end
end
collect(M._traces[1], 0)
Snacks.notify.warn(lines, { title = "Traces" })
end
---@param opts? {min?: number, show?:boolean}
---@return {summary:table<string, snacks.debug.Stat>, trace:snacks.debug.Stat[], traces:snacks.debug.Trace[]}
function M.stats(opts)
opts = opts or {}
local stack, lines, trace = {}, {}, {} ---@type string[], string[], snacks.debug.Stat[]
local summary = {} ---@type table<string, snacks.debug.Stat>
---@param stat snacks.debug.Trace
local function collect(stat)
if #stack > 0 then
local recursive = vim.list_contains(stack, stat.name)
summary[stat.name] = summary[stat.name] or { time = 0, count = 0, name = stat.name }
summary[stat.name].time = summary[stat.name].time + (recursive and 0 or stat.time)
summary[stat.name].count = summary[stat.name].count + 1
table.insert(trace, { name = stat.name, time = stat.time or 0, depth = #stack - 1 })
end
table.insert(stack, stat.name)
for _, entry in ipairs(stat) do
collect(entry)
end
table.remove(stack)
end
collect(M._traces[1])
---@param entries snacks.debug.Stat[]
local function add(entries)
for _, stat in ipairs(entries) do
local ms = math.floor(stat.time / 1e4) / 1e2
if ms >= (opts.min or 0) then
local line = ("%s- `%s`: **%.2f**ms"):format((" "):rep(stat.depth or 0), stat.name, ms)
table.insert(lines, line .. (stat.count and (" ([%d])"):format(stat.count) or ""))
end
end
end
if opts.show ~= false then
lines[#lines + 1] = "# Summary"
summary = vim.tbl_values(summary)
table.sort(summary, function(a, b)
return a.time > b.time
end)
add(summary)
lines[#lines + 1] = "\n# Trace"
add(trace)
Snacks.notify.warn(lines, { title = "Traces" })
end
return { summary = summary, trace = trace, tree = M._traces }
end
return M