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

feat(debug): Snacks.debug.trace and Snacks.debug.stats for hierarchical traces (like lazy profile)

Browse source
This commit is contained in:
Folke Lemaitre 2024-11-18 08:55:15 +01:00
parent 3f24e0a477
commit b593598859
No known key found for this signature in database
GPG key ID: 41F8B1FBACAE2040
1 changed files with 40 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

40
lua/snacks/debug.lua
View file

@ -93,4 +93,44 @@ function M.log(...)
fd:close()
end
---@alias snacks.debug.Trace {name: string, time: number, [number]:snacks.debug.Trace}
---@type snacks.debug.Trace[]
M._traces = { { name = "snacks" } }
---@param name string?
function M.trace(name)
if name then
local entry = { name = name, time = uv.hrtime() } ---@type snacks.debug.Trace
table.insert(M._traces[#M._traces], entry)
table.insert(M._traces, entry)
return entry
else
local entry = assert(table.remove(M._traces), "trace not ended?") ---@type snacks.debug.Trace
entry.time = uv.hrtime() - entry.time
return entry
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
end
end
end
collect(M._traces[1], 0)
Snacks.notify.warn(lines, { title = "Traces" })
end
return M