fix: fix load image failed

Signed-off-by: ningmingxiao <ning.mingxiao@zte.com.cn>
Signed-off-by: 宁明晓10296073 <ning.mingxiao@zte.com.cn>

Author: ningmingxiao

pkg/imgutil/load/load.go

@@ -22,6 +22,7 @@ import (
 	"fmt"
 	"os"
 	"strings"
+	"sync"
 
 	containerd "github.com/containerd/containerd/v2/client"
 	"github.com/containerd/containerd/v2/core/images"
@@ -60,14 +61,6 @@ func FromArchive(ctx context.Context, client *containerd.Client, options types.I
 	}
 
 	imageService := client.ImageService()
-	beforeImages, err := imageService.List(ctx)
-	if err != nil {
-		return nil, err
-	}
-	beforeSet := make(map[string]bool)
-	for _, img := range beforeImages {
-		beforeSet[img.Name] = true
-	}
 
 	var storeOpts []transferimage.StoreOpt
 	platUnpack := platforms.DefaultSpec()
@@ -86,16 +79,16 @@ func FromArchive(ctx context.Context, client *containerd.Client, options types.I
 
 	var loadedImages []images.Image
 	pf, done := transferutil.ProgressHandler(ctx, options.Stdout)
-
-	err = client.Transfer(ctx,
+	var mu sync.Mutex
+	err := client.Transfer(ctx,
 		tarchive.NewImageImportStream(options.Stdin, ""),
 		transferimage.NewStore("", storeOpts...),
 		transfer.WithProgress(func(p transfer.Progress) {
+			mu.Lock()
+			defer mu.Unlock()
 			if p.Event == "saved" {
 				if img, err := imageService.Get(ctx, p.Name); err == nil {
-					if !beforeSet[img.Name] {
-						loadedImages = append(loadedImages, img)
-					}
+					loadedImages = append(loadedImages, img)
 				}
 			}
 			pf(p)
@@ -105,9 +98,11 @@ func FromArchive(ctx context.Context, client *containerd.Client, options types.I
 	done()
 
 	if !options.Quiet {
+		mu.Lock()
 		for _, img := range loadedImages {
 			fmt.Fprintf(options.Stdout, "Loaded image: %s\n", img.Name)
 		}
+		mu.Unlock()
 	}
 
 	return loadedImages, err