feat(build): Add simple api to compile file descriptor set (#1921)

hyperium · Sep 4, 2024 · 77b8223 · 77b8223
1 parent e4c71b8
commit 77b8223
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/tonic-build/src/lib.rs b/tonic-build/src/lib.rs
@@ -83,7 +83,7 @@ use quote::TokenStreamExt;
 mod prost;
 
 #[cfg(feature = "prost")]
-pub use prost::{compile_protos, configure, Builder};
+pub use prost::{compile_fds, compile_protos, configure, Builder};
 
 pub mod manual;
 

diff --git a/tonic-build/src/prost.rs b/tonic-build/src/prost.rs
@@ -61,6 +61,11 @@ pub fn compile_protos(proto: impl AsRef<Path>) -> io::Result<()> {
     self::configure().compile_protos(&[proto_path], &[proto_dir])
 }
 
+/// Simple file descriptor set compiling. Use [`configure`] instead if you need more options.
+pub fn compile_fds(fds: prost_types::FileDescriptorSet) -> io::Result<()> {
+    self::configure().compile_fds(fds)
+}
+
 /// Non-path Rust types allowed for request/response types.
 const NON_PATH_TYPE_ALLOWLIST: &[&str] = &["()"];