From 155f8a2ba2c2584f20a6258edca21e9321b72b12 Mon Sep 17 00:00:00 2001
From: "sijie.sun" <sijie.sun@smartx.com>
Date: Thu, 6 Mar 2025 09:59:41 +0800
Subject: [PATCH] make prost build smaller

---
 easytier/build.rs | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/easytier/build.rs b/easytier/build.rs
index 0a6a87ab..489a234b 100644
--- a/easytier/build.rs
+++ b/easytier/build.rs
@@ -128,16 +128,16 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
     #[cfg(target_os = "windows")]
     WindowsBuild::check_for_win();
 
+    let proto_files_reflect = ["src/proto/peer_rpc.proto", "src/proto/common.proto"];
+
     let proto_files = [
-        "src/proto/peer_rpc.proto",
-        "src/proto/common.proto",
         "src/proto/error.proto",
         "src/proto/tests.proto",
         "src/proto/cli.proto",
         "src/proto/web.proto",
     ];
 
-    for proto_file in &proto_files {
+    for proto_file in proto_files.iter().chain(proto_files_reflect.iter()) {
         println!("cargo:rerun-if-changed={}", proto_file);
     }
 
@@ -159,9 +159,11 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
         .service_generator(Box::new(rpc_build::ServiceGenerator::new()))
         .btree_map(["."]);
 
+    config.compile_protos(&proto_files, &["src/proto/"])?;
+
     prost_reflect_build::Builder::new()
         .file_descriptor_set_bytes("crate::proto::DESCRIPTOR_POOL_BYTES")
-        .compile_protos_with_config(config, &proto_files, &["src/proto/"])?;
+        .compile_protos_with_config(config, &proto_files_reflect, &["src/proto/"])?;
 
     check_locale();
     Ok(())