Set Content-Length for blob uploads (#1781)

Sometimes we know it, sometimes we don't. When we do, set it.
google · Sep 5, 2023 · a748190 · a748190
1 parent 190ad0e
commit a748190
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/v1/remote/write.go b/pkg/v1/remote/write.go
@@ -280,6 +280,11 @@ func (w *writer) streamBlob(ctx context.Context, layer v1.Layer, streamLocation
 	if _, ok := layer.(*stream.Layer); !ok {
 		// We can't retry streaming layers.
 		req.GetBody = getBody
+
+		// If we know the size, set it.
+		if size, err := layer.Size(); err == nil {
+			req.ContentLength = size
+		}
 	}
 	req.Header.Set("Content-Type", "application/octet-stream")