compression: the identity body should respect is_end_stream (#535)

mattklein123 · web-flow · commit d0c522b6da26 · 2024-12-18T19:09:12.000-05:00
This breaks certain gRPC failure scenarios when not compressing.

Signed-off-by: Matt Klein &lt;mklein@bitdrift.io&gt;
diff --git a/tower-http/src/compression/body.rs b/tower-http/src/compression/body.rs
@@ -277,6 +277,14 @@ where
             http_body::SizeHint::new()
         }
     }
+
+    fn is_end_stream(&self) -> bool {
+        if let BodyInner::Identity { inner } = &self.inner {
+            inner.is_end_stream()
+        } else {
+            false
+        }
+    }
 }
 
 #[cfg(feature = "compression-gzip")]

Original file line number	Diff line number	Diff line change
`@@ -277,6 +277,14 @@ where`
`277`	`277`	`http_body::SizeHint::new()`
`278`	`278`	`}`
`279`	`279`	`}`
	`280`	`+`
	`281`	`+ fn is_end_stream(&self) -> bool {`
	`282`	`+ if let BodyInner::Identity { inner } = &self.inner {`
	`283`	`+ inner.is_end_stream()`
	`284`	`+ } else {`
	`285`	`+ false`
	`286`	`+ }`
	`287`	`+ }`
`280`	`288`	`}`
`281`	`289`
`282`	`290`	`#[cfg(feature = "compression-gzip")]`