Struct InMemoryTaskStore 

pub struct InMemoryTaskStore { /* private fields */ }

In-memory TaskStore backed by a BTreeMap under a RwLock.

Suitable for testing and single-process deployments. Data is lost when the process exits.

Supports TTL-based eviction of terminal tasks and a maximum capacity limit to prevent unbounded memory growth.

Eviction behavior

Eviction runs as a background task every N writes (configurable via TaskStoreConfig::eviction_interval) and whenever the store exceeds max_capacity. The eviction sweep is decoupled from the save() write lock so that writers are not blocked during the O(n) cleanup. However, if the system goes idle (no save() calls), completed tasks may persist in memory longer than their TTL.

Operators should call run_eviction() periodically (e.g. every 60 seconds via tokio::time::interval) to ensure timely cleanup of terminal tasks during idle periods.

Concurrency

For high-concurrency production deployments, consider SqliteTaskStore which uses a connection pool and row-level locking. The in-memory store uses a single RwLock and is optimized for testing and moderate load.

Implementations

impl InMemoryTaskStore

pub async fn run_eviction(&self)

Runs background eviction of expired and over-capacity entries.

Call this periodically (e.g. every 60 seconds) to clean up terminal tasks that would otherwise persist until the next save() call.

impl InMemoryTaskStore

pub fn new() -> Self

Creates a new empty in-memory task store with default configuration.

Default: max 10,000 tasks, 1-hour TTL for terminal tasks.

pub fn with_config(config: TaskStoreConfig) -> Self

Creates a new in-memory task store with custom configuration.

Trait Implementations

impl Debug for InMemoryTaskStore

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

Formats the value using the given formatter.

impl Default for InMemoryTaskStore

fn default() -> Self

Returns the “default value” for a type.

impl TaskStore for InMemoryTaskStore

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

Saves (creates or updates) a task.

fn get<'a>( &'a self, id: &'a TaskId, ) -> Pin<Box<dyn Future<Output = A2aResult<Option<Task>>> + 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 = A2aResult<TaskListResponse>> + Send + 'a>>

Lists tasks matching the given filter parameters.

fn insert_if_absent<'a>( &'a self, task: Task, ) -> Pin<Box<dyn Future<Output = A2aResult<bool>> + 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 = A2aResult<()>> + Send + 'a>>

Deletes a task by its ID.

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

Returns the total number of tasks in the store.