feat(build): Expose Prost generation plugin (#947)

hyperium · Mar 31, 2022 · d4bd475 · d4bd475
1 parent 9dab8be
commit d4bd475
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tonic-build/src/prost.rs b/tonic-build/src/prost.rs
@@ -412,10 +412,16 @@ impl Builder {
             config.protoc_arg(arg);
         }
 
-        config.service_generator(Box::new(ServiceGenerator::new(self)));
+        config.service_generator(self.service_generator());
 
         config.compile_protos(protos, includes)?;
 
         Ok(())
     }
+
+    /// Turn the builder into a `ServiceGenerator` ready to be passed to `prost-build`s
+    /// `Config::service_generator`.
+    pub fn service_generator(self) -> Box<dyn prost_build::ServiceGenerator> {
+        Box::new(ServiceGenerator::new(self))
+    }
 }