Add EASEMESH_TAGS env for agent

megaease · Jan 10, 2022 · e056619 · e056619
1 parent c6f6f4c
commit e056619
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/add-ons/shadow/pkg/handler/clone_handler.go b/add-ons/shadow/pkg/handler/clone_handler.go
@@ -52,6 +52,7 @@ const (
 
 	shadowDeploymentNameSuffix = "-shadow"
 
+	easemeshTagsConfigEnv        = "EASEMESH_TAGS"
 	databaseShadowConfigEnv      = "EASE_RESOURCE_DATABASE"
 	kafkaShadowConfigEnv         = "EASE_RESOURCE_KAFKA"
 	rabbitmqShadowConfigEnv      = "EASE_RESOURCE_RABBITMQ"

diff --git a/add-ons/shadow/pkg/handler/deployment_cloner.go b/add-ons/shadow/pkg/handler/deployment_cloner.go
@@ -207,15 +207,29 @@ func shadowConfigurationMap(shadowService *object.ShadowService) map[string]inte
 	shadowConfigs[redisShadowConfigEnv] = shadowService.Redis
 	shadowConfigs[kafkaShadowConfigEnv] = shadowService.Kafka
 	shadowConfigs[rabbitmqShadowConfigEnv] = shadowService.RabbitMQ
+	shadowConfigs[easemeshTagsConfigEnv] = defaultEaseMeshTags()
 	return shadowConfigs
 }
 
 func shadowConfigurationKeys() []string {
-	configKeys := []string{databaseShadowConfigEnv, elasticsearchShadowConfigEnv, redisShadowConfigEnv, kafkaShadowConfigEnv, rabbitmqShadowConfigEnv}
+	configKeys := []string{
+		databaseShadowConfigEnv,
+		elasticsearchShadowConfigEnv,
+		redisShadowConfigEnv,
+		kafkaShadowConfigEnv,
+		rabbitmqShadowConfigEnv,
+		easemeshTagsConfigEnv,
+	}
 	sort.Strings(configKeys)
 	return configKeys
 }
 
+func defaultEaseMeshTags() map[string]string {
+	m := map[string]string{
+		"label.local": "shadow",
+	}
+	return m
+}
 func sourceName(name string) string {
 	return strings.TrimSuffix(name, shadowDeploymentNameSuffix)
 }