token_bucket.h

Go to the documentation of this file.

 
#ifndef CPPCOMMON_ALGORITHMS_TOKEN_BUCKET_H
#define CPPCOMMON_ALGORITHMS_TOKEN_BUCKET_H
 
#include <atomic>
#include <cstdint>
 
namespace CppCommon {
 
 
class TokenBucket
{
public:
 
    TokenBucket(uint64_t rate, uint64_t burst);
    TokenBucket(const TokenBucket& tb);
    TokenBucket(TokenBucket&&) = delete;
    ~TokenBucket() = default;
 
    TokenBucket& operator=(const TokenBucket& tb);
    TokenBucket& operator=(TokenBucket&&) = delete;
 
 
    bool Consume(uint64_t tokens = 1);
 
private:
    std::atomic<uint64_t> _time;
    std::atomic<uint64_t> _time_per_token;
    std::atomic<uint64_t> _time_per_burst;
};
 
} // namespace CppCommon
 
#include "token_bucket.inl"
 
#endif // CPPCOMMON_ALGORITHMS_TOKEN_BUCKET_H