update ds names on relation removal

canonical · Nov 28, 2024 · 3c2a2ee · 3c2a2ee
1 parent 82dc57d
commit 3c2a2ee
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lib/charms/grafana_k8s/v0/grafana_source.py b/lib/charms/grafana_k8s/v0/grafana_source.py
@@ -676,6 +676,12 @@ def _remove_source_from_datastore(self, event: RelationDepartedEvent) -> bool:
                     self._remove_source(host["source_name"])
 
             self.set_peer_data("sources", stored_sources)
+
+            # remove this datasource name from the published datasource names shared with the datasource units
+            self._publish_source_uids(
+                event.relation, {ds["unit"]: ds["source_name"] for ds in stored_sources[rel_id]}
+            )
+
             return True
         return False