vitess_grpc/generated/
automationservice.rs

1/// Generated client implementations.
2pub mod automation_client {
3    #![allow(unused_variables, dead_code, missing_docs, clippy::let_unit_value)]
4    use tonic::codegen::*;
5    use tonic::codegen::http::Uri;
6    #[derive(Debug, Clone)]
7    pub struct AutomationClient<T> {
8        inner: tonic::client::Grpc<T>,
9    }
10    impl AutomationClient<tonic::transport::Channel> {
11        /// Attempt to create a new client by connecting to a given endpoint.
12        pub async fn connect<D>(dst: D) -> Result<Self, tonic::transport::Error>
13        where
14            D: TryInto<tonic::transport::Endpoint>,
15            D::Error: Into<StdError>,
16        {
17            let conn = tonic::transport::Endpoint::new(dst)?.connect().await?;
18            Ok(Self::new(conn))
19        }
20    }
21    impl<T> AutomationClient<T>
22    where
23        T: tonic::client::GrpcService<tonic::body::BoxBody>,
24        T::Error: Into<StdError>,
25        T::ResponseBody: Body<Data = Bytes> + Send + 'static,
26        <T::ResponseBody as Body>::Error: Into<StdError> + Send,
27    {
28        pub fn new(inner: T) -> Self {
29            let inner = tonic::client::Grpc::new(inner);
30            Self { inner }
31        }
32        pub fn with_origin(inner: T, origin: Uri) -> Self {
33            let inner = tonic::client::Grpc::with_origin(inner, origin);
34            Self { inner }
35        }
36        pub fn with_interceptor<F>(
37            inner: T,
38            interceptor: F,
39        ) -> AutomationClient<InterceptedService<T, F>>
40        where
41            F: tonic::service::Interceptor,
42            T::ResponseBody: Default,
43            T: tonic::codegen::Service<
44                http::Request<tonic::body::BoxBody>,
45                Response = http::Response<
46                    <T as tonic::client::GrpcService<tonic::body::BoxBody>>::ResponseBody,
47                >,
48            >,
49            <T as tonic::codegen::Service<
50                http::Request<tonic::body::BoxBody>,
51            >>::Error: Into<StdError> + Send + Sync,
52        {
53            AutomationClient::new(InterceptedService::new(inner, interceptor))
54        }
55        /// Compress requests with the given encoding.
56        ///
57        /// This requires the server to support it otherwise it might respond with an
58        /// error.
59        #[must_use]
60        pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self {
61            self.inner = self.inner.send_compressed(encoding);
62            self
63        }
64        /// Enable decompressing responses.
65        #[must_use]
66        pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self {
67            self.inner = self.inner.accept_compressed(encoding);
68            self
69        }
70        /// Limits the maximum size of a decoded message.
71        ///
72        /// Default: `4MB`
73        #[must_use]
74        pub fn max_decoding_message_size(mut self, limit: usize) -> Self {
75            self.inner = self.inner.max_decoding_message_size(limit);
76            self
77        }
78        /// Limits the maximum size of an encoded message.
79        ///
80        /// Default: `usize::MAX`
81        #[must_use]
82        pub fn max_encoding_message_size(mut self, limit: usize) -> Self {
83            self.inner = self.inner.max_encoding_message_size(limit);
84            self
85        }
86        /// Start a cluster operation.
87        pub async fn enqueue_cluster_operation(
88            &mut self,
89            request: impl tonic::IntoRequest<
90                super::super::automation::EnqueueClusterOperationRequest,
91            >,
92        ) -> std::result::Result<
93            tonic::Response<super::super::automation::EnqueueClusterOperationResponse>,
94            tonic::Status,
95        > {
96            self.inner
97                .ready()
98                .await
99                .map_err(|e| {
100                    tonic::Status::new(
101                        tonic::Code::Unknown,
102                        format!("Service was not ready: {}", e.into()),
103                    )
104                })?;
105            let codec = tonic::codec::ProstCodec::default();
106            let path = http::uri::PathAndQuery::from_static(
107                "/automationservice.Automation/EnqueueClusterOperation",
108            );
109            let mut req = request.into_request();
110            req.extensions_mut()
111                .insert(
112                    GrpcMethod::new(
113                        "automationservice.Automation",
114                        "EnqueueClusterOperation",
115                    ),
116                );
117            self.inner.unary(req, path, codec).await
118        }
119        /// TODO(mberlin): Polling this is bad. Implement a subscribe mechanism to wait for changes?
120        /// Get all details of an active cluster operation.
121        pub async fn get_cluster_operation_details(
122            &mut self,
123            request: impl tonic::IntoRequest<
124                super::super::automation::GetClusterOperationDetailsRequest,
125            >,
126        ) -> std::result::Result<
127            tonic::Response<
128                super::super::automation::GetClusterOperationDetailsResponse,
129            >,
130            tonic::Status,
131        > {
132            self.inner
133                .ready()
134                .await
135                .map_err(|e| {
136                    tonic::Status::new(
137                        tonic::Code::Unknown,
138                        format!("Service was not ready: {}", e.into()),
139                    )
140                })?;
141            let codec = tonic::codec::ProstCodec::default();
142            let path = http::uri::PathAndQuery::from_static(
143                "/automationservice.Automation/GetClusterOperationDetails",
144            );
145            let mut req = request.into_request();
146            req.extensions_mut()
147                .insert(
148                    GrpcMethod::new(
149                        "automationservice.Automation",
150                        "GetClusterOperationDetails",
151                    ),
152                );
153            self.inner.unary(req, path, codec).await
154        }
155    }
156}