X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=installers%2Fcharm%2Fosm-mon%2Fsrc%2Fcharm.py;h=45874624d8c893b8f18645890813691aeca5cf9e;hb=refs%2Fchanges%2F56%2F13556%2F27;hp=045fd981856726b4938a62a0205fe6db370c587a;hpb=7da7a3343ea03386d498eaa6b6a93eb3f770d6cd;p=osm%2Fdevops.git diff --git a/installers/charm/osm-mon/src/charm.py b/installers/charm/osm-mon/src/charm.py index 045fd981..45874624 100755 --- a/installers/charm/osm-mon/src/charm.py +++ b/installers/charm/osm-mon/src/charm.py @@ -22,14 +22,16 @@ # # Learn more at: https://juju.is/docs/sdk -"""OSM NBI charm. +"""OSM MON charm. See more: https://charmhub.io/osm """ 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 KafkaRequires, _KafkaAvailableEvent from charms.observability_libs.v1.kubernetes_service_patch import KubernetesServicePatch from charms.osm_libs.v0.utils import ( @@ -46,7 +48,12 @@ from ops.framework import EventSource, StoredState from ops.main import main from ops.model import ActiveStatus, Container -from legacy_interfaces import KeystoneClient, MongoClient, PrometheusClient +from grafana_datasource_handler import ( + DatasourceConfig, + GrafanaConfig, + GrafanaDataSourceHandler, +) +from legacy_interfaces import KeystoneClient, PrometheusClient HOSTPATHS = [ HostPath( @@ -85,7 +92,7 @@ class OsmMonCharm(CharmBase): def __init__(self, *args): super().__init__(*args) self.kafka = KafkaRequires(self) - self.mongodb_client = MongoClient(self, "mongodb") + self.mongodb_client = DatabaseRequires(self, "mongodb", database_name="osm") self.prometheus_client = PrometheusClient(self, "prometheus") self.keystone_client = KeystoneClient(self, "keystone") self.vca = VcaRequires(self) @@ -154,9 +161,53 @@ class OsmMonCharm(CharmBase): event.fail("debug-mode has not started. Hint: juju config mon debug-mode=true") return - debug_info = {"command": self.debug_mode.command, "password": self.debug_mode.password} + debug_info = { + "command": self.debug_mode.command, + "password": self.debug_mode.password, + } event.set_results(debug_info) + def _on_create_datasource_action(self, event: ActionEvent) -> None: + """Handler for the create-datasource action event.""" + url = event.params["url"] + if not self._is_valid_url(url): + event.fail(f"Invalid datasource url '{url}'") + return + grafana_config = self._get_grafana_config() + datasource_config = DatasourceConfig(event.params["name"], url) + response = GrafanaDataSourceHandler.create_datasource(grafana_config, datasource_config) + logger.debug(response) + if response.is_success: + event.set_results(response.results) + return + event.fail(response.message) + + def _on_list_datasources_action(self, event: ActionEvent) -> None: + """Handler for the list-datasource action event.""" + grafana_config = self._get_grafana_config() + response = GrafanaDataSourceHandler.list_datasources(grafana_config) + logger.debug(response) + if response.is_success: + event.set_results(response.results) + return + event.fail(response.message) + + def _on_delete_datasource_action(self, event: ActionEvent) -> None: + """Handler for the delete-datasource action event.""" + datasource_name = event.params["name"] + grafana_config = self._get_grafana_config() + response = GrafanaDataSourceHandler.delete_datasource(grafana_config, datasource_name) + logger.debug(response) + if not response.is_success: + event.fail(response.message) + + def _get_grafana_config(self) -> GrafanaConfig: + return GrafanaConfig( + self.config.get("grafana-user", ""), + self.config.get("grafana-password", ""), + self.config.get("grafana-url", ""), + ) + # --------------------------------------------------------------------------- # Validation and configuration and more # --------------------------------------------------------------------------- @@ -171,16 +222,27 @@ class OsmMonCharm(CharmBase): self.on.vca_data_changed: self._on_config_changed, self.on.kafka_available: self._on_config_changed, 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, # Action events self.on.get_debug_mode_information_action: self._on_get_debug_mode_information_action, + self.on.create_datasource_action: self._on_create_datasource_action, + self.on.list_datasources_action: self._on_list_datasources_action, + self.on.delete_datasource_action: self._on_delete_datasource_action, } - for relation in [self.on[rel_name] for rel_name in ["mongodb", "prometheus", "keystone"]]: + 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) + def _is_database_available(self) -> bool: + try: + return self.mongodb_client.is_resource_created() + except KeyError: + return False + def _validate_config(self) -> None: """Validate charm configuration. @@ -188,6 +250,24 @@ class OsmMonCharm(CharmBase): CharmError: if charm configuration is invalid. """ logger.debug("validating charm config") + missing_configs = [] + grafana_configs = ["grafana-url", "grafana-user", "grafana-password"] + for config in grafana_configs: + if not self.config.get(config): + missing_configs.append(config) + + if missing_configs: + config_str = ", ".join(missing_configs) + error_msg = f"need {config_str} config" + logger.warning(error_msg) + raise CharmError(error_msg) + + grafana_url = self.config["grafana-url"] + if not self._is_valid_url(grafana_url): + raise CharmError(f"Invalid value for grafana-url config: '{grafana_url}'") + + def _is_valid_url(self, url) -> bool: + return urlparse(url).hostname is not None def _check_relations(self) -> None: """Validate charm relations. @@ -200,7 +280,7 @@ class OsmMonCharm(CharmBase): if not self.kafka.host or not self.kafka.port: missing_relations.append("kafka") - if self.mongodb_client.is_missing_data_in_unit(): + if not self._is_database_available(): missing_relations.append("mongodb") if self.prometheus_client.is_missing_data_in_app(): missing_relations.append("prometheus") @@ -229,20 +309,31 @@ class OsmMonCharm(CharmBase): "OSMMON_GLOBAL_REQUEST_TIMEOUT": self.config["global-request-timeout"], "OSMMON_COLLECTOR_INTERVAL": self.config["collector-interval"], "OSMMON_EVALUATOR_INTERVAL": self.config["evaluator-interval"], + "OSMMON_COLLECTOR_VM_INFRA_METRICS": self.config["vm-infra-metrics"], # Kafka configuration "OSMMON_MESSAGE_DRIVER": "kafka", "OSMMON_MESSAGE_HOST": self.kafka.host, "OSMMON_MESSAGE_PORT": self.kafka.port, # Database configuration "OSMMON_DATABASE_DRIVER": "mongo", - "OSMMON_DATABASE_URI": self.mongodb_client.connection_string, + "OSMMON_DATABASE_URI": self._get_mongodb_uri(), "OSMMON_DATABASE_COMMONKEY": self.config["database-commonkey"], # Prometheus/grafana configuration "OSMMON_PROMETHEUS_URL": f"http://{self.prometheus_client.hostname}:{self.prometheus_client.port}", - "OSMMON_GRAFANA_URL": self.config["grafana-url"], - "OSMMON_GRAFANA_USER": self.config["grafana-user"], - "OSMMON_GRAFANA_PASSWORD": self.config["grafana-password"], + "OSMMON_PROMETHEUS_USER": self.prometheus_client.user, + "OSMMON_PROMETHEUS_PASSWORD": self.prometheus_client.password, + "OSMMON_GRAFANA_URL": self.config.get("grafana-url", ""), + "OSMMON_GRAFANA_USER": self.config.get("grafana-user", ""), + "OSMMON_GRAFANA_PASSWORD": self.config.get("grafana-password", ""), + "OSMMON_KEYSTONE_ENABLED": self.config["keystone-enabled"], + "OSMMON_KEYSTONE_URL": self.keystone_client.host, + "OSMMON_KEYSTONE_DOMAIN_NAME": self.keystone_client.user_domain_name, + "OSMMON_KEYSTONE_SERVICE_PROJECT": self.keystone_client.service, + "OSMMON_KEYSTONE_SERVICE_USER": self.keystone_client.username, + "OSMMON_KEYSTONE_SERVICE_PASSWORD": self.keystone_client.password, + "OSMMON_KEYSTONE_SERVICE_PROJECT_DOMAIN_NAME": self.keystone_client.project_domain_name, } + logger.info(f"{environment}") if self.vca.data: environment["OSMMON_VCA_HOST"] = self.vca.data.endpoints environment["OSMMON_VCA_SECRET"] = self.vca.data.secret @@ -264,6 +355,9 @@ class OsmMonCharm(CharmBase): }, } + def _get_mongodb_uri(self): + return list(self.mongodb_client.fetch_relation_data().values())[0]["uris"] + def _patch_k8s_service(self) -> None: port = ServicePort(SERVICE_PORT, name=f"{self.app.name}") self.service_patcher = KubernetesServicePatch(self, [port])