[MISC] Fix Juju Spaces lack of addresses (II) (#657)

Author: sinclert-canonical

src/charm.py

@@ -722,6 +722,18 @@ def update_endpoint_addresses(self) -> None:
             f"{RELATION_CONSUMER}-address": self.replication_consumer_address,
         })
 
+    def update_endpoint_address(self, relation_name: str) -> None:
+        """Update ip address for the provided relation on unit peer databag."""
+        logger.debug(f"Updating {relation_name} endpoint address")
+
+        relation_binding = self.model.get_binding(relation_name)
+        if not relation_binding:
+            return
+
+        self.unit_peer_data.update({
+            f"{relation_name}-address": str(relation_binding.network.bind_address)
+        })
+
     def install_workload(self) -> bool:
         """Exponential backoff retry to install and configure MySQL.
 

src/relations/mysql_provider.py

@@ -157,7 +157,7 @@ def _update_endpoints(self, relation_id: int, remote_app: str) -> None:
             relation_id (int): The id of the relation
             remote_app (str): The name of the remote application
         """
-        self.charm.update_endpoint_addresses()
+        self.charm.update_endpoint_address(DB_RELATION_NAME)
 
         try:
             rw_endpoints, ro_endpoints, _ = self.charm.get_cluster_endpoints(DB_RELATION_NAME)
@@ -230,6 +230,9 @@ def _on_database_requested(self, event: DatabaseRequestedEvent):
 
         remote_app = event.app.name
 
+        # Update endpoint addresses
+        self.charm.update_endpoint_address(DB_RELATION_NAME)
+
         try:
             db_version = self.charm._mysql.get_mysql_version()
             rw_endpoints, ro_endpoints, _ = self.charm.get_cluster_endpoints(DB_RELATION_NAME)