icydb-core 0.253.3

IcyDB — A schema-first typed query engine and persistence runtime for Internet Computer canisters
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

//! Module: db::registry::readers
//! Responsibility: structural row and index reader adapters for registered stores.
//! Does not own: executor scan policy or index-entry decode semantics.
//! Boundary: lets StoreHandle satisfy structural preflight reader traits.

use crate::{
    db::{
        data::{DecodedDataStoreKey, RawRow},
        direction::Direction,
        index::{
            IndexEntryValue, IndexStore, RawIndexStoreKey, StructuralIndexEntryReader,
            StructuralPrimaryRowReader, push_structural_index_entry_primary_key_values_limited,
        },
        key_taxonomy::PrimaryKeyValue,
        registry::StoreHandle,
    },
    error::InternalError,
};
use std::{cell::RefCell, ops::Bound, thread::LocalKey};

impl StructuralPrimaryRowReader for StoreHandle {
    fn read_primary_row(&self, key: &DecodedDataStoreKey) -> Result<Option<RawRow>, InternalError> {
        let raw_key = key.to_raw()?;

        Ok(self.with_data(|store| store.get(&raw_key)))
    }
}

impl StructuralIndexEntryReader for StoreHandle {
    fn read_index_entry(
        &self,
        index_store: &'static LocalKey<RefCell<IndexStore>>,
        key: &RawIndexStoreKey,
    ) -> Result<Option<IndexEntryValue>, InternalError> {
        Ok(index_store.with_borrow(|store| store.get(key)))
    }

    fn read_index_keys_in_raw_range(
        &self,
        index_store: &'static LocalKey<RefCell<IndexStore>>,
        bounds: (&Bound<RawIndexStoreKey>, &Bound<RawIndexStoreKey>),
        limit: usize,
    ) -> Result<Vec<PrimaryKeyValue>, InternalError> {
        let mut out = Vec::with_capacity(limit.min(32));
        index_store.with_borrow(|store| {
            store.visit_raw_entries_in_range(bounds, Direction::Asc, |raw_key, raw_entry| {
                push_structural_index_entry_primary_key_values_limited(
                    raw_key, raw_entry, &mut out, limit,
                )
            })
        })?;

        Ok(out)
    }
}