feat: Add vector monitoring

openedx · Mar 18, 2024 · 4d45504 · 4d45504
1 parent 83dc67e
commit 4d45504
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/platform_plugin_aspects/management/commands/monitor_load_test_tracking.py b/platform_plugin_aspects/management/commands/monitor_load_test_tracking.py
@@ -234,16 +234,16 @@ def get_vector_stats(self):
         metrics = r.json()["data"]["sinks"]["edges"][0]["node"]["metrics"]
 
         # These will be null until events start arriving
-        received = metrics.get("receivedEventsTotal", {}).get("receivedEventsTotal", 0)
-        sent = metrics.get("sentEventsTotal", {}).get("sentEventsTotal", 0)
+        received = metrics["receivedEventsTotal"]["receivedEventsTotal"] if metrics["receivedEventsTotal"] else 0
+        sent = metrics["sentEventsTotal"]["sentEventsTotal"] if metrics["sentEventsTotal"] else 0
 
         rtn = {
             "events_received": received,
             "events_sent": sent,
             "lag": received - sent
         }
 
-        log.info(f"Vector lag: {rtn.get('lag')}")
+        log.info(f"Vector received: {rtn['events_received']} sent: {rtn['events_sent']} lag: {rtn.get('lag')}")
         return rtn