Struct InMemoryBudgetTracker 

pub struct InMemoryBudgetTracker { /* private fields */ }

Process-local budget tracker. Used by adapter tests and any future in-memory dispatcher; production code wires a real BudgetStore-backed implementation in the binary.

Implementations

impl InMemoryBudgetTracker

pub fn new() -> Self

pub fn set_cap(&self, run_id: &str, kind: BudgetKind, cap_usd_micros: i64)

Pre-seed the cap for (run_id, kind). Subsequent add_spend calls accumulate against this cap.

pub fn spent(&self, run_id: &str, kind: BudgetKind) -> i64

Trait Implementations

impl BudgetTracker for InMemoryBudgetTracker

fn cap<'life0, 'life1, 'async_trait>( &'life0 self, run_id: &'life1 str, kind: BudgetKind, ) -> Pin<Box<dyn Future<Output = Result<Option<i64>, AiError>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Return the cap for (run_id, kind) in USD micros, or None if no cap is configured.

fn current_spend<'life0, 'life1, 'async_trait>( &'life0 self, run_id: &'life1 str, kind: BudgetKind, ) -> Pin<Box<dyn Future<Output = Result<i64, AiError>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Read the current spent_usd_micros for (run_id, kind). A non-existent row reads as 0 so pre-call cap checks against a brand-new run do not need to seed the row first.

fn add_spend<'life0, 'life1, 'async_trait>( &'life0 self, run_id: &'life1 str, kind: BudgetKind, micros: i64, ) -> Pin<Box<dyn Future<Output = Result<i64, AiError>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Atomically increment spent_usd_micros by micros and return the new total.

impl Default for InMemoryBudgetTracker

fn default() -> InMemoryBudgetTracker

Returns the “default value” for a type.