tiny.sql.RelationSession

1pub const RelationSession = struct {2    allocator: Allocator,3    catalog: catalog_mod.Catalog,4    name: []u8,5    schema: catalog_mod.Schema,6    handle: catalog_mod.RelationHandle,7    stats: ?catalog_mod.RelationStats,8    root: version.RelationRoot,9    edits: std.ArrayList(relation_mod.Edit),1011    pub fn open(allocator: Allocator, catalog: *const catalog_mod.Catalog, name: []const u8) Error!RelationSession {12        const owned_name = try allocator.dupe(u8, name);13        errdefer allocator.free(owned_name);14        var state = try catalog.readRelation(allocator, name);15        errdefer state.deinit();16        var root = try version.relationRootMaintained(17            allocator,18            name,19            state.schema,20            &state.handle,21            state.relationStats(),22        );23        errdefer root.deinit();24        return .{25            .allocator = allocator,26            .catalog = catalog.*,27            .name = owned_name,28            .schema = state.schema,29            .handle = state.handle,30            .stats = state.stats,31            .root = root,32            .edits = .empty,33        };34    }3536    pub fn deinit(self: *RelationSession) void {37        self.clearEdits();38        self.edits.deinit(self.allocator);39        self.root.deinit();40        if (self.stats) |*stats| stats.deinit();41        self.handle.deinit();42        self.allocator.free(self.name);43        self.* = undefined;44    }4546    pub fn put(self: *RelationSession, rowid: i64, values: []const row.Value) Error!void {47        const phase = trace.scope("session.relation.put");48        defer phase.end();4950        try appendPutEdit(self.allocator, &self.edits, rowid, values);51    }5253    pub fn putEncoded(self: *RelationSession, rowid: i64, bytes: []const u8) Error!void {54        const phase = trace.scope("session.relation.put_encoded");55        defer phase.end();5657        _ = try row.View.init(bytes);58        const owned = try self.allocator.dupe(u8, bytes);59        errdefer self.allocator.free(owned);60        try self.edits.append(self.allocator, .{ .put = .{61            .rowid = rowid,62            .bytes = owned,63        } });64    }6566    pub fn update(self: *RelationSession, rowid: i64, assignments: []const relation_mod.Edit.Assignment) Error!void {67        const phase = trace.scope("session.relation.update");68        defer phase.end();6970        try appendUpdateEdit(self.allocator, &self.edits, rowid, assignments);71    }7273    pub fn delete(self: *RelationSession, rowid: i64) Error!void {74        const phase = trace.scope("session.relation.delete");75        defer phase.end();7677        try appendDeleteEdit(self.allocator, &self.edits, rowid);78    }7980    pub fn pendingEdits(self: *const RelationSession) usize {81        return self.edits.items.len;82    }8384    pub fn stagingLimits(self: *const RelationSession) Error!staging.Limits {85        const demand = try staging.Storage.demandForEdits(self.edits.items);86        return .{87            .relations = 1,88            .edits = demand.edits,89            .payload_bytes = demand.payload_bytes,90            .assignments = demand.assignments,91        };92    }9394    pub fn refresh(self: *RelationSession) Error!void {95        const phase = trace.scope("session.relation.refresh");96        defer phase.end();9798        var state = try self.catalog.readRelation(self.allocator, self.name);99        errdefer state.deinit();100        try state.handle.relation.validateIndexes(self.allocator);101        var root = try version.relationRootMaintained(102            self.allocator,103            self.name,104            state.schema,105            &state.handle,106            state.relationStats(),107        );108        errdefer root.deinit();109110        self.root.deinit();111        if (self.stats) |*old_stats| old_stats.deinit();112        self.handle.deinit();113        self.schema = state.schema;114        self.handle = state.handle;115        self.stats = state.stats;116        self.root = root;117    }118119    fn clearEdits(self: *RelationSession) void {120        self.discardEditsFrom(0);121        self.edits.clearRetainingCapacity();122    }123124    fn discardEditsFrom(self: *RelationSession, start: usize) void {125        for (self.edits.items[start..]) |edit| switch (edit) {126            .put => |put_edit| self.allocator.free(put_edit.bytes),127            .update => |update_edit| relation_mod.freeAssignments(self.allocator, update_edit.assignments),128            .delete => {},129        };130        self.edits.items.len = start;131    }132};

1pub const RelationSession = session.RelationSession;