import logging
from typing import Any, Dict
-from urllib.parse import urlparse
from charms.data_platform_libs.v0.data_interfaces import DatabaseRequires
from charms.kafka_k8s.v0.kafka import KafkaEvents, KafkaRequires
from ops.main import main
from ops.model import ActiveStatus, Container
-from legacy_interfaces import KeystoneClient
+from legacy_interfaces import KeystoneClient, PrometheusClient
HOSTPATHS = [
HostPath(
self.mongodb_client = DatabaseRequires(
self, "mongodb", database_name="osm", extra_user_roles="admin"
)
+ self.prometheus_client = PrometheusClient(self, "prometheus")
self.keystone_client = KeystoneClient(self, "keystone")
self._observe_charm_events()
self.container: Container = self.unit.get_container("nbi")
def _on_update_status(self, _=None) -> None:
"""Handler for the update-status event."""
try:
- self._validate_config()
self._check_relations()
if self.debug_mode.started:
return
self.on["kafka"].relation_broken: self._on_required_relation_broken,
self.mongodb_client.on.database_created: self._on_config_changed,
self.on["mongodb"].relation_broken: self._on_required_relation_broken,
- self.on["keystone"].relation_changed: self._on_config_changed,
- self.on["keystone"].relation_broken: self._on_required_relation_broken,
# Action events
self.on.get_debug_mode_information_action: self._on_get_debug_mode_information_action,
self.on.nbi_relation_joined: self._update_nbi_relation,
}
+ for relation in [self.on[rel_name] for rel_name in ["prometheus", "keystone"]]:
+ event_handler_mapping[relation.relation_changed] = self._on_config_changed
+ event_handler_mapping[relation.relation_broken] = self._on_required_relation_broken
for event, handler in event_handler_mapping.items():
self.framework.observe(event, handler)
CharmError: if charm configuration is invalid.
"""
logger.debug("validating charm config")
- url = self.config.get("prometheus-url")
- if not url:
- raise CharmError("need prometheus-url config")
- if not self._is_valid_url(url):
- raise CharmError(f"Invalid value for prometheus-url config: '{url}'")
-
- def _is_valid_url(self, url) -> bool:
- return urlparse(url).hostname is not None
def _check_relations(self) -> None:
"""Validate charm relations.
missing_relations.append("kafka")
if not self._is_database_available():
missing_relations.append("mongodb")
+ if self.prometheus_client.is_missing_data_in_app():
+ missing_relations.append("prometheus")
if self.keystone_client.is_missing_data_in_app():
missing_relations.append("keystone")
def _get_layer(self) -> Dict[str, Any]:
"""Get layer for Pebble."""
- prometheus = urlparse(self.config["prometheus-url"])
return {
"summary": "nbi layer",
"description": "pebble config layer for nbi",
"OSMNBI_STORAGE_COLLECTION": "files",
"OSMNBI_STORAGE_URI": self._get_mongodb_uri(),
# Prometheus configuration
- "OSMNBI_PROMETHEUS_HOST": prometheus.hostname,
- "OSMNBI_PROMETHEUS_PORT": prometheus.port if prometheus.port else "",
+ "OSMNBI_PROMETHEUS_HOST": self.prometheus_client.hostname,
+ "OSMNBI_PROMETHEUS_PORT": self.prometheus_client.port,
# Log configuration
"OSMNBI_LOG_LEVEL": self.config["log-level"],
# Authentication environments