mirror of https://github.com/waku-org/nwaku.git
25 lines
602 B
Nim
25 lines
602 B
Nim
|
when (NimMajor, NimMinor) < (1, 4):
|
||
|
|
{.push raises: [Defect].}
|
||
|
|
else:
|
||
|
|
{.push raises: [].}
|
||
|
|
|
||
|
|
import std/options
|
||
|
|
import chronos/timer
|
||
|
|
import ./tokenbucket
|
||
|
|
|
||
|
|
export tokenbucket
|
||
|
|
|
||
|
|
type RateLimitSetting* = tuple[volume: int, period: Duration]
|
||
|
|
|
||
|
|
let DefaultGlobalNonRelayRateLimit*: RateLimitSetting = (60, 1.minutes)
|
||
|
|
|
||
|
|
proc newTokenBucket*(setting: Option[RateLimitSetting]): Option[TokenBucket] =
|
||
|
|
if setting.isNone:
|
||
|
|
return none[TokenBucket]()
|
||
|
|
|
||
|
|
let (volume, period) = setting.get()
|
||
|
|
if volume <= 0 or period <= 0.seconds:
|
||
|
|
return none[TokenBucket]()
|
||
|
|
|
||
|
|
return some(TokenBucket.new(volume, period))
|