fix: fix load image failed

Signed-off-by: ningmingxiao <ning.mingxiao@zte.com.cn>

Author: ningmingxiao

pkg/imgutil/load/load.go

@@ -60,14 +60,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,24 +78,23 @@ 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,
+	ch := make(chan struct{})
+	err := client.Transfer(ctx,
 		tarchive.NewImageImportStream(options.Stdin, ""),
 		transferimage.NewStore("", storeOpts...),
 		transfer.WithProgress(func(p transfer.Progress) {
 			if p.Event == "saved" {
+				defer close(ch)
 				if img, err := imageService.Get(ctx, p.Name); err == nil {
-					if !beforeSet[img.Name] {
-						loadedImages = append(loadedImages, img)
-					}
+					loadedImages = append(loadedImages, img)
 				}
 			}
 			pf(p)
 		}),
 	)
 
 	done()
-
+	<-ch
 	if !options.Quiet {
 		for _, img := range loadedImages {
 			fmt.Fprintf(options.Stdout, "Loaded image: %s\n", img.Name)