Check if content-range is a bytes range

microsoft · Jul 16, 2024 · 9b6c5cf · 9b6c5cf
1 parent e2aac71
commit 9b6c5cf
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/compression_handler.go b/compression_handler.go
@@ -74,7 +74,7 @@ func (c *CompressionHandler) Intercept(pipeline Pipeline, middlewareIndex int, r
 		req = req.WithContext(ctx)
 	}
 
-	if !reqOption.ShouldCompress() || contentRangeIsPresent(req.Header) || req.Body == nil {
+	if !reqOption.ShouldCompress() || contentRangeBytesIsPresent(req.Header) || req.Body == nil {
 		return pipeline.Next(req, middlewareIndex)
 	}
 	if span != nil {
@@ -129,9 +129,14 @@ func (c *CompressionHandler) Intercept(pipeline Pipeline, middlewareIndex int, r
 	return resp, nil
 }
 
-func contentRangeIsPresent(header http.Header) bool {
-	_, contentRangePresent := header["Content-Range"]
-	return contentRangePresent
+func contentRangeBytesIsPresent(header http.Header) bool {
+	contentRanges, _ := header["Content-Range"]
+	for _, contentRange := range contentRanges {
+		if strings.Contains(strings.ToLower(contentRange), "bytes") {
+			return true
+		}
+	}
+	return false
 }
 
 func compressReqBody(reqBody []byte) (io.ReadCloser, int, error) {