aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/messagebus/misc/tokenquota.h
diff options
context:
space:
mode:
authorAnton Samokhvalov <pg83@yandex.ru>2022-02-10 16:45:15 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:45:15 +0300
commit72cb13b4aff9bc9cf22e49251bc8fd143f82538f (patch)
treeda2c34829458c7d4e74bdfbdf85dff449e9e7fb8 /library/cpp/messagebus/misc/tokenquota.h
parent778e51ba091dc39e7b7fcab2b9cf4dbedfb6f2b5 (diff)
downloadydb-72cb13b4aff9bc9cf22e49251bc8fd143f82538f.tar.gz
Restoring authorship annotation for Anton Samokhvalov <pg83@yandex.ru>. Commit 1 of 2.
Diffstat (limited to 'library/cpp/messagebus/misc/tokenquota.h')
-rw-r--r--library/cpp/messagebus/misc/tokenquota.h30
1 files changed, 15 insertions, 15 deletions
diff --git a/library/cpp/messagebus/misc/tokenquota.h b/library/cpp/messagebus/misc/tokenquota.h
index 190547fa54..d5bdd79625 100644
--- a/library/cpp/messagebus/misc/tokenquota.h
+++ b/library/cpp/messagebus/misc/tokenquota.h
@@ -25,26 +25,26 @@ namespace NBus {
Y_UNUSED(padd_);
}
- bool Acquire(TAtomic level = 1, bool force = false) {
- level = Max(TAtomicBase(level), TAtomicBase(1));
+ bool Acquire(TAtomic level = 1, bool force = false) {
+ level = Max(TAtomicBase(level), TAtomicBase(1));
- if (Enabled && (Acquired < level || force)) {
+ if (Enabled && (Acquired < level || force)) {
Acquired += AtomicSwap(&Tokens_, 0);
}
return !Enabled || Acquired >= level;
}
- void Consume(size_t items) {
- if (Enabled) {
+ void Consume(size_t items) {
+ if (Enabled) {
Y_ASSERT(Acquired >= TAtomicBase(items));
Acquired -= items;
}
}
- bool Return(size_t items_) noexcept {
- if (!Enabled || items_ == 0)
+ bool Return(size_t items_) noexcept {
+ if (!Enabled || items_ == 0)
return false;
const TAtomic items = items_;
@@ -53,31 +53,31 @@ namespace NBus {
return (value - items < WakeLev && value >= WakeLev);
}
- bool IsEnabled() const noexcept {
+ bool IsEnabled() const noexcept {
return Enabled;
}
- bool IsAboveWake() const noexcept {
+ bool IsAboveWake() const noexcept {
return !Enabled || (WakeLev <= AtomicGet(Tokens_));
}
- size_t Tokens() const noexcept {
+ size_t Tokens() const noexcept {
return Acquired + AtomicGet(Tokens_);
}
- size_t Check(const TAtomic level) const noexcept {
+ size_t Check(const TAtomic level) const noexcept {
return !Enabled || level <= Acquired;
}
private:
- bool Enabled;
- TAtomicBase Acquired;
+ bool Enabled;
+ TAtomicBase Acquired;
const TAtomicBase WakeLev;
- TAtomic Tokens_;
+ TAtomic Tokens_;
/* This padd requires for align Tokens_ member on its own
CPU cacheline. */
- ui64 padd_;
+ ui64 padd_;
};
}