atomo 0.0.5

reserved for atomic snapshots
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

use crate::{batch::BoxedVec, db::TableId};

pub type ImKeyCollection = im::HashSet<BoxedVec, fxhash::FxBuildHasher>;

pub type MaybeImKeyCollection = Option<ImKeyCollection>;

/// Analogues to the [`VerticalBatch`] this data structure is responsible for
/// holding an immutable data containing all of the keys for each table, given
/// that the iterator functionality is enabled for the table.
#[derive(Default, Clone)]
pub struct VerticalKeys(Vec<MaybeImKeyCollection>);

impl VerticalKeys {
    /// Enable the indexer on the given table index.
    pub fn enable(&mut self, index: usize) {
        if index < self.0.len() {
            self.0[index] = Some(ImKeyCollection::default());
        } else {
            self.0.resize(index, None);
            self.0.push(Some(ImKeyCollection::default()));
        }
    }

    #[inline(always)]
    pub fn update<F>(&mut self, tid: TableId, closure: F)
    where
        F: FnOnce(&mut ImKeyCollection),
    {
        let tid = tid as usize;
        if tid < self.0.len() {
            if let Some(data) = &mut self.0[tid] {
                closure(data);
            }
        }
    }

    #[inline(always)]
    pub fn get(&self, tid: TableId) -> &MaybeImKeyCollection {
        let tid = tid as usize;
        if tid >= self.0.len() {
            return &None;
        }
        &self.0[tid]
    }
}