Add segment count to GET /:index

Author: lalinsky

src/IndexReader.zig

@@ -68,6 +68,10 @@ pub fn getVersion(self: *Self) u64 {
     return 0;
 }
 
+pub fn getSegmentCount(self: *Self) usize {
+    return self.memory_segments.value.count() + self.file_segments.value.count();
+}
+
 pub fn getAttributes(self: *Self, allocator: std.mem.Allocator) !std.StringHashMapUnmanaged(u64) {
     var attributes: std.StringHashMapUnmanaged(u64) = .{};
     errdefer attributes.deinit(allocator);

src/segment_list.zig

@@ -149,6 +149,10 @@ pub fn SegmentList(Segment: type) type {
             return false;
         }
 
+        pub fn count(self: Self) usize {
+            return self.nodes.items.len;
+        }
+
         pub fn getFirst(self: Self) ?Node {
             return if (self.nodes.items.len > 0) self.nodes.items[0] else null;
         }

src/server.zig

@@ -412,6 +412,7 @@ const Attributes = struct {
 
 const GetIndexResponse = struct {
     version: u64,
+    segments: usize,
     attributes: Attributes,
 
     pub fn msgpackFormat() msgpack.StructFormat {
@@ -428,6 +429,7 @@ fn handleGetIndex(ctx: *Context, req: *httpz.Request, res: *httpz.Response) !voi
 
     const response = GetIndexResponse{
         .version = index_reader.getVersion(),
+        .segments = index_reader.getSegmentCount(),
         .attributes = .{
             .attributes = try index_reader.getAttributes(req.arena),
         },

tests/test_index_api.py

@@ -33,9 +33,9 @@ def test_get_index(client, index_name, create_index, fmt):
     req = client.get(f'/{index_name}', headers=headers(fmt))
     assert req.status_code == 200, req.content
     if fmt == 'json':
-        expected = {'version': 1, 'attributes': {'foo': 1234}}
+        expected = {'version': 1, 'segments': 1, 'attributes': {'foo': 1234}}
     else:
-        expected = {'v': 1, 'a': {'foo': 1234}}
+        expected = {'v': 1, 's': 1, 'a': {'foo': 1234}}
     assert decode(fmt, req.content) == expected