Add docs to transport and refactor some api's
This commit is contained in:
@@ -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()))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user