Mongod-exporter charm: fixing connection string in HA

[osm/devops.git] / installers / charm / mongodb-exporter / src / pod_spec.py

diff --git a/installers/charm/mongodb-exporter/src/pod_spec.py b/installers/charm/mongodb-exporter/src/pod_spec.py
index 781c689..8255b20 100644 (file)
--- a/installers/charm/mongodb-exporter/src/pod_spec.py
+++ b/installers/charm/mongodb-exporter/src/pod_spec.py
@@ -122,10 +122,15 @@ def _make_pod_envconfig(
     Returns:
         Dict[str, Any]: pod environment configuration.
     """
+    parsed = urlparse(relation_state.get("mongodb_connection_string"))
+
     envconfig = {
-        "MONGODB_URI": relation_state.get("mongodb_connection_string"),
+        "MONGODB_URI": f"mongodb://{parsed.netloc.split(',')[0]}{parsed.path}",
     }
 
+    if parsed.query:
+        envconfig["MONGODB_URI"] += f"?{parsed.query}"
+
     return envconfig