tokenbucket.h

// SPDX-FileCopyrightText: Copyright (C) 2014 swift Project Community / Contributors
// SPDX-License-Identifier: GPL-3.0-or-later OR LicenseRef-swift-pilot-client-1
 
#ifndef SWIFT_MISC_TOKENBUCKET_H
#define SWIFT_MISC_TOKENBUCKET_H
 
#include <QDateTime>
 
#include "misc/swiftmiscexport.h"
 
namespace swift::misc
{
    class SWIFT_MISC_EXPORT CTokenBucket
    {
    public:
        CTokenBucket(int capacity, qint64 intervalMs, int numTokensToRefill);
 
        bool tryConsume(int numTokens = 1, qint64 msSinceEpoch = -1);
 
        void setNumberOfTokensToRefill(int numTokens);
 
        void setCapacity(int capacity);
 
        void setCapacityAndTokensToRefill(int numTokens);
 
        void setInterval(qint64 intervalMs) { m_intervalMs = intervalMs; }
 
        int getTokensPerSecond() const;
 
    private:
        int getTokens(qint64 msSinceEpoch);
 
        int m_capacity = 10; 
        int m_availableTokens = 10; 
        int m_numTokensToRefill = 1; 
        qint64 m_intervalMs = 5000; 
        qint64 m_lastReplenishmentTime = QDateTime::currentMSecsSinceEpoch(); 
    };
} // namespace swift::misc
 
#endif // SWIFT_MISC_TOKENBUCKET_H