praxis-proxy-filter 0.5.4

Filter pipeline engine and built-in filters for Praxis
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
50
51

// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2024 Praxis Contributors

//! Protocol-agnostic load-balancing strategies and endpoint types.

use std::sync::atomic::{AtomicU64, Ordering};

pub(crate) mod consistent_hash;
pub(crate) mod endpoint;
pub(crate) mod hash;
pub(crate) mod least_connections;
pub(crate) mod maglev;
pub(crate) mod p2c;
pub(crate) mod priority;
pub(crate) mod random;
pub(crate) mod ring_hash;
pub(crate) mod round_robin;
pub(crate) mod strategy;
pub(crate) mod subset;
pub(crate) mod zone_aware;

// -----------------------------------------------------------------------------
// Shared LCG RNG
// -----------------------------------------------------------------------------

/// Multiplier for the LCG RNG (Knuth MMIX, truncated to 64 bits).
const LCG_A: u64 = 6_364_136_223_846_793_005;

/// Increment for the LCG RNG.
const LCG_C: u64 = 1_442_695_040_888_963_407;

/// Advance an atomic LCG state and return the new value.
fn next_random(rng: &AtomicU64) -> u64 {
    rng.fetch_update(Ordering::Relaxed, Ordering::Relaxed, |s| {
        Some(s.wrapping_mul(LCG_A).wrapping_add(LCG_C))
    })
    .unwrap_or(0)
}

// -----------------------------------------------------------------------------
// Shared Helpers
// -----------------------------------------------------------------------------

/// Whether `addr` appears in a retry-exclusion list.
///
/// Exclusion lists hold endpoints already attempted for this request, so a
/// retry lands somewhere new. Kept here so composite and hash-ring
/// strategies share one definition.
pub(crate) fn is_excluded(addr: &str, exclude: &[std::sync::Arc<str>]) -> bool {
    exclude.iter().any(|e| e.as_ref() == addr)
}