talksik/tonic
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix(codegen): Use wellknown types from prost-types (#49)

Browse Source 
This change introduces the ability to let prost-build
provide the fully qualifed type path for types that
start with `google.protobuf`.
This commit is contained in:
Lucio Franco
2019-10-05 13:16:09 -04:00
committed by GitHub
parent ed3e7e95a5
commit 4e1fcece15
6 changed files with 42 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tonic-build/src/client.rs
+4 -8
View File
@@ -96,8 +96,7 @@ fn generate_methods(service: &Service, proto: &str) -> TokenStream {
fn generate_unary(method: &Method, proto: &str, path: String) -> TokenStream {
let ident = format_ident!("{}", method.name);
let request = crate::replace_wellknown(proto, &method.input_type);
let response = crate::replace_wellknown(proto, &method.output_type);
let (request, response) = crate::replace_wellknown(proto, &method);
quote! {
pub async fn #ident(&mut self, request: tonic::Request<#request>)
@@ -113,8 +112,7 @@ fn generate_unary(method: &Method, proto: &str, path: String) -> TokenStream {
fn generate_server_streaming(method: &Method, proto: &str, path: String) -> TokenStream {
let ident = format_ident!("{}", method.name);
let request = crate::replace_wellknown(proto, &method.input_type);
let response = crate::replace_wellknown(proto, &method.output_type);
let (request, response) = crate::replace_wellknown(proto, &method);
quote! {
pub async fn #ident(&mut self, request: tonic::Request<#request>)
@@ -130,8 +128,7 @@ fn generate_server_streaming(method: &Method, proto: &str, path: String) -> Toke
fn generate_client_streaming(method: &Method, proto: &str, path: String) -> TokenStream {
let ident = format_ident!("{}", method.name);
let request = crate::replace_wellknown(proto, &method.input_type);
let response = crate::replace_wellknown(proto, &method.output_type);
let (request, response) = crate::replace_wellknown(proto, &method);
quote! {
pub async fn #ident<S>(&mut self, request: tonic::Request<S>)
@@ -149,8 +146,7 @@ fn generate_client_streaming(method: &Method, proto: &str, path: String) -> Toke
fn generate_streaming(method: &Method, proto: &str, path: String) -> TokenStream {
let ident = format_ident!("{}", method.name);
let request = crate::replace_wellknown(proto, &method.input_type);
let response = crate::replace_wellknown(proto, &method.output_type);
let (request, response) = crate::replace_wellknown(proto, &method);
quote! {
pub async fn #ident<S>(&mut self, request: tonic::Request<S>)