architect_api/utils/
rate_limit.rs

1use crate::utils::NonZeroDurationAsStr;
2use anyhow::Result;
3use governor::Quota;
4use schemars::JsonSchema;
5use serde::{Deserialize, Serialize};
6use serde_with::{serde_as, serde_conv};
7use std::{num::NonZeroU32, time::Duration};
8
9#[serde_as]
10#[derive(Debug, Clone, Copy, Serialize, Deserialize, JsonSchema)]
11pub struct RateLimit {
12    pub max: NonZeroU32,
13    #[serde_as(as = "NonZeroDurationAsStr")]
14    #[schemars(with = "NonZeroDurationAsStr")]
15    pub per: Duration,
16}
17
18impl RateLimit {
19    pub fn as_quota(&self) -> governor::Quota {
20        governor::Quota::with_period(self.per)
21            .unwrap() // NonZeroDurationAsStr ensures this is non-zero
22            .allow_burst(self.max)
23    }
24}
25
26impl From<&Quota> for RateLimit {
27    fn from(quota: &Quota) -> Self {
28        RateLimit { max: quota.burst_size(), per: quota.replenish_interval() }
29    }
30}
31
32serde_conv!(
33    pub QuotaAsRateLimit,
34    Quota,
35    RateLimit::from,
36    try_into_quota
37);
38
39fn try_into_quota(rate_limit: RateLimit) -> Result<Quota, std::convert::Infallible> {
40    Ok(rate_limit.as_quota())
41}
architect_api/utils/rate_limit.rs

architect_api/utils/
rate_limit.rs
