Add docs to transport and refactor some api's

This commit is contained in:
Lucio Franco
2019-09-29 17:04:42 -04:00
parent 1d9e3daa07
commit a35ef96a03
16 changed files with 485 additions and 233 deletions
+2 -8
View File
@@ -1,4 +1,4 @@
use crate::{generate_doc_comment, generate_doc_comments};
use crate::generate_doc_comments;
use proc_macro2::TokenStream;
use prost_build::{Method, Service};
use quote::{format_ident, quote};
@@ -52,12 +52,6 @@ pub(crate) fn generate(service: &Service, proto: &str) -> TokenStream {
#[cfg(feature = "transport")]
fn generate_connect(service_ident: &syn::Ident) -> TokenStream {
let doc_example = format!(
"let client = {}::connect(\"http://[::1]:50051\")?;",
service_ident
);
let doc_example = generate_doc_comment(&doc_example);
quote! {
impl #service_ident<tonic::transport::Channel> {
/// Attempt to create a new client by connecting to a given endpoint.
@@ -66,7 +60,7 @@ fn generate_connect(service_ident: &syn::Ident) -> TokenStream {
D: std::convert::TryInto<tonic::transport::Endpoint>,
D::Error: Into<StdError>,
{
tonic::transport::Channel::builder().build(dst).map(|c| Self::new(c))
tonic::transport::Endpoint::new(dst).map(|c| Self::new(c.channel()))
}
}
}