From 1110808a9d39d4b808aef724c861a2e1a38d2a69 Mon Sep 17 00:00:00 2001 From: Devtools Arcadia <arcadia-devtools@yandex-team.ru> Date: Mon, 7 Feb 2022 18:08:42 +0300 Subject: intermediate changes ref:cde9a383711a11544ce7e107a78147fb96cc4029 --- library/cpp/monlib/metrics/atomics_array.h | 52 ++++++++++++++++++++++++++++++ 1 file changed, 52 insertions(+) create mode 100644 library/cpp/monlib/metrics/atomics_array.h (limited to 'library/cpp/monlib/metrics/atomics_array.h') diff --git a/library/cpp/monlib/metrics/atomics_array.h b/library/cpp/monlib/metrics/atomics_array.h new file mode 100644 index 0000000000..f19aebf291 --- /dev/null +++ b/library/cpp/monlib/metrics/atomics_array.h @@ -0,0 +1,52 @@ +#pragma once + +#include <util/generic/ptr.h> +#include <util/generic/vector.h> + +#include <atomic> + +namespace NMonitoring { + class TAtomicsArray { + public: + explicit TAtomicsArray(size_t size) + : Values_(new std::atomic<ui64>[size]) + , Size_(size) + { + for (size_t i = 0; i < Size_; i++) { + Values_[i].store(0, std::memory_order_relaxed); + } + } + + ui64 operator[](size_t index) const noexcept { + Y_VERIFY_DEBUG(index < Size_); + return Values_[index].load(std::memory_order_relaxed); + } + + size_t Size() const noexcept { + return Size_; + } + + void Add(size_t index, ui32 count) noexcept { + Y_VERIFY_DEBUG(index < Size_); + Values_[index].fetch_add(count, std::memory_order_relaxed); + } + + void Reset() noexcept { + for (size_t i = 0; i < Size_; i++) { + Values_[i].store(0, std::memory_order_relaxed); + } + } + + TVector<ui64> Copy() const { + TVector<ui64> copy(Reserve(Size_)); + for (size_t i = 0; i < Size_; i++) { + copy.push_back(Values_[i].load(std::memory_order_relaxed)); + } + return copy; + } + + private: + TArrayHolder<std::atomic<ui64>> Values_; + size_t Size_; + }; +} -- cgit v1.2.3