prob-rate-limiter

ProbRateLimiter is a probabilistic rate limiter. When load approaches the configured limit, the struct chooses randomly whether to accept or reject each request. It adjusts the probability of rejection so throughput is steady around the limit.

Use Cases

• Shed load to prevent overload
• Avoid overloading the services you depend on
• Control costs

Features

• Tiny, uses 44 bytes
• 100% test coverage
• Optimized: 32ns per check, 31M checks per second on an i5-8259U
• No unsafe or unsafe deps

Limitations

• Requires a mutable reference.
• Not fair. Treats all requests equally, regardless of source. A client that overloads the server will consume most of the throughput.

Alternatives

• r8limit
  • Uses a sliding window
  • No unsafe or deps
  • Optimized: 48ns per check, 21M checks per second on an i5-8259U
  • Requires a mutable reference.
• governor
  • Uses a non-mutable reference, easy to share between threads
  • Popular
  • Good docs
  • Optimized: 29ns per check on an i5-8259U.
  • Unnecessary unsafe
  • Uses non-standard mutex library parking_lot
  • Uses a complicated algorithm
• leaky-bucket
  • Async tasks can wait for their turn to use a resource.
  • Unsuitable for load shedding because there is no try_acquire.

Related Crates

• safe-dns uses this

Example

rust let mut limiter = ProbRateLimiter::new(10.0).unwrap(); let mut now = Instant::now(); assert!(limiter.check(5, now)); assert!(limiter.check(5, now)); now += Duration::from_secs(1); assert!(limiter.check(5, now)); assert!(limiter.check(5, now)); now += Duration::from_secs(1); assert!(limiter.check(5, now)); assert!(limiter.check(5, now)); now += Duration::from_secs(1); assert!(limiter.check(5, now)); assert!(limiter.check(5, now)); now += Duration::from_secs(1); assert!(limiter.check(5, now)); assert!(limiter.check(5, now)); assert!(!limiter.check(5, now));

Cargo Geiger Safety Report

```

Metric output format: x/y x = unsafe code used by the build y = total unsafe code found in the crate

Symbols: 🔒 = No unsafe usage found, declares #![forbid(unsafecode)] ❓ = No unsafe usage found, missing #![forbid(unsafecode)] ☢️ = unsafe usage found

Functions Expressions Impls Traits Methods Dependency

0/0 0/0 0/0 0/0 0/0 🔒 prob-rate-limiter 0.1.0 0/0 0/0 0/0 0/0 0/0 🔒 └── oorandom 11.1.3

0/0 0/0 0/0 0/0 0/0

```

Changelog

• v0.1.0 - Initial version

TO DO

• Publish
• Add graph from the benchmark.

License: Apache-2.0