Remove "primary" label on losing leadership

yuha0 · Jan 28, 2024 · d024064 · d024064
1 parent c6cd4d1
commit d024064
Showing 1 changed file with 15 additions and 7 deletions.
diff --git a/cloudflare-ddns/leaderelection.py b/cloudflare-ddns/leaderelection.py
@@ -49,13 +49,21 @@ def _get_namespace(self):
     def _prepare_callback(self, isleader, cb):
         def wrapper():
             logging.info("I am %s", "the leader" if isleader else "a follower")
-            patches = [
-                {
-                    "op": "replace",
-                    "path": "/metadata/labels/primary",
-                    "value": str(isleader).lower(),
-                }
-            ]
+            if isleader:
+                patches = [
+                    {
+                        "op": "replace",
+                        "path": "/metadata/labels/primary",
+                        "value": "true",
+                    }
+                ]
+            else:
+                patches = [
+                    {
+                        "op": "remove",
+                        "path": "/metadata/labels/primary",
+                    }
+                ]
             self.kclient.patch_namespaced_pod(self.candidate_id, self.ns, body=patches)
             cb()