mirror of
https://github.com/swift-project/pilotclient.git
synced 2026-03-23 23:45:35 +08:00
45 lines
1.5 KiB
C++
45 lines
1.5 KiB
C++
/* Copyright (C) 2014
|
|
* swift project Community / Contributors
|
|
*
|
|
* This file is part of swift project. It is subject to the license terms in the LICENSE file found in the top-level
|
|
* directory of this distribution and at http://www.swift-project.org/license.html. No part of swift project,
|
|
* including this file, may be copied, modified, propagated, or distributed except according to the terms
|
|
* contained in the LICENSE file.
|
|
*/
|
|
|
|
#include "token_bucket.h"
|
|
|
|
namespace BlackCore
|
|
{
|
|
bool CTokenBucket::tryConsume(int numTokens)
|
|
{
|
|
Q_ASSERT(numTokens > 0 && numTokens < m_capacity);
|
|
|
|
// Replenish maximal up to capacity
|
|
int replenishedTokens = qMin(m_capacity, getTokens());
|
|
|
|
// Take care of overflows
|
|
m_availableTokens = qMin(m_availableTokens + replenishedTokens, m_capacity);
|
|
|
|
if (numTokens <= m_availableTokens)
|
|
{
|
|
m_availableTokens -= numTokens;
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
int CTokenBucket::getTokens()
|
|
{
|
|
auto now = QDateTime::currentDateTime();
|
|
auto deltaSeconds = m_lastReplenishmentTime.secsTo(now);
|
|
int numberOfTokens = static_cast<int>( m_numTokensToRefill * deltaSeconds / m_interval.value(BlackMisc::PhysicalQuantities::CTimeUnit::s()));
|
|
|
|
// Update the time only when replenishment actually took place. We will end up in a infinite loop otherwise.
|
|
if (numberOfTokens > 0 ) m_lastReplenishmentTime = now;
|
|
return numberOfTokens;
|
|
}
|
|
|
|
} // namespace
|