+++ /dev/null
-# Copyright 2020 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.
-import yaml
-
-from charmhelpers.core.hookenv import log, metadata, config
-from charms import layer
-from charms.layer.caas_base import pod_spec_set
-from charms.osm.k8s import get_service_ip
-from charms.reactive import endpoint_from_flag
-from charms.reactive import when, when_not, hook
-from charms.reactive.flags import set_flag, clear_flag
-
-
-@hook("upgrade-charm")
-@when("leadership.is_leader")
-def upgrade():
- clear_flag("ro-k8s.configured")
- clear_flag("ro-k8s.ready")
-
-
-@when("config.changed")
-@when("leadership.is_leader")
-def restart():
- clear_flag("ro-k8s.configured")
- clear_flag("ro-k8s.ready")
-
-
-@when_not("mongo.ready")
-@when_not("mysql.available")
-@when_not("ro-k8s.configured")
-@when("leadership.is_leader")
-def waiting_for_mysql():
- layer.status.waiting("Waiting for db to be ready")
- clear_flag("ro-k8s.ready")
-
-
-@when_not("kafka.ready")
-@when_not("ro-k8s.configured")
-@when("leadership.is_leader")
-def waiting_for_kafka():
- layer.status.waiting("Waiting for kafka to be ready")
- clear_flag("ro-k8s.ready")
-
-
-@when_not("ro-k8s.ready")
-@when("mysql.available")
-def ro_k8s_mysql_ready():
- set_flag("ro-k8s.ready")
-
-
-@when_not("ro-k8s.ready")
-@when("kafka.ready")
-@when("mongo.ready")
-def ro_k8s_kafka_mongo_ready():
- set_flag("ro-k8s.ready")
-
-
-@when("ro-k8s.ready")
-@when_not("ro-k8s.configured")
-@when("leadership.is_leader")
-def configure():
- layer.status.maintenance("Configuring ro container")
- try:
- mysql = endpoint_from_flag("mysql.available")
- kafka = endpoint_from_flag("kafka.ready")
- mongo = endpoint_from_flag("mongo.ready")
- spec = None
- if mysql:
- if mysql.host() is not None:
- spec = make_pod_spec(
- mysql.host(),
- mysql.port(),
- mysql.user(),
- mysql.password(),
- mysql.root_password(),
- )
- elif kafka and mongo:
- kafka_units = kafka.kafkas()
- kafka_unit = kafka_units[0]
- mongo_uri = mongo.connection_string()
-
- if (
- mongo_uri
- and kafka_unit["host"]
- # and kafka_unit["port"]
- ):
- spec = make_pod_spec_new_ro(
- kafka_unit["host"],
- # kafka_unit["port"],
- mongo_uri,
- )
- if spec:
- log("set pod spec:\n{}".format(spec))
- pod_spec_set(spec)
- layer.status.active("creating container")
- set_flag("ro-k8s.configured")
- except Exception as e:
- layer.status.blocked("k8s spec failed to deploy: {}".format(e))
-
-
-@when("ro-k8s.ready")
-@when_not("leadership.is_leader")
-def non_leaders_active():
- layer.status.active("ready")
-
-
-@when("ro-k8s.configured")
-def set_ro_active():
- layer.status.active("ready")
-
-
-@when("ro-k8s.configured", "ro.joined")
-def send_config():
- layer.status.maintenance("Sending RO configuration")
- try:
- ro = endpoint_from_flag("ro.joined")
- if ro:
- service_ip = get_service_ip("ro")
- if service_ip:
- ro.send_connection(
- service_ip,
- get_ro_port(),
- )
- clear_flag("ro.joined")
- except Exception as e:
- log("Fail sending RO configuration: {}".format(e))
-
-
-def make_pod_spec(
- mysql_host, mysql_port, mysql_user, mysql_password, mysql_root_password
-):
- """Make pod specification for Kubernetes
-
- Args:
- mysql_name (str): RO DB name
- mysql_host (str): RO DB host
- mysql_port (int): RO DB port
- mysql_user (str): RO DB user
- mysql_password (str): RO DB password
- Returns:
- pod_spec: Pod specification for Kubernetes
- """
-
- with open("reactive/spec_template.yaml") as spec_file:
- pod_spec_template = spec_file.read()
-
- md = metadata()
- cfg = config()
-
- data = {
- "name": md.get("name"),
- }
- data.update(cfg)
- spec = yaml.safe_load(pod_spec_template % data)
- spec["containers"][0]["config"].update(
- {
- "RO_DB_HOST": mysql_host,
- "RO_DB_PORT": mysql_port,
- "RO_DB_NAME": cfg.get("ro_database"),
- "RO_DB_USER": mysql_user,
- "RO_DB_ROOT_PASSWORD": mysql_root_password,
- "RO_DB_PASSWORD": mysql_password,
- "RO_DB_OVIM_PASSWORD": mysql_password,
- "RO_DB_OVIM_HOST": mysql_host,
- "RO_DB_OVIM_PORT": mysql_port,
- "RO_DB_OVIM_USER": mysql_user,
- "RO_DB_OVIM_ROOT_PASSWORD": mysql_root_password,
- "RO_DB_OVIM_NAME": cfg.get("vim_database"),
- }
- )
- return spec
-
-
-def make_pod_spec_new_ro(kafka_host, mongodb_uri):
- """Make pod specification for Kubernetes
-
- Args:
- kafka_host (str): Kafka host
- mongodb_uri (str): Mongodb URI
- Returns:
- pod_spec: Pod specification for Kubernetes
- """
-
- with open("reactive/spec_template.yaml") as spec_file:
- pod_spec_template = spec_file.read()
-
- md = metadata()
- cfg = config()
-
- data = {
- "name": md.get("name"),
- }
- data.update(cfg)
- spec = yaml.safe_load(pod_spec_template % data)
- spec["containers"][0]["config"].update(
- {
- "OSMRO_DATABASE_URI": mongodb_uri,
- "OSMRO_MESSAGE_HOST": kafka_host,
- "OSMRO_DATABASE_COMMONKEY": cfg.get("database_commonkey"),
- }
- )
- return spec
-
-
-def get_ro_port():
- """Returns RO port"""
- cfg = config()
- return cfg.get("advertised-port")