ydb 0.12.0

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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152

use tracing::trace;

use ydb_grpc::ydb_proto::topic::v1::topic_service_client::TopicServiceClient;
use ydb_grpc::ydb_proto::topic::{stream_read_message, stream_write_message};

use crate::grpc_wrapper::grpc_stream_wrapper::AsyncGrpcStreamWrapper;
use crate::grpc_wrapper::raw_errors::RawResult;
use crate::grpc_wrapper::raw_services::{GrpcServiceForDiscovery, Service};
use crate::grpc_wrapper::raw_topic_service::alter_topic::RawAlterTopicRequest;
use crate::grpc_wrapper::raw_topic_service::create_topic::RawCreateTopicRequest;
use crate::grpc_wrapper::raw_topic_service::describe_consumer::{
    RawDescribeConsumerRequest, RawDescribeConsumerResult,
};
use crate::grpc_wrapper::raw_topic_service::describe_topic::{
    RawDescribeTopicRequest, RawDescribeTopicResult,
};
use crate::grpc_wrapper::raw_topic_service::drop_topic::RawDropTopicRequest;
use crate::grpc_wrapper::raw_topic_service::stream_read;
use crate::grpc_wrapper::raw_topic_service::update_offsets_in_transaction::RawUpdateOffsetsInTransactionRequest;
use crate::grpc_wrapper::runtime_interceptors::InterceptedChannel;

pub(crate) struct RawTopicClient {
    service: TopicServiceClient<InterceptedChannel>,
}

impl RawTopicClient {
    pub fn new(service: InterceptedChannel) -> Self {
        Self {
            service: TopicServiceClient::new(service),
        }
    }

    pub async fn create_topic(&mut self, req: RawCreateTopicRequest) -> RawResult<()> {
        request_without_result!(
            self.service.create_topic,
            req => ydb_grpc::ydb_proto::topic::CreateTopicRequest
        );
    }

    pub async fn alter_topic(&mut self, req: RawAlterTopicRequest) -> RawResult<()> {
        request_without_result!(
            self.service.alter_topic,
            req => ydb_grpc::ydb_proto::topic::AlterTopicRequest
        );
    }

    pub async fn describe_consumer(
        &mut self,
        req: RawDescribeConsumerRequest,
    ) -> RawResult<RawDescribeConsumerResult> {
        request_with_result!(
            self.service.describe_consumer,
            req => ydb_grpc::ydb_proto::topic::DescribeConsumerRequest,
            ydb_grpc::ydb_proto::topic::DescribeConsumerResult => RawDescribeConsumerResult
        );
    }

    pub async fn describe_topic(
        &mut self,
        req: RawDescribeTopicRequest,
    ) -> RawResult<RawDescribeTopicResult> {
        request_with_result!(
            self.service.describe_topic,
            req => ydb_grpc::ydb_proto::topic::DescribeTopicRequest,
            ydb_grpc::ydb_proto::topic::DescribeTopicResult => RawDescribeTopicResult
        );
    }

    pub async fn delete_topic(&mut self, req: RawDropTopicRequest) -> RawResult<()> {
        request_without_result!(
            self.service.drop_topic,
            req => ydb_grpc::ydb_proto::topic::DropTopicRequest
        );
    }

    pub async fn update_offsets_in_transaction(
        &mut self,
        req: RawUpdateOffsetsInTransactionRequest,
    ) -> RawResult<()> {
        request_without_result!(
            self.service.update_offsets_in_transaction,
            req => ydb_grpc::ydb_proto::topic::UpdateOffsetsInTransactionRequest
        );
    }

    pub async fn stream_read(
        &mut self,
        init_req_body: stream_read::messages::RawInitRequest,
    ) -> RawResult<
        AsyncGrpcStreamWrapper<stream_read_message::FromClient, stream_read_message::FromServer>,
    > {
        let (tx, rx): (
            tokio::sync::mpsc::UnboundedSender<stream_read_message::FromClient>,
            tokio::sync::mpsc::UnboundedReceiver<stream_read_message::FromClient>,
        ) = tokio::sync::mpsc::unbounded_channel();

        let mess = stream_read_message::FromClient {
            client_message: Some(
                stream_read_message::from_client::ClientMessage::InitRequest(init_req_body.into()),
            ),
        };
        tx.send(mess)?;

        let request_stream = tokio_stream::wrappers::UnboundedReceiverStream::new(rx);
        let stream_reader_result = self.service.stream_read(request_stream).await;
        let response_stream = stream_reader_result?.into_inner();

        Ok(AsyncGrpcStreamWrapper::<
            stream_read_message::FromClient,
            stream_read_message::FromServer,
        >::new(tx, response_stream))
    }

    pub async fn stream_write(
        &mut self,
        init_req_body: stream_write_message::InitRequest,
    ) -> RawResult<
        AsyncGrpcStreamWrapper<stream_write_message::FromClient, stream_write_message::FromServer>,
    > {
        let (tx, rx): (
            tokio::sync::mpsc::UnboundedSender<stream_write_message::FromClient>,
            tokio::sync::mpsc::UnboundedReceiver<stream_write_message::FromClient>,
        ) = tokio::sync::mpsc::unbounded_channel();

        let mess = stream_write_message::FromClient {
            client_message: Some(
                stream_write_message::from_client::ClientMessage::InitRequest(init_req_body),
            ),
        };
        tx.send(mess)?;

        let request_stream = tokio_stream::wrappers::UnboundedReceiverStream::new(rx);
        let stream_writer_result = self.service.stream_write(request_stream).await;
        let response_stream = stream_writer_result?.into_inner();

        Ok(AsyncGrpcStreamWrapper::<
            stream_write_message::FromClient,
            stream_write_message::FromServer,
        >::new(tx, response_stream)) // pass tx instead of mock_tx in case of proper solution
    }

    // use for tests only, while reader not ready
    pub(crate) fn get_grpc_service(&self) -> TopicServiceClient<InterceptedChannel> {
        self.service.clone()
    }
}

impl GrpcServiceForDiscovery for RawTopicClient {
    fn get_grpc_discovery_service() -> Service {
        Service::Topic
    }
}