fix(build): Fix service and rpc name conflict

Closes #89
hyperium · Oct 23, 2019 · 0483cef · 0483cef
1 parent 81c1bb3
commit 0483cef
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tonic-build/src/server.rs b/tonic-build/src/server.rs
@@ -223,7 +223,7 @@ fn generate_unary(
     proto_path: &str,
     server_trait: Ident,
 ) -> TokenStream {
-    let service_ident = Ident::new(&method.proto_name, Span::call_site());
+    let service_ident = quote::format_ident!("{}Svc", method.proto_name);
 
     let (request, response) = crate::replace_wellknown(proto_path, &method);
 
@@ -262,7 +262,7 @@ fn generate_server_streaming(
     proto_path: &str,
     server_trait: Ident,
 ) -> TokenStream {
-    let service_ident = Ident::new(&method.proto_name, Span::call_site());
+    let service_ident = quote::format_ident!("{}Svc", method.proto_name);
 
     let (request, response) = crate::replace_wellknown(proto_path, &method);
 
@@ -305,7 +305,7 @@ fn generate_client_streaming(
     proto_path: &str,
     server_trait: Ident,
 ) -> TokenStream {
-    let service_ident = Ident::new(&method.proto_name, Span::call_site());
+    let service_ident = quote::format_ident!("{}Svc", method.proto_name);
 
     let (request, response) = crate::replace_wellknown(proto_path, &method);
 
@@ -346,7 +346,7 @@ fn generate_streaming(
     proto_path: &str,
     server_trait: Ident,
 ) -> TokenStream {
-    let service_ident = Ident::new(&method.proto_name, Span::call_site());
+    let service_ident = quote::format_ident!("{}Svc", method.proto_name);
 
     let (request, response) = crate::replace_wellknown(proto_path, &method);