Feature 11016: Service KPI Metric Based Scaling of VNF using exporter endpoint in...

[osm/NG-SA.git] / src / osm_ngsa / dags / multivim_vim_status.py

diff --git a/src/osm_ngsa/dags/multivim_vim_status.py b/src/osm_ngsa/dags/multivim_vim_status.py
index 5307879..d9f54bb 100644 (file)
--- a/src/osm_ngsa/dags/multivim_vim_status.py
+++ b/src/osm_ngsa/dags/multivim_vim_status.py
@@ -85,7 +85,7 @@ def create_dag(dag_id, dag_number, dag_description, vim_id):
                 vim_type = vim_account["config"]["vim_type"].lower()
                 if vim_type == "vio" and "vrops_site" not in vim_account["config"]:
                     vim_type = "openstack"
-            if vim_type == "openstack":
+            if vim_type == "openstack" or vim_type == "vio":
                 return OpenStackCollector(vim_account)
             if vim_type == "gcp":
                 return GcpCollector(vim_account)
@@ -122,7 +122,8 @@ def create_dag(dag_id, dag_number, dag_description, vim_id):
             if collector:
                 status = collector.is_vim_ok()
                 logger.info(f"VIM status: {status}")
-                metric.labels(vim_id).set(1)
+                if status:
+                    metric.labels(vim_id).set(1)
             else:
                 logger.info("Error creating VIM collector")
             # Push to Prometheus