agent: Fix init logging

Fix bug when logger was initialized in the wrong place. This commit puts the init phase in the _Execute_ method of the run command. Signed-off-by: Cosmin Tupangiu <[email protected]>
kubev2v · Dec 5, 2024 · e772cf4 · e772cf4
1 parent 3d33aa9
commit e772cf4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/cmd/planner-agent/main.go b/cmd/planner-agent/main.go
@@ -54,6 +54,10 @@ func NewAgentCommand() *agentCmd {
 		zap.S().Fatalf("Error validating config: %v", err)
 	}
 
+	return a
+}
+
+func (a *agentCmd) Execute() error {
 	logLvl, err := zap.ParseAtomicLevel(a.config.LogLevel)
 	if err != nil {
 		logLvl = zap.NewAtomicLevelAt(zapcore.InfoLevel)
@@ -65,10 +69,6 @@ func NewAgentCommand() *agentCmd {
 	undo := zap.ReplaceGlobals(logger)
 	defer undo()
 
-	return a
-}
-
-func (a *agentCmd) Execute() error {
 	agentID, err := a.getAgentID()
 	if err != nil {
 		zap.S().Fatalf("failed to retreive agent_id: %v", err)