exmaple: Update stream example to verify #208

Add test for issues: #169, #207.

Signed-off-by: Tim Zhang <[email protected]>

Author: Tim-Zhang

example/async-stream-client.rs

@@ -41,9 +41,12 @@ async fn main() {
     let sc1 = sc.clone();
     let t5 = tokio::spawn(echo_null(sc1));
 
-    let t6 = tokio::spawn(echo_null_stream(sc));
+    let sc1 = sc.clone();
+    let t6 = tokio::spawn(echo_null_stream(sc1));
+
+    let t7 = tokio::spawn(echo_default_value(sc));
 
-    let _ = tokio::join!(t1, t2, t3, t4, t5, t6);
+    let _ = tokio::join!(t1, t2, t3, t4, t5, t6, t7);
 }
 
 fn default_ctx() -> Context {
@@ -185,3 +188,16 @@ async fn echo_null_stream(cli: streaming_ttrpc::StreamingClient) {
         .unwrap()
         .unwrap();
 }
+
+#[cfg(unix)]
+async fn echo_default_value(cli: streaming_ttrpc::StreamingClient) {
+    let mut stream = cli
+        .echo_default_value(default_ctx(), &Default::default()) // send default value to verify #208
+        .await
+        .unwrap();
+
+    let received = stream.recv().await.unwrap().unwrap();
+
+    assert_eq!(received.seq, 0);
+    assert_eq!(received.msg, "");
+}

example/async-stream-server.rs

@@ -13,7 +13,7 @@ use log::{info, LevelFilter};
 #[cfg(unix)]
 use protocols::r#async::{empty, streaming, streaming_ttrpc};
 #[cfg(unix)]
-use ttrpc::asynchronous::Server;
+use ttrpc::{asynchronous::Server, Error};
 
 #[cfg(unix)]
 use async_trait::async_trait;
@@ -134,6 +134,24 @@ impl streaming_ttrpc::Streaming for StreamingService {
         }
         Ok(())
     }
+
+    // It verifies PR #208
+    async fn echo_default_value(
+        &self,
+        _ctx: &::ttrpc::r#async::TtrpcContext,
+        e: streaming::EchoPayload,
+        s: ::ttrpc::r#async::ServerStreamSender<streaming::EchoPayload>,
+    ) -> ::ttrpc::Result<()> {
+        if e.seq != 0 || !e.msg.is_empty() {
+            return Err(Error::Others(
+                "Expect a request with empty payload to verify #208".to_string(),
+            ));
+        }
+
+        s.send(&e).await.unwrap();
+
+        Ok(())
+    }
 }
 
 #[cfg(windows)]

example/protocols/protos/streaming.proto

@@ -32,6 +32,7 @@ service Streaming {
 	rpc DivideStream(Sum) returns (stream Part);
 	rpc EchoNull(stream EchoPayload) returns (google.protobuf.Empty);
 	rpc EchoNullStream(stream EchoPayload) returns (stream google.protobuf.Empty);
+	rpc EchoDefaultValue(EchoPayload) returns (stream EchoPayload);
 }
 
 message EchoPayload {