tiny.coz.Registry

1pub const Registry = struct {2    mutex: std.atomic.Mutex = .unlocked,3    throughput_points: std.StringHashMapUnmanaged(*progress_point.ThroughputPoint) = .empty,4    latency_points: std.StringHashMapUnmanaged(*progress_point.LatencyPoint) = .empty,56    pub fn deinit(self: *Registry, allocator: std.mem.Allocator) void {7        lockMutex(&self.mutex);8        defer self.mutex.unlock();910        var throughput_iter = self.throughput_points.iterator();11        while (throughput_iter.next()) |entry| {12            allocator.free(entry.key_ptr.*);13            allocator.destroy(entry.value_ptr.*);14        }15        self.throughput_points.deinit(allocator);1617        var latency_iter = self.latency_points.iterator();18        while (latency_iter.next()) |entry| {19            allocator.free(entry.key_ptr.*);20            allocator.destroy(entry.value_ptr.*);21        }22        self.latency_points.deinit(allocator);2324        self.throughput_points = .empty;25        self.latency_points = .empty;26    }2728    pub fn getThroughputPoint(29        self: *Registry,30        allocator: std.mem.Allocator,31        name: []const u8,32    ) !*progress_point.ThroughputPoint {33        lockMutex(&self.mutex);34        defer self.mutex.unlock();3536        if (self.throughput_points.get(name)) |point| return point;3738        const owned_name = try allocator.dupe(u8, name);39        errdefer allocator.free(owned_name);4041        const point = try allocator.create(progress_point.ThroughputPoint);42        errdefer allocator.destroy(point);43        point.* = progress_point.ThroughputPoint.init(owned_name);4445        try self.throughput_points.put(allocator, owned_name, point);46        return point;47    }4849    pub fn getLatencyPoint(50        self: *Registry,51        allocator: std.mem.Allocator,52        name: []const u8,53    ) !*progress_point.LatencyPoint {54        lockMutex(&self.mutex);55        defer self.mutex.unlock();5657        if (self.latency_points.get(name)) |point| return point;5859        const owned_name = try allocator.dupe(u8, name);60        errdefer allocator.free(owned_name);6162        const point = try allocator.create(progress_point.LatencyPoint);63        errdefer allocator.destroy(point);64        point.* = progress_point.LatencyPoint.init(owned_name);6566        try self.latency_points.put(allocator, owned_name, point);67        return point;68    }6970    pub fn getCounter(71        self: *Registry,72        allocator: std.mem.Allocator,73        kind: abi.CounterKind,74        name: []const u8,75    ) !*abi.Counter {76        return switch (kind) {77            .throughput => (try self.getThroughputPoint(allocator, name)).counterStruct(),78            .begin => (try self.getLatencyPoint(allocator, name)).beginCounterStruct(),79            .end => (try self.getLatencyPoint(allocator, name)).endCounterStruct(),80        };81    }8283    pub fn saveThroughputSnapshots(84        self: *Registry,85        allocator: std.mem.Allocator,86    ) ![]progress_point.ThroughputSnapshot {87        lockMutex(&self.mutex);88        defer self.mutex.unlock();8990        var snapshots: std.ArrayListUnmanaged(progress_point.ThroughputSnapshot) = .empty;91        errdefer snapshots.deinit(allocator);9293        var iter = self.throughput_points.valueIterator();94        while (iter.next()) |point| {95            try snapshots.append(allocator, point.*.save());96        }9798        return snapshots.toOwnedSlice(allocator);99    }100101    pub fn saveLatencySnapshots(102        self: *Registry,103        allocator: std.mem.Allocator,104    ) ![]progress_point.LatencySnapshot {105        lockMutex(&self.mutex);106        defer self.mutex.unlock();107108        var snapshots: std.ArrayListUnmanaged(progress_point.LatencySnapshot) = .empty;109        errdefer snapshots.deinit(allocator);110111        var iter = self.latency_points.valueIterator();112        while (iter.next()) |point| {113            try snapshots.append(allocator, point.*.save());114        }115116        return snapshots.toOwnedSlice(allocator);117    }118};

1pub const Registry = registry.Registry;