87 lines
2.5 KiB
Rust
87 lines
2.5 KiB
Rust
pub mod pb {
|
|
include!(concat!(env!("OUT_DIR"), "/grpc.examples.echo.rs"));
|
|
}
|
|
|
|
use pb::{EchoRequest, EchoResponse};
|
|
use std::collections::VecDeque;
|
|
use tonic::{body::BoxBody, transport::Server, Request, Response, Status, Streaming};
|
|
use tower::Service;
|
|
|
|
type EchoResult<T> = Result<Response<T>, Status>;
|
|
type Stream = VecDeque<Result<EchoResponse, Status>>;
|
|
|
|
#[derive(Default)]
|
|
pub struct EchoServer;
|
|
|
|
#[tonic::async_trait]
|
|
impl pb::server::Echo for EchoServer {
|
|
async fn unary_echo(&self, request: Request<EchoRequest>) -> EchoResult<EchoResponse> {
|
|
let message = request.into_inner().message;
|
|
Ok(Response::new(EchoResponse { message }))
|
|
}
|
|
|
|
type ServerStreamingEchoStream = Stream;
|
|
|
|
async fn server_streaming_echo(
|
|
&self,
|
|
_: Request<EchoRequest>,
|
|
) -> EchoResult<Self::ServerStreamingEchoStream> {
|
|
Err(Status::unimplemented("not implemented"))
|
|
}
|
|
|
|
async fn client_streaming_echo(
|
|
&self,
|
|
_: Request<Streaming<EchoRequest>>,
|
|
) -> EchoResult<EchoResponse> {
|
|
Err(Status::unimplemented("not implemented"))
|
|
}
|
|
|
|
type BidirectionalStreamingEchoStream = Stream;
|
|
|
|
async fn bidirectional_streaming_echo(
|
|
&self,
|
|
_: Request<Streaming<EchoRequest>>,
|
|
) -> EchoResult<Self::BidirectionalStreamingEchoStream> {
|
|
Err(Status::unimplemented("not implemented"))
|
|
}
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
let addr = "[::1]:50051".parse().unwrap();
|
|
let server = EchoServer::default();
|
|
|
|
Server::builder()
|
|
.interceptor_fn(move |svc, req| {
|
|
let auth_header = req.headers().get("authorization").clone();
|
|
|
|
let authed = if let Some(auth_header) = auth_header {
|
|
auth_header == "Bearer some-secret-token"
|
|
} else {
|
|
false
|
|
};
|
|
|
|
let fut = svc.call(req);
|
|
|
|
async move {
|
|
if authed {
|
|
fut.await
|
|
} else {
|
|
// Cancel the inner future since we never await it
|
|
// the IO never gets registered.
|
|
drop(fut);
|
|
let res = http::Response::builder()
|
|
.header("grpc-status", "16")
|
|
.body(BoxBody::empty())
|
|
.unwrap();
|
|
Ok(res)
|
|
}
|
|
}
|
|
})
|
|
.clone()
|
|
.serve(addr, pb::server::EchoServer::new(server))
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|