fix: donot remove images plugin inuse

Signed-off-by: zwtop <[email protected]>

Author: zwtop

plugin/executor.go

@@ -224,7 +224,14 @@ func (w *executor) Apply(ctx context.Context) error {
 		return fmt.Errorf("start post containers: %s", err)
 	}
 
-	err = w.removeUnusedImages(ctx, w.instance.Containers...)
+	inuseImages := lo.Union(
+		lo.Map(w.instance.PrecheckContainers, func(c model.ContainerDefinition, _ int) string { return c.Image }),
+		lo.Map(w.instance.InitContainers, func(c model.ContainerDefinition, _ int) string { return c.Image }),
+		lo.Map(w.instance.Containers, func(c model.ContainerDefinition, _ int) string { return c.Image }),
+		lo.Map(w.instance.PostContainers, func(c model.ContainerDefinition, _ int) string { return c.Image }),
+		lo.Map(w.instance.CleanContainers, func(c model.ContainerDefinition, _ int) string { return c.Image }),
+	)
+	err = w.removeUnusedImages(ctx, inuseImages...)
 	if err != nil {
 		return fmt.Errorf("remove unused images: %s", err)
 	}
@@ -543,7 +550,7 @@ func (w *executor) removeAllInNamespace(ctx context.Context) error {
 	return nil
 }
 
-func (w *executor) removeUnusedImages(ctx context.Context, exceptImagesInContainer ...model.ContainerDefinition) error {
+func (w *executor) removeUnusedImages(ctx context.Context, inuseImages ...string) error {
 	images, err := w.runtime.ListImages(ctx)
 	if err != nil {
 		return err
@@ -553,10 +560,7 @@ func (w *executor) removeUnusedImages(ctx context.Context, exceptImagesInContain
 	for _, i := range images {
 		imageSet.Insert(i.Name)
 	}
-
-	for _, c := range exceptImagesInContainer {
-		imageSet.Delete(c.Image)
-	}
+	imageSet.Delete(inuseImages...)
 
 	for _, image := range imageSet.List() {
 		w.Infof("remove image %s from containerd", image)