put prometheus behind a node-monitoring flag

antithesishq · Sep 11, 2023 · a16337c · a16337c
1 parent 83a3d7e
commit a16337c
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/src/node_watch.py b/src/node_watch.py
@@ -166,8 +166,8 @@ def perform_action(self):
 
 
 class PrometheusAction(TestnetMonitorAction):
-    def __init__(self):
-        super().__init__(name="prometheus", interval=TestnetMonitorActionInterval.EVERY_SLOT)
+    def __init__(self, interval: TestnetMonitorActionInterval):
+        super().__init__(name="prometheus", interval=interval)
 
     def perform_action(self) -> None:
         queries = {
@@ -183,20 +183,20 @@ def _one_request(key, query):
                 params=dict(query=query)
             )
             if resp.status_code != 200:
-                logging.info(f"prometheus query {query} responded with status code {resp.status_code}, skipping")
+                logging.debug(f"prometheus query {query} responded with status code {resp.status_code}, skipping")
                 return
 
             result_all = resp.json()
             if (result_status := result_all["status"]) != "success":
-                logging.info(f"prometheus query {query} has result status {result_status!r}, skipping")
+                logging.debug(f"prometheus query {query} has result status {result_status!r}, skipping")
                 return
 
             return key, result_all["data"]
 
         with ThreadPoolExecutor(max_workers=8) as exc:
             results = dict(exc.map(lambda item: _one_request(*item), queries.items()))
 
-        logging.info(
+        logging(
             json.dumps({"prometheus_metrics": results})
         )
 
@@ -240,7 +240,8 @@ def build_testnet_monitor(self, cli_args) -> TestnetMonitor:
             "peers": PeersMonitorAction,
             "blob": BlobMonitorAction,
             "execution_availability": HeadsMonitorExecutionAvailabilityCheckAction,
-            "consensus_availability": HeadsMonitorConsensusAvailabilityCheckAction
+            "consensus_availability": HeadsMonitorConsensusAvailabilityCheckAction,
+            "prometheus": PrometheusAction,
         }
 
         intervals = {
@@ -256,6 +257,7 @@ def build_testnet_monitor(self, cli_args) -> TestnetMonitor:
                 raise Exception(f"Unknown metric: {metric}")
             if interval not in intervals:
                 raise Exception(f"Unknown interval: {interval}")
+
 
             _interval = intervals[interval]
             testnet_monitor.add_action(
@@ -268,8 +270,6 @@ def build_testnet_monitor(self, cli_args) -> TestnetMonitor:
                 )
             )
 
-        testnet_monitor.add_action(PrometheusAction())
-
         return testnet_monitor
 
     def get_testnet_info_str(self) -> str: