rrelayer_core 0.7.0

Core types and functionality for rrelayer - a powerful blockchain transaction relay service
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

use crate::common_types::EvmAddress;
use axum::http::HeaderMap;

use super::types::{RateLimitDetectContext, RateLimitError};

pub const RATE_LIMIT_HEADER_NAME: &str = "x-rrelayer-rate-limit-key";

pub struct RateLimitDetector {
    fallback_to_relayer: bool,
}

impl RateLimitDetector {
    pub fn new(fallback_to_relayer: bool) -> Self {
        Self { fallback_to_relayer }
    }

    pub fn detect(
        &self,
        headers: &HeaderMap,
        relayer_address: &EvmAddress,
    ) -> Result<RateLimitDetectContext, RateLimitError> {
        if let Some(header_value) =
            headers.get(RATE_LIMIT_HEADER_NAME).and_then(|v| v.to_str().ok())
        {
            return Ok(RateLimitDetectContext { key: header_value.to_string() });
        }

        if self.fallback_to_relayer {
            return Ok(RateLimitDetectContext { key: relayer_address.hex() });
        }

        Err(RateLimitError::NoRateLimitKey)
    }
}