From: Aleš Mrázek <ales.mrazek@nic.cz>
Date: Fri, 19 Apr 2024 16:19:02 +0000 (+0200)
Subject: manager: statistics: prometheus format for 'predict' module
X-Git-Tag: v6.0.8~25^2
X-Git-Url: http://git.ipfire.org/gitweb/gitweb.cgi?a=commitdiff_plain;h=refs%2Fenvironments%2Fdocs-develop-cach-aroclx%2Fdeployments%2F3861;p=thirdparty%2Fknot-resolver.git

manager: statistics: prometheus format for 'predict' module
---

diff --git a/manager/knot_resolver_manager/statistics.py b/manager/knot_resolver_manager/statistics.py
index 51e8708d6..0dcfdac08 100644
--- a/manager/knot_resolver_manager/statistics.py
+++ b/manager/knot_resolver_manager/statistics.py
@@ -204,6 +204,27 @@ if _prometheus_support:
             value=metrics["query"]["dnssec"],
         )
 
+        if "predict" in metrics:
+            if "epoch" in metrics["predict"]:
+                yield _counter(
+                    "resolver_predict_epoch",
+                    "current prediction epoch (based on time of day and sampling window)",
+                    label=("instance_id", sid),
+                    value=metrics["predict"]["epoch"],
+                )
+            yield _counter(
+                "resolver_predict_queue",
+                "number of queued queries in current window",
+                label=("instance_id", sid),
+                value=metrics["predict"]["queue"],
+            )
+            yield _counter(
+                "resolver_predict_learned",
+                "number of learned queries in current window",
+                label=("instance_id", sid),
+                value=metrics["predict"]["learned"],
+            )
+
     def _create_resolver_metrics_loaded_gauge(kresid: "KresID", loaded: bool) -> GaugeMetricFamily:
         return _gauge(
             "resolver_metrics_loaded",