From 77b8223a222060afad972e2b65d67f785315da62 Mon Sep 17 00:00:00 2001 From: tottoto Date: Thu, 5 Sep 2024 00:02:01 +0900 Subject: [PATCH] feat(build): Add simple api to compile file descriptor set (#1921) --- tonic-build/src/lib.rs | 2 +- tonic-build/src/prost.rs | 5 +++++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/tonic-build/src/lib.rs b/tonic-build/src/lib.rs index 49316cbee..b9cc48e48 100644 --- 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 index b743cbade..79462a0dd 100644 --- a/tonic-build/src/prost.rs +++ b/tonic-build/src/prost.rs @@ -61,6 +61,11 @@ pub fn compile_protos(proto: impl AsRef) -> 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] = &["()"];