# 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
-from charmhelpers.core.hookenv import log, metadata, config
-from charms import layer
-from charms.osm.k8s import get_service_ip
@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 mysql to be ready")
+ 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")
-
- if mysql.host() is not None:
- spec = make_pod_spec(
- mysql.host(),
- mysql.port(),
- mysql.user(),
- mysql.password(),
- mysql.root_password(),
- )
-
+ 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")
layer.status.blocked("k8s spec failed to deploy: {}".format(e))
-@when("mysql.available")
+@when("ro-k8s.configured")
@when_not("leadership.is_leader")
def non_leaders_active():
layer.status.active("ready")
service_ip = get_service_ip("ro")
if service_ip:
ro.send_connection(
- service_ip, get_ro_port(),
+ service_ip,
+ get_ro_port(),
)
clear_flag("ro.joined")
except Exception as e:
data = {
"name": md.get("name"),
- "mysql_host": mysql_host,
- "mysql_port": mysql_port,
- "mysql_user": mysql_user,
- "mysql_password": mysql_password,
- "mysql_root_password": mysql_root_password,
}
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()
- return pod_spec_template % data
+ 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,
+ }
+ )
+ return spec
def get_ro_port():
projects / osm / devops.git / commitdiff