Verkehrssteuerung

Middleware, die Funktionen zur Verkehrssteuerung bereitstellt.

Hauptfunktionen

RateIssuer bietet eine Abstraktion für die Zuweisung von Schlüsselwerten zur Identifizierung von Besuchern. RemoteIpIssuer ist eine Implementierung davon, die Besucher anhand ihrer IP-Adresse identifiziert. Der Schlüssel muss nicht unbedingt ein String sein – jeder Typ, der die Einschränkungen Hash + Eq + Send + Sync + 'static erfüllt, kann als Schlüssel verwendet werden.
RateGuard bietet eine Abstraktion für den Verkehrssteuerungsalgorithmus. Standardmäßig sind zwei Implementierungen verfügbar: Festes Fenster (FixedGuard) und Gleitendes Fenster (SlidingGuard).
RateStore stellt Operationen zum Speichern und Abrufen von Daten bereit. MokaStore ist eine integrierte, speicherbasierte Cache-Implementierung auf Basis von moka. Sie können auch Ihre eigene Implementierung definieren.
RateLimiter ist eine Struktur, die Handler implementiert und intern ein skipper-Feld enthält, mit dem Anfragen übersprungen werden können, die nicht zwischengespeichert werden müssen. Standardmäßig wird none_skipper verwendet, der keine Anfragen überspringt.
QuotaGetter bietet eine Abstraktion für die Kontingentbeschaffung. Basierend auf dem Key des Besuchers kann ein Kontingentobjekt abgerufen werden, was bedeutet, dass wir Benutzerkontingente und ähnliche Informationen in einer Datenbank konfigurieren und dynamisch ändern oder abrufen können.

Beispielcode

Beispiel für statisches Kontingent

main.rs

Cargo.toml

rate-limiter-static/src/main.rs

use salvo::prelude::*;
use salvo::rate_limiter::{BasicQuota, FixedGuard, MokaStore, RateLimiter, RemoteIpIssuer};

#[handler]
async fn hello() -> &'static str {
    "Hello World"
}

#[tokio::main]
async fn main() {
    tracing_subscriber::fmt().init();

    let limiter = RateLimiter::new(
        FixedGuard::new(),
        MokaStore::new(),
        RemoteIpIssuer,
        BasicQuota::per_second(1),
    );
    let router = Router::with_hoop(limiter).get(hello);
    let acceptor = TcpListener::new("0.0.0.0:5800").bind().await;
    Server::new(acceptor).serve(router).await;
}

Beispiel für dynamisches Kontingent

main.rs

Cargo.toml

rate-limiter-dynamic/src/main.rs

use std::borrow::Borrow;
use std::collections::HashMap;
use std::hash::Hash;
use std::sync::LazyLock;

use salvo::Error;
use salvo::prelude::*;
use salvo::rate_limiter::{
    CelledQuota, MokaStore, QuotaGetter, RateIssuer, RateLimiter, SlidingGuard,
};

static USER_QUOTAS: LazyLock<HashMap<String, CelledQuota>> = LazyLock::new(|| {
    let mut map = HashMap::new();
    map.insert("user1".into(), CelledQuota::per_second(1, 1));
    map.insert("user2".into(), CelledQuota::set_seconds(1, 1, 5));
    map.insert("user3".into(), CelledQuota::set_seconds(1, 1, 10));
    map
});

struct UserIssuer;
impl RateIssuer for UserIssuer {
    type Key = String;
    async fn issue(&self, req: &mut Request, _depot: &Depot) -> Option<Self::Key> {
        req.query::<Self::Key>("user")
    }
}

struct CustomQuotaGetter;
impl QuotaGetter<String> for CustomQuotaGetter {
    type Quota = CelledQuota;
    type Error = Error;

    async fn get<Q>(&self, key: &Q) -> Result<Self::Quota, Self::Error>
    where
        String: Borrow<Q>,
        Q: Hash + Eq + Sync,
    {
        USER_QUOTAS
            .get(key)
            .cloned()
            .ok_or_else(|| Error::other("user not found"))
    }
}

#[handler]
async fn limited() -> &'static str {
    "Limited page"
}
#[handler]
async fn home() -> Text<&'static str> {
    Text::Html(HOME_HTML)
}

#[tokio::main]
async fn main() {
    tracing_subscriber::fmt().init();

    let limiter = RateLimiter::new(
        SlidingGuard::new(),
        MokaStore::new(),
        UserIssuer,
        CustomQuotaGetter,
    );
    let router = Router::new()
        .get(home)
        .push(Router::with_path("limited").hoop(limiter).get(limited));
    let acceptor = TcpListener::new("0.0.0.0:5800").bind().await;
    Server::new(acceptor).serve(router).await;
}

static HOME_HTML: &str = r#"
<!DOCTYPE html>
<html>
    <head>
        <title>Rate Limiter Dynmaic</title>
    </head>
    <body>
        <h2>Rate Limiter Dynamic</h2>
        <p>
            This example shows how to set limit for different users.
        </p>
        <p>
            <a href="/limited?user=user1" target="_blank">Limited page for user1: 1/second</a>
        </p>
        <p>
            <a href="/limited?user=user2" target="_blank">Limited page for user2: 1/5seconds</a>
        </p>
        <p>
            <a href="/limited?user=user3" target="_blank">Limited page for user3: 1/10seconds</a>
        </p>
    </body>
</html>
"#;

Auf dieser Seite

#Verkehrssteuerung

#Hauptfunktionen

#Beispiel für statisches Kontingent

#Beispiel für dynamisches Kontingent

Verkehrssteuerung

Hauptfunktionen

Beispiel für statisches Kontingent

Beispiel für dynamisches Kontingent