feat: support protobuf in workspace mode

cloudwego · Jul 29, 2024 · cc1e4e1 · cc1e4e1
1 parent 21efbba
commit cc1e4e1
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 1 deletion.
diff --git a/pilota-build/src/codegen/protobuf/mod.rs b/pilota-build/src/codegen/protobuf/mod.rs
@@ -350,6 +350,8 @@ impl ProtobufBackend {
 }
 
 impl CodegenBackend for ProtobufBackend {
+    const PROTOCOL: &'static str = "protobuf";
+
     fn codegen_struct_impl(&self, def_id: DefId, stream: &mut String, s: &rir::Message) {
         let name = self.cx.rust_name(def_id);
         let encoded_len = s

diff --git a/pilota-build/src/codegen/thrift/mod.rs b/pilota-build/src/codegen/thrift/mod.rs
@@ -450,6 +450,8 @@ impl ThriftBackend {
 }
 
 impl CodegenBackend for ThriftBackend {
+    const PROTOCOL: &'static str = "thrift";
+
     fn codegen_struct_impl(&self, def_id: DefId, stream: &mut String, s: &Message) {
         let keep = self.keep_unknown_fields.contains(&def_id);
         let name = self.cx.rust_name(def_id);

diff --git a/pilota-build/src/codegen/traits.rs b/pilota-build/src/codegen/traits.rs
@@ -5,6 +5,8 @@ use crate::{
 };
 
 pub trait CodegenBackend: Clone {
+    const PROTOCOL: &'static str;
+
     fn cx(&self) -> &Context;
     fn codegen_struct_impl(&self, _def_id: DefId, _stream: &mut String, _s: &rir::Message) {}
     fn codegen_service_impl(&self, _def_id: DefId, _stream: &mut String, _s: &rir::Service) {}

diff --git a/pilota-build/src/codegen/workspace.rs b/pilota-build/src/codegen/workspace.rs
@@ -101,7 +101,14 @@ where
     pilota = "*"
     anyhow = "1"
     volo = "*"
-    volo-thrift = "*""#
+    volo-{} = "*""#,
+                if B::PROTOCOL == "thrift" {
+                    "thrift"
+                } else if B::PROTOCOL == "protobuf" {
+                    "grpc"
+                } else {
+                    panic!("unknown protocol")
+                }
             ))
             .unwrap(),
         );