feature(main): fix check image type after image pull (#1066)

labring · May 27, 2022 · 485db0c · 485db0c
1 parent ee17bd0
commit 485db0c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/pkg/apply/processor/create.go b/pkg/apply/processor/create.go
@@ -81,6 +81,10 @@ func (c *CreateProcessor) Check(cluster *v2.Cluster) error {
 	if err != nil {
 		return err
 	}
+
+	return nil
+}
+func (c *CreateProcessor) CheckImageType(cluster *v2.Cluster) error {
 	ociList, err := c.ImageManager.Inspect(cluster.Spec.Image...)
 	if err != nil {
 		return err
@@ -98,12 +102,14 @@ func (c *CreateProcessor) Check(cluster *v2.Cluster) error {
 	}
 	return nil
 }
-
 func (c *CreateProcessor) PreProcess(cluster *v2.Cluster) error {
 	err := c.RegistryManager.Pull(cluster.Spec.Image...)
 	if err != nil {
 		return err
 	}
+	if err = c.CheckImageType(cluster); err != nil {
+		return err
+	}
 	for _, img := range cluster.Spec.Image {
 		clusterManifest, err := c.ClusterManager.Create(fmt.Sprintf("%s-%s", cluster.Name, rand.Generator(8)), img)
 		if err != nil {