quill-sql 0.2.0

A tiny yet serious SQL database in Rust with ARIES-style WAL, 2PL, and B+Tree indexes.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

use std::sync::Arc;

use crate::{
    catalog::Catalog,
    error::QuillSQLResult,
    storage::{index::btree_index::BPlusTreeIndex, table_heap::TableHeap, tuple::Tuple},
    transaction::{CommandId, TransactionId},
    utils::table_ref::TableReference,
};

pub trait StorageEngine: Send + Sync {
    fn table_heap(
        &self,
        catalog: &Catalog,
        table: &TableReference,
    ) -> QuillSQLResult<Arc<TableHeap>>;

    fn table_indexes(
        &self,
        catalog: &Catalog,
        table: &TableReference,
    ) -> QuillSQLResult<Vec<Arc<BPlusTreeIndex>>>;

    fn mvcc_insert(
        &self,
        catalog: &Catalog,
        table: &TableReference,
        tuple: &Tuple,
        txn_id: TransactionId,
        cid: CommandId,
    ) -> QuillSQLResult<(Arc<TableHeap>, crate::storage::page::RecordId)>;

    fn mvcc_update(
        &self,
        heap: &Arc<TableHeap>,
        rid: crate::storage::page::RecordId,
        new_tuple: Tuple,
        txn_id: TransactionId,
        cid: CommandId,
    ) -> QuillSQLResult<(
        crate::storage::page::RecordId,
        crate::storage::page::TupleMeta,
    )>;

    fn mvcc_delete(
        &self,
        heap: &Arc<TableHeap>,
        rid: crate::storage::page::RecordId,
        txn_id: TransactionId,
        cid: CommandId,
    ) -> QuillSQLResult<crate::storage::page::TupleMeta>;
}

#[derive(Default)]
pub struct DefaultStorageEngine;

impl StorageEngine for DefaultStorageEngine {
    fn table_heap(
        &self,
        catalog: &Catalog,
        table: &TableReference,
    ) -> QuillSQLResult<Arc<TableHeap>> {
        catalog.table_heap(table)
    }

    fn table_indexes(
        &self,
        catalog: &Catalog,
        table: &TableReference,
    ) -> QuillSQLResult<Vec<Arc<BPlusTreeIndex>>> {
        catalog.table_indexes(table)
    }

    fn mvcc_insert(
        &self,
        catalog: &Catalog,
        table: &TableReference,
        tuple: &Tuple,
        txn_id: crate::transaction::TransactionId,
        cid: CommandId,
    ) -> QuillSQLResult<(Arc<TableHeap>, crate::storage::page::RecordId)> {
        let heap = catalog.table_heap(table)?;
        let (rid, _) = heap.mvcc_insert_version(tuple, txn_id, cid, None)?;
        Ok((heap, rid))
    }

    fn mvcc_update(
        &self,
        heap: &Arc<TableHeap>,
        rid: crate::storage::page::RecordId,
        new_tuple: Tuple,
        txn_id: crate::transaction::TransactionId,
        cid: CommandId,
    ) -> QuillSQLResult<(
        crate::storage::page::RecordId,
        crate::storage::page::TupleMeta,
    )> {
        heap.mvcc_update(rid, new_tuple, txn_id, cid)
    }

    fn mvcc_delete(
        &self,
        heap: &Arc<TableHeap>,
        rid: crate::storage::page::RecordId,
        txn_id: crate::transaction::TransactionId,
        cid: CommandId,
    ) -> QuillSQLResult<crate::storage::page::TupleMeta> {
        heap.mvcc_mark_deleted(rid, txn_id, cid)
    }
}