raphtory-storage 0.17.0

Raphtory storage interface
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

use super::node_ref::NodeStorageRef;
use crate::graph::nodes::nodes_ref::NodesStorageEntry;
use raphtory_api::core::entities::VID;
use raphtory_core::storage::ReadLockedStorage;
use std::sync::Arc;

#[cfg(feature = "storage")]
use crate::disk::storage_interface::nodes::DiskNodesOwned;

pub enum NodesStorage {
    Mem(Arc<ReadLockedStorage>),
    #[cfg(feature = "storage")]
    Disk(DiskNodesOwned),
}

impl NodesStorage {
    #[inline]
    pub fn as_ref(&self) -> NodesStorageEntry<'_> {
        match self {
            NodesStorage::Mem(storage) => NodesStorageEntry::Mem(storage),
            #[cfg(feature = "storage")]
            NodesStorage::Disk(storage) => NodesStorageEntry::Disk(storage.as_ref()),
        }
    }

    #[inline]
    pub fn node_entry(&self, vid: VID) -> NodeStorageRef<'_> {
        match self {
            NodesStorage::Mem(storage) => NodeStorageRef::Mem(storage.get_entry(vid)),
            #[cfg(feature = "storage")]
            NodesStorage::Disk(storage) => NodeStorageRef::Disk(storage.node(vid)),
        }
    }

    pub fn len(&self) -> usize {
        self.as_ref().len()
    }
}