rs-zero 0.2.6

Rust-first microservice framework inspired by go-zero engineering practices
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

use std::sync::Arc;

use axum::{
    body::Body,
    extract::Request,
    middleware::Next,
    response::{IntoResponse, Response},
};
use tokio::sync::Semaphore;

use crate::rest::{
    RestError,
    middleware::{MiddlewareMetrics, record_resilience_event},
};

/// Applies a fail-fast concurrency limit to one request.
pub async fn concurrency_middleware(
    semaphore: Arc<Semaphore>,
    metrics: MiddlewareMetrics,
    request: Request<Body>,
    next: Next,
) -> Response {
    let permit = match semaphore.try_acquire_owned() {
        Ok(permit) => permit,
        Err(_) => {
            record_resilience_event(&metrics, "concurrency", "rejected");
            return RestError::ConcurrencyLimit.into_response();
        }
    };

    record_resilience_event(&metrics, "concurrency", "allowed");
    let response = next.run(request).await;
    drop(permit);
    response
}