[agent-operator] - 增加镜像同步的消息通知内容

agent-go/a_executor/K8sFunction.go

@@ -14,7 +14,6 @@ import (
 	"sync"
 	"time"
 	image2 "wdd.io/agent-common/image"
-	"wdd.io/agent-common/utils"
 )
 
 var k8sConfigFilePath = "/root/wdd/kube_config_cluster.yml"
@@ -142,36 +141,51 @@ func K8sDeploymentUpdateTag(supreme, appName, newTag string) (bool, string) {
 	updateResultLog := ""
 
 	containers := deployment.Spec.Template.Spec.Containers
-	if len(containers) == 1 {
-		// only update this kind
-		container := containers[0]
 
-		oldName := container.Image
+	// 只支持container的数量为1的形式
+	container := containers[0]
 
-		split := strings.Split(container.Image, ":")
-		if strings.HasPrefix(container.Image, image2.CmiiHarborPrefix) {
-			// harbor
-			container.Image = split[0] + ":" + newTag
-		} else if strings.Contains(container.Image, "8033") {
-			// 192.168.6.6:8033/rancher/k8s-dns-sidecar:v1.0.2
-			container.Image = split[0] + ":" + split[1] + ":" + newTag
+	oldName := container.Image
+	split := strings.Split(container.Image, ":")
+	if strings.HasPrefix(container.Image, image2.CmiiHarborPrefix) {
+		// harbor
+		container.Image = split[0] + ":" + newTag
+	} else if strings.Contains(container.Image, "8033") {
+		// 192.168.6.6:8033/rancher/k8s-dns-sidecar:v1.0.2
+		// 重新拼接
+		container.Image = split[0] + ":" + split[1] + ":" + newTag
+	}
+	log.DebugF("[DeploymentUpdateTag] - update [%s] [%s] from [%s] to [%s]", deployment.Namespace, appName, oldName, container.Image)
+
+	// 更新Cmii BIZ_GROUP
+	tagVersion := newTag
+	if strings.Contains(newTag, "-") {
+		tagVersion = strings.Split(newTag, "-")[0]
+	}
+	envList := container.Env
+	for _, envVar := range envList {
+		if envVar.Name == "IMAGE_VERSION" {
+			envVar.Value = tagVersion
 		}
-
-		updateResultLog = fmt.Sprintf(" [%s] [%s] [%s] from [%s] to [%s]", utils.TimeSplitFormatString(), supreme, appName, oldName, container.Image)
-		log.Info(updateResultLog)
-
-		// re assign
-		deployment.Spec.Template.Spec.Containers[0] = container
-
-		// update
-		_, err := k8sClient.AppsV1().Deployments(deployment.Namespace).Update(context.TODO(), deployment, metav1.UpdateOptions{})
-		if err != nil {
-			log.ErrorF("[K8sDeploymentUpdateTag] - update [%s] [%s] from [%s] to [%s] error ! %s", supreme, appName, split[1], container.Image, err.Error())
-			return false, ""
+		if envVar.Name == "BIZ_CONFIG_GROUP" {
+			envVar.Value = tagVersion
 		}
-	} else if len(containers) == 2 {
-		log.ErrorF("[K8sDeploymentUpdateTag] - cant update app with 2 containers !")
-		return false, ""
+		if envVar.Name == "SYS_CONFIG_GROUP" {
+			envVar.Value = tagVersion
+		}
+	}
+	log.DebugF("[DeploymentUpdateTag] -  update env IMAGE_VERSION to [%s]", tagVersion)
+
+	// 赋值回去 很关键
+	deployment.Spec.Template.Spec.Containers[0] = container
+
+	// update
+	_, err := k8sClient.AppsV1().Deployments(deployment.Namespace).Update(context.TODO(), deployment, metav1.UpdateOptions{})
+	if err != nil {
+		sprintf := fmt.Sprintf("[DeploymentUpdateTag] - update [%s] [%s] from [%s] to [%s] error ! %s", deployment.Namespace, appName, split[1], container.Image, err.Error())
+		log.Error(sprintf)
+		return false, sprintf
+
 	}
 
 	return true, updateResultLog