nebula-client 0.7.3

Nebula Graph/Meta/Storage client
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119

use bytes::Bytes;
use fbthrift::{BinaryProtocol, Transport};
use nebula_fbthrift_meta_v1::{
    client::{MetaService, MetaServiceImpl},
    errors::meta_service::{GetSpaceError, ListPartsError, ListSpacesError, ListTagsError},
    types::{
        GetSpaceReq, GetSpaceResp, ListEdgesReq, ListEdgesResp, ListPartsReq, ListPartsResp,
        ListSpacesReq, ListSpacesResp, ListTagsReq, ListTagsResp,
    },
};

//
//
//
struct MetaConnection<T>
where
    T: Transport,
    Bytes: ::fbthrift::Framing<DecBuf = ::fbthrift::FramingDecoded<T>>,
    ::fbthrift::ProtocolEncoded<BinaryProtocol>:
        ::fbthrift::BufMutExt<Final = ::fbthrift::FramingEncodedFinal<T>>,
{
    service: MetaServiceImpl<BinaryProtocol, T>,
}

impl<T> MetaConnection<T>
where
    T: Transport,
    Bytes: ::fbthrift::Framing<DecBuf = ::fbthrift::FramingDecoded<T>>,
    ::fbthrift::ProtocolEncoded<BinaryProtocol>:
        ::fbthrift::BufMutExt<Final = ::fbthrift::FramingEncodedFinal<T>>,
{
    fn new(transport: T) -> Self {
        Self {
            service: MetaServiceImpl::<BinaryProtocol, _>::new(transport),
        }
    }
}

//
//
//
pub struct MetaClient<T>
where
    T: Transport,
    Bytes: ::fbthrift::Framing<DecBuf = ::fbthrift::FramingDecoded<T>>,
    ::fbthrift::ProtocolEncoded<BinaryProtocol>:
        ::fbthrift::BufMutExt<Final = ::fbthrift::FramingEncodedFinal<T>>,
{
    connection: MetaConnection<T>,
}

impl<T> MetaClient<T>
where
    T: Transport,
    Bytes: ::fbthrift::Framing<DecBuf = ::fbthrift::FramingDecoded<T>>,
    ::fbthrift::ProtocolEncoded<BinaryProtocol>:
        ::fbthrift::BufMutExt<Final = ::fbthrift::FramingEncodedFinal<T>>,
{
    pub fn new(transport: T) -> Self {
        Self {
            connection: MetaConnection::new(transport),
        }
    }

    pub async fn list_spaces(&self) -> Result<ListSpacesResp, ListSpacesError> {
        self.connection
            .service
            .listSpaces(&ListSpacesReq {
                ..Default::default()
            })
            .await
    }

    pub async fn get_space(&self, space_name: &str) -> Result<GetSpaceResp, GetSpaceError> {
        self.connection
            .service
            .getSpace(&GetSpaceReq {
                space_name: space_name.to_owned(),
                ..Default::default()
            })
            .await
    }

    // part_id from 1
    pub async fn list_parts(
        &self,
        space_id: i32,
        part_ids: Vec<i32>,
    ) -> Result<ListPartsResp, ListPartsError> {
        self.connection
            .service
            .listParts(&ListPartsReq {
                space_id,
                part_ids,
                ..Default::default()
            })
            .await
    }

    pub async fn list_tags(&self, space_id: i32) -> Result<ListTagsResp, ListTagsError> {
        self.connection
            .service
            .listTags(&ListTagsReq {
                space_id,
                ..Default::default()
            })
            .await
    }

    pub async fn list_edges(&self, space_id: i32) -> Result<ListEdgesResp, ListTagsError> {
        self.connection
            .service
            .listEdges(&ListEdgesReq {
                space_id,
                ..Default::default()
            })
            .await
    }
}