X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=installers%2Fcharm%2Fmongodb-k8s%2Freactive%2Fmongo.py;fp=installers%2Fcharm%2Fmongodb-k8s%2Freactive%2Fmongo.py;h=026f5ed04a2186477bf152a47fae0d688aae3ebf;hb=aef4259b77168e06aedc8e5a71f83cea946b2167;hp=0000000000000000000000000000000000000000;hpb=f1492c324f3be50c6a7b3d421b7f040ac8d1c8fe;p=osm%2Fdevops.git

diff --git a/installers/charm/mongodb-k8s/reactive/mongo.py b/installers/charm/mongodb-k8s/reactive/mongo.py
new file mode 100644
index 00000000..026f5ed0
--- /dev/null
+++ b/installers/charm/mongodb-k8s/reactive/mongo.py
@@ -0,0 +1,128 @@
+# Copyright 2021 Canonical Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+#
+# For those usages not covered by the Apache License, Version 2.0 please
+# contact: legal@canonical.com
+#
+# To get in touch with the maintainers, please contact:
+# osm-charmers@lists.launchpad.net
+##
+
+from charms.layer.caas_base import pod_spec_set
+from charms.reactive import endpoint_from_flag
+from charms.reactive import when, when_not, hook
+from charms.reactive.flags import set_flag, clear_flag
+from charmhelpers.core.hookenv import log, metadata, config, goal_state
+from charms import layer
+
+
+@hook("upgrade-charm")
+@when("leadership.is_leader")
+def upgrade():
+    clear_flag("mongodb-k8s.configured")
+
+
+@when("config.changed")
+@when("leadership.is_leader")
+def restart():
+    clear_flag("mongodb-k8s.configured")
+
+
+@when_not("mongodb-k8s.configured")
+@when("leadership.is_leader")
+def configure():
+    layer.status.maintenance("Configuring MongoDB container")
+    try:
+        spec = make_pod_spec()
+        log("set pod spec:\n{}".format(spec))
+        pod_spec_set(spec)
+        set_flag("mongodb-k8s.configured")
+    except Exception as e:
+        layer.status.blocked("k8s spec failed to deploy: {}".format(e))
+
+
+@when("mongodb-k8s.configured")
+def set_mongodb_active():
+    layer.status.active("ready")
+
+
+@when_not("leadership.is_leader")
+def non_leaders_active():
+    layer.status.active("ready")
+
+
+@when("mongodb-k8s.configured", "mongo.joined")
+def send_config():
+    layer.status.maintenance("Sending mongo configuration")
+    try:
+        mongo = endpoint_from_flag("mongo.joined")
+        if mongo:
+            cfg = config()
+            mongo_uri = "mongodb://"
+            for i, unit in enumerate(goal_state()["units"]):
+                pod_base_name = unit.split("/")[0]
+                service_name = cfg.get("service-name")
+                pod_name = "{}-{}".format(pod_base_name, i)
+                if i:
+                    mongo_uri += ","
+                mongo_uri += "{}.{}:{}".format(
+                    pod_name, service_name, get_mongodb_port()
+                )
+            if cfg.get("enable-sidecar"):
+                mongo_uri += "/?replicaSet={}".format(get_mongodb_replset())
+            log("Mongo URI: {}".format(mongo_uri))
+            mongo.send_connection_string(mongo_uri)
+            clear_flag("mongo.joined")
+    except Exception as e:
+        log("Exception sending config: {}".format(e))
+        clear_flag("mongo.joined")
+
+
+def make_pod_spec():
+    """Make pod specification for Kubernetes
+
+    Returns:
+        pod_spec: Pod specification for Kubernetes
+    """
+    md = metadata()
+    cfg = config()
+
+    if cfg.get("enable-sidecar"):
+        with open("reactive/spec_template_ha.yaml") as spec_file:
+            pod_spec_template = spec_file.read()
+    else:
+        with open("reactive/spec_template.yaml") as spec_file:
+            pod_spec_template = spec_file.read()
+
+    data = {
+        "name": md.get("name"),
+        "docker_image": cfg.get("mongodb-image"),
+        "sc_docker_image": cfg.get("sidecar-image"),
+        "pod_labels": "juju-app={}".format(cfg.get("advertised-hostname")),
+    }
+
+    data.update(cfg)
+    return pod_spec_template % data
+
+
+def get_mongodb_port():
+    """Returns MongoDB port"""
+    cfg = config()
+    return cfg.get("advertised-port")
+
+
+def get_mongodb_replset():
+    """Returns MongoDB port"""
+    cfg = config()
+    return cfg.get("replica-set")