roxy-proxy 0.1.0

High-performance Ethereum JSON-RPC proxy
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

//! Benchmarks for rate limiter
#![allow(missing_docs)]

use std::time::Duration;

use criterion::{Criterion, black_box, criterion_group, criterion_main};
use roxy_rpc::{RateLimiterConfig, SlidingWindowRateLimiter};
use roxy_traits::RateLimiter;

fn bench_rate_limiter_check(c: &mut Criterion) {
    let config =
        RateLimiterConfig { requests_per_window: 1000, window_duration: Duration::from_secs(1) };
    let limiter = SlidingWindowRateLimiter::new(config);

    c.bench_function("rate_limiter_check_allowed", |b| {
        b.iter(|| {
            let _ = limiter.check(black_box("client_1"));
        })
    });
}

fn bench_rate_limiter_many_clients(c: &mut Criterion) {
    let config =
        RateLimiterConfig { requests_per_window: 1000, window_duration: Duration::from_secs(1) };
    let limiter = SlidingWindowRateLimiter::new(config);

    // Pre-populate with many clients
    for i in 0..1000 {
        let _ = limiter.check_and_record(&format!("client_{}", i));
    }

    c.bench_function("rate_limiter_many_clients", |b| {
        let mut i = 0;
        b.iter(|| {
            let client = format!("client_{}", i % 1000);
            i += 1;
            let _ = limiter.check(black_box(&client));
        })
    });
}

criterion_group!(benches, bench_rate_limiter_check, bench_rate_limiter_many_clients);
criterion_main!(benches);