Require order of request ID layers

Signed-off-by: Daniele Ahmed <[email protected]>
smithy-lang · Mar 13, 2023 · 0b817b9 · 0b817b9
1 parent ac8c02f
commit 0b817b9
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/rust-runtime/aws-smithy-http-server/src/request/request_id.rs b/rust-runtime/aws-smithy-http-server/src/request/request_id.rs
@@ -135,13 +135,14 @@ impl<S> Layer<S> for ServerRequestIdProviderLayer {
     }
 }
 
-impl<Body, S> Service<http::Request<Body>> for ServerRequestIdProvider<S>
+impl<Body, S> Service<http::Request<Body>> for ServerRequestIdProvider<ServerRequestIdResponseProvider<S>>
 where
-    S: Service<http::Request<Body>>,
+    S: Service<http::Request<Body>, Response = Response<crate::body::BoxBody>>,
+    S::Future: std::marker::Send + 'static,
 {
     type Response = S::Response;
     type Error = S::Error;
-    type Future = S::Future;
+    type Future = ServerRequestIdResponseFuture<S::Future>;
 
     fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
         self.inner.poll_ready(cx)