Ydb stable 23-1-1923.1.19

x-stable-origin-commit: c5d5a396e89d0a72e0267a55e93d8404d4fb54fe

1 files changed, 41 insertions, 0 deletions

diff --git a/library/cpp/unified_agent_client/duration_counter.cpp b/library/cpp/unified_agent_client/duration_counter.cpp
new file mode 100644
index 0000000000..118778a226
--- /dev/null
+++ b/library/cpp/unified_agent_client/duration_counter.cpp
@@ -0,0 +1,41 @@
+#include "duration_counter.h"
+
+namespace NUnifiedAgent {
+    using namespace NMonitoring;
+
+    TDurationUsCounter::TDurationUsCounter(const TString& name, TDynamicCounters& owner)
+        : Counter(*owner.GetCounter(name, true))
+        , ActiveTimers()
+        , Lock()
+    {
+    }
+
+    NHPTimer::STime* TDurationUsCounter::Begin() {
+        with_lock (Lock) {
+            ActiveTimers.push_back(0);
+            auto& result = ActiveTimers.back();
+            NHPTimer::GetTime(&result);
+            return &result;
+        }
+    }
+
+    void TDurationUsCounter::End(NHPTimer::STime* startTime) {
+        with_lock (Lock) {
+            Counter += static_cast<ui64>(NHPTimer::GetTimePassed(startTime) * 1000000);
+            *startTime = 0;
+            while (!ActiveTimers.empty() && ActiveTimers.front() == 0) {
+                ActiveTimers.pop_front();
+            }
+        }
+    }
+
+    void TDurationUsCounter::Update() {
+        with_lock (Lock) {
+            for (auto& startTime : ActiveTimers) {
+                if (startTime != 0) {
+                    Counter += static_cast<ui64>(NHPTimer::GetTimePassed(&startTime) * 1000000);
+                }
+            }
+        }
+    }
+}