4 releases

0.4.1 Jan 5, 2025
0.4.0 Nov 26, 2024
0.3.10 Sep 28, 2024
0.3.9 Sep 6, 2024

#552 in Database interfaces

Download history 191/week @ 2024-09-24 36/week @ 2024-10-01 1/week @ 2024-10-08 145/week @ 2024-11-26 18/week @ 2024-12-03 13/week @ 2024-12-10 109/week @ 2024-12-31 30/week @ 2025-01-07

139 downloads per month

MIT license

60KB
1.5K SLoC

Absolutely not a Database (WorkTable)

WorkTable is in-memory (on-disk persistence is in progress currently) storage.

Usage

WorkTable can be used just in user's code with worktable! macro. It will generate table struct and other related structs that will be used for table logic.

worktable!(
    name: Test,
    columns: {
        id: u64 primary_key autoincrement,
        test: i64,
        another: u64 optional,
        exchange: String
    },
    indexes: {
        test_idx: test unique,
        exchnage_idx: exchange,
    }
    queries: {
        update: {
            AnotherByExchange(another) by exchange,
            AnotherByTest(another) by test,
            AnotherById(another) by id,
        },
        delete: {
            ByAnother() by another,
            ByExchange() by exchange,
            ByTest() by test,
        }
    }
);

Declaration parts

name declaration

name field is used to define table's name, and is a prefix for generated objects. For example declaration above will generate struct TestWorkTable, so table struct will always have name as <name>WorkTable.

let table = TestWorkTable::default ();
let name = table.name();
assert_eq!(name, "Test");

columns declaration

columns field is used to define table's row schema. Default usage is <column_name>: <type>. But also there are some flags that can be applied to columns as <column_name>: <type> <flags>*.

Flags list:

  • primary_key flag and related to it.
  • optional flag.

primary_key flag declaration

If user want to mark column as primary key primary_key flag is used. This flag can be used on multiple columns at a time. Primary key generation is also supported. For some basic types autoincrement is supported. Also custom generation is available. In this case user must provide his own implementation.

#[derive(
    Archive,
    Debug,
    Default,
    Deserialize,
    Clone,
    Eq,
    From,
    PartialOrd,
    PartialEq,
    Ord,
    Serialize,
    SizeMeasure,
)]
#[rkyv(compare(PartialEq), derive(Debug))]
struct CustomId(u64);

#[derive(Debug, Default)]
pub struct Generator(AtomicU64);

impl PrimaryKeyGenerator<TestPrimaryKey> for Generator {
    fn next(&self) -> TestPrimaryKey {
        let res = self.0.fetch_add(1, Ordering::Relaxed);
        if res >= 10 {
            self.0.store(0, Ordering::Relaxed);
        }
        CustomId::from(res).into()
    }
}

impl TablePrimaryKey for TestPrimaryKey {
    type Generator = Generator;
}

worktable!(
  name: Test,
  columns: {
    id: CustomId primary_key custom,
    test: u64
  }
);

For primary key newtype is generated for declared type:

// Generated code
#[derive(
    Clone,
    rkyv::Archive,
    Debug,
    rkyv::Deserialize,
    rkyv::Serialize,
    From,
    Eq,
    Into,
    PartialEq,
    PartialOrd,
    Ord
)]
pub struct TestPrimaryKey(u64);

optional flag declaration

If column field is Option<T>, optional flag can be used like it was done in declaration.

another: u64 optional,

Row type generation

For described column row type struct is generated:

// Generated code
#[derive(
    rkyv::Archive,
    Debug,
    rkyv::Deserialize,
    Clone,
    rkyv::Serialize,
    PartialEq
)]
#[rkyv(derive(Debug))]
#[repr(C)]
pub struct TestRow {
    pub id: u64,
    pub test: i64,
    pub another: Option<u64>,
    pub exchange: String,
}

This struct is used in WorkTable interface and will be used by users.

indexes declaration

indexes field is used to define table's index schema. Default usage is <index_name>: <column_name> <unique>?.

Index allows faster access to data by some field. Adding indexes field adds methods to generated WorkTable. This method for now is select_by_<indexed_column_name>. It will be described below.

Default implemented queries

There are some default query implementations that are available for all WorkTable's:

  • select(&self, pk: <Name>PrimaryKey) -> Option<<Name>Row>;
  • insert(&self, row: <Name>Row) -> Result<<Name>PrimaryKey, WorkTableError>;
  • upsert(&self, row: <Name>Row) -> Result<(), WorkTableError>;
  • update(&self, row: <Name>Row) -> Result<(), WorkTableError>;
  • delete(&self, pk: <Name>PrimaryKey) -> Result<(), WorkTableError>;
  • select_all<'a>(&'a self) -> SelectQueryBuilder<'a, <Name>Row, Self>;

queries declaration

indexes field is used to define table's queries schema. Queries are used to update/select/delete data.

queries: {
    update: {
        AnotherByExchange(another) by exchange,
        AnotherByTest(another) by test,
        AnotherById(another) by id,
    },
    delete: {
        ByAnother() by another,
        ByExchange() by exchange,
        ByTest() by test,
    }
}

Default query declaration is <QueryName>(<column_name>*) by <column_name>. It is same for update/select/delete.

For each query <QueryName>Query and <QueryName>By structs are generated. They will be used by user to call the query.

update query declaration

update queries are used to update row's data partially. Default generated update allows only full update of the row. But if user's logic needs some simultaneous update of row parts from different code parts. update logic supports smart lock logic that allows simultaneous update of not overlapping row fields.

WorkTable internals structure

TODO

Dependencies

~10–17MB
~231K SLoC