Struct CanaryTokenStore 

pub struct CanaryTokenStore { /* private fields */ }

A thread-safe, per-tenant canary token store.

Uses Arc<RwLock<HashMap<String, Vec<CanaryToken>>>> internally so it can be shared across async tasks and threads.

Examples

use llmtrace_security::canary::{CanaryConfig, CanaryToken, CanaryTokenStore};

let store = CanaryTokenStore::new();
let config = CanaryConfig::default();
let token = CanaryToken::generate(&config);
let token_string = token.token.clone();

store.add("tenant-1", token);
assert_eq!(store.get("tenant-1").len(), 1);

store.remove("tenant-1", &token_string);
assert!(store.get("tenant-1").is_empty());

Implementations

impl CanaryTokenStore

pub fn new() -> Self

Create a new, empty token store.

pub fn add(&self, tenant_id: &str, token: CanaryToken)

Add a canary token for the given tenant.

pub fn remove(&self, tenant_id: &str, token_str: &str) -> bool

Remove a canary token (by token string) for the given tenant.

Returns true if a token was removed, false otherwise.

pub fn get(&self, tenant_id: &str) -> Vec<CanaryToken>

Get all canary tokens for a tenant (cloned).

Returns an empty Vec if the tenant has no tokens.

pub fn tenant_count(&self) -> usize

Return the number of tenants with registered tokens.

pub fn token_count(&self) -> usize

Return the total number of tokens across all tenants.

Trait Implementations

impl Clone for CanaryTokenStore

fn clone(&self) -> CanaryTokenStore

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for CanaryTokenStore

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for CanaryTokenStore

fn default() -> Self

Returns the “default value” for a type.