From a6aae19706829e6489f32bcf253fe2b5dd5e0a9d Mon Sep 17 00:00:00 2001
From: Idriss Neumann <idriss.neumann@comwork.io>
Date: Fri, 1 Nov 2024 13:05:35 +0100
Subject: [PATCH] Add kind as label

---
 VERSION                           |  2 +-
 ansible-imalive/defaults/main.yml |  2 +-
 src/utils/monitor.py              | 20 ++++++++++----------
 3 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/VERSION b/VERSION
index a2cec7a..7919852 100644
--- a/VERSION
+++ b/VERSION
@@ -1 +1 @@
-4.0.8
+4.0.9
diff --git a/ansible-imalive/defaults/main.yml b/ansible-imalive/defaults/main.yml
index 60dcacd..3326c5c 100644
--- a/ansible-imalive/defaults/main.yml
+++ b/ansible-imalive/defaults/main.yml
@@ -1,5 +1,5 @@
 ---
-imalive_api_version: 4.0.8
+imalive_api_version: 4.0.9
 imalive_port: 8099
 imalive_wait_time: 300
 imalive_force_recreate: true
diff --git a/src/utils/monitor.py b/src/utils/monitor.py
index c6b43d5..4d5f906 100644
--- a/src/utils/monitor.py
+++ b/src/utils/monitor.py
@@ -33,7 +33,7 @@ def check_http_monitor(monitor, gauges):
             "message": "Not an http monitor",
             "monitor": monitor 
         })
-        set_gauge(gauges['result'], 0, labels)
+        set_gauge(gauges['result'], 0, {**labels, 'kind': 'result'})
         return
 
     if is_empty_key(monitor, 'url'):
@@ -44,7 +44,7 @@ def check_http_monitor(monitor, gauges):
             "message": "Missing mandatory url",
             "monitor": monitor 
         })
-        set_gauge(gauges['result'], 0, labels)
+        set_gauge(gauges['result'], 0, {**labels, 'kind': 'result'})
         return
 
     method = get_or_else(monitor, 'method', 'GET')
@@ -65,11 +65,11 @@ def check_http_monitor(monitor, gauges):
         if method == "GET":
             response = requests.get(monitor['url'], auth=auth, timeout=timeout)
             duration = response.elapsed.total_seconds()
-            set_gauge(gauges['duration'], duration, labels)
+            set_gauge(gauges['duration'], duration, {**labels, 'kind': 'duration'})
         elif method == "POST":
             response = requests.post(monitor['url'], auth=auth, timeout=timeout)
             duration = response.elapsed.total_seconds()
-            set_gauge(gauges['duration'], duration, labels)
+            set_gauge(gauges['duration'], duration, {**labels, 'kind': 'duration'})
         else:
             log_msg("ERROR", { 
                 "status": "ko",
@@ -78,7 +78,7 @@ def check_http_monitor(monitor, gauges):
                 "message": "Not supported http method: actual = {}".format(method),
                 "monitor": pmonitor
             })
-            set_gauge(gauges['result'], 0, labels)
+            set_gauge(gauges['result'], 0, {**labels, 'kind': 'result'})
             return
 
         if response.status_code != expected_http_code:
@@ -90,7 +90,7 @@ def check_http_monitor(monitor, gauges):
                 "message": "Not expected status code: expected = {}, actual = {}".format(expected_http_code, response.status_code),
                 "monitor": pmonitor
             })
-            set_gauge(gauges['result'], 0, labels)
+            set_gauge(gauges['result'], 0, {**labels, 'kind': 'result'})
             return
 
         if is_not_empty(expected_contain) and expected_contain not in response.text:
@@ -102,10 +102,10 @@ def check_http_monitor(monitor, gauges):
                 "message": "Response not valid: expected = {}, actual = {}".format(expected_contain, response.text),
                 "monitor": pmonitor
             })
-            set_gauge(gauges['result'], 0, labels)
+            set_gauge(gauges['result'], 0, {**labels, 'kind': 'result'})
             return
 
-        set_gauge(gauges['result'], 1, labels)
+        set_gauge(gauges['result'], 1, {**labels, 'kind': 'result'})
         log_msg("INFO", { 
             "status": "ok",
             "type": "monitor",
@@ -116,7 +116,7 @@ def check_http_monitor(monitor, gauges):
         })
 
     except Exception as e:
-        set_gauge(gauges['result'], 0, labels)
+        set_gauge(gauges['result'], 0, {**labels, 'kind': 'result'})
         log_msg("ERROR", {
             "status": "ko",
             "type": "monitor",
@@ -129,7 +129,7 @@ def check_http_monitor(monitor, gauges):
 
 gauges = {}
 def monitors():
-    labels = ['name', 'family']
+    labels = ['name', 'family', 'kind']
     def loop_monitors():
         config_path = os.path.realpath(os.path.join(os.path.dirname(__file__), '..', '..', 'imalive.yml'))
         with open(config_path, "r") as stream: