1 pub mod pb { 2 tonic::include_proto!("grpc.examples.echo"); 3 } 4 5 use futures::Stream; 6 use pb::{EchoRequest, EchoResponse}; 7 use std::pin::Pin; 8 use tonic::transport::{Certificate, Identity, Server, ServerTlsConfig}; 9 use tonic::{Request, Response, Status}; 10 11 type EchoResult<T> = Result<Response<T>, Status>; 12 type ResponseStream = Pin<Box<dyn Stream<Item = Result<EchoResponse, Status>> + Send + Sync>>; 13 14 #[derive(Default)] 15 pub struct EchoServer; 16 17 #[tonic::async_trait] 18 impl pb::echo_server::Echo for EchoServer { 19 async fn unary_echo(&self, request: Request<EchoRequest>) -> EchoResult<EchoResponse> { 20 let message = request.into_inner().message; 21 Ok(Response::new(EchoResponse { message })) 22 } 23 24 type ServerStreamingEchoStream = ResponseStream; 25 type BidirectionalStreamingEchoStream = ResponseStream; 26 } 27 28 #[tokio::main] 29 async fn main() -> Result<(), Box<dyn std::error::Error>> { 30 let cert = tokio::fs::read("tonic-examples/data/tls/server.pem").await?; 31 let key = tokio::fs::read("tonic-examples/data/tls/server.key").await?; 32 let server_identity = Identity::from_pem(cert, key); 33 34 let client_ca_cert = tokio::fs::read("tonic-examples/data/tls/client_ca.pem").await?; 35 let client_ca_cert = Certificate::from_pem(client_ca_cert); 36 37 let addr = "[::1]:50051".parse().unwrap(); 38 let server = EchoServer::default(); 39 40 let tls = ServerTlsConfig::new() 41 .identity(server_identity) 42 .client_ca_root(client_ca_cert); 43 44 Server::builder() 45 .tls_config(tls) 46 .add_service(pb::echo_server::EchoServer::new(server)) 47 .serve(addr) 48 .await?; 49 50 Ok(()) 51 } 52