Add support for attributes in MemorySegment

Author: lalinsky

src/FileSegment.zig

@@ -8,6 +8,7 @@ const KeepOrDelete = common.KeepOrDelete;
 
 const Item = @import("segment.zig").Item;
 const SegmentInfo = @import("segment.zig").SegmentInfo;
+const SegmentBase = @import("segment.zig").SegmentBase;
 
 const filefmt = @import("filefmt.zig");
 
@@ -20,6 +21,7 @@ pub const Options = struct {
 allocator: std.mem.Allocator,
 dir: std.fs.Dir,
 info: SegmentInfo = .{},
+attributes: std.StringHashMapUnmanaged([]const u8) = .{},
 docs: std.AutoHashMap(u32, bool),
 index: std.ArrayList(u32),
 block_size: usize = 0,
@@ -30,6 +32,8 @@ delete_in_deinit: bool = false,
 
 raw_data: ?[]align(std.mem.page_size) u8 = null,
 
+usingnamespace SegmentBase(Self);
+
 pub fn init(allocator: std.mem.Allocator, options: Options) Self {
     return Self{
         .allocator = allocator,
@@ -41,6 +45,7 @@ pub fn init(allocator: std.mem.Allocator, options: Options) Self {
 }
 
 pub fn deinit(self: *Self, delete_file: KeepOrDelete) void {
+    self.deinitAttributes(self.allocator);
     self.docs.deinit();
     self.index.deinit();
 

src/MemorySegment.zig

@@ -6,6 +6,7 @@ const SearchResults = common.SearchResults;
 const KeepOrDelete = common.KeepOrDelete;
 const SegmentInfo = @import("segment.zig").SegmentInfo;
 const Item = @import("segment.zig").Item;
+const SegmentBase = @import("segment.zig").SegmentBase;
 
 const Change = @import("change.zig").Change;
 
@@ -24,6 +25,8 @@ docs: std.AutoHashMap(u32, bool),
 items: std.ArrayList(Item),
 frozen: bool = false,
 
+usingnamespace SegmentBase(Self);
+
 pub fn init(allocator: std.mem.Allocator, opts: Options) Self {
     _ = opts;
     return .{
@@ -36,15 +39,7 @@ pub fn init(allocator: std.mem.Allocator, opts: Options) Self {
 pub fn deinit(self: *Self, delete_file: KeepOrDelete) void {
     _ = delete_file;
 
-    {
-        var iter = self.attributes.iterator();
-        while (iter.next()) |entry| {
-            self.allocator.free(entry.key_ptr.*);
-            self.allocator.free(entry.value_ptr.*);
-        }
-    }
-    self.attributes.deinit(self.allocator);
-
+    self.deinitAttributes(self.allocator);
     self.docs.deinit();
     self.items.deinit();
 }

src/segment.zig

@@ -97,3 +97,16 @@ test "Item array sort" {
         Item{ .hash = 2, .id = 200 },
     }, items);
 }
+
+pub fn SegmentBase(comptime T: type) type {
+    return struct {
+        pub fn deinitAttributes(self: *T, allocator: std.mem.Allocator) void {
+            var iter = self.attributes.iterator();
+            while (iter.next()) |entry| {
+                allocator.free(entry.key_ptr.*);
+                allocator.free(entry.value_ptr.*);
+            }
+            self.attributes.deinit(allocator);
+        }
+    };
+}