| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | // ====================================================================== | ||
| 2 | // \title TokenBucket.cpp | ||
| 3 | // \author vwong | ||
| 4 | // \brief cpp file for a rate limiter utility class | ||
| 5 | // | ||
| 6 | // \copyright | ||
| 7 | // | ||
| 8 | // Copyright (C) 2009-2020 California Institute of Technology. | ||
| 9 | // | ||
| 10 | // ALL RIGHTS RESERVED. United States Government Sponsorship | ||
| 11 | // acknowledged. | ||
| 12 | // ====================================================================== | ||
| 13 | |||
| 14 | #include <Utils/TokenBucket.hpp> | ||
| 15 | #include <algorithm> | ||
| 16 | |||
| 17 | namespace Utils { | ||
| 18 | |||
| 19 | 4 | TokenBucket ::TokenBucket(U32 replenishInterval, U32 maxTokens, U32 replenishRate, U32 startTokens, Fw::Time startTime) | |
| 20 | 4 | : m_replenishInterval(replenishInterval), | |
| 21 | 4 | m_maxTokens(maxTokens), | |
| 22 | 4 | m_replenishRate(replenishRate), | |
| 23 | 4 | m_tokens(startTokens), | |
| 24 | 4 | m_time(startTime) {} | |
| 25 | |||
| 26 | 6 | TokenBucket ::TokenBucket(U32 replenishInterval, U32 maxTokens) | |
| 27 | 6 | : m_replenishInterval(replenishInterval), | |
| 28 | 6 | m_maxTokens(maxTokens), | |
| 29 | 6 | m_replenishRate(1), | |
| 30 | 6 | m_tokens(maxTokens), | |
| 31 | 6 | m_time(0, 0) { | |
| 32 | 6 | FW_ASSERT(this->m_maxTokens <= MAX_TOKEN_BUCKET_TOKENS, static_cast<FwAssertArgType>(this->m_maxTokens)); | |
| 33 | 6 | } | |
| 34 | |||
| 35 | 2 | void TokenBucket ::setReplenishInterval(U32 replenishInterval) { | |
| 36 | 2 | this->m_replenishInterval = replenishInterval; | |
| 37 | 2 | } | |
| 38 | |||
| 39 | 1 | void TokenBucket ::setMaxTokens(U32 maxTokens) { | |
| 40 | 1 | this->m_maxTokens = maxTokens; | |
| 41 | 1 | } | |
| 42 | |||
| 43 | 1 | void TokenBucket ::setReplenishRate(U32 replenishRate) { | |
| 44 | 1 | this->m_replenishRate = replenishRate; | |
| 45 | 1 | } | |
| 46 | |||
| 47 | 5 | void TokenBucket ::replenish() { | |
| 48 |
2/2✓ Branch 4 taken 4 times.
✓ Branch 5 taken 1 times.
|
5 | if (this->m_tokens < this->m_maxTokens) { |
| 49 | 4 | this->m_tokens = this->m_maxTokens; | |
| 50 | } | ||
| 51 | 5 | } | |
| 52 | |||
| 53 | 2 | U32 TokenBucket ::getReplenishInterval() const { | |
| 54 | 2 | return this->m_replenishInterval; | |
| 55 | } | ||
| 56 | |||
| 57 | 2 | U32 TokenBucket ::getMaxTokens() const { | |
| 58 | 2 | return this->m_maxTokens; | |
| 59 | } | ||
| 60 | |||
| 61 | 2 | U32 TokenBucket ::getReplenishRate() const { | |
| 62 | 2 | return this->m_replenishRate; | |
| 63 | } | ||
| 64 | |||
| 65 | 900 | U32 TokenBucket ::getTokens() const { | |
| 66 | 900 | return this->m_tokens; | |
| 67 | } | ||
| 68 | |||
| 69 | 5352 | bool TokenBucket ::trigger(const Fw::Time time) { | |
| 70 | // attempt replenishing | ||
| 71 |
2/2✓ Branch 2 taken 5349 times.
✓ Branch 3 taken 3 times.
|
5352 | if (this->m_replenishRate > 0) { |
| 72 |
1/1✓ Branch 4 taken 5349 times.
|
5349 | Fw::Time replenishInterval = Fw::Time(this->m_replenishInterval / 1000000, this->m_replenishInterval % 1000000); |
| 73 |
1/1✓ Branch 5 taken 5349 times.
|
5349 | Fw::Time nextTime = Fw::Time::add(this->m_time, replenishInterval); |
| 74 |
7/7✓ Branch 4 taken 6454 times.
✓ Branch 5 taken 13 times.
✓ Branch 8 taken 6454 times.
✓ Branch 10 taken 1118 times.
✓ Branch 11 taken 5336 times.
✓ Branch 12 taken 1118 times.
✓ Branch 13 taken 5349 times.
|
6467 | while (this->m_tokens < this->m_maxTokens && nextTime <= time) { |
| 75 | // replenish by replenish rate, or up to maxTokens | ||
| 76 | 1118 | this->m_tokens += std::min(this->m_replenishRate, this->m_maxTokens - this->m_tokens); | |
| 77 |
1/1✓ Branch 6 taken 1118 times.
|
1118 | this->m_time = nextTime; |
| 78 |
2/2✓ Branch 5 taken 1118 times.
✓ Branch 10 taken 1118 times.
|
1118 | nextTime = Fw::Time::add(this->m_time, replenishInterval); |
| 79 | } | ||
| 80 |
7/7✓ Branch 4 taken 13 times.
✓ Branch 5 taken 5336 times.
✓ Branch 12 taken 13 times.
✓ Branch 14 taken 1 times.
✓ Branch 15 taken 12 times.
✓ Branch 16 taken 1 times.
✓ Branch 17 taken 5348 times.
|
5349 | if (this->m_tokens >= this->m_maxTokens && this->m_time < time) { |
| 81 |
1/1✓ Branch 6 taken 1 times.
|
1 | this->m_time = time; |
| 82 | } | ||
| 83 | 5349 | } | |
| 84 | |||
| 85 | // attempt consuming token | ||
| 86 |
2/2✓ Branch 2 taken 2904 times.
✓ Branch 3 taken 2448 times.
|
5352 | if (this->m_tokens > 0) { |
| 87 | 2904 | this->m_tokens--; | |
| 88 | 2904 | return true; | |
| 89 | |||
| 90 | } else { | ||
| 91 | 2448 | return false; | |
| 92 | } | ||
| 93 | } | ||
| 94 | |||
| 95 | } // end namespace Utils | ||
| 96 |