koios-sdk 0.1.1

A Rust SDK for the Koios Cardano API
Documentation 
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
44
45
46
47
48
49

// src/client/rate_limit.rs

use governor::{
    clock::{Clock, DefaultClock},
    state::{InMemoryState, NotKeyed},
    RateLimiter as Governor,
};
use std::sync::Arc;
use std::time::Duration;

/// Custom rate limit error type
#[derive(Debug)]
pub struct RateLimitError {
    wait_time: Duration,
}

impl RateLimitError {
    pub fn wait_time(&self) -> Duration {
        self.wait_time
    }
}

/// Rate limiter implementation for the API client
#[derive(Debug, Clone)]
pub struct RateLimiter {
    limiter: Arc<Governor<NotKeyed, InMemoryState, DefaultClock>>,
}

impl RateLimiter {
    /// Create a new rate limiter with the specified quota
    pub fn new(quota: governor::Quota) -> Self {
        Self {
            limiter: Arc::new(Governor::direct(quota)),
        }
    }

    /// Check if a request can be made
    pub async fn check(&self) -> Result<(), RateLimitError> {
        self.limiter.check().map_err(|negative| RateLimitError {
            wait_time: negative.wait_time_from(self.limiter.clock().now()),
        })
    }
}

impl Default for RateLimiter {
    fn default() -> Self {
        Self::new(governor::Quota::per_second(100u32.try_into().unwrap()))
    }
}