Wait for kubelet port to be ready before setting

Signed-off-by: Daishan Peng <daishan@acorn.io>

Wait for kubelet port to be ready before setting
abda5307 · Daishan Peng · Brad Davidson · 0cf6b03d · abda5307
Commit abda5307 authored Mar 09, 2023 by Daishan Peng Committed by Brad Davidson Mar 13, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

tunnel.go pkg/agent/tunnel/tunnel.go +6 -1

No files found.
--- a/pkg/agent/tunnel/tunnel.go
+++ b/pkg/agent/tunnel/tunnel.go
@@ -160,7 +160,12 @@ func (a *agentTunnel) setKubeletPort(ctx context.Context, apiServerReady <-chan 
 			logrus.Debugf("Tunnel authorizer failed to get Kubelet Port: %v", err)
 			return false, nil
 		}
-		a.kubeletPort = strconv.FormatInt(int64(node.Status.DaemonEndpoints.KubeletEndpoint.Port), 10)
+		kubeletPort := strconv.FormatInt(int64(node.Status.DaemonEndpoints.KubeletEndpoint.Port), 10)
+		if kubeletPort == "0" {
+			logrus.Debugf("Waiting for the kubelet port to be populated")
+			return false, nil
+		}
+		a.kubeletPort = kubeletPort
 		logrus.Infof("Tunnel authorizer set Kubelet Port %s", a.kubeletPort)
 		return true, nil
 	})