simple-triplestore 0.1.1

A simple graph database for storing triples with support for custom node and edge properties.
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

use serde::{de::DeserializeOwned, Serialize};
use sled::Transactional;

use crate::{
    prelude::*,
    traits::{IdType, Property},
    Triple,
};

use super::{SledTripleStore, SledTripleStoreError};

impl<
        Id: IdType,
        NodeProps: Property + Serialize + DeserializeOwned,
        EdgeProps: Property + Serialize + DeserializeOwned,
    > TripleStoreInsert<Id, NodeProps, EdgeProps> for SledTripleStore<Id, NodeProps, EdgeProps>
{
    fn insert_node(&mut self, node: Id, props: NodeProps) -> Result<(), SledTripleStoreError> {
        let key_bytes = &node.to_be_bytes();
        let data_bytes = bincode::serialize(&props)?;
        self.node_props.insert(key_bytes, data_bytes)?;
        Ok(())
    }

    fn insert_edge(
        &mut self,
        triple: Triple<Id>,
        props: EdgeProps,
    ) -> Result<(), SledTripleStoreError> {
        let prop_key = self.id_generator.fresh();
        let prop_key_bytes = prop_key.to_be_bytes();

        let data_bytes = bincode::serialize(&props)?;

        (
            &self.edge_props,
            &self.spo_data,
            &self.pos_data,
            &self.osp_data,
        )
            .transaction(move |(edge_props, spo_data, pos_data, osp_data)| {
                edge_props.insert(prop_key_bytes.as_ref(), data_bytes.as_slice())?;
                spo_data.insert(
                    Id::encode_spo_triple(&triple).as_ref(),
                    prop_key_bytes.as_ref(),
                )?;
                pos_data.insert(
                    Id::encode_pos_triple(&triple).as_ref(),
                    prop_key_bytes.as_ref(),
                )?;
                osp_data.insert(
                    Id::encode_osp_triple(&triple).as_ref(),
                    prop_key_bytes.as_ref(),
                )?;
                Ok(())
            })
            .map_err(|e| match e {
                sled::transaction::TransactionError::Abort(e) => SledTripleStoreError::SledError(e),
                sled::transaction::TransactionError::Storage(e) => {
                    SledTripleStoreError::SledError(e)
                }
            })?;

        Ok(())
    }
}

#[cfg(test)]
mod test {
    use crate::{SledTripleStore, UlidIdGenerator};

    #[test]
    fn test_insert_node() {
        let (_tempdir, db) = crate::sled::create_test_db().expect("ok");
        let sled_db = SledTripleStore::new(&db, UlidIdGenerator::new()).expect("ok");
        crate::conformance::insert::test_insert_node(sled_db);
    }

    #[test]
    fn test_insert_edge() {
        let (_tempdir, db) = crate::sled::create_test_db().expect("ok");
        let sled_db = SledTripleStore::new(&db, UlidIdGenerator::new()).expect("ok");
        crate::conformance::insert::test_insert_edge(sled_db);
    }
}