Struct UsageRecorder 

pub struct UsageRecorder { /* private fields */ }

Combines in-memory UsageStats with an optional persistent UsageStore.

Every record_* call updates the in-memory counters immediately and, if a store is configured, sends the record to a single background task that batches writes to reduce spawn overhead and SQLite write contention.

Implementations

impl UsageRecorder

pub fn new(store: Option<Arc<dyn UsageStore>>) -> Self

Creates a new recorder, optionally backed by a persistent store.

When a store is provided a background flush loop is spawned that drains records from an mpsc channel in micro-batches (up to 64 at a time).

pub fn record_success( &self, model: &str, provider: &str, input_tokens: u64, output_tokens: u64, )

Record a successful request with token counts.

Uses DEFAULT_ACCOUNT as the account attribution; call record_success_for when the specific OAuth account is known.

pub fn record_success_for( &self, model: &str, provider: &str, account_id: &str, input_tokens: u64, output_tokens: u64, )

Record a successful request with token counts, attributing it to a specific account.

pub fn record_failure(&self, model: &str, provider: &str)

Record a failed request.

pub fn record_failure_for(&self, model: &str, provider: &str, account_id: &str)

Record a failed request, attributing it to a specific account.

pub fn snapshot(&self) -> UsageSnapshot

Take a snapshot of in-memory stats.

pub fn preload( &self, model: &str, requests: u64, input_tokens: u64, output_tokens: u64, )

Pre-load cumulative counters from historical totals (e.g. on startup).

pub fn store(&self) -> Option<&Arc<dyn UsageStore>>

Returns a reference to the backing store (if configured).