infra
/
node-external-ip-controller
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compare commits

base: infra:363c514f8a078256b3048fffecf5c502425577b5
Branches Tags
infra:main
...
compare: infra:1f9f45baf57e8ccf8a528a688650e4329dee9981
Branches Tags
infra:main

2 Commits
363c514f8a ... 1f9f45baf5

Author SHA1 Message Date
Malar Invention 1f9f45baf5 update service annotation with ip of node that is set as gateway 2024-10-31 22:32:17 +05:30
Malar Invention 564dc1d9a1 update service ip based on external ip 2024-10-31 21:24:39 +05:30
1 changed files with 9 additions and 5 deletions
Show Stats Expand all files Collapse all files

14
node_external_ip_controller.py
View File

@ -8,8 +8,10 @@ config.load_incluster_config()
v1 = client.CoreV1Api() v1 = client.CoreV1Api()
# Configuration # Configuration
SERVICE_NAME = "YOUR_SERVICE_NAME" SERVICE_NAME = "traefik"
NAMESPACE = "YOUR_NAMESPACE" NAMESPACE = "kube-system"
ANNOTATION_KEY = "kube-vip.io/loadbalancerIPs"
NODE_ANNOTATION_KEY = "svccontroller.k3s.cattle.io/enablelb"
def update_service_annotation(external_ip): def update_service_annotation(external_ip):
@ -17,10 +19,10 @@ def update_service_annotation(external_ip):
service = v1.read_namespaced_service(SERVICE_NAME, NAMESPACE) service = v1.read_namespaced_service(SERVICE_NAME, NAMESPACE)
# Check if the annotation needs to be updated # Check if the annotation needs to be updated
current_annotation = service.metadata.annotations.get("external-ip") current_annotation = service.metadata.annotations.get(ANNOTATION_KEY)
if current_annotation != external_ip: if current_annotation != external_ip:
# Update the annotation # Update the annotation
body = {"metadata": {"annotations": {"external-ip": external_ip}}} body = {"metadata": {"annotations": {ANNOTATION_KEY: external_ip}}}
v1.patch_namespaced_service(SERVICE_NAME, NAMESPACE, body) v1.patch_namespaced_service(SERVICE_NAME, NAMESPACE, body)
print(f"Updated service {SERVICE_NAME} with new external IP: {external_ip}") print(f"Updated service {SERVICE_NAME} with new external IP: {external_ip}")
@ -30,11 +32,13 @@ def main():
for event in w.stream(v1.list_node, _request_timeout=60): for event in w.stream(v1.list_node, _request_timeout=60):
node = event["object"] node = event["object"]
node_name = node.metadata.name 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)}")
# Extract the external IP if it exists # Extract the external IP if it exists
external_ip = None external_ip = None
for address in node.status.addresses: for address in node.status.addresses:
if address.type == "ExternalIP": if address.type == "ExternalIP" and is_gateway == True:
external_ip = address.address external_ip = address.address
break break