use util.hostNameEnv instead KUBE_NODE_NAME

kubeovn · Aug 21, 2021 · 0380d64 · 0380d64
1 parent 38e04f3
commit 0380d64
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/daemon/gateway.go b/pkg/daemon/gateway.go
@@ -641,7 +641,7 @@ func (c *Controller) setExGateway() error {
 
 func (c *Controller) getLocalPodIPsNeedNAT(protocol string) ([]string, error) {
 	var localPodIPs []string
-	hostname := os.Getenv("KUBE_NODE_NAME")
+	hostname := os.Getenv(util.HostnameEnv)
 	allPods, err := c.podsLister.List(labels.Everything())
 	if err != nil {
 		klog.Errorf("list pods failed, %+v", err)
@@ -684,7 +684,7 @@ func (c *Controller) getLocalPodIPsNeedPR(protocol string) (map[policyRouteMeta]
 		return nil, err
 	}
 
-	hostname := os.Getenv("KUBE_NODE_NAME")
+	hostname := os.Getenv(util.HostnameEnv)
 	localPodIPs := make(map[policyRouteMeta][]string)
 	for _, pod := range allPods {
 		if pod.Spec.HostNetwork ||