fix: Agent exposes the readinessProbe on port 6080 by default if it's…

… not in a container
bytedance · Dec 31, 2024 · b32be51 · b32be51
1 parent c416990
commit b32be51
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/internal/agent/agent.go b/internal/agent/agent.go
@@ -151,8 +151,14 @@ func NewAgent(
 	})
 
 	go func() {
-		if err := r.Run(fmt.Sprintf(":%d", varmorconfig.AgentServicePort)); err != nil {
-			log.Error(err, "fatal error: agent service failed to start")
+		if inContainer {
+			if err := r.Run(fmt.Sprintf(":%d", varmorconfig.AgentServicePort)); err != nil {
+				log.Error(err, "fatal error: agent service failed to start")
+			}
+		} else {
+			if err := r.Run(":6080"); err != nil {
+				log.Error(err, "fatal error: agent service failed to start")
+			}
 		}
 	}()