simply api

codeitlikemiley · Dec 16, 2023 · ecb5fbe · ecb5fbe
1 parent 5976fb5
commit ecb5fbe
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -1,14 +1,15 @@
 use {{service_name}}_service::{{service_name}}::{{service_name}}_service_server::{{service_name | pascal_case}}ServiceServer;
 use {{service_name}}_service::{{service_name}}_impl::{{service_name | pascal_case}}ServiceImpl;
+use std::net::SocketAddr;
 use tonic::transport::Server;
 
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error>> {
-    let addr = "[::1]:50051".parse()?;
-    let {{service_name}}_service = {{service_name | pascal_case}}ServiceImpl::default();
+    let addr = SocketAddr::from(([0, 0, 0, 0], 50051));
 
     Server::builder()
-        .add_service({{service_name | pascal_case}}ServiceServer::new({{service_name}}_service))
+        .add_service({{service_name | pascal_case}}ServiceImpl::attach_reflection()?)
+        .add_service({{service_name | pascal_case}}ServiceImpl::new())
         .serve(addr)
         .await?;
     Ok(())