Moved logic to lib/

lib/api_metrics.py

@@ -0,0 +1,58 @@
+import os
+import time
+import openstack
+import datetime
+import traceback
+import prometheus_client as prom
+
+api_metrics = prom.Gauge('openstack_api_response_seconds', 'Time for openstack api to execute.', ['api_name'])
+api_status = prom.Gauge('openstack_api_status', 'API current status. 1 = up 0 = down.',['api_name'])
+
+def generate_nova_metrics(connection):
+    try:
+        start_time = datetime.datetime.now()
+        for server in connection.compute.servers():
+            name = server
+        end_time = datetime.datetime.now()
+        time_took = end_time - start_time
+        seconds_took = time_took.seconds
+        print(f'Nova took {seconds_took} seconds')
+        api_metrics.labels('nova').set(seconds_took)
+        api_status.labels('nova').set(1)
+    except:
+        print(traceback.print_exc())
+        print("Nova api is down.")
+        api_status.labels('nova').set(0)
+
+def generate_neutron_metrics(connection):
+    try:
+        project = connection.current_project
+        start_time = datetime.datetime.now()
+        for network in connection.network.networks(project_id=project.id):
+            name = network
+        end_time = datetime.datetime.now()
+        time_took = end_time - start_time
+        seconds_took = time_took.seconds
+        print(f'Neutron took {seconds_took} seconds')
+        api_metrics.labels('neutron').set(seconds_took)
+        api_status.labels('neutron').set(1)
+    except:
+        print(traceback.print_exc())
+        print("Neutron api is down.")
+        api_status.labels('neutron').set(0)
+
+def generate_cinder_metrics(connection):
+    try:
+        start_time = datetime.datetime.now()
+        for volume in  connection.volume.volumes():
+            name = volume
+        end_time = datetime.datetime.now()
+        time_took = end_time - start_time
+        seconds_took = time_took.seconds
+        print(f'Cinder took {seconds_took} seconds')
+        api_metrics.labels('cinder').set(seconds_took)
+        api_status.labels('cinder').set(1)
+    except:
+        print(traceback.print_exc())
+        print("Cinder api is down.")
+        api_status.labels('cinder').set(0)