Automatically decompress archives for transfer service import

hinshun · hinshun · commit 2aec52493f3d · 2024-03-24T02:17:19.000+08:00
Signed-off-by: Edgar Lee &lt;edgarhinshunlee@gmail.com&gt;
diff --git a/pkg/transfer/archive/importer.go b/pkg/transfer/archive/importer.go
@@ -24,6 +24,7 @@ import (
 	ocispec "github.com/opencontainers/image-spec/specs-go/v1"
 
 	transferapi "github.com/containerd/containerd/api/types/transfer"
+	"github.com/containerd/containerd/archive/compression"
 	"github.com/containerd/containerd/content"
 	"github.com/containerd/containerd/images/archive"
 	"github.com/containerd/containerd/log"
@@ -64,7 +65,18 @@ func (iis *ImageImportStream) Import(ctx context.Context, store content.Store) (
 	if iis.forceCompress {
 		opts = append(opts, archive.WithImportCompression())
 	}
-	return archive.ImportIndex(ctx, store, iis.stream, opts...)
+
+	r := iis.stream
+	if iis.mediaType == "" {
+		d, err := compression.DecompressStream(iis.stream)
+		if err != nil {
+			return ocispec.Descriptor{}, err
+		}
+		defer d.Close()
+		r = d
+	}
+
+	return archive.ImportIndex(ctx, store, r, opts...)
 }
 
 func (iis *ImageImportStream) MarshalAny(ctx context.Context, sm streaming.StreamCreator) (typeurl.Any, error) {
