update

apecloud · Aug 28, 2024 · 9700082 · 9700082
1 parent 82ea5d3
commit 9700082
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/pkg/controller/component/kbagent.go b/pkg/controller/component/kbagent.go
@@ -132,7 +132,7 @@ func buildKBAgentContainer(synthesizedComp *SynthesizedComponent) error {
 		}
 	}
 
-	eye, err := buildKBAgentEyeContainer(discovery)
+	eye, err := buildKBAgentEyeContainer(discovery, envVars)
 	if err != nil {
 		return err
 	}
@@ -496,7 +496,7 @@ func mountPathExists(volumeMounts []corev1.VolumeMount, mountPath string) bool {
 	return false
 }
 
-func buildKBAgentEyeContainer(discovery map[string]string) (*corev1.Container, error) {
+func buildKBAgentEyeContainer(discovery map[string]string, envVar []corev1.EnvVar) (*corev1.Container, error) {
 	var env corev1.EnvVar
 	dd, err := json.Marshal(discovery)
 	if err != nil {
@@ -511,6 +511,7 @@ func buildKBAgentEyeContainer(discovery map[string]string) (*corev1.Container, e
 		SetImage(viper.GetString(constant.KBToolsImage)).
 		SetImagePullPolicy(corev1.PullIfNotPresent).
 		AddCommands(kbAgentCommand).
+		AddEnv(envVar...).
 		AddEnv(env).
 		GetObject(), nil
 }
diff --git a/pkg/kbagent/service/service.go b/pkg/kbagent/service/service.go
@@ -37,6 +37,9 @@ type Service interface {
 }
 
 func New(logger logr.Logger, actions []proto.Action, probes []proto.Probe) ([]Service, error) {
+	actions = append(actions, proto.Action{
+		Name: "eye",
+	})
 	sa, err := newActionService(logger, actions)
 	if err != nil {
 		return nil, err

diff --git a/pkg/kbagent/setup.go b/pkg/kbagent/setup.go
@@ -33,7 +33,7 @@ import (
 const (
 	ContainerName     = "kbagent"
 	InitContainerName = "init-kbagent"
-	EyeContainerName  = "kbagent-eye"
+	EyeContainerName  = "kba-eye"
 	DefaultPortName   = "http"
 
 	actionEnvName = "KB_AGENT_ACTION"