Add an option to treat missing data as critical state

gollux · Jun 5, 2020 · 8d6a127 · 8d6a127
1 parent f32c6c1
commit 8d6a127
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/check_influxdb_query.py b/check_influxdb_query.py
@@ -114,6 +114,8 @@ def serialize_influx_series_key(key):
                     help="warning thresholds, Nagios format")
 parser.add_argument("--critical", type=str, required=False,
                     help="critical thresholds, Nagios format")
+parser.add_argument("--missing-as-critical", action='store_const', const=True, required=False,
+                    help="treat missing data as critical state instead of unknown")
 
 parser.add_argument("--output-template", type=str, required=False, default="Values: %s",
                     help="Template for outputing message")
@@ -201,8 +203,12 @@ def serialize_influx_series_key(key):
 # EDGE CASE: NO DATA
 
 if is_resultset_empty:
-    nagios_status = 3
-    nagios_status_desc = 'UNKNOWN'
+    if args.missing_as_critical:
+        nagios_status = 2
+        nagios_status_desc = 'CRITICAL'
+    else:
+        nagios_status = 3
+        nagios_status_desc = 'UNKNOWN'
     message = 'No data returned from query'
     message = nagios_status_desc + " - " + message
     print(message)