tonic: move mapping responsibility to caller

hyperium · Sep 21, 2024 · 168ba46 · 168ba46
1 parent 4b5415f
commit 168ba46
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tonic/src/client/grpc.rs b/tonic/src/client/grpc.rs
@@ -297,7 +297,7 @@ impl<T> Grpc<T> {
             .map(|s| {
                 encode_client(
                     codec.encoder(),
-                    s,
+                    s.map(Ok),
                     self.config.send_compression_encodings,
                     self.config.max_encoding_message_size,
                 )

diff --git a/tonic/src/codec/encode.rs b/tonic/src/codec/encode.rs
@@ -20,14 +20,14 @@ pub fn encode_client<T, U>(
     source: U,
     compression_encoding: Option<CompressionEncoding>,
     max_message_size: Option<usize>,
-) -> EncodeBody<T, impl Stream<Item = Result<T::Item, Status>>>
+) -> EncodeBody<T, U>
 where
     T: Encoder<Error = Status>,
-    U: Stream<Item = T::Item>,
+    U: Stream,
 {
     let stream = EncodedBytes::new(
         encoder,
-        source.map(Ok),
+        source,
         compression_encoding,
         SingleMessageCompressionOverride::default(),
         max_message_size,