pub mod state_service_client {
#![allow(unused_variables, dead_code, missing_docs, clippy::let_unit_value)]
use tonic::codegen::*;
use tonic::codegen::http::Uri;
#[derive(Debug, Clone)]
pub struct StateServiceClient<T> {
inner: tonic::client::Grpc<T>,
}
impl StateServiceClient<tonic::transport::Channel> {
pub async fn connect<D>(dst: D) -> Result<Self, tonic::transport::Error>
where
D: TryInto<tonic::transport::Endpoint>,
D::Error: Into<StdError>,
{
let conn = tonic::transport::Endpoint::new(dst)?.connect().await?;
Ok(Self::new(conn))
}
}
impl<T> StateServiceClient<T>
where
T: tonic::client::GrpcService<tonic::body::BoxBody>,
T::Error: Into<StdError>,
T::ResponseBody: Body<Data = Bytes> + Send + 'static,
<T::ResponseBody as Body>::Error: Into<StdError> + Send,
{
pub fn new(inner: T) -> Self {
let inner = tonic::client::Grpc::new(inner);
Self { inner }
}
pub fn with_origin(inner: T, origin: Uri) -> Self {
let inner = tonic::client::Grpc::with_origin(inner, origin);
Self { inner }
}
pub fn with_interceptor<F>(
inner: T,
interceptor: F,
) -> StateServiceClient<InterceptedService<T, F>>
where
F: tonic::service::Interceptor,
T::ResponseBody: Default,
T: tonic::codegen::Service<
http::Request<tonic::body::BoxBody>,
Response = http::Response<
<T as tonic::client::GrpcService<tonic::body::BoxBody>>::ResponseBody,
>,
>,
<T as tonic::codegen::Service<
http::Request<tonic::body::BoxBody>,
>>::Error: Into<StdError> + Send + Sync,
{
StateServiceClient::new(InterceptedService::new(inner, interceptor))
}
#[must_use]
pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.inner = self.inner.send_compressed(encoding);
self
}
#[must_use]
pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.inner = self.inner.accept_compressed(encoding);
self
}
#[must_use]
pub fn max_decoding_message_size(mut self, limit: usize) -> Self {
self.inner = self.inner.max_decoding_message_size(limit);
self
}
#[must_use]
pub fn max_encoding_message_size(mut self, limit: usize) -> Self {
self.inner = self.inner.max_encoding_message_size(limit);
self
}
pub async fn put(
&mut self,
request: impl tonic::IntoRequest<super::PutRequest>,
) -> std::result::Result<tonic::Response<super::PutResponse>, tonic::Status> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.StateService/Put",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(GrpcMethod::new("flyteidl2.workflow.StateService", "Put"));
self.inner.unary(req, path, codec).await
}
pub async fn get(
&mut self,
request: impl tonic::IntoRequest<super::GetRequest>,
) -> std::result::Result<tonic::Response<super::GetResponse>, tonic::Status> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.StateService/Get",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(GrpcMethod::new("flyteidl2.workflow.StateService", "Get"));
self.inner.unary(req, path, codec).await
}
pub async fn watch(
&mut self,
request: impl tonic::IntoRequest<super::WatchRequest>,
) -> std::result::Result<
tonic::Response<tonic::codec::Streaming<super::WatchResponse>>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.StateService/Watch",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(GrpcMethod::new("flyteidl2.workflow.StateService", "Watch"));
self.inner.server_streaming(req, path, codec).await
}
}
}
pub mod state_service_server {
#![allow(unused_variables, dead_code, missing_docs, clippy::let_unit_value)]
use tonic::codegen::*;
#[async_trait]
pub trait StateService: Send + Sync + 'static {
async fn put(
&self,
request: tonic::Request<super::PutRequest>,
) -> std::result::Result<tonic::Response<super::PutResponse>, tonic::Status>;
async fn get(
&self,
request: tonic::Request<super::GetRequest>,
) -> std::result::Result<tonic::Response<super::GetResponse>, tonic::Status>;
type WatchStream: tonic::codegen::tokio_stream::Stream<
Item = std::result::Result<super::WatchResponse, tonic::Status>,
>
+ Send
+ 'static;
async fn watch(
&self,
request: tonic::Request<super::WatchRequest>,
) -> std::result::Result<tonic::Response<Self::WatchStream>, tonic::Status>;
}
#[derive(Debug)]
pub struct StateServiceServer<T: StateService> {
inner: Arc<T>,
accept_compression_encodings: EnabledCompressionEncodings,
send_compression_encodings: EnabledCompressionEncodings,
max_decoding_message_size: Option<usize>,
max_encoding_message_size: Option<usize>,
}
impl<T: StateService> StateServiceServer<T> {
pub fn new(inner: T) -> Self {
Self::from_arc(Arc::new(inner))
}
pub fn from_arc(inner: Arc<T>) -> Self {
Self {
inner,
accept_compression_encodings: Default::default(),
send_compression_encodings: Default::default(),
max_decoding_message_size: None,
max_encoding_message_size: None,
}
}
pub fn with_interceptor<F>(
inner: T,
interceptor: F,
) -> InterceptedService<Self, F>
where
F: tonic::service::Interceptor,
{
InterceptedService::new(Self::new(inner), interceptor)
}
#[must_use]
pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.accept_compression_encodings.enable(encoding);
self
}
#[must_use]
pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.send_compression_encodings.enable(encoding);
self
}
#[must_use]
pub fn max_decoding_message_size(mut self, limit: usize) -> Self {
self.max_decoding_message_size = Some(limit);
self
}
#[must_use]
pub fn max_encoding_message_size(mut self, limit: usize) -> Self {
self.max_encoding_message_size = Some(limit);
self
}
}
impl<T, B> tonic::codegen::Service<http::Request<B>> for StateServiceServer<T>
where
T: StateService,
B: Body + Send + 'static,
B::Error: Into<StdError> + Send + 'static,
{
type Response = http::Response<tonic::body::BoxBody>;
type Error = std::convert::Infallible;
type Future = BoxFuture<Self::Response, Self::Error>;
fn poll_ready(
&mut self,
_cx: &mut Context<'_>,
) -> Poll<std::result::Result<(), Self::Error>> {
Poll::Ready(Ok(()))
}
fn call(&mut self, req: http::Request<B>) -> Self::Future {
match req.uri().path() {
"/flyteidl2.workflow.StateService/Put" => {
#[allow(non_camel_case_types)]
struct PutSvc<T: StateService>(pub Arc<T>);
impl<T: StateService> tonic::server::UnaryService<super::PutRequest>
for PutSvc<T> {
type Response = super::PutResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::PutRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as StateService>::put(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = PutSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.StateService/Get" => {
#[allow(non_camel_case_types)]
struct GetSvc<T: StateService>(pub Arc<T>);
impl<T: StateService> tonic::server::UnaryService<super::GetRequest>
for GetSvc<T> {
type Response = super::GetResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::GetRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as StateService>::get(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = GetSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.StateService/Watch" => {
#[allow(non_camel_case_types)]
struct WatchSvc<T: StateService>(pub Arc<T>);
impl<
T: StateService,
> tonic::server::ServerStreamingService<super::WatchRequest>
for WatchSvc<T> {
type Response = super::WatchResponse;
type ResponseStream = T::WatchStream;
type Future = BoxFuture<
tonic::Response<Self::ResponseStream>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::WatchRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as StateService>::watch(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = WatchSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.server_streaming(method, req).await;
Ok(res)
};
Box::pin(fut)
}
_ => {
Box::pin(async move {
Ok(
http::Response::builder()
.status(200)
.header("grpc-status", tonic::Code::Unimplemented as i32)
.header(
http::header::CONTENT_TYPE,
tonic::metadata::GRPC_CONTENT_TYPE,
)
.body(empty_body())
.unwrap(),
)
})
}
}
}
}
impl<T: StateService> Clone for StateServiceServer<T> {
fn clone(&self) -> Self {
let inner = self.inner.clone();
Self {
inner,
accept_compression_encodings: self.accept_compression_encodings,
send_compression_encodings: self.send_compression_encodings,
max_decoding_message_size: self.max_decoding_message_size,
max_encoding_message_size: self.max_encoding_message_size,
}
}
}
impl<T: StateService> tonic::server::NamedService for StateServiceServer<T> {
const NAME: &'static str = "flyteidl2.workflow.StateService";
}
}
pub mod queue_service_client {
#![allow(unused_variables, dead_code, missing_docs, clippy::let_unit_value)]
use tonic::codegen::*;
use tonic::codegen::http::Uri;
#[derive(Debug, Clone)]
pub struct QueueServiceClient<T> {
inner: tonic::client::Grpc<T>,
}
impl QueueServiceClient<tonic::transport::Channel> {
pub async fn connect<D>(dst: D) -> Result<Self, tonic::transport::Error>
where
D: TryInto<tonic::transport::Endpoint>,
D::Error: Into<StdError>,
{
let conn = tonic::transport::Endpoint::new(dst)?.connect().await?;
Ok(Self::new(conn))
}
}
impl<T> QueueServiceClient<T>
where
T: tonic::client::GrpcService<tonic::body::BoxBody>,
T::Error: Into<StdError>,
T::ResponseBody: Body<Data = Bytes> + Send + 'static,
<T::ResponseBody as Body>::Error: Into<StdError> + Send,
{
pub fn new(inner: T) -> Self {
let inner = tonic::client::Grpc::new(inner);
Self { inner }
}
pub fn with_origin(inner: T, origin: Uri) -> Self {
let inner = tonic::client::Grpc::with_origin(inner, origin);
Self { inner }
}
pub fn with_interceptor<F>(
inner: T,
interceptor: F,
) -> QueueServiceClient<InterceptedService<T, F>>
where
F: tonic::service::Interceptor,
T::ResponseBody: Default,
T: tonic::codegen::Service<
http::Request<tonic::body::BoxBody>,
Response = http::Response<
<T as tonic::client::GrpcService<tonic::body::BoxBody>>::ResponseBody,
>,
>,
<T as tonic::codegen::Service<
http::Request<tonic::body::BoxBody>,
>>::Error: Into<StdError> + Send + Sync,
{
QueueServiceClient::new(InterceptedService::new(inner, interceptor))
}
#[must_use]
pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.inner = self.inner.send_compressed(encoding);
self
}
#[must_use]
pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.inner = self.inner.accept_compressed(encoding);
self
}
#[must_use]
pub fn max_decoding_message_size(mut self, limit: usize) -> Self {
self.inner = self.inner.max_decoding_message_size(limit);
self
}
#[must_use]
pub fn max_encoding_message_size(mut self, limit: usize) -> Self {
self.inner = self.inner.max_encoding_message_size(limit);
self
}
pub async fn enqueue_action(
&mut self,
request: impl tonic::IntoRequest<super::EnqueueActionRequest>,
) -> std::result::Result<
tonic::Response<super::EnqueueActionResponse>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.QueueService/EnqueueAction",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(
GrpcMethod::new("flyteidl2.workflow.QueueService", "EnqueueAction"),
);
self.inner.unary(req, path, codec).await
}
pub async fn abort_queued_run(
&mut self,
request: impl tonic::IntoRequest<super::AbortQueuedRunRequest>,
) -> std::result::Result<
tonic::Response<super::AbortQueuedRunResponse>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.QueueService/AbortQueuedRun",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(
GrpcMethod::new("flyteidl2.workflow.QueueService", "AbortQueuedRun"),
);
self.inner.unary(req, path, codec).await
}
pub async fn abort_queued_action(
&mut self,
request: impl tonic::IntoRequest<super::AbortQueuedActionRequest>,
) -> std::result::Result<
tonic::Response<super::AbortQueuedActionResponse>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.QueueService/AbortQueuedAction",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(
GrpcMethod::new(
"flyteidl2.workflow.QueueService",
"AbortQueuedAction",
),
);
self.inner.unary(req, path, codec).await
}
}
}
pub mod queue_service_server {
#![allow(unused_variables, dead_code, missing_docs, clippy::let_unit_value)]
use tonic::codegen::*;
#[async_trait]
pub trait QueueService: Send + Sync + 'static {
async fn enqueue_action(
&self,
request: tonic::Request<super::EnqueueActionRequest>,
) -> std::result::Result<
tonic::Response<super::EnqueueActionResponse>,
tonic::Status,
>;
async fn abort_queued_run(
&self,
request: tonic::Request<super::AbortQueuedRunRequest>,
) -> std::result::Result<
tonic::Response<super::AbortQueuedRunResponse>,
tonic::Status,
>;
async fn abort_queued_action(
&self,
request: tonic::Request<super::AbortQueuedActionRequest>,
) -> std::result::Result<
tonic::Response<super::AbortQueuedActionResponse>,
tonic::Status,
>;
}
#[derive(Debug)]
pub struct QueueServiceServer<T: QueueService> {
inner: Arc<T>,
accept_compression_encodings: EnabledCompressionEncodings,
send_compression_encodings: EnabledCompressionEncodings,
max_decoding_message_size: Option<usize>,
max_encoding_message_size: Option<usize>,
}
impl<T: QueueService> QueueServiceServer<T> {
pub fn new(inner: T) -> Self {
Self::from_arc(Arc::new(inner))
}
pub fn from_arc(inner: Arc<T>) -> Self {
Self {
inner,
accept_compression_encodings: Default::default(),
send_compression_encodings: Default::default(),
max_decoding_message_size: None,
max_encoding_message_size: None,
}
}
pub fn with_interceptor<F>(
inner: T,
interceptor: F,
) -> InterceptedService<Self, F>
where
F: tonic::service::Interceptor,
{
InterceptedService::new(Self::new(inner), interceptor)
}
#[must_use]
pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.accept_compression_encodings.enable(encoding);
self
}
#[must_use]
pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.send_compression_encodings.enable(encoding);
self
}
#[must_use]
pub fn max_decoding_message_size(mut self, limit: usize) -> Self {
self.max_decoding_message_size = Some(limit);
self
}
#[must_use]
pub fn max_encoding_message_size(mut self, limit: usize) -> Self {
self.max_encoding_message_size = Some(limit);
self
}
}
impl<T, B> tonic::codegen::Service<http::Request<B>> for QueueServiceServer<T>
where
T: QueueService,
B: Body + Send + 'static,
B::Error: Into<StdError> + Send + 'static,
{
type Response = http::Response<tonic::body::BoxBody>;
type Error = std::convert::Infallible;
type Future = BoxFuture<Self::Response, Self::Error>;
fn poll_ready(
&mut self,
_cx: &mut Context<'_>,
) -> Poll<std::result::Result<(), Self::Error>> {
Poll::Ready(Ok(()))
}
fn call(&mut self, req: http::Request<B>) -> Self::Future {
match req.uri().path() {
"/flyteidl2.workflow.QueueService/EnqueueAction" => {
#[allow(non_camel_case_types)]
struct EnqueueActionSvc<T: QueueService>(pub Arc<T>);
impl<
T: QueueService,
> tonic::server::UnaryService<super::EnqueueActionRequest>
for EnqueueActionSvc<T> {
type Response = super::EnqueueActionResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::EnqueueActionRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as QueueService>::enqueue_action(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = EnqueueActionSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.QueueService/AbortQueuedRun" => {
#[allow(non_camel_case_types)]
struct AbortQueuedRunSvc<T: QueueService>(pub Arc<T>);
impl<
T: QueueService,
> tonic::server::UnaryService<super::AbortQueuedRunRequest>
for AbortQueuedRunSvc<T> {
type Response = super::AbortQueuedRunResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::AbortQueuedRunRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as QueueService>::abort_queued_run(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = AbortQueuedRunSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.QueueService/AbortQueuedAction" => {
#[allow(non_camel_case_types)]
struct AbortQueuedActionSvc<T: QueueService>(pub Arc<T>);
impl<
T: QueueService,
> tonic::server::UnaryService<super::AbortQueuedActionRequest>
for AbortQueuedActionSvc<T> {
type Response = super::AbortQueuedActionResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::AbortQueuedActionRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as QueueService>::abort_queued_action(&inner, request)
.await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = AbortQueuedActionSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
_ => {
Box::pin(async move {
Ok(
http::Response::builder()
.status(200)
.header("grpc-status", tonic::Code::Unimplemented as i32)
.header(
http::header::CONTENT_TYPE,
tonic::metadata::GRPC_CONTENT_TYPE,
)
.body(empty_body())
.unwrap(),
)
})
}
}
}
}
impl<T: QueueService> Clone for QueueServiceServer<T> {
fn clone(&self) -> Self {
let inner = self.inner.clone();
Self {
inner,
accept_compression_encodings: self.accept_compression_encodings,
send_compression_encodings: self.send_compression_encodings,
max_decoding_message_size: self.max_decoding_message_size,
max_encoding_message_size: self.max_encoding_message_size,
}
}
}
impl<T: QueueService> tonic::server::NamedService for QueueServiceServer<T> {
const NAME: &'static str = "flyteidl2.workflow.QueueService";
}
}
pub mod run_logs_service_client {
#![allow(unused_variables, dead_code, missing_docs, clippy::let_unit_value)]
use tonic::codegen::*;
use tonic::codegen::http::Uri;
#[derive(Debug, Clone)]
pub struct RunLogsServiceClient<T> {
inner: tonic::client::Grpc<T>,
}
impl RunLogsServiceClient<tonic::transport::Channel> {
pub async fn connect<D>(dst: D) -> Result<Self, tonic::transport::Error>
where
D: TryInto<tonic::transport::Endpoint>,
D::Error: Into<StdError>,
{
let conn = tonic::transport::Endpoint::new(dst)?.connect().await?;
Ok(Self::new(conn))
}
}
impl<T> RunLogsServiceClient<T>
where
T: tonic::client::GrpcService<tonic::body::BoxBody>,
T::Error: Into<StdError>,
T::ResponseBody: Body<Data = Bytes> + Send + 'static,
<T::ResponseBody as Body>::Error: Into<StdError> + Send,
{
pub fn new(inner: T) -> Self {
let inner = tonic::client::Grpc::new(inner);
Self { inner }
}
pub fn with_origin(inner: T, origin: Uri) -> Self {
let inner = tonic::client::Grpc::with_origin(inner, origin);
Self { inner }
}
pub fn with_interceptor<F>(
inner: T,
interceptor: F,
) -> RunLogsServiceClient<InterceptedService<T, F>>
where
F: tonic::service::Interceptor,
T::ResponseBody: Default,
T: tonic::codegen::Service<
http::Request<tonic::body::BoxBody>,
Response = http::Response<
<T as tonic::client::GrpcService<tonic::body::BoxBody>>::ResponseBody,
>,
>,
<T as tonic::codegen::Service<
http::Request<tonic::body::BoxBody>,
>>::Error: Into<StdError> + Send + Sync,
{
RunLogsServiceClient::new(InterceptedService::new(inner, interceptor))
}
#[must_use]
pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.inner = self.inner.send_compressed(encoding);
self
}
#[must_use]
pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.inner = self.inner.accept_compressed(encoding);
self
}
#[must_use]
pub fn max_decoding_message_size(mut self, limit: usize) -> Self {
self.inner = self.inner.max_decoding_message_size(limit);
self
}
#[must_use]
pub fn max_encoding_message_size(mut self, limit: usize) -> Self {
self.inner = self.inner.max_encoding_message_size(limit);
self
}
pub async fn tail_logs(
&mut self,
request: impl tonic::IntoRequest<super::TailLogsRequest>,
) -> std::result::Result<
tonic::Response<tonic::codec::Streaming<super::TailLogsResponse>>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.RunLogsService/TailLogs",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(
GrpcMethod::new("flyteidl2.workflow.RunLogsService", "TailLogs"),
);
self.inner.server_streaming(req, path, codec).await
}
}
}
pub mod run_logs_service_server {
#![allow(unused_variables, dead_code, missing_docs, clippy::let_unit_value)]
use tonic::codegen::*;
#[async_trait]
pub trait RunLogsService: Send + Sync + 'static {
type TailLogsStream: tonic::codegen::tokio_stream::Stream<
Item = std::result::Result<super::TailLogsResponse, tonic::Status>,
>
+ Send
+ 'static;
async fn tail_logs(
&self,
request: tonic::Request<super::TailLogsRequest>,
) -> std::result::Result<tonic::Response<Self::TailLogsStream>, tonic::Status>;
}
#[derive(Debug)]
pub struct RunLogsServiceServer<T: RunLogsService> {
inner: Arc<T>,
accept_compression_encodings: EnabledCompressionEncodings,
send_compression_encodings: EnabledCompressionEncodings,
max_decoding_message_size: Option<usize>,
max_encoding_message_size: Option<usize>,
}
impl<T: RunLogsService> RunLogsServiceServer<T> {
pub fn new(inner: T) -> Self {
Self::from_arc(Arc::new(inner))
}
pub fn from_arc(inner: Arc<T>) -> Self {
Self {
inner,
accept_compression_encodings: Default::default(),
send_compression_encodings: Default::default(),
max_decoding_message_size: None,
max_encoding_message_size: None,
}
}
pub fn with_interceptor<F>(
inner: T,
interceptor: F,
) -> InterceptedService<Self, F>
where
F: tonic::service::Interceptor,
{
InterceptedService::new(Self::new(inner), interceptor)
}
#[must_use]
pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.accept_compression_encodings.enable(encoding);
self
}
#[must_use]
pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.send_compression_encodings.enable(encoding);
self
}
#[must_use]
pub fn max_decoding_message_size(mut self, limit: usize) -> Self {
self.max_decoding_message_size = Some(limit);
self
}
#[must_use]
pub fn max_encoding_message_size(mut self, limit: usize) -> Self {
self.max_encoding_message_size = Some(limit);
self
}
}
impl<T, B> tonic::codegen::Service<http::Request<B>> for RunLogsServiceServer<T>
where
T: RunLogsService,
B: Body + Send + 'static,
B::Error: Into<StdError> + Send + 'static,
{
type Response = http::Response<tonic::body::BoxBody>;
type Error = std::convert::Infallible;
type Future = BoxFuture<Self::Response, Self::Error>;
fn poll_ready(
&mut self,
_cx: &mut Context<'_>,
) -> Poll<std::result::Result<(), Self::Error>> {
Poll::Ready(Ok(()))
}
fn call(&mut self, req: http::Request<B>) -> Self::Future {
match req.uri().path() {
"/flyteidl2.workflow.RunLogsService/TailLogs" => {
#[allow(non_camel_case_types)]
struct TailLogsSvc<T: RunLogsService>(pub Arc<T>);
impl<
T: RunLogsService,
> tonic::server::ServerStreamingService<super::TailLogsRequest>
for TailLogsSvc<T> {
type Response = super::TailLogsResponse;
type ResponseStream = T::TailLogsStream;
type Future = BoxFuture<
tonic::Response<Self::ResponseStream>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::TailLogsRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as RunLogsService>::tail_logs(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = TailLogsSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.server_streaming(method, req).await;
Ok(res)
};
Box::pin(fut)
}
_ => {
Box::pin(async move {
Ok(
http::Response::builder()
.status(200)
.header("grpc-status", tonic::Code::Unimplemented as i32)
.header(
http::header::CONTENT_TYPE,
tonic::metadata::GRPC_CONTENT_TYPE,
)
.body(empty_body())
.unwrap(),
)
})
}
}
}
}
impl<T: RunLogsService> Clone for RunLogsServiceServer<T> {
fn clone(&self) -> Self {
let inner = self.inner.clone();
Self {
inner,
accept_compression_encodings: self.accept_compression_encodings,
send_compression_encodings: self.send_compression_encodings,
max_decoding_message_size: self.max_decoding_message_size,
max_encoding_message_size: self.max_encoding_message_size,
}
}
}
impl<T: RunLogsService> tonic::server::NamedService for RunLogsServiceServer<T> {
const NAME: &'static str = "flyteidl2.workflow.RunLogsService";
}
}
pub mod run_service_client {
#![allow(unused_variables, dead_code, missing_docs, clippy::let_unit_value)]
use tonic::codegen::*;
use tonic::codegen::http::Uri;
#[derive(Debug, Clone)]
pub struct RunServiceClient<T> {
inner: tonic::client::Grpc<T>,
}
impl RunServiceClient<tonic::transport::Channel> {
pub async fn connect<D>(dst: D) -> Result<Self, tonic::transport::Error>
where
D: TryInto<tonic::transport::Endpoint>,
D::Error: Into<StdError>,
{
let conn = tonic::transport::Endpoint::new(dst)?.connect().await?;
Ok(Self::new(conn))
}
}
impl<T> RunServiceClient<T>
where
T: tonic::client::GrpcService<tonic::body::BoxBody>,
T::Error: Into<StdError>,
T::ResponseBody: Body<Data = Bytes> + Send + 'static,
<T::ResponseBody as Body>::Error: Into<StdError> + Send,
{
pub fn new(inner: T) -> Self {
let inner = tonic::client::Grpc::new(inner);
Self { inner }
}
pub fn with_origin(inner: T, origin: Uri) -> Self {
let inner = tonic::client::Grpc::with_origin(inner, origin);
Self { inner }
}
pub fn with_interceptor<F>(
inner: T,
interceptor: F,
) -> RunServiceClient<InterceptedService<T, F>>
where
F: tonic::service::Interceptor,
T::ResponseBody: Default,
T: tonic::codegen::Service<
http::Request<tonic::body::BoxBody>,
Response = http::Response<
<T as tonic::client::GrpcService<tonic::body::BoxBody>>::ResponseBody,
>,
>,
<T as tonic::codegen::Service<
http::Request<tonic::body::BoxBody>,
>>::Error: Into<StdError> + Send + Sync,
{
RunServiceClient::new(InterceptedService::new(inner, interceptor))
}
#[must_use]
pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.inner = self.inner.send_compressed(encoding);
self
}
#[must_use]
pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.inner = self.inner.accept_compressed(encoding);
self
}
#[must_use]
pub fn max_decoding_message_size(mut self, limit: usize) -> Self {
self.inner = self.inner.max_decoding_message_size(limit);
self
}
#[must_use]
pub fn max_encoding_message_size(mut self, limit: usize) -> Self {
self.inner = self.inner.max_encoding_message_size(limit);
self
}
pub async fn create_run(
&mut self,
request: impl tonic::IntoRequest<super::CreateRunRequest>,
) -> std::result::Result<
tonic::Response<super::CreateRunResponse>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.RunService/CreateRun",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(GrpcMethod::new("flyteidl2.workflow.RunService", "CreateRun"));
self.inner.unary(req, path, codec).await
}
pub async fn abort_run(
&mut self,
request: impl tonic::IntoRequest<super::AbortRunRequest>,
) -> std::result::Result<
tonic::Response<super::AbortRunResponse>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.RunService/AbortRun",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(GrpcMethod::new("flyteidl2.workflow.RunService", "AbortRun"));
self.inner.unary(req, path, codec).await
}
pub async fn get_run_details(
&mut self,
request: impl tonic::IntoRequest<super::GetRunDetailsRequest>,
) -> std::result::Result<
tonic::Response<super::GetRunDetailsResponse>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.RunService/GetRunDetails",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(
GrpcMethod::new("flyteidl2.workflow.RunService", "GetRunDetails"),
);
self.inner.unary(req, path, codec).await
}
pub async fn watch_run_details(
&mut self,
request: impl tonic::IntoRequest<super::WatchRunDetailsRequest>,
) -> std::result::Result<
tonic::Response<tonic::codec::Streaming<super::WatchRunDetailsResponse>>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.RunService/WatchRunDetails",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(
GrpcMethod::new("flyteidl2.workflow.RunService", "WatchRunDetails"),
);
self.inner.server_streaming(req, path, codec).await
}
pub async fn get_action_details(
&mut self,
request: impl tonic::IntoRequest<super::GetActionDetailsRequest>,
) -> std::result::Result<
tonic::Response<super::GetActionDetailsResponse>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.RunService/GetActionDetails",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(
GrpcMethod::new("flyteidl2.workflow.RunService", "GetActionDetails"),
);
self.inner.unary(req, path, codec).await
}
pub async fn watch_action_details(
&mut self,
request: impl tonic::IntoRequest<super::WatchActionDetailsRequest>,
) -> std::result::Result<
tonic::Response<tonic::codec::Streaming<super::WatchActionDetailsResponse>>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.RunService/WatchActionDetails",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(
GrpcMethod::new(
"flyteidl2.workflow.RunService",
"WatchActionDetails",
),
);
self.inner.server_streaming(req, path, codec).await
}
pub async fn get_action_data(
&mut self,
request: impl tonic::IntoRequest<super::GetActionDataRequest>,
) -> std::result::Result<
tonic::Response<super::GetActionDataResponse>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.RunService/GetActionData",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(
GrpcMethod::new("flyteidl2.workflow.RunService", "GetActionData"),
);
self.inner.unary(req, path, codec).await
}
pub async fn list_runs(
&mut self,
request: impl tonic::IntoRequest<super::ListRunsRequest>,
) -> std::result::Result<
tonic::Response<super::ListRunsResponse>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.RunService/ListRuns",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(GrpcMethod::new("flyteidl2.workflow.RunService", "ListRuns"));
self.inner.unary(req, path, codec).await
}
pub async fn watch_runs(
&mut self,
request: impl tonic::IntoRequest<super::WatchRunsRequest>,
) -> std::result::Result<
tonic::Response<tonic::codec::Streaming<super::WatchRunsResponse>>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.RunService/WatchRuns",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(GrpcMethod::new("flyteidl2.workflow.RunService", "WatchRuns"));
self.inner.server_streaming(req, path, codec).await
}
pub async fn list_actions(
&mut self,
request: impl tonic::IntoRequest<super::ListActionsRequest>,
) -> std::result::Result<
tonic::Response<super::ListActionsResponse>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.RunService/ListActions",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(GrpcMethod::new("flyteidl2.workflow.RunService", "ListActions"));
self.inner.unary(req, path, codec).await
}
pub async fn watch_actions(
&mut self,
request: impl tonic::IntoRequest<super::WatchActionsRequest>,
) -> std::result::Result<
tonic::Response<tonic::codec::Streaming<super::WatchActionsResponse>>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.RunService/WatchActions",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(
GrpcMethod::new("flyteidl2.workflow.RunService", "WatchActions"),
);
self.inner.server_streaming(req, path, codec).await
}
pub async fn watch_cluster_events(
&mut self,
request: impl tonic::IntoRequest<super::WatchClusterEventsRequest>,
) -> std::result::Result<
tonic::Response<tonic::codec::Streaming<super::WatchClusterEventsResponse>>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.RunService/WatchClusterEvents",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(
GrpcMethod::new(
"flyteidl2.workflow.RunService",
"WatchClusterEvents",
),
);
self.inner.server_streaming(req, path, codec).await
}
pub async fn abort_action(
&mut self,
request: impl tonic::IntoRequest<super::AbortActionRequest>,
) -> std::result::Result<
tonic::Response<super::AbortActionResponse>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.RunService/AbortAction",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(GrpcMethod::new("flyteidl2.workflow.RunService", "AbortAction"));
self.inner.unary(req, path, codec).await
}
pub async fn watch_groups(
&mut self,
request: impl tonic::IntoRequest<super::WatchGroupsRequest>,
) -> std::result::Result<
tonic::Response<tonic::codec::Streaming<super::WatchGroupsResponse>>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.RunService/WatchGroups",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(GrpcMethod::new("flyteidl2.workflow.RunService", "WatchGroups"));
self.inner.server_streaming(req, path, codec).await
}
}
}
pub mod run_service_server {
#![allow(unused_variables, dead_code, missing_docs, clippy::let_unit_value)]
use tonic::codegen::*;
#[async_trait]
pub trait RunService: Send + Sync + 'static {
async fn create_run(
&self,
request: tonic::Request<super::CreateRunRequest>,
) -> std::result::Result<
tonic::Response<super::CreateRunResponse>,
tonic::Status,
>;
async fn abort_run(
&self,
request: tonic::Request<super::AbortRunRequest>,
) -> std::result::Result<
tonic::Response<super::AbortRunResponse>,
tonic::Status,
>;
async fn get_run_details(
&self,
request: tonic::Request<super::GetRunDetailsRequest>,
) -> std::result::Result<
tonic::Response<super::GetRunDetailsResponse>,
tonic::Status,
>;
type WatchRunDetailsStream: tonic::codegen::tokio_stream::Stream<
Item = std::result::Result<super::WatchRunDetailsResponse, tonic::Status>,
>
+ Send
+ 'static;
async fn watch_run_details(
&self,
request: tonic::Request<super::WatchRunDetailsRequest>,
) -> std::result::Result<
tonic::Response<Self::WatchRunDetailsStream>,
tonic::Status,
>;
async fn get_action_details(
&self,
request: tonic::Request<super::GetActionDetailsRequest>,
) -> std::result::Result<
tonic::Response<super::GetActionDetailsResponse>,
tonic::Status,
>;
type WatchActionDetailsStream: tonic::codegen::tokio_stream::Stream<
Item = std::result::Result<
super::WatchActionDetailsResponse,
tonic::Status,
>,
>
+ Send
+ 'static;
async fn watch_action_details(
&self,
request: tonic::Request<super::WatchActionDetailsRequest>,
) -> std::result::Result<
tonic::Response<Self::WatchActionDetailsStream>,
tonic::Status,
>;
async fn get_action_data(
&self,
request: tonic::Request<super::GetActionDataRequest>,
) -> std::result::Result<
tonic::Response<super::GetActionDataResponse>,
tonic::Status,
>;
async fn list_runs(
&self,
request: tonic::Request<super::ListRunsRequest>,
) -> std::result::Result<
tonic::Response<super::ListRunsResponse>,
tonic::Status,
>;
type WatchRunsStream: tonic::codegen::tokio_stream::Stream<
Item = std::result::Result<super::WatchRunsResponse, tonic::Status>,
>
+ Send
+ 'static;
async fn watch_runs(
&self,
request: tonic::Request<super::WatchRunsRequest>,
) -> std::result::Result<tonic::Response<Self::WatchRunsStream>, tonic::Status>;
async fn list_actions(
&self,
request: tonic::Request<super::ListActionsRequest>,
) -> std::result::Result<
tonic::Response<super::ListActionsResponse>,
tonic::Status,
>;
type WatchActionsStream: tonic::codegen::tokio_stream::Stream<
Item = std::result::Result<super::WatchActionsResponse, tonic::Status>,
>
+ Send
+ 'static;
async fn watch_actions(
&self,
request: tonic::Request<super::WatchActionsRequest>,
) -> std::result::Result<
tonic::Response<Self::WatchActionsStream>,
tonic::Status,
>;
type WatchClusterEventsStream: tonic::codegen::tokio_stream::Stream<
Item = std::result::Result<
super::WatchClusterEventsResponse,
tonic::Status,
>,
>
+ Send
+ 'static;
async fn watch_cluster_events(
&self,
request: tonic::Request<super::WatchClusterEventsRequest>,
) -> std::result::Result<
tonic::Response<Self::WatchClusterEventsStream>,
tonic::Status,
>;
async fn abort_action(
&self,
request: tonic::Request<super::AbortActionRequest>,
) -> std::result::Result<
tonic::Response<super::AbortActionResponse>,
tonic::Status,
>;
type WatchGroupsStream: tonic::codegen::tokio_stream::Stream<
Item = std::result::Result<super::WatchGroupsResponse, tonic::Status>,
>
+ Send
+ 'static;
async fn watch_groups(
&self,
request: tonic::Request<super::WatchGroupsRequest>,
) -> std::result::Result<
tonic::Response<Self::WatchGroupsStream>,
tonic::Status,
>;
}
#[derive(Debug)]
pub struct RunServiceServer<T: RunService> {
inner: Arc<T>,
accept_compression_encodings: EnabledCompressionEncodings,
send_compression_encodings: EnabledCompressionEncodings,
max_decoding_message_size: Option<usize>,
max_encoding_message_size: Option<usize>,
}
impl<T: RunService> RunServiceServer<T> {
pub fn new(inner: T) -> Self {
Self::from_arc(Arc::new(inner))
}
pub fn from_arc(inner: Arc<T>) -> Self {
Self {
inner,
accept_compression_encodings: Default::default(),
send_compression_encodings: Default::default(),
max_decoding_message_size: None,
max_encoding_message_size: None,
}
}
pub fn with_interceptor<F>(
inner: T,
interceptor: F,
) -> InterceptedService<Self, F>
where
F: tonic::service::Interceptor,
{
InterceptedService::new(Self::new(inner), interceptor)
}
#[must_use]
pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.accept_compression_encodings.enable(encoding);
self
}
#[must_use]
pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.send_compression_encodings.enable(encoding);
self
}
#[must_use]
pub fn max_decoding_message_size(mut self, limit: usize) -> Self {
self.max_decoding_message_size = Some(limit);
self
}
#[must_use]
pub fn max_encoding_message_size(mut self, limit: usize) -> Self {
self.max_encoding_message_size = Some(limit);
self
}
}
impl<T, B> tonic::codegen::Service<http::Request<B>> for RunServiceServer<T>
where
T: RunService,
B: Body + Send + 'static,
B::Error: Into<StdError> + Send + 'static,
{
type Response = http::Response<tonic::body::BoxBody>;
type Error = std::convert::Infallible;
type Future = BoxFuture<Self::Response, Self::Error>;
fn poll_ready(
&mut self,
_cx: &mut Context<'_>,
) -> Poll<std::result::Result<(), Self::Error>> {
Poll::Ready(Ok(()))
}
fn call(&mut self, req: http::Request<B>) -> Self::Future {
match req.uri().path() {
"/flyteidl2.workflow.RunService/CreateRun" => {
#[allow(non_camel_case_types)]
struct CreateRunSvc<T: RunService>(pub Arc<T>);
impl<
T: RunService,
> tonic::server::UnaryService<super::CreateRunRequest>
for CreateRunSvc<T> {
type Response = super::CreateRunResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::CreateRunRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as RunService>::create_run(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = CreateRunSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.RunService/AbortRun" => {
#[allow(non_camel_case_types)]
struct AbortRunSvc<T: RunService>(pub Arc<T>);
impl<
T: RunService,
> tonic::server::UnaryService<super::AbortRunRequest>
for AbortRunSvc<T> {
type Response = super::AbortRunResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::AbortRunRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as RunService>::abort_run(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = AbortRunSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.RunService/GetRunDetails" => {
#[allow(non_camel_case_types)]
struct GetRunDetailsSvc<T: RunService>(pub Arc<T>);
impl<
T: RunService,
> tonic::server::UnaryService<super::GetRunDetailsRequest>
for GetRunDetailsSvc<T> {
type Response = super::GetRunDetailsResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::GetRunDetailsRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as RunService>::get_run_details(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = GetRunDetailsSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.RunService/WatchRunDetails" => {
#[allow(non_camel_case_types)]
struct WatchRunDetailsSvc<T: RunService>(pub Arc<T>);
impl<
T: RunService,
> tonic::server::ServerStreamingService<
super::WatchRunDetailsRequest,
> for WatchRunDetailsSvc<T> {
type Response = super::WatchRunDetailsResponse;
type ResponseStream = T::WatchRunDetailsStream;
type Future = BoxFuture<
tonic::Response<Self::ResponseStream>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::WatchRunDetailsRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as RunService>::watch_run_details(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = WatchRunDetailsSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.server_streaming(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.RunService/GetActionDetails" => {
#[allow(non_camel_case_types)]
struct GetActionDetailsSvc<T: RunService>(pub Arc<T>);
impl<
T: RunService,
> tonic::server::UnaryService<super::GetActionDetailsRequest>
for GetActionDetailsSvc<T> {
type Response = super::GetActionDetailsResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::GetActionDetailsRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as RunService>::get_action_details(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = GetActionDetailsSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.RunService/WatchActionDetails" => {
#[allow(non_camel_case_types)]
struct WatchActionDetailsSvc<T: RunService>(pub Arc<T>);
impl<
T: RunService,
> tonic::server::ServerStreamingService<
super::WatchActionDetailsRequest,
> for WatchActionDetailsSvc<T> {
type Response = super::WatchActionDetailsResponse;
type ResponseStream = T::WatchActionDetailsStream;
type Future = BoxFuture<
tonic::Response<Self::ResponseStream>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::WatchActionDetailsRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as RunService>::watch_action_details(&inner, request)
.await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = WatchActionDetailsSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.server_streaming(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.RunService/GetActionData" => {
#[allow(non_camel_case_types)]
struct GetActionDataSvc<T: RunService>(pub Arc<T>);
impl<
T: RunService,
> tonic::server::UnaryService<super::GetActionDataRequest>
for GetActionDataSvc<T> {
type Response = super::GetActionDataResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::GetActionDataRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as RunService>::get_action_data(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = GetActionDataSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.RunService/ListRuns" => {
#[allow(non_camel_case_types)]
struct ListRunsSvc<T: RunService>(pub Arc<T>);
impl<
T: RunService,
> tonic::server::UnaryService<super::ListRunsRequest>
for ListRunsSvc<T> {
type Response = super::ListRunsResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::ListRunsRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as RunService>::list_runs(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = ListRunsSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.RunService/WatchRuns" => {
#[allow(non_camel_case_types)]
struct WatchRunsSvc<T: RunService>(pub Arc<T>);
impl<
T: RunService,
> tonic::server::ServerStreamingService<super::WatchRunsRequest>
for WatchRunsSvc<T> {
type Response = super::WatchRunsResponse;
type ResponseStream = T::WatchRunsStream;
type Future = BoxFuture<
tonic::Response<Self::ResponseStream>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::WatchRunsRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as RunService>::watch_runs(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = WatchRunsSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.server_streaming(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.RunService/ListActions" => {
#[allow(non_camel_case_types)]
struct ListActionsSvc<T: RunService>(pub Arc<T>);
impl<
T: RunService,
> tonic::server::UnaryService<super::ListActionsRequest>
for ListActionsSvc<T> {
type Response = super::ListActionsResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::ListActionsRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as RunService>::list_actions(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = ListActionsSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.RunService/WatchActions" => {
#[allow(non_camel_case_types)]
struct WatchActionsSvc<T: RunService>(pub Arc<T>);
impl<
T: RunService,
> tonic::server::ServerStreamingService<super::WatchActionsRequest>
for WatchActionsSvc<T> {
type Response = super::WatchActionsResponse;
type ResponseStream = T::WatchActionsStream;
type Future = BoxFuture<
tonic::Response<Self::ResponseStream>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::WatchActionsRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as RunService>::watch_actions(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = WatchActionsSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.server_streaming(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.RunService/WatchClusterEvents" => {
#[allow(non_camel_case_types)]
struct WatchClusterEventsSvc<T: RunService>(pub Arc<T>);
impl<
T: RunService,
> tonic::server::ServerStreamingService<
super::WatchClusterEventsRequest,
> for WatchClusterEventsSvc<T> {
type Response = super::WatchClusterEventsResponse;
type ResponseStream = T::WatchClusterEventsStream;
type Future = BoxFuture<
tonic::Response<Self::ResponseStream>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::WatchClusterEventsRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as RunService>::watch_cluster_events(&inner, request)
.await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = WatchClusterEventsSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.server_streaming(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.RunService/AbortAction" => {
#[allow(non_camel_case_types)]
struct AbortActionSvc<T: RunService>(pub Arc<T>);
impl<
T: RunService,
> tonic::server::UnaryService<super::AbortActionRequest>
for AbortActionSvc<T> {
type Response = super::AbortActionResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::AbortActionRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as RunService>::abort_action(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = AbortActionSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.RunService/WatchGroups" => {
#[allow(non_camel_case_types)]
struct WatchGroupsSvc<T: RunService>(pub Arc<T>);
impl<
T: RunService,
> tonic::server::ServerStreamingService<super::WatchGroupsRequest>
for WatchGroupsSvc<T> {
type Response = super::WatchGroupsResponse;
type ResponseStream = T::WatchGroupsStream;
type Future = BoxFuture<
tonic::Response<Self::ResponseStream>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::WatchGroupsRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as RunService>::watch_groups(&inner, request).await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = WatchGroupsSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.server_streaming(method, req).await;
Ok(res)
};
Box::pin(fut)
}
_ => {
Box::pin(async move {
Ok(
http::Response::builder()
.status(200)
.header("grpc-status", tonic::Code::Unimplemented as i32)
.header(
http::header::CONTENT_TYPE,
tonic::metadata::GRPC_CONTENT_TYPE,
)
.body(empty_body())
.unwrap(),
)
})
}
}
}
}
impl<T: RunService> Clone for RunServiceServer<T> {
fn clone(&self) -> Self {
let inner = self.inner.clone();
Self {
inner,
accept_compression_encodings: self.accept_compression_encodings,
send_compression_encodings: self.send_compression_encodings,
max_decoding_message_size: self.max_decoding_message_size,
max_encoding_message_size: self.max_encoding_message_size,
}
}
}
impl<T: RunService> tonic::server::NamedService for RunServiceServer<T> {
const NAME: &'static str = "flyteidl2.workflow.RunService";
}
}
pub mod translator_service_client {
#![allow(unused_variables, dead_code, missing_docs, clippy::let_unit_value)]
use tonic::codegen::*;
use tonic::codegen::http::Uri;
#[derive(Debug, Clone)]
pub struct TranslatorServiceClient<T> {
inner: tonic::client::Grpc<T>,
}
impl TranslatorServiceClient<tonic::transport::Channel> {
pub async fn connect<D>(dst: D) -> Result<Self, tonic::transport::Error>
where
D: TryInto<tonic::transport::Endpoint>,
D::Error: Into<StdError>,
{
let conn = tonic::transport::Endpoint::new(dst)?.connect().await?;
Ok(Self::new(conn))
}
}
impl<T> TranslatorServiceClient<T>
where
T: tonic::client::GrpcService<tonic::body::BoxBody>,
T::Error: Into<StdError>,
T::ResponseBody: Body<Data = Bytes> + Send + 'static,
<T::ResponseBody as Body>::Error: Into<StdError> + Send,
{
pub fn new(inner: T) -> Self {
let inner = tonic::client::Grpc::new(inner);
Self { inner }
}
pub fn with_origin(inner: T, origin: Uri) -> Self {
let inner = tonic::client::Grpc::with_origin(inner, origin);
Self { inner }
}
pub fn with_interceptor<F>(
inner: T,
interceptor: F,
) -> TranslatorServiceClient<InterceptedService<T, F>>
where
F: tonic::service::Interceptor,
T::ResponseBody: Default,
T: tonic::codegen::Service<
http::Request<tonic::body::BoxBody>,
Response = http::Response<
<T as tonic::client::GrpcService<tonic::body::BoxBody>>::ResponseBody,
>,
>,
<T as tonic::codegen::Service<
http::Request<tonic::body::BoxBody>,
>>::Error: Into<StdError> + Send + Sync,
{
TranslatorServiceClient::new(InterceptedService::new(inner, interceptor))
}
#[must_use]
pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.inner = self.inner.send_compressed(encoding);
self
}
#[must_use]
pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.inner = self.inner.accept_compressed(encoding);
self
}
#[must_use]
pub fn max_decoding_message_size(mut self, limit: usize) -> Self {
self.inner = self.inner.max_decoding_message_size(limit);
self
}
#[must_use]
pub fn max_encoding_message_size(mut self, limit: usize) -> Self {
self.inner = self.inner.max_encoding_message_size(limit);
self
}
pub async fn literals_to_launch_form_json(
&mut self,
request: impl tonic::IntoRequest<super::LiteralsToLaunchFormJsonRequest>,
) -> std::result::Result<
tonic::Response<super::LiteralsToLaunchFormJsonResponse>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.TranslatorService/LiteralsToLaunchFormJson",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(
GrpcMethod::new(
"flyteidl2.workflow.TranslatorService",
"LiteralsToLaunchFormJson",
),
);
self.inner.unary(req, path, codec).await
}
pub async fn launch_form_json_to_literals(
&mut self,
request: impl tonic::IntoRequest<super::LaunchFormJsonToLiteralsRequest>,
) -> std::result::Result<
tonic::Response<super::LaunchFormJsonToLiteralsResponse>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.TranslatorService/LaunchFormJsonToLiterals",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(
GrpcMethod::new(
"flyteidl2.workflow.TranslatorService",
"LaunchFormJsonToLiterals",
),
);
self.inner.unary(req, path, codec).await
}
pub async fn task_spec_to_launch_form_json(
&mut self,
request: impl tonic::IntoRequest<super::TaskSpecToLaunchFormJsonRequest>,
) -> std::result::Result<
tonic::Response<super::TaskSpecToLaunchFormJsonResponse>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.TranslatorService/TaskSpecToLaunchFormJson",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(
GrpcMethod::new(
"flyteidl2.workflow.TranslatorService",
"TaskSpecToLaunchFormJson",
),
);
self.inner.unary(req, path, codec).await
}
pub async fn json_values_to_literals(
&mut self,
request: impl tonic::IntoRequest<super::JsonValuesToLiteralsRequest>,
) -> std::result::Result<
tonic::Response<super::JsonValuesToLiteralsResponse>,
tonic::Status,
> {
self.inner
.ready()
.await
.map_err(|e| {
tonic::Status::new(
tonic::Code::Unknown,
format!("Service was not ready: {}", e.into()),
)
})?;
let codec = tonic::codec::ProstCodec::default();
let path = http::uri::PathAndQuery::from_static(
"/flyteidl2.workflow.TranslatorService/JsonValuesToLiterals",
);
let mut req = request.into_request();
req.extensions_mut()
.insert(
GrpcMethod::new(
"flyteidl2.workflow.TranslatorService",
"JsonValuesToLiterals",
),
);
self.inner.unary(req, path, codec).await
}
}
}
pub mod translator_service_server {
#![allow(unused_variables, dead_code, missing_docs, clippy::let_unit_value)]
use tonic::codegen::*;
#[async_trait]
pub trait TranslatorService: Send + Sync + 'static {
async fn literals_to_launch_form_json(
&self,
request: tonic::Request<super::LiteralsToLaunchFormJsonRequest>,
) -> std::result::Result<
tonic::Response<super::LiteralsToLaunchFormJsonResponse>,
tonic::Status,
>;
async fn launch_form_json_to_literals(
&self,
request: tonic::Request<super::LaunchFormJsonToLiteralsRequest>,
) -> std::result::Result<
tonic::Response<super::LaunchFormJsonToLiteralsResponse>,
tonic::Status,
>;
async fn task_spec_to_launch_form_json(
&self,
request: tonic::Request<super::TaskSpecToLaunchFormJsonRequest>,
) -> std::result::Result<
tonic::Response<super::TaskSpecToLaunchFormJsonResponse>,
tonic::Status,
>;
async fn json_values_to_literals(
&self,
request: tonic::Request<super::JsonValuesToLiteralsRequest>,
) -> std::result::Result<
tonic::Response<super::JsonValuesToLiteralsResponse>,
tonic::Status,
>;
}
#[derive(Debug)]
pub struct TranslatorServiceServer<T: TranslatorService> {
inner: Arc<T>,
accept_compression_encodings: EnabledCompressionEncodings,
send_compression_encodings: EnabledCompressionEncodings,
max_decoding_message_size: Option<usize>,
max_encoding_message_size: Option<usize>,
}
impl<T: TranslatorService> TranslatorServiceServer<T> {
pub fn new(inner: T) -> Self {
Self::from_arc(Arc::new(inner))
}
pub fn from_arc(inner: Arc<T>) -> Self {
Self {
inner,
accept_compression_encodings: Default::default(),
send_compression_encodings: Default::default(),
max_decoding_message_size: None,
max_encoding_message_size: None,
}
}
pub fn with_interceptor<F>(
inner: T,
interceptor: F,
) -> InterceptedService<Self, F>
where
F: tonic::service::Interceptor,
{
InterceptedService::new(Self::new(inner), interceptor)
}
#[must_use]
pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.accept_compression_encodings.enable(encoding);
self
}
#[must_use]
pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self {
self.send_compression_encodings.enable(encoding);
self
}
#[must_use]
pub fn max_decoding_message_size(mut self, limit: usize) -> Self {
self.max_decoding_message_size = Some(limit);
self
}
#[must_use]
pub fn max_encoding_message_size(mut self, limit: usize) -> Self {
self.max_encoding_message_size = Some(limit);
self
}
}
impl<T, B> tonic::codegen::Service<http::Request<B>> for TranslatorServiceServer<T>
where
T: TranslatorService,
B: Body + Send + 'static,
B::Error: Into<StdError> + Send + 'static,
{
type Response = http::Response<tonic::body::BoxBody>;
type Error = std::convert::Infallible;
type Future = BoxFuture<Self::Response, Self::Error>;
fn poll_ready(
&mut self,
_cx: &mut Context<'_>,
) -> Poll<std::result::Result<(), Self::Error>> {
Poll::Ready(Ok(()))
}
fn call(&mut self, req: http::Request<B>) -> Self::Future {
match req.uri().path() {
"/flyteidl2.workflow.TranslatorService/LiteralsToLaunchFormJson" => {
#[allow(non_camel_case_types)]
struct LiteralsToLaunchFormJsonSvc<T: TranslatorService>(pub Arc<T>);
impl<
T: TranslatorService,
> tonic::server::UnaryService<super::LiteralsToLaunchFormJsonRequest>
for LiteralsToLaunchFormJsonSvc<T> {
type Response = super::LiteralsToLaunchFormJsonResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<
super::LiteralsToLaunchFormJsonRequest,
>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as TranslatorService>::literals_to_launch_form_json(
&inner,
request,
)
.await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = LiteralsToLaunchFormJsonSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.TranslatorService/LaunchFormJsonToLiterals" => {
#[allow(non_camel_case_types)]
struct LaunchFormJsonToLiteralsSvc<T: TranslatorService>(pub Arc<T>);
impl<
T: TranslatorService,
> tonic::server::UnaryService<super::LaunchFormJsonToLiteralsRequest>
for LaunchFormJsonToLiteralsSvc<T> {
type Response = super::LaunchFormJsonToLiteralsResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<
super::LaunchFormJsonToLiteralsRequest,
>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as TranslatorService>::launch_form_json_to_literals(
&inner,
request,
)
.await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = LaunchFormJsonToLiteralsSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.TranslatorService/TaskSpecToLaunchFormJson" => {
#[allow(non_camel_case_types)]
struct TaskSpecToLaunchFormJsonSvc<T: TranslatorService>(pub Arc<T>);
impl<
T: TranslatorService,
> tonic::server::UnaryService<super::TaskSpecToLaunchFormJsonRequest>
for TaskSpecToLaunchFormJsonSvc<T> {
type Response = super::TaskSpecToLaunchFormJsonResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<
super::TaskSpecToLaunchFormJsonRequest,
>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as TranslatorService>::task_spec_to_launch_form_json(
&inner,
request,
)
.await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = TaskSpecToLaunchFormJsonSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
"/flyteidl2.workflow.TranslatorService/JsonValuesToLiterals" => {
#[allow(non_camel_case_types)]
struct JsonValuesToLiteralsSvc<T: TranslatorService>(pub Arc<T>);
impl<
T: TranslatorService,
> tonic::server::UnaryService<super::JsonValuesToLiteralsRequest>
for JsonValuesToLiteralsSvc<T> {
type Response = super::JsonValuesToLiteralsResponse;
type Future = BoxFuture<
tonic::Response<Self::Response>,
tonic::Status,
>;
fn call(
&mut self,
request: tonic::Request<super::JsonValuesToLiteralsRequest>,
) -> Self::Future {
let inner = Arc::clone(&self.0);
let fut = async move {
<T as TranslatorService>::json_values_to_literals(
&inner,
request,
)
.await
};
Box::pin(fut)
}
}
let accept_compression_encodings = self.accept_compression_encodings;
let send_compression_encodings = self.send_compression_encodings;
let max_decoding_message_size = self.max_decoding_message_size;
let max_encoding_message_size = self.max_encoding_message_size;
let inner = self.inner.clone();
let fut = async move {
let method = JsonValuesToLiteralsSvc(inner);
let codec = tonic::codec::ProstCodec::default();
let mut grpc = tonic::server::Grpc::new(codec)
.apply_compression_config(
accept_compression_encodings,
send_compression_encodings,
)
.apply_max_message_size_config(
max_decoding_message_size,
max_encoding_message_size,
);
let res = grpc.unary(method, req).await;
Ok(res)
};
Box::pin(fut)
}
_ => {
Box::pin(async move {
Ok(
http::Response::builder()
.status(200)
.header("grpc-status", tonic::Code::Unimplemented as i32)
.header(
http::header::CONTENT_TYPE,
tonic::metadata::GRPC_CONTENT_TYPE,
)
.body(empty_body())
.unwrap(),
)
})
}
}
}
}
impl<T: TranslatorService> Clone for TranslatorServiceServer<T> {
fn clone(&self) -> Self {
let inner = self.inner.clone();
Self {
inner,
accept_compression_encodings: self.accept_compression_encodings,
send_compression_encodings: self.send_compression_encodings,
max_decoding_message_size: self.max_decoding_message_size,
max_encoding_message_size: self.max_encoding_message_size,
}
}
}
impl<T: TranslatorService> tonic::server::NamedService
for TranslatorServiceServer<T> {
const NAME: &'static str = "flyteidl2.workflow.TranslatorService";
}
}