From 168ba46ada29985a98da1934de9079a4097d0283 Mon Sep 17 00:00:00 2001
From: Dirkjan Ochtman <dirkjan@ochtman.nl>
Date: Thu, 5 Sep 2024 13:25:20 +0200
Subject: [PATCH] tonic: move mapping responsibility to caller

---
 tonic/src/client/grpc.rs  | 2 +-
 tonic/src/codec/encode.rs | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/tonic/src/client/grpc.rs b/tonic/src/client/grpc.rs
index d582fc33c..2aed93737 100644
--- 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
index 23d18f7cc..4c94f0e8f 100644
--- 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,