prometheus: track calls per system

USA-RedDragon · Nov 8, 2023 · 7c981b2 · 7c981b2
1 parent e625628
commit 7c981b2
Showing 1 changed file with 28 additions and 2 deletions.
diff --git a/prometheus_plugin.cc b/prometheus_plugin.cc
@@ -42,6 +42,7 @@ class Prometheus : public Plugin_Api
 
   prometheus::Family<prometheus::Counter> *http_requests_counter;
   prometheus::Family<prometheus::Gauge> *active_calls;
+  prometheus::Family<prometheus::Counter> *calls_counter;
 
 public:
   // Factory method
@@ -69,6 +70,11 @@ class Prometheus : public Plugin_Api
                           .Help("Number of active calls")
                           .Register(*registry);
 
+    this->calls_counter = &BuildCounter()
+                          .Name("calls")
+                          .Help("Call history")
+                          .Register(*registry);
+
     this->http_requests_counter = &BuildCounter()
                                   .Name("http_requests_total")
                                   .Help("Number of HTTP requests served")
@@ -89,23 +95,43 @@ class Prometheus : public Plugin_Api
     int num = calls.size();
     std::string s = std::to_string(num);
     this->active_calls->Add({}).Set(num);
+    std::map<std::string, int> callsPerSystem;
 
     auto ret = 0;
     for (std::vector<Call *>::iterator it = calls.begin(); it != calls.end(); it++)
     {
       Call *call = *it;
-      this->active_calls->Add({
+
+      std::string callSystem = call->get_system()->get_short_name();
+      if (callsPerSystem.find(callSystem) == callsPerSystem.end())
+      {
+        callsPerSystem[callSystem] = 1;
+      }
+      else
+      {
+        callsPerSystem[callSystem] += 1;
+      }
+
+      this->calls_counter->Add({
         {"call_id", std::to_string(call->get_call_num())},
-        {"call_system", call->get_system()->get_short_name()},
+        {"call_system", callSystem},
         {"encrypted", std::to_string(call->get_encrypted())},
         {"talkgroup", std::to_string(call->get_talkgroup())},
       }).Increment();
+
       ret = this->update_call_metrics(call);
       if (ret != 0)
       {
         return ret;
       }
     }
+
+    for (auto& callPerSystem : callsPerSystem)
+    {
+      this->active_calls->Add({
+        {"system", callPerSystem.first},
+      }).Set(callPerSystem.second);
+    }
     return ret;
   }