diff options
| author | Devtools Arcadia <arcadia-devtools@yandex-team.ru> | 2022-02-07 18:08:42 +0300 |
|---|---|---|
| committer | Devtools Arcadia <arcadia-devtools@mous.vla.yp-c.yandex.net> | 2022-02-07 18:08:42 +0300 |
| commit | 1110808a9d39d4b808aef724c861a2e1a38d2a69 (patch) | |
| tree | e26c9fed0de5d9873cce7e00bc214573dc2195b7 /library/cpp/histogram/adaptive/merger.h | |
| download | ydb-1110808a9d39d4b808aef724c861a2e1a38d2a69.tar.gz | |
intermediate changes
ref:cde9a383711a11544ce7e107a78147fb96cc4029
Diffstat (limited to 'library/cpp/histogram/adaptive/merger.h')
| -rw-r--r-- | library/cpp/histogram/adaptive/merger.h | 68 |
1 files changed, 68 insertions, 0 deletions
diff --git a/library/cpp/histogram/adaptive/merger.h b/library/cpp/histogram/adaptive/merger.h new file mode 100644 index 0000000000..fc9a6b6a4f --- /dev/null +++ b/library/cpp/histogram/adaptive/merger.h @@ -0,0 +1,68 @@ +#pragma once + +#include <util/generic/buffer.h> + +namespace NKiwiAggr { + class IMerger { + private: + bool IsMerged; + ui32 AutoMergeInterval; // Call Merge() after each AutoMergeInterval calls of Add(); zero means no autoMerge + ui32 NotMergedCount; + + public: + IMerger(ui32 autoMergeInterval = 0) + : IsMerged(true) + , AutoMergeInterval(autoMergeInterval) + , NotMergedCount(0) + { + } + + virtual ~IMerger() { + } + + // returns true if something is added + virtual bool Add(const void* data, size_t size) { + if (AddImpl(data, size)) { + AutoMerge(); + return true; + } + return false; + } + + virtual void Merge() { + if (!IsMerged) { + MergeImpl(); + IsMerged = true; + } + } + + virtual void Reset() { + ResetImpl(); + IsMerged = true; + } + + // You can add some more result-getters if you want. + // Do not forget to call Merge() in the beginning of each merger. + virtual void GetResult(TBuffer& buffer) = 0; + + protected: + // AutoMerge() is called in Add() after each AddImpl() + void AutoMerge() { + IsMerged = false; + if (AutoMergeInterval) { + ++NotMergedCount; + if (NotMergedCount >= AutoMergeInterval) { + MergeImpl(); + IsMerged = true; + NotMergedCount = 0; + } + } + } + + // Implementation of merger: define it in derivatives + virtual bool AddImpl(const void* data, size_t size) = 0; // returns true if something is added + virtual void MergeImpl() = 0; + virtual void ResetImpl() = 0; + }; + +} |