-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Ryan Vacek
authored and
Ryan Vacek
committed
Dec 9, 2023
1 parent
0a99535
commit 9b66dd0
Showing
19 changed files
with
651 additions
and
3 deletions.
There are no files selected for viewing
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,323 @@ | ||
#[allow(clippy::derive_partial_eq_without_eq)] | ||
#[derive(Clone, PartialEq, ::prost::Message)] | ||
pub struct QuoterRequest {} | ||
#[allow(clippy::derive_partial_eq_without_eq)] | ||
#[derive(Clone, PartialEq, ::prost::Message)] | ||
pub struct Deal { | ||
#[prost(int32, tag = "1")] | ||
pub location: i32, | ||
#[prost(double, tag = "2")] | ||
pub price: f64, | ||
#[prost(double, tag = "3")] | ||
pub quantity: f64, | ||
} | ||
#[allow(clippy::derive_partial_eq_without_eq)] | ||
#[derive(Clone, PartialEq, ::prost::Message)] | ||
pub struct QuoterResponse { | ||
#[prost(int32, tag = "1")] | ||
pub spread: i32, | ||
#[prost(message, repeated, tag = "2")] | ||
pub ask_deals: ::prost::alloc::vec::Vec<Deal>, | ||
#[prost(message, repeated, tag = "3")] | ||
pub bid_deals: ::prost::alloc::vec::Vec<Deal>, | ||
} | ||
/// Generated client implementations. | ||
pub mod quoter_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 QuoterClient<T> { | ||
inner: tonic::client::Grpc<T>, | ||
} | ||
impl QuoterClient<tonic::transport::Channel> { | ||
/// Attempt to create a new client by connecting to a given endpoint. | ||
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> QuoterClient<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, | ||
) -> QuoterClient<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, | ||
{ | ||
QuoterClient::new(InterceptedService::new(inner, interceptor)) | ||
} | ||
/// Compress requests with the given encoding. | ||
/// | ||
/// This requires the server to support it otherwise it might respond with an | ||
/// error. | ||
#[must_use] | ||
pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self { | ||
self.inner = self.inner.send_compressed(encoding); | ||
self | ||
} | ||
/// Enable decompressing responses. | ||
#[must_use] | ||
pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self { | ||
self.inner = self.inner.accept_compressed(encoding); | ||
self | ||
} | ||
/// Limits the maximum size of a decoded message. | ||
/// | ||
/// Default: `4MB` | ||
#[must_use] | ||
pub fn max_decoding_message_size(mut self, limit: usize) -> Self { | ||
self.inner = self.inner.max_decoding_message_size(limit); | ||
self | ||
} | ||
/// Limits the maximum size of an encoded message. | ||
/// | ||
/// Default: `usize::MAX` | ||
#[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 server_streaming_quoter( | ||
&mut self, | ||
request: impl tonic::IntoRequest<super::QuoterRequest>, | ||
) -> std::result::Result< | ||
tonic::Response<tonic::codec::Streaming<super::QuoterResponse>>, | ||
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( | ||
"/lib.Quoter/ServerStreamingQuoter", | ||
); | ||
let mut req = request.into_request(); | ||
req.extensions_mut() | ||
.insert(GrpcMethod::new("lib.Quoter", "ServerStreamingQuoter")); | ||
self.inner.server_streaming(req, path, codec).await | ||
} | ||
} | ||
} | ||
/// Generated server implementations. | ||
pub mod quoter_server { | ||
#![allow(unused_variables, dead_code, missing_docs, clippy::let_unit_value)] | ||
use tonic::codegen::*; | ||
/// Generated trait containing gRPC methods that should be implemented for use with QuoterServer. | ||
#[async_trait] | ||
pub trait Quoter: Send + Sync + 'static { | ||
/// Server streaming response type for the ServerStreamingQuoter method. | ||
type ServerStreamingQuoterStream: futures_core::Stream< | ||
Item = std::result::Result<super::QuoterResponse, tonic::Status>, | ||
> | ||
+ Send | ||
+ 'static; | ||
async fn server_streaming_quoter( | ||
&self, | ||
request: tonic::Request<super::QuoterRequest>, | ||
) -> std::result::Result< | ||
tonic::Response<Self::ServerStreamingQuoterStream>, | ||
tonic::Status, | ||
>; | ||
} | ||
#[derive(Debug)] | ||
pub struct QuoterServer<T: Quoter> { | ||
inner: _Inner<T>, | ||
accept_compression_encodings: EnabledCompressionEncodings, | ||
send_compression_encodings: EnabledCompressionEncodings, | ||
max_decoding_message_size: Option<usize>, | ||
max_encoding_message_size: Option<usize>, | ||
} | ||
struct _Inner<T>(Arc<T>); | ||
impl<T: Quoter> QuoterServer<T> { | ||
pub fn new(inner: T) -> Self { | ||
Self::from_arc(Arc::new(inner)) | ||
} | ||
pub fn from_arc(inner: Arc<T>) -> Self { | ||
let inner = _Inner(inner); | ||
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) | ||
} | ||
/// Enable decompressing requests with the given encoding. | ||
#[must_use] | ||
pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self { | ||
self.accept_compression_encodings.enable(encoding); | ||
self | ||
} | ||
/// Compress responses with the given encoding, if the client supports it. | ||
#[must_use] | ||
pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self { | ||
self.send_compression_encodings.enable(encoding); | ||
self | ||
} | ||
/// Limits the maximum size of a decoded message. | ||
/// | ||
/// Default: `4MB` | ||
#[must_use] | ||
pub fn max_decoding_message_size(mut self, limit: usize) -> Self { | ||
self.max_decoding_message_size = Some(limit); | ||
self | ||
} | ||
/// Limits the maximum size of an encoded message. | ||
/// | ||
/// Default: `usize::MAX` | ||
#[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 QuoterServer<T> | ||
where | ||
T: Quoter, | ||
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 { | ||
let inner = self.inner.clone(); | ||
match req.uri().path() { | ||
"/lib.Quoter/ServerStreamingQuoter" => { | ||
#[allow(non_camel_case_types)] | ||
struct ServerStreamingQuoterSvc<T: Quoter>(pub Arc<T>); | ||
impl< | ||
T: Quoter, | ||
> tonic::server::ServerStreamingService<super::QuoterRequest> | ||
for ServerStreamingQuoterSvc<T> { | ||
type Response = super::QuoterResponse; | ||
type ResponseStream = T::ServerStreamingQuoterStream; | ||
type Future = BoxFuture< | ||
tonic::Response<Self::ResponseStream>, | ||
tonic::Status, | ||
>; | ||
fn call( | ||
&mut self, | ||
request: tonic::Request<super::QuoterRequest>, | ||
) -> Self::Future { | ||
let inner = Arc::clone(&self.0); | ||
let fut = async move { | ||
(*inner).server_streaming_quoter(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 inner = inner.0; | ||
let method = ServerStreamingQuoterSvc(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", "12") | ||
.header("content-type", "application/grpc") | ||
.body(empty_body()) | ||
.unwrap(), | ||
) | ||
}) | ||
} | ||
} | ||
} | ||
} | ||
impl<T: Quoter> Clone for QuoterServer<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: Quoter> Clone for _Inner<T> { | ||
fn clone(&self) -> Self { | ||
Self(Arc::clone(&self.0)) | ||
} | ||
} | ||
impl<T: std::fmt::Debug> std::fmt::Debug for _Inner<T> { | ||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { | ||
write!(f, "{:?}", self.0) | ||
} | ||
} | ||
impl<T: Quoter> tonic::server::NamedService for QuoterServer<T> { | ||
const NAME: &'static str = "lib.Quoter"; | ||
} | ||
} |
Oops, something went wrong.