Ensure that the lead control-plane unit allocates a cos-token for itself

canonical · Aug 29, 2024 · b5f91be · b5f91be
1 parent 673a306
commit b5f91be
Show file tree

Hide file tree

Showing 4 changed files with 194 additions and 105 deletions.
diff --git a/charms/worker/k8s/src/charm.py b/charms/worker/k8s/src/charm.py
@@ -448,10 +448,18 @@ def _create_cos_tokens(self):
 
         log.info("Prepare cos tokens")
         if rel := self.model.get_relation("cos-tokens"):
-            self.distributor.allocate_tokens(relation=rel, token_strategy=TokenStrategy.COS)
+            self.distributor.allocate_tokens(
+                relation=rel,
+                token_strategy=TokenStrategy.COS,
+                token_type=ClusterTokenType.CONTROL_PLANE,
+            )
 
         if rel := self.model.get_relation("cos-worker-tokens"):
-            self.distributor.allocate_tokens(relation=rel, token_strategy=TokenStrategy.COS)
+            self.distributor.allocate_tokens(
+                relation=rel,
+                token_strategy=TokenStrategy.COS,
+                token_type=ClusterTokenType.WORKER,
+            )
 
     @on_error(
         WaitingStatus("Waiting to enable features"),