@@ -6,7 +6,7 @@ use crate::{Attributes, Service};
|
|||||||
|
|
||||||
/// Builder for the generic code generation of server and clients.
|
/// Builder for the generic code generation of server and clients.
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct CodeGen8uilder {
|
pub struct CodeGenBuilder {
|
||||||
emit_package: bool,
|
emit_package: bool,
|
||||||
compile_well_known_types: bool,
|
compile_well_known_types: bool,
|
||||||
attributes: Attributes,
|
attributes: Attributes,
|
||||||
@@ -14,7 +14,7 @@ pub struct CodeGen8uilder {
|
|||||||
disable_comments: HashSet<String>,
|
disable_comments: HashSet<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl CodeGen8uilder {
|
impl CodeGenBuilder {
|
||||||
/// Create a new code gen builder with default options.
|
/// Create a new code gen builder with default options.
|
||||||
pub fn new() -> Self {
|
pub fn new() -> Self {
|
||||||
Default::default()
|
Default::default()
|
||||||
@@ -89,7 +89,7 @@ impl CodeGen8uilder {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for CodeGen8uilder {
|
impl Default for CodeGenBuilder {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Self {
|
Self {
|
||||||
emit_package: true,
|
emit_package: true,
|
||||||
|
|||||||
@@ -95,7 +95,7 @@ pub mod client;
|
|||||||
pub mod server;
|
pub mod server;
|
||||||
|
|
||||||
mod code_gen;
|
mod code_gen;
|
||||||
pub use code_gen::CodeGen8uilder;
|
pub use code_gen::CodeGenBuilder;
|
||||||
|
|
||||||
/// Service generation trait.
|
/// Service generation trait.
|
||||||
///
|
///
|
||||||
|
|||||||
@@ -28,7 +28,7 @@
|
|||||||
//! }
|
//! }
|
||||||
//! ```
|
//! ```
|
||||||
|
|
||||||
use crate::code_gen::CodeGen8uilder;
|
use crate::code_gen::CodeGenBuilder;
|
||||||
|
|
||||||
use proc_macro2::TokenStream;
|
use proc_macro2::TokenStream;
|
||||||
use quote::ToTokens;
|
use quote::ToTokens;
|
||||||
@@ -352,7 +352,7 @@ struct ServiceGenerator {
|
|||||||
impl ServiceGenerator {
|
impl ServiceGenerator {
|
||||||
fn generate(&mut self, service: &Service) {
|
fn generate(&mut self, service: &Service) {
|
||||||
if self.builder.build_server {
|
if self.builder.build_server {
|
||||||
let server = CodeGen8uilder::new()
|
let server = CodeGenBuilder::new()
|
||||||
.emit_package(true)
|
.emit_package(true)
|
||||||
.compile_well_known_types(false)
|
.compile_well_known_types(false)
|
||||||
.generate_server(service, "");
|
.generate_server(service, "");
|
||||||
@@ -361,7 +361,7 @@ impl ServiceGenerator {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if self.builder.build_client {
|
if self.builder.build_client {
|
||||||
let client = CodeGen8uilder::new()
|
let client = CodeGenBuilder::new()
|
||||||
.emit_package(true)
|
.emit_package(true)
|
||||||
.compile_well_known_types(false)
|
.compile_well_known_types(false)
|
||||||
.build_transport(self.builder.build_transport)
|
.build_transport(self.builder.build_transport)
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
use crate::code_gen::CodeGen8uilder;
|
use crate::code_gen::CodeGenBuilder;
|
||||||
|
|
||||||
use super::Attributes;
|
use super::Attributes;
|
||||||
use proc_macro2::TokenStream;
|
use proc_macro2::TokenStream;
|
||||||
@@ -161,7 +161,7 @@ impl ServiceGenerator {
|
|||||||
impl prost_build::ServiceGenerator for ServiceGenerator {
|
impl prost_build::ServiceGenerator for ServiceGenerator {
|
||||||
fn generate(&mut self, service: prost_build::Service, _buf: &mut String) {
|
fn generate(&mut self, service: prost_build::Service, _buf: &mut String) {
|
||||||
if self.builder.build_server {
|
if self.builder.build_server {
|
||||||
let server = CodeGen8uilder::new()
|
let server = CodeGenBuilder::new()
|
||||||
.emit_package(self.builder.emit_package)
|
.emit_package(self.builder.emit_package)
|
||||||
.compile_well_known_types(self.builder.compile_well_known_types)
|
.compile_well_known_types(self.builder.compile_well_known_types)
|
||||||
.attributes(self.builder.server_attributes.clone())
|
.attributes(self.builder.server_attributes.clone())
|
||||||
@@ -172,7 +172,7 @@ impl prost_build::ServiceGenerator for ServiceGenerator {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if self.builder.build_client {
|
if self.builder.build_client {
|
||||||
let client = CodeGen8uilder::new()
|
let client = CodeGenBuilder::new()
|
||||||
.emit_package(self.builder.emit_package)
|
.emit_package(self.builder.emit_package)
|
||||||
.compile_well_known_types(self.builder.compile_well_known_types)
|
.compile_well_known_types(self.builder.compile_well_known_types)
|
||||||
.attributes(self.builder.client_attributes.clone())
|
.attributes(self.builder.client_attributes.clone())
|
||||||
|
|||||||
Reference in New Issue
Block a user