aurelia 0.2.0

Embeddable service mesh for Rust distributed applications.
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

// This file is part of the Aurelia workspace.
// SPDX-FileCopyrightText: 2026 Zivatar Limited
// SPDX-License-Identifier: Apache-2.0

use crate::peering::config::DomusConfigAccess;
use std::sync::atomic::{AtomicUsize, Ordering};

pub(super) struct PreAuthGate {
    inflight: AtomicUsize,
}

impl PreAuthGate {
    pub(super) fn new() -> Self {
        Self {
            inflight: AtomicUsize::new(0),
        }
    }

    pub(super) async fn try_acquire(
        &self,
        config: &DomusConfigAccess,
    ) -> Option<PreAuthPermit<'_>> {
        let limit = config.snapshot().await.inbound_handshake_limit_total.max(1);
        let next = self.inflight.fetch_add(1, Ordering::SeqCst) + 1;
        if next > limit {
            let registry = config.limited_registry();
            crate::warn_limited!(
                registry,
                crate::ids::LogId::HandshakeTotalLimit,
                inflight = next,
                limit,
                "pre-auth handshake limit reached; closing inbound connection"
            );
            self.inflight.fetch_sub(1, Ordering::SeqCst);
            return None;
        }
        Some(PreAuthPermit { gate: self })
    }
}

pub(super) struct PreAuthPermit<'a> {
    gate: &'a PreAuthGate,
}

impl Drop for PreAuthPermit<'_> {
    fn drop(&mut self) {
        self.gate.inflight.fetch_sub(1, Ordering::SeqCst);
    }
}