filter based on node label being set to enablelb

node_external_ip_controller.py

@@ -11,7 +11,7 @@ v1 = client.CoreV1Api()
 SERVICE_NAME = "traefik"
 NAMESPACE = "kube-system"
 ANNOTATION_KEY = "kube-vip.io/loadbalancerIPs"
-NODE_ANNOTATION_KEY = "svccontroller.k3s.cattle.io/enablelb"
+NODE_LABEL = "svccontroller.k3s.cattle.io/enablelb=true"
 
 
 def update_service_annotation(external_ip):
@@ -29,22 +29,32 @@ def update_service_annotation(external_ip):
 
 def main():
     w = watch.Watch()
-    for event in w.stream(v1.list_node, _request_timeout=60):
-        node = event["object"]
-        node_name = node.metadata.name
-        is_gateway = node.metadata.annotations.get(NODE_ANNOTATION_KEY)
-        print(f"{node_name} is gateway?: {is_gateway}, type:{type(is_gateway)}")
+    while True:
+        try:
+            for event in w.stream(
+                v1.list_node, label_selector=NODE_LABEL, _request_timeout=300
+            ):
+                node = event["object"]
+                node_name = node.metadata.name
 
-        # Extract the external IP if it exists
-        external_ip = None
-        for address in node.status.addresses:
-            if address.type == "ExternalIP" and is_gateway == True:
-                external_ip = address.address
-                break
+                # Extract the external IP if it exists
+                external_ip = None
+                for address in node.status.addresses:
+                    if address.type == "ExternalIP" and is_gateway == True:
+                        external_ip = address.address
+                        break
 
-        if external_ip:
-            print(f"Detected external IP {external_ip} for node {node_name}")
-            update_service_annotation(external_ip)
+                if external_ip:
+                    print(f"Detected external IP {external_ip} for node {node_name}")
+                    update_service_annotation(external_ip)
+
+        except client.exceptions.ApiException as e:
+            print(f"API Exception: {e}")
+            time.sleep(5)  # Wait before retrying
+
+        except Exception as e:
+            print(f"Unexpected error: {e}")
+            time.sleep(5)
 
 
 if __name__ == "__main__":