bctx-conductor 0.1.6

bctx-conductor — Spiral Cycle agent runtime, SignalGraph, PassageRun
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

use std::collections::HashMap;
use std::time::Instant;

pub struct RateLimiter {
    counts: HashMap<String, (usize, Instant)>,
    per_second: usize,
}

impl RateLimiter {
    pub fn new(per_second: usize) -> Self {
        Self {
            counts: HashMap::new(),
            per_second,
        }
    }

    pub fn allow(&mut self, agent_id: &str) -> bool {
        let entry = self
            .counts
            .entry(agent_id.to_string())
            .or_insert((0, Instant::now()));
        if entry.1.elapsed().as_secs() >= 1 {
            *entry = (0, Instant::now());
        }
        if entry.0 < self.per_second {
            entry.0 += 1;
            true
        } else {
            false
        }
    }
}