xref: /tonic/examples/src/autoreload/server.rs (revision 8508f369)
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 {
say_hello( &self, request: Request<HelloRequest>, ) -> Result<Response<HelloReply>, Status>15     async fn say_hello(
16         &self,
17         request: Request<HelloRequest>,
18     ) -> Result<Response<HelloReply>, Status> {
19         println!("Got a request from {:?}", request.remote_addr());
20 
21         let reply = hello_world::HelloReply {
22             message: format!("Hello {}!", request.into_inner().name),
23         };
24         Ok(Response::new(reply))
25     }
26 }
27 
28 #[tokio::main]
main() -> Result<(), Box<dyn std::error::Error>>29 async fn main() -> Result<(), Box<dyn std::error::Error>> {
30     let addr = "[::1]:50051".parse().unwrap();
31     let greeter = MyGreeter::default();
32 
33     println!("GreeterServer listening on {}", addr);
34 
35     let server = Server::builder().add_service(GreeterServer::new(greeter));
36 
37     match listenfd::ListenFd::from_env().take_tcp_listener(0)? {
38         Some(listener) => {
39             listener.set_nonblocking(true)?;
40             let listener = tokio_stream::wrappers::TcpListenerStream::new(
41                 tokio::net::TcpListener::from_std(listener)?,
42             );
43 
44             server.serve_with_incoming(listener).await?;
45         }
46         None => {
47             server.serve(addr).await?;
48         }
49     }
50 
51     Ok(())
52 }
53