tiny.tldr.trace

1pub const trace = @import("trace.zig");

1const std = @import("std");2const coz = @import("coz");3const model = @import("model.zig");45const stage_count = @typeInfo(model.ProductStage).@"enum".field_names.len;67pub const PhaseTimers = struct {8    totals_ns: [stage_count]u64 = @as([stage_count]u64, @splat(0)),9    calls: [stage_count]u64 = @as([stage_count]u64, @splat(0)),1011    pub fn record(self: *PhaseTimers, stage: model.ProductStage, elapsed_ns: u64) void {12        const index = @backingInt(stage);13        _ = @atomicRmw(u64, &self.totals_ns[index], .Add, elapsed_ns, .monotonic);14        _ = @atomicRmw(u64, &self.calls[index], .Add, 1, .monotonic);15    }1617    pub fn totalNs(self: *const PhaseTimers, stage: model.ProductStage) u64 {18        return self.totals_ns[@backingInt(stage)];19    }2021    pub fn callCount(self: *const PhaseTimers, stage: model.ProductStage) u64 {22        return self.calls[@backingInt(stage)];23    }2425    pub fn sumNs(self: *const PhaseTimers) u64 {26        var sum: u64 = 0;27        for (self.totals_ns) |value| sum += value;28        return sum;29    }30};3132pub const Clock = *const fn () i128;3334var active_timers: ?*PhaseTimers = null;35var active_clock: ?Clock = null;3637pub fn beginPhaseTiming(timers: *PhaseTimers, clock: Clock) void {38    active_timers = timers;39    active_clock = clock;40}4142pub fn endPhaseTiming() void {43    active_timers = null;44    active_clock = null;45}4647pub inline fn scope(comptime name: []const u8) coz.Scope("tldr." ++ name) {48    return coz.scope("tldr." ++ name);49}5051pub inline fn product(comptime stage: model.ProductStage) ProductScope(stage) {52    return ProductScope(stage).begin();53}5455fn ProductScope(comptime stage: model.ProductStage) type {56    const trace_name = "tldr." ++ stage.traceName();57    const CozScope = coz.Scope(trace_name);58    return struct {59        coz_scope: CozScope,60        start_ns: i128,6162        const Self = @This();6364        inline fn begin() Self {65            return .{66                .coz_scope = coz.scope(trace_name),67                .start_ns = if (active_clock) |clock| clock() else 0,68            };69        }7071        pub inline fn end(self: Self) void {72            if (active_timers) |timers| {73                if (active_clock) |clock| {74                    const elapsed = clock() - self.start_ns;75                    timers.record(stage, if (elapsed < 0) 0 else @intCast(elapsed));76                }77            }78            self.coz_scope.end();79        }80    };81}8283test "trace phase markers compile" {84    const phase = scope("test");85    defer phase.end();86    const product_phase = product(.manifest_recording);87    defer product_phase.end();88}8990fn zeroClock() i128 {91    return 0;92}9394test "phase timing accumulates while active" {95    var timers = PhaseTimers{};96    beginPhaseTiming(&timers, zeroClock);97    {98        const phase = product(.output_writing);99        defer phase.end();100    }101    endPhaseTiming();102    try std.testing.expectEqual(@as(u64, 1), timers.callCount(.output_writing));103    try std.testing.expectEqual(@as(u64, 0), timers.callCount(.input_discovery));104}