77 lines
1.7 KiB
C++
77 lines
1.7 KiB
C++
|
/***
|
||
|
Copyright (C) 20212J Reece Wilson (a/k/a "Reece"). All rights reserved.
|
||
|
|
||
|
File: RateLimiter.hpp
|
||
|
Date: 2022-2-8
|
||
|
Author: Reece
|
||
|
***/
|
||
|
#pragma once
|
||
|
|
||
|
namespace Aurora::Utility
|
||
|
{
|
||
|
struct RateLimiter
|
||
|
{
|
||
|
AuUInt64 nextTriggerTime;
|
||
|
AuUInt64 nsTimeStep;
|
||
|
|
||
|
inline auline void SetNextStep(AuUInt64 nsTimeStep)
|
||
|
{
|
||
|
this->nsTimeStep = nsTimeStep;
|
||
|
this->nextTriggerTime = nsTimeStep + Aurora::Time::CurrentInternalClockNS();
|
||
|
}
|
||
|
|
||
|
inline auline bool CheckExchangePass()
|
||
|
{
|
||
|
auto cur = this->nextTriggerTime;
|
||
|
|
||
|
if (!cur)
|
||
|
{
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
if (cur <= Aurora::Time::CurrentInternalClockNS())
|
||
|
{
|
||
|
if (this->nsTimeStep)
|
||
|
{
|
||
|
return AuAtomicCompareExchange(&this->nextTriggerTime, GetLatch(), cur) == cur;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
return true;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
inline auline bool HasPassed()
|
||
|
{
|
||
|
if (!this->nextTriggerTime)
|
||
|
{
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
if (this->nextTriggerTime <= Aurora::Time::CurrentInternalClockNS())
|
||
|
{
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
inline auline void Relatch()
|
||
|
{
|
||
|
if (this->nsTimeStep)
|
||
|
{
|
||
|
AuAtomicAdd(&this->nextTriggerTime, GetLatch());
|
||
|
}
|
||
|
}
|
||
|
|
||
|
private:
|
||
|
|
||
|
inline auline AuUInt64 GetLatch()
|
||
|
{
|
||
|
return this->nextTriggerTime + this->nsTimeStep;
|
||
|
}
|
||
|
};
|
||
|
};
|