rust-nebula 0.0.2

A Rust client for NebulaGraph database.
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

use std::collections::{BTreeMap, HashMap};

use crate::nebula_fbthrift_meta_v3::{EdgeItem, TagItem};
use crate::common::types::HostAddr;

use super::client::MetaClientError;

pub struct SpaceCache {
    pub space_id: i32,
    pub space_name: Vec<u8>,
    pub tag_items: HashMap<Vec<u8>, TagItem>,
    pub edge_items: HashMap<Vec<u8>, EdgeItem>,
    pub parts_alloc: BTreeMap<i32, Vec<HostAddr>>,
}

pub struct MetaCache {
    pub space_caches: HashMap<Vec<u8>, SpaceCache>,
    pub space_id_names: HashMap<i32, Vec<u8>>,
    pub storage_addrs: Option<Vec<HostAddr>>,
    pub storage_leader: HashMap<Vec<u8>, HashMap<i32, HostAddr>>,
}

impl MetaCache {
    pub fn new() -> Self {
        MetaCache {
            space_caches: HashMap::new(),
            space_id_names: HashMap::new(),
            storage_addrs: None,
            storage_leader: HashMap::new(),
        }
    }

    pub(super) fn contains_space(&self, space_name: &Vec<u8>) -> bool {
        self.space_caches.contains_key(space_name)
    }

    pub(super) fn get_space_cache(
        &self,
        space_name: &Vec<u8>,
    ) -> Result<&SpaceCache, MetaClientError> {
        if !self.contains_space(space_name) {
            Err(MetaClientError::SpaceNotFoundError(space_name.to_vec()))
        } else {
            Ok(&self.space_caches[space_name])
        }
    }

    pub(super) fn contains_tag(&self, space_name: &Vec<u8>, tag_name: &Vec<u8>) -> bool {
        if !self.contains_space(space_name) {
            return false;
        }
        self.space_caches[space_name]
            .tag_items
            .contains_key(tag_name)
    }

    pub(super) fn get_tag_item(
        &self,
        space_name: &Vec<u8>,
        tag_name: &Vec<u8>,
    ) -> Result<&TagItem, MetaClientError> {
        if !self.contains_space(space_name) {
            Err(MetaClientError::SpaceNotFoundError(space_name.to_vec()))
        } else {
            if !self.space_caches[space_name]
                .tag_items
                .contains_key(tag_name)
            {
                Err(MetaClientError::TagNotFoundError(tag_name.to_vec()))
            } else {
                Ok(&self.space_caches[space_name].tag_items[tag_name])
            }
        }
    }

    pub(super) fn contains_edge(&self, space_name: &Vec<u8>, edge_name: &Vec<u8>) -> bool {
        if !self.contains_space(space_name) {
            return false;
        }
        self.space_caches[space_name]
            .edge_items
            .contains_key(edge_name)
    }

    pub(super) fn get_edge_item(
        &self,
        space_name: &Vec<u8>,
        edge_name: &Vec<u8>,
    ) -> Result<&EdgeItem, MetaClientError> {
        if !self.contains_space(space_name) {
            Err(MetaClientError::SpaceNotFoundError(space_name.to_vec()))
        } else {
            if !self.space_caches[space_name]
                .edge_items
                .contains_key(edge_name)
            {
                Err(MetaClientError::EdgeNotFoundError(edge_name.to_vec()))
            } else {
                Ok(&self.space_caches[space_name].edge_items[edge_name])
            }
        }
    }
}