Add an option to split generated pilota files

Fixes cloudwego#454

Cargo.toml

@@ -24,12 +24,12 @@ license = "MIT OR Apache-2.0"
 rust-version = "1.80.0"
 
 [workspace.dependencies]
-pilota = "0.11"
-pilota-build = "0.11"
-pilota-thrift-parser = "0.11"
-# pilota = { git = "https://github.com/cloudwego/pilota", branch = "main" }
-# pilota-build = { git = "https://github.com/cloudwego/pilota", branch = "main" }
-# pilota-thrift-parser = { git = "https://github.com/cloudwego/pilota", branch = "main" }
+#pilota = "0.11"
+#pilota-build = "0.11"
+#pilota-thrift-parser = "0.11"
+pilota = { git = "https://github.com/cloudwego/pilota", branch = "main" }
+pilota-build = { git = "https://github.com/cloudwego/pilota", branch = "main" }
+pilota-thrift-parser = { git = "https://github.com/cloudwego/pilota", branch = "main" }
 
 motore = "0.4"
 # motore = { git = "https://github.com/cloudwego/motore", branch = "main" }

examples/volo-gen/volo.yml

@@ -22,6 +22,7 @@ entries:
   thrift:
     filename: thrift_gen.rs
     protocol: thrift
+    split_generated_files: true
     services:
       - idl:
           source: local

volo-build/src/config_builder.rs

@@ -125,6 +125,15 @@ impl InnerBuilder {
         }
     }
 
+    pub fn split_generated_files(self, split_generated_files: bool) -> Self {
+        match self {
+            InnerBuilder::Protobuf(inner) => {
+                InnerBuilder::Protobuf(inner.split_generated_files(split_generated_files))
+            }
+            InnerBuilder::Thrift(inner) => InnerBuilder::Thrift(inner.split_generated_files(split_generated_files)),
+        }
+    }
+
     pub fn common_crate_name(self, name: FastStr) -> Self {
         match self {
             InnerBuilder::Protobuf(inner) => InnerBuilder::Protobuf(inner.common_crate_name(name)),
@@ -192,6 +201,7 @@ impl ConfigBuilder {
                     .add_services(service_builders)
                     .ignore_unused(!entry.common_option.touch_all)
                     .special_namings(entry.common_option.special_namings)
+                    .split_generated_files(entry.common_option.split_generated_files)
                     .dedup(entry.common_option.dedups)
                     .write()?;
 

volo-build/src/lib.rs

@@ -112,6 +112,14 @@ impl<MkB, Parser> Builder<MkB, Parser> {
         self
     }
 
+    pub fn split_generated_files(
+        mut self,
+        split_generated_files: bool
+    ) -> Self {
+        self.pilota_builder = self.pilota_builder.split_generated_files(split_generated_files);
+        self
+    }
+
     pub fn special_namings(mut self, namings: impl IntoIterator<Item = FastStr>) -> Self {
         self.pilota_builder = self.pilota_builder.special_namings(namings);
         self

volo-build/src/model.rs

@@ -22,6 +22,8 @@ pub struct CommonOption {
     pub dedups: Vec<FastStr>,
     #[serde(default, skip_serializing_if = "Vec::is_empty")]
     pub special_namings: Vec<FastStr>,
+    #[serde(default, skip_serializing_if = "is_false")]
+    pub split_generated_files: bool,
 }
 
 #[derive(Serialize, Deserialize, Debug, Clone)]

volo-build/src/workspace.rs

@@ -106,6 +106,7 @@ where
             .dedup(config.common_option.dedups)
             .special_namings(config.common_option.special_namings)
             .common_crate_name(config.common_crate_name)
+            .split_generated_files(config.common_option.split_generated_files)
             .pilota_builder
             .compile_with_config(idl_services, pilota_build::Output::Workspace(work_dir));
     }
@@ -148,6 +149,11 @@ where
         self
     }
 
+    pub fn split_generated_files(mut self, split_generated_files: bool) -> Self {
+        self.pilota_builder = self.pilota_builder.split_generated_files(split_generated_files);
+        self
+    }
+
     pub fn touch(
         mut self,
         items: impl IntoIterator<Item = (PathBuf, Vec<impl Into<String>>)>,