Modify and Merge request compression middleware logic

aws · Nov 2, 2023 · 560c9cc · 560c9cc
1 parent 80428e2
commit 560c9cc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/aws/middleware/request_compression.go b/aws/middleware/request_compression.go
@@ -54,11 +54,11 @@ func (m requestCompression) HandleBuild(
 			return out, metadata, fmt.Errorf("failed to compress request stream, %v", err)
 		}
 
-		if newReq, err := req.SetStream(bytes.NewReader(compressedBytes)); err != nil {
+		var newReq *smithyhttp.Request
+		if newReq, err = req.SetStream(bytes.NewReader(compressedBytes)); err != nil {
 			return out, metadata, fmt.Errorf("failed to set request stream, %v", err)
-		} else {
-			*req = *newReq
 		}
+		*req = *newReq
 		isCompressed = true
 	} else if req.ContentLength >= m.requestMinCompressSizeBytes {
 		compressedBytes, err := compress(req.Body)