ydb 0.13.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
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
120
121

use std::fmt::Debug;
use ydb_grpc::ydb_proto::auth::LoginResponse;
use ydb_grpc::ydb_proto::coordination::{
    AlterNodeResponse, CreateNodeResponse, DescribeNodeResponse, DropNodeResponse,
};
use ydb_grpc::ydb_proto::discovery::{ListEndpointsResponse, WhoAmIResponse};
use ydb_grpc::ydb_proto::issue::IssueMessage;
use ydb_grpc::ydb_proto::operations::Operation as YdbOperation;
use ydb_grpc::ydb_proto::scheme::{
    DescribePathResponse, ListDirectoryResponse, MakeDirectoryResponse, RemoveDirectoryResponse,
};
use ydb_grpc::ydb_proto::status_ids::StatusCode;
use ydb_grpc::ydb_proto::table::{
    BulkUpsertResponse, CommitTransactionResponse, CopyTableResponse, CopyTablesResponse,
    CreateSessionResponse, DeleteSessionResponse, DescribeTableResponse, ExecuteDataQueryResponse,
    ExecuteSchemeQueryResponse, ExplainDataQueryResponse, KeepAliveResponse, ReadRowsResponse,
    RollbackTransactionResponse,
};
use ydb_grpc::ydb_proto::topic::{
    AlterTopicResponse, CreateTopicResponse, DescribeConsumerResponse, DescribeTopicResponse,
    DropTopicResponse, UpdateOffsetsInTransactionResponse,
};

use crate::grpc_wrapper::raw_errors::{RawError, RawResult};

pub(crate) trait YdbGrpcStatus<T>: Debug {
    fn status(&self) -> RawResult<StatusCode>;
    fn issues(&self) -> RawResult<&[IssueMessage]>;
    fn into_result(self) -> RawResult<T>;
}

impl YdbGrpcStatus<ReadRowsResponse> for ReadRowsResponse {
    fn status(&self) -> RawResult<StatusCode> {
        Ok(self.status())
    }

    fn issues(&self) -> RawResult<&[IssueMessage]> {
        Ok(&self.issues)
    }

    fn into_result(self) -> RawResult<ReadRowsResponse> {
        Ok(self)
    }
}

pub(crate) trait Operation: Debug {
    fn operation(&self) -> Option<YdbOperation>;
}

macro_rules! operation_impl_for {
    ($t:ty) => {
        impl Operation for $t {
            fn operation(&self) -> Option<YdbOperation> {
                return self.operation.clone();
            }
        }

        impl<T: Default + prost::Message> YdbGrpcStatus<T> for $t {
            fn status(&self) -> RawResult<StatusCode> {
                let operation = self
                    .operation
                    .as_ref()
                    .ok_or_else(|| RawError::Custom("no operation to fetch status".to_string()))?;

                Ok(operation.status())
            }

            fn issues(&self) -> RawResult<&[IssueMessage]> {
                let operation = self
                    .operation
                    .as_ref()
                    .ok_or_else(|| RawError::Custom("no operation to fetch status".to_string()))?;

                Ok(&operation.issues)
            }

            fn into_result(self) -> RawResult<T> {
                let operation = self
                    .operation
                    .ok_or_else(|| RawError::Custom("no operation object in result".to_string()))?;

                let result = operation
                    .result
                    .ok_or_else(|| RawError::Custom("no result data in operation".into()))?;

                let decoded = T::decode(result.value.as_slice())?;
                Ok(decoded)
            }
        }
    };
}

operation_impl_for!(CommitTransactionResponse);
operation_impl_for!(CreateSessionResponse);
operation_impl_for!(DeleteSessionResponse);
operation_impl_for!(ExecuteDataQueryResponse);
operation_impl_for!(ExecuteSchemeQueryResponse);
operation_impl_for!(ExplainDataQueryResponse);
operation_impl_for!(KeepAliveResponse);
operation_impl_for!(ListEndpointsResponse);
operation_impl_for!(RollbackTransactionResponse);
operation_impl_for!(WhoAmIResponse);
operation_impl_for!(MakeDirectoryResponse);
operation_impl_for!(ListDirectoryResponse);
operation_impl_for!(DescribePathResponse);
operation_impl_for!(RemoveDirectoryResponse);
operation_impl_for!(AlterTopicResponse);
operation_impl_for!(CreateTopicResponse);
operation_impl_for!(DescribeTopicResponse);
operation_impl_for!(DropTopicResponse);
operation_impl_for!(CreateNodeResponse);
operation_impl_for!(DescribeNodeResponse);
operation_impl_for!(AlterNodeResponse);
operation_impl_for!(DropNodeResponse);
operation_impl_for!(CopyTableResponse);
operation_impl_for!(CopyTablesResponse);
operation_impl_for!(DescribeTableResponse);
operation_impl_for!(LoginResponse);
operation_impl_for!(DescribeConsumerResponse);
operation_impl_for!(UpdateOffsetsInTransactionResponse);
operation_impl_for!(BulkUpsertResponse);