tiny.sql.Index

1pub const Index = struct {2    entries: tree.Tree,3    columns: []const row.Column,45    pub fn open(database: *file.Database, options: Options) Error!Index {6        return .{7            .entries = try tree.Tree.open(database, options.tree),8            .columns = options.columns,9        };10    }1112    pub fn reader(self: *const Index, snapshot: file.Snapshot) Error!Reader {13        return .{14            .entries = try self.entries.reader(snapshot),15            .columns = self.columns,16        };17    }1819    pub fn put(self: *Index, rowid: i64, values: []const row.Value, options: file.CommitOptions) Error!file.Commit {20        const phase = trace.scope("index.put");21        defer phase.end();2223        var write = try tree.Write.beginTree(&self.entries);24        defer write.deinit();25        try self.putPayloadIn(&write, rowid, values, "");26        return try write.commit(options);27    }2829    pub fn putIn(self: *Index, write: *tree.Write, rowid: i64, values: []const row.Value) Error!void {30        try self.putPayloadIn(write, rowid, values, "");31    }3233    pub fn putPayload(self: *Index, rowid: i64, values: []const row.Value, payload: []const u8, options: file.CommitOptions) Error!file.Commit {34        const phase = trace.scope("index.put_payload");35        defer phase.end();3637        var write = try tree.Write.beginTree(&self.entries);38        defer write.deinit();39        try self.putPayloadIn(&write, rowid, values, payload);40        return try write.commit(options);41    }4243    pub fn putPayloadIn(self: *Index, write: *tree.Write, rowid: i64, values: []const row.Value, payload: []const u8) Error!void {44        var key_bytes: [page.size]u8 = undefined;45        const encoded = key.encodeIndex(&key_bytes, values, self.columns, rowid) catch |err| switch (err) {46            error.OutputTooSmall => return error.KeyTooLarge,47            else => return err,48        };49        try write.put(&self.entries, encoded, payload);50    }5152    pub fn delete(self: *Index, rowid: i64, values: []const row.Value, options: file.CommitOptions) Error!file.Commit {53        const phase = trace.scope("index.delete");54        defer phase.end();5556        var write = try tree.Write.beginTree(&self.entries);57        defer write.deinit();58        try self.deleteIn(&write, rowid, values);59        return try write.commit(options);60    }6162    pub fn deleteIn(self: *Index, write: *tree.Write, rowid: i64, values: []const row.Value) Error!void {63        var key_bytes: [page.size]u8 = undefined;64        const encoded = key.encodeIndex(&key_bytes, values, self.columns, rowid) catch |err| switch (err) {65            error.OutputTooSmall => return error.KeyTooLarge,66            else => return err,67        };68        try write.delete(&self.entries, encoded);69    }7071    pub fn scan(72        self: *const Index,73        target: *Scan,74        allocator: Allocator,75        start: ?[]const row.Value,76        end: ?[]const row.Value,77    ) Error!void {78        var read = try self.entries.database.beginRead();79        defer read.deinit();80        const opened = try self.reader(read.snapshot());81        try opened.scan(target, allocator, start, end);82    }8384    pub fn range(85        self: *const Index,86        target: *Scan,87        allocator: Allocator,88        start: ?Bound,89        end: ?Bound,90    ) Error!void {91        var read = try self.entries.database.beginRead();92        defer read.deinit();93        const opened = try self.reader(read.snapshot());94        try opened.range(target, allocator, start, end);95    }9697    pub fn scanPayloads(98        self: *const Index,99        target: *Scan,100        allocator: Allocator,101        start: ?[]const row.Value,102        end: ?[]const row.Value,103    ) Error!void {104        var read = try self.entries.database.beginRead();105        defer read.deinit();106        const opened = try self.reader(read.snapshot());107        try opened.scanPayloads(target, allocator, start, end);108    }109110    pub fn lookup(111        self: *const Index,112        target: *Scan,113        allocator: Allocator,114        prefix: []const row.Value,115    ) Error!void {116        var read = try self.entries.database.beginRead();117        defer read.deinit();118        const opened = try self.reader(read.snapshot());119        try opened.lookup(target, allocator, prefix);120    }121122    pub fn lookupPayloads(123        self: *const Index,124        target: *Scan,125        allocator: Allocator,126        prefix: []const row.Value,127    ) Error!void {128        var read = try self.entries.database.beginRead();129        defer read.deinit();130        const opened = try self.reader(read.snapshot());131        try opened.lookupPayloads(target, allocator, prefix);132    }133134    pub fn summarize(self: *const Index) Error!tree.Summary {135        var read = try self.entries.database.beginRead();136        defer read.deinit();137        const opened = try self.reader(read.snapshot());138        return try opened.summarize();139    }140141    pub fn summarizeIn(self: *const Index, write: *const tree.Write) Error!tree.Summary {142        const phase = trace.scope("index.summarize_in");143        defer phase.end();144145        return try self.entries.summarizeIn(write);146    }147};

1pub const Index = index.Index;