ydb 0.17.1

Crate contains generated low-level grpc code from YDB API protobuf, used as base for ydb crate
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

use crate::client::TimeoutSettings;
use crate::client_scheme::list_types::SchemeEntry;
use crate::grpc_connection_manager::GrpcConnectionManager;
use crate::grpc_wrapper::raw_scheme_client::client::{
    RawMakeDirectoryRequest, RawRemoveDirectoryRequest,
};
use crate::grpc_wrapper::raw_scheme_client::describe_path_types::RawDescribePathRequest;
use crate::grpc_wrapper::raw_scheme_client::list_directory_types::RawListDirectoryRequest;

use crate::{YdbResult, grpc_wrapper};
use tracing::instrument;

#[derive(Clone)]
pub struct SchemeClient {
    timeouts: TimeoutSettings,
    connection_manager: GrpcConnectionManager,
}

impl SchemeClient {
    pub(crate) fn new(connection_manager: GrpcConnectionManager) -> Self {
        Self {
            timeouts: TimeoutSettings::default(),
            connection_manager,
        }
    }

    #[instrument(name = "ydb.SchemeClient.MakeDirectory", skip_all, fields(db.system.name = "ydb", ydb.path = %path))]
    pub async fn make_directory(&mut self, path: String) -> YdbResult<()> {
        let req = RawMakeDirectoryRequest {
            operation_params: self.timeouts.operation_params(),
            path,
        };
        let mut service = self.connection().await?;
        service.make_directory(req).await?;
        Ok(())
    }

    #[instrument(name = "ydb.SchemeClient.DescribePath", skip_all, fields(db.system.name = "ydb", ydb.path = %path))]
    pub async fn describe_path(&mut self, path: String) -> YdbResult<SchemeEntry> {
        let req = RawDescribePathRequest {
            operation_params: self.timeouts.operation_params(),
            path,
        };

        let mut service = self.connection().await?;
        let res = service.describe_path(req).await?;

        Ok(res.entry)
    }

    #[instrument(name = "ydb.SchemeClient.ListDirectory", skip_all, fields(db.system.name = "ydb", ydb.path = %path))]
    pub async fn list_directory(&mut self, path: String) -> YdbResult<Vec<SchemeEntry>> {
        let req = RawListDirectoryRequest {
            operation_params: self.timeouts.operation_params(),
            path,
        };

        let mut service = self.connection().await?;
        let res = service.list_directory(req).await?;

        Ok(res.children.into_iter().collect())
    }

    #[instrument(name = "ydb.SchemeClient.RemoveDirectory", skip_all, fields(db.system.name = "ydb", ydb.path = %path))]
    pub async fn remove_directory(&mut self, path: String) -> YdbResult<()> {
        let req = RawRemoveDirectoryRequest {
            operation_params: self.timeouts.operation_params(),
            path,
        };
        let mut service = self.connection().await?;
        service.remove_directory(req).await?;
        Ok(())
    }

    async fn connection(
        &self,
    ) -> YdbResult<grpc_wrapper::raw_scheme_client::client::RawSchemeClient> {
        self.connection_manager
            .get_auth_service(grpc_wrapper::raw_scheme_client::client::RawSchemeClient::new)
            .await
    }
}