feat: use afit to optimize gen code

cloudwego · Oct 21, 2023 · dcd9eb7 · dcd9eb7
1 parent d669325
commit dcd9eb7
Show file tree

Hide file tree

Showing 11 changed files with 60 additions and 40 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/pilota-build/Cargo.toml b/pilota-build/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "pilota-build"
-version = "0.8.6"
+version = "0.9.0"
 edition = "2021"
 description = "Compile thrift and protobuf idl into rust code at compile-time."
 documentation = "https://docs.rs/pilota-build"
@@ -17,9 +17,9 @@ keywords = ["serialization", "thrift", "protobuf", "volo"]
 maintenance = { status = "actively-developed" }
 
 [dependencies]
-pilota-thrift-parser = { path = "../pilota-thrift-parser", version = "0.4" }
+pilota-thrift-parser = { path = "../pilota-thrift-parser", version = "0.9" }
 
-toml = "0.7"
+toml = "0.8"
 heck = "0.4"
 syn = "2"
 normpath = "1"
@@ -36,7 +36,7 @@ tracing = "0.1"
 dashmap = "5"
 tracing-subscriber = { version = "0.3", features = ["env-filter"] }
 phf = { version = "0.11", features = ["macros"] }
-itertools = "0.10"
+itertools = "0.11"
 paste = "1"
 petgraph = "0.6"
 anyhow = "1"
@@ -47,9 +47,6 @@ protobuf-parse = { package = "protobuf-parse2", version = "4.0.0-alpha.3" }
 protobuf = { package = "protobuf2", version = "4.0.0-alpha.2" }
 faststr = "0.2"
 
-[build-dependencies]
-itertools = "0.10"
-
 [dev-dependencies]
 async-trait = "0.1"
 pilota = { path = "../pilota" }

diff --git a/pilota-build/src/codegen/mod.rs b/pilota-build/src/codegen/mod.rs
@@ -323,7 +323,6 @@ where
 
         stream.push_str(&format! {
             r#"
-            #[::async_trait::async_trait]
             pub trait {name} {{
                 {methods}
             }}

diff --git a/pilota-build/test_data/protobuf/service.rs b/pilota-build/test_data/protobuf/service.rs
@@ -51,7 +51,6 @@ pub mod service {
             }
         }
 
-        #[::async_trait::async_trait]
         pub trait Echo {}
         #[derive(PartialOrd, Hash, Eq, Ord, Debug, Default, Clone, PartialEq)]
         pub struct EchoResponse {

diff --git a/pilota-build/test_data/thrift/normal.rs b/pilota-build/test_data/thrift/normal.rs
@@ -136,7 +136,6 @@ pub mod normal {
                     + protocol.struct_end_len()
             }
         }
-        #[::async_trait::async_trait]
         pub trait Test {}
         #[derive(PartialOrd, Hash, Eq, Ord, Debug, Default, Clone, PartialEq)]
         pub struct TestTest123ArgsSend {}

diff --git a/pilota-build/test_data/thrift/pilota_name.rs b/pilota-build/test_data/thrift/pilota_name.rs
@@ -773,7 +773,6 @@ pub mod pilota_name {
             }
         }
         pub const LANG_ID: &'static str = "id";
-        #[::async_trait::async_trait]
         pub trait TestService {}
         #[derive(PartialOrd, Hash, Eq, Ord, Debug, ::pilota::derivative::Derivative)]
         #[derivative(Default)]

diff --git a/pilota-build/test_data/thrift/void.rs b/pilota-build/test_data/thrift/void.rs
@@ -440,7 +440,6 @@ pub mod void {
                     + protocol.struct_end_len()
             }
         }
-        #[::async_trait::async_trait]
         pub trait Test {}
     }
 }
diff --git a/pilota-build/test_data/thrift/wrapper_arc.rs b/pilota-build/test_data/thrift/wrapper_arc.rs
@@ -118,7 +118,6 @@ pub mod wrapper_arc {
                     + protocol.struct_end_len()
             }
         }
-        #[::async_trait::async_trait]
         pub trait TestService {}
         #[derive(Debug, ::pilota::derivative::Derivative)]
         #[derivative(Default)]

diff --git a/pilota-build/test_data/unknown_fields.rs b/pilota-build/test_data/unknown_fields.rs
@@ -717,7 +717,6 @@ pub mod unknown_fields {
                     + protocol.struct_end_len()
             }
         }
-        #[::async_trait::async_trait]
         pub trait Test {}
         #[derive(
             PartialOrd,
@@ -3462,7 +3461,6 @@ pub mod unknown_fields {
 
     pub mod void {
 
-        #[::async_trait::async_trait]
         pub trait Test {}
         #[derive(PartialOrd, Hash, Eq, Ord, Debug, ::pilota::derivative::Derivative)]
         #[derivative(Default)]

diff --git a/pilota-thrift-parser/Cargo.toml b/pilota-thrift-parser/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "pilota-thrift-parser"
-version = "0.4.2"
+version = "0.9.0"
 edition = "2021"
 description = "Pilota thrift Parser."
 documentation = "https://docs.rs/pilota"

diff --git a/pilota/Cargo.toml b/pilota/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "pilota"
-version = "0.8.1"
+version = "0.9.0"
 edition = "2021"
 description = "Pilota is a thrift and protobuf implementation in pure rust with high performance and extensibility."
 documentation = "https://docs.rs/pilota"