vecdb 0.12.2

High-performance mutable persistent vectors built on rawdb
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

use std::collections::BTreeSet;

use rawdb::Database;

use crate::{Bytes, Error, ImportOptions, ImportableVec, Version};

use super::{MutableRawVec, MutableVec};

impl<V> MutableVec<V>
where
    V: MutableRawVec,
{
    fn import_inner(options: ImportOptions) -> crate::Result<Self> {
        let holes = options
            .db
            .get_region(&Self::holes_region_name_with(options.name))
            .map(|region| {
                region
                    .create_reader()
                    .read_all()
                    .chunks(size_of::<usize>())
                    .map(usize::from_bytes)
                    .collect::<crate::Result<BTreeSet<usize>>>()
            })
            .transpose()?;
        let has_stored_holes = holes.is_some();
        Ok(Self::from_parts(
            V::import_with(options)?,
            holes.unwrap_or_default(),
            has_stored_holes,
        ))
    }
}

impl<V> ImportableVec for MutableVec<V>
where
    V: MutableRawVec,
{
    fn import(db: &Database, name: &str, version: Version) -> crate::Result<Self> {
        Self::import_with((db, name, version).into())
    }

    fn import_with(options: ImportOptions) -> crate::Result<Self> {
        Self::import_inner(options)
    }

    fn forced_import(db: &Database, name: &str, version: Version) -> crate::Result<Self> {
        Self::forced_import_with((db, name, version).into())
    }

    fn forced_import_with(options: ImportOptions) -> crate::Result<Self> {
        match Self::import_inner(options) {
            Err(Error::WrongEndian)
            | Err(Error::WrongLength { .. })
            | Err(Error::DifferentFormat { .. })
            | Err(Error::DifferentVersion { .. }) => {
                options
                    .db
                    .remove_region_if_exists(&Self::holes_region_name_with(options.name))?;
                Ok(Self::new(V::forced_import_with(options)?))
            }
            result => result,
        }
    }
}