tiny.profiling.ingest.matching

1const std = @import("std");2const ingest = @import("root.zig");34pub const StringPart = enum(u8) {5    partial,6    terminal,7};89const Family = enum(u8) {10    kind,11    schema,12    event,13    summary,14};1516comptime {17    std.debug.assert(kind_matches.len <= @bitSizeOf(u32));18    std.debug.assert(kind_matches[0].name.len > 0);19}2021pub const Matcher = struct {22    index: usize,23    mask: u32,24    family: Family,2526    pub fn kinds() Matcher {27        const result = Matcher{ .index = 0, .mask = fullMask(kind_matches.len), .family = .kind };28        std.debug.assert(result.mask == fullMask(kind_matches.len));29        std.debug.assert(result.index == 0);30        return result;31    }3233    pub fn values(source_field: ingest.event.Field) Matcher {34        return switch (source_field) {35            .kind => kinds(),36            .schema => single(.schema),37            .event => single(.event),38            .summary => single(.summary),39            else => single(.summary),40        };41    }4243    pub fn feed(self: *Matcher, bytes: []const u8) void {44        std.debug.assert(bytes.len <= std.math.maxInt(usize) - self.index);45        const initial_index = self.index;46        const initial_mask = self.mask;47        for (bytes) |byte| {48            self.feedByte(byte);49            self.index += 1;50        }51        std.debug.assert(self.index == initial_index + bytes.len);52        std.debug.assert(self.mask & initial_mask == self.mask);53    }5455    pub fn kind(self: Matcher) ?ingest.event.EventKind {56        std.debug.assert(self.family == .kind);57        inline for (kind_matches, 0..) |entry, index| {58            if (self.matches(entry.name, index)) return entry.value;59        }60        return null;61    }6263    pub fn schemaMetric(self: Matcher) bool {64        std.debug.assert(self.family == .schema);65        return self.matches(metric_schema, 0);66    }6768    pub fn eventBenchEnd(self: Matcher) bool {69        std.debug.assert(self.family == .event);70        return self.matches(bench_end, 0);71    }7273    pub fn summaryMemory(self: Matcher) bool {74        std.debug.assert(self.family == .summary);75        return self.matches(memory, 0);76    }7778    fn feedByte(self: *Matcher, byte: u8) void {79        std.debug.assert(self.mask <= fullMask(kind_matches.len));80        std.debug.assert(self.index <= std.math.maxInt(usize) - 1);81        switch (self.family) {82            .kind => inline for (kind_matches, 0..) |entry, index| {83                if (self.index >= entry.name.len or entry.name[self.index] != byte) {84                    self.remove(index);85                }86            },87            .schema => if (self.index >= metric_schema.len or88                metric_schema[self.index] != byte) self.remove(0),89            .event => if (self.index >= bench_end.len or90                bench_end[self.index] != byte) self.remove(0),91            .summary => if (self.index >= memory.len or92                memory[self.index] != byte) self.remove(0),93        }94    }9596    fn matches(self: Matcher, name: []const u8, index: usize) bool {97        std.debug.assert(index < @bitSizeOf(u32));98        std.debug.assert(name.len > 0);99        return self.mask & (@as(u32, 1) << @intCast(index)) != 0 and self.index == name.len;100    }101102    fn remove(self: *Matcher, index: usize) void {103        std.debug.assert(index < @bitSizeOf(u32));104        self.mask &= ~(@as(u32, 1) << @intCast(index));105    }106107    fn single(family: Family) Matcher {108        std.debug.assert(family != .kind);109        return .{ .index = 0, .mask = 1, .family = family };110    }111};112113pub fn feedToken(matcher: *Matcher, token: std.json.Token) ?StringPart {114    switch (matcher.family) {115        .kind => std.debug.assert(matcher.family == .kind),116        .schema => std.debug.assert(matcher.family == .schema),117        .event => std.debug.assert(matcher.family == .event),118        .summary => std.debug.assert(matcher.family == .summary),119    }120    switch (token) {121        .partial_string => |bytes| matcher.feed(bytes),122        .partial_string_escaped_1 => |bytes| matcher.feed(&bytes),123        .partial_string_escaped_2 => |bytes| matcher.feed(&bytes),124        .partial_string_escaped_3 => |bytes| matcher.feed(&bytes),125        .partial_string_escaped_4 => |bytes| matcher.feed(&bytes),126        .string => |bytes| {127            matcher.feed(bytes);128            return .terminal;129        },130        else => return null,131    }132    return .partial;133}134135fn fullMask(count: usize) u32 {136    std.debug.assert(count > 0);137    std.debug.assert(count < @bitSizeOf(u32));138    return (@as(u32, 1) << @intCast(count)) - 1;139}140141const KindMatch = struct {142    name: []const u8,143    value: ingest.event.EventKind,144};145const kind_matches = [_]KindMatch{146    .{ .name = "alloc", .value = .alloc },147    .{ .name = "free", .value = .free },148    .{ .name = "resize", .value = .resize },149    .{ .name = "remap", .value = .remap },150    .{ .name = "allocator", .value = .allocator },151    .{ .name = "summary", .value = .summary },152};153const metric_schema = "tiny.profiling.metric/v1";154const bench_end = "bench_end";155const memory = "memory";

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