fix: set client and server streaming

stepancheg · Feb 25, 2024 · b6d0542 · b6d0542
1 parent 56e74cb
commit b6d0542
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/protobuf-parse/src/pure/convert/mod.rs b/protobuf-parse/src/pure/convert/mod.rs
@@ -341,6 +341,14 @@ impl<'a> Resolver<'a> {
                 .full_name
                 .to_string(),
         );
+
+        if input.client_streaming {
+            output.set_client_streaming(input.client_streaming);
+        }
+
+        if input.server_streaming {
+            output.set_server_streaming(input.server_streaming);
+        }
         Ok(output)
     }
 

diff --git a/test-crates/protobuf-codegen-protoc-test/src/v2/test_stream_pb.proto b/test-crates/protobuf-codegen-protoc-test/src/v2/test_stream_pb.proto
@@ -0,0 +1,18 @@
+syntax = "proto2";
+
+
+message Req {
+    required int32 a = 1;
+    repeated string b = 2;
+}
+
+message Resp {
+    required int32 status = 1;
+}
+
+service TestService {
+    rpc test_client_streaming(stream Req) returns (Resp);
+    rpc test_server_streaming(Req) returns (stream Resp);
+    rpc test_bi_streaming(stream Req) returns (stream Resp);
+}
+