1 use tonic::{transport::Server, Request, Response, Status}; 2 3 use hello_world::greeter_server::{Greeter, GreeterServer}; 4 use hello_world::{HelloReply, HelloRequest}; 5 6 pub mod hello_world { 7 tonic::include_proto!("helloworld"); 8 } 9 10 #[derive(Default)] 11 pub struct MyGreeter {} 12 13 #[tonic::async_trait] 14 impl Greeter for MyGreeter { 15 async fn say_hello( 16 &self, 17 request: Request<HelloRequest>, 18 ) -> Result<Response<HelloReply>, Status> { 19 let extension = request.extensions().get::<MyExtension>().unwrap(); 20 println!("extension data = {}", extension.some_piece_of_data); 21 22 let reply = hello_world::HelloReply { 23 message: format!("Hello {}!", request.into_inner().name), 24 }; 25 Ok(Response::new(reply)) 26 } 27 } 28 29 #[tokio::main] 30 async fn main() -> Result<(), Box<dyn std::error::Error>> { 31 let addr = "[::1]:50051".parse().unwrap(); 32 let greeter = MyGreeter::default(); 33 34 // See examples/src/interceptor/client.rs for an example of how to create a 35 // named interceptor that can be returned from functions or stored in 36 // structs. 37 let svc = GreeterServer::with_interceptor(greeter, intercept); 38 39 println!("GreeterServer listening on {}", addr); 40 41 Server::builder().add_service(svc).serve(addr).await?; 42 43 Ok(()) 44 } 45 46 /// This function will get called on each inbound request, if a `Status` 47 /// is returned, it will cancel the request and return that status to the 48 /// client. 49 fn intercept(mut req: Request<()>) -> Result<Request<()>, Status> { 50 println!("Intercepting request: {:?}", req); 51 52 // Set an extension that can be retrieved by `say_hello` 53 req.extensions_mut().insert(MyExtension { 54 some_piece_of_data: "foo".to_string(), 55 }); 56 57 Ok(req) 58 } 59 60 struct MyExtension { 61 some_piece_of_data: String, 62 } 63