Struct TenantAwareInMemoryTaskStore 

pub struct TenantAwareInMemoryTaskStore { /* private fields */ }

Tenant-isolated in-memory TaskStore.

Maintains a separate InMemoryTaskStore per tenant, providing full data isolation between tenants. The current tenant is determined from TenantContext.

Usage

use a2a_protocol_server::store::tenant::{TenantAwareInMemoryTaskStore, TenantContext};
use a2a_protocol_server::store::TaskStore;

let store = TenantAwareInMemoryTaskStore::new();

// Tenant A saves a task
TenantContext::scope("tenant-a", async {
    let task = Task {
        id: TaskId::new("task-1"),
        context_id: ContextId::new("ctx-1"),
        status: TaskStatus::with_timestamp(TaskState::Submitted),
        history: None,
        artifacts: None,
        metadata: None,
    };
    store.save(task).await.unwrap();
}).await;

// Tenant B cannot see tenant A's task
TenantContext::scope("tenant-b", async {
    let result = store.get(&TaskId::new("task-1")).await.unwrap();
    assert!(result.is_none());
}).await;

Implementations

impl TenantAwareInMemoryTaskStore

pub fn new() -> TenantAwareInMemoryTaskStore

Creates a new tenant-aware store with default configuration.

pub fn with_config(config: TenantStoreConfig) -> TenantAwareInMemoryTaskStore

Creates a new tenant-aware store with custom configuration.

pub async fn tenant_count(&self) -> usize

Returns the number of active tenant partitions.

pub async fn run_eviction_all(&self)

Runs eviction on all tenant stores.

Call periodically to clean up terminal tasks in idle tenants.

pub async fn prune_empty_tenants(&self)

Removes empty tenant partitions to reclaim memory.

A partition is considered empty when its task count is zero.

Trait Implementations

impl Debug for TenantAwareInMemoryTaskStore

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

Formats the value using the given formatter.

impl Default for TenantAwareInMemoryTaskStore

fn default() -> TenantAwareInMemoryTaskStore

Returns the “default value” for a type.

impl TaskStore for TenantAwareInMemoryTaskStore

fn save<'a>( &'a self, task: Task, ) -> Pin<Box<dyn Future<Output = Result<(), A2aError>> + Send + 'a>>

Saves (creates or updates) a task.

fn get<'a>( &'a self, id: &'a TaskId, ) -> Pin<Box<dyn Future<Output = Result<Option<Task>, A2aError>> + Send + 'a>>

Retrieves a task by its ID, returning None if not found.

fn list<'a>( &'a self, params: &'a ListTasksParams, ) -> Pin<Box<dyn Future<Output = Result<TaskListResponse, A2aError>> + Send + 'a>>

Lists tasks matching the given filter parameters.

fn insert_if_absent<'a>( &'a self, task: Task, ) -> Pin<Box<dyn Future<Output = Result<bool, A2aError>> + Send + 'a>>

Atomically inserts a task only if no task with the same ID exists.

fn delete<'a>( &'a self, id: &'a TaskId, ) -> Pin<Box<dyn Future<Output = Result<(), A2aError>> + Send + 'a>>

Deletes a task by its ID.

fn count<'a>( &'a self, ) -> Pin<Box<dyn Future<Output = Result<u64, A2aError>> + Send + 'a>>

Returns the total number of tasks in the store.