[Agent][Deploy] - optimize cmii image

agent-common/image/ImageNameConvert.go

@@ -1,6 +1,8 @@
 package image
 
 import (
+	"os"
+	"path/filepath"
 	"strings"
 	"wdd.io/agent-common/logger"
 )
@@ -172,3 +174,31 @@ func GzipFileNameToImageNameAndTag(gzipFileName string) (imageName, imageTag str
 
 	return "", ""
 }
+
+func GzipFolderPathToCmiiImageTagMaps(gzipFolderPath string) (frontendImageVersionMap, backendImageVersionMap, srsImageVersionMap map[string]string) {
+
+	frontendImageVersionMap = make(map[string]string)
+	backendImageVersionMap = make(map[string]string)
+	srsImageVersionMap = make(map[string]string)
+
+	filepath.WalkDir(gzipFolderPath, func(path string, d os.DirEntry, err error) error {
+		//fmt.Println(path)
+		name := d.Name()
+		if strings.HasSuffix(name, ".tar.gz") {
+			imageName, imageTag := GzipFileNameToImageNameAndTag(name)
+			if strings.Contains(imageName, "platform") {
+				frontendImageVersionMap[imageName] = imageTag
+			} else if strings.Contains(imageName, "srs") {
+				srsImageVersionMap[imageName] = imageTag
+			} else if strings.Contains(imageName, "operator") {
+				srsImageVersionMap[imageName] = imageTag
+			} else {
+				backendImageVersionMap[imageName] = imageTag
+			}
+		}
+
+		return nil
+	})
+
+	return frontendImageVersionMap, backendImageVersionMap, srsImageVersionMap
+}

agent-common/image/ImageNameConvert_test.go

@@ -2,7 +2,11 @@ package image
 
 import (
 	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
 	"testing"
+	"wdd.io/agent-common/utils"
 )
 
 func TestImageFullNameToGzipFileName(t *testing.T) {
@@ -195,3 +199,34 @@ func TestGzipFileNameToImageFullName(t *testing.T) {
 		fmt.Println("--------------------")
 	}
 }
+
+func TestImageGzipFileNameToImageFullName(t *testing.T) {
+
+	frontendMap := make(map[string]string)
+	backendMap := make(map[string]string)
+	srsMap := make(map[string]string)
+
+	gzipFilePrefix := "/root/octopus_image/xjyd/"
+	filepath.WalkDir(gzipFilePrefix, func(path string, d os.DirEntry, err error) error {
+		//fmt.Println(path)
+		name := d.Name()
+		if strings.HasSuffix(name, ".tar.gz") {
+			imageName, imageTag := GzipFileNameToImageNameAndTag(name)
+			if strings.Contains(imageName, "platform") {
+				frontendMap[imageName] = imageTag
+			} else if strings.Contains(imageName, "srs") {
+				srsMap[imageName] = imageTag
+			} else if strings.Contains(imageName, "operator") {
+				srsMap[imageName] = imageTag
+			} else {
+				backendMap[imageName] = imageTag
+			}
+		}
+
+		return nil
+	})
+
+	utils.BeautifulPrint(backendMap)
+	utils.BeautifulPrint(frontendMap)
+	utils.BeautifulPrint(srsMap)
+}