Package io.a2a.server.tasks

Class InMemoryTaskStore

java.lang.Object
io.a2a.server.tasks.InMemoryTaskStore
All Implemented Interfaces:
TaskStateProvider, TaskStore
@ApplicationScoped public class InMemoryTaskStore extends Object implements TaskStore, TaskStateProvider

  • Constructor Details

    • InMemoryTaskStore

      public InMemoryTaskStore()

  • Method Details

    • save

      public void save(Task task)
      Specified by:
      save in interface TaskStore

    • get

      public Task get(String taskId)
      Specified by:
      get in interface TaskStore

    • delete

      public void delete(String taskId)
      Specified by:
      delete in interface TaskStore

    • isTaskActive

      public boolean isTaskActive(String taskId)
      Description copied from interface: TaskStateProvider
      Determines whether a task is considered active for queue management purposes.

      This method includes the grace period in its check. A task is considered active if:

      • Its state is not final, OR
      • Its state is final but finalized within the grace period (now < finalizedAt + gracePeriod)

      This method is used to decide whether to process late-arriving events.

      Specified by:
      isTaskActive in interface TaskStateProvider
      Parameters:
      taskId - the ID of the task to check
      Returns:
      true if the task is active (or recently finalized within grace period), false otherwise

    • isTaskFinalized

      public boolean isTaskFinalized(String taskId)
      Description copied from interface: TaskStateProvider
      Determines whether a task is in a final state, ignoring the grace period.

      This method performs an immediate check: returns true only if the task is in a final state (COMPLETED, CANCELED, FAILED, etc.), regardless of when it was finalized.

      This method is used by cleanup callbacks and MainQueue closing logic to decide whether a queue can be closed and removed. By ignoring the grace period, it ensures responsive cleanup while late in-flight events are still handled by the grace period mechanism for isTaskActive.

      Specified by:
      isTaskFinalized in interface TaskStateProvider
      Parameters:
      taskId - the ID of the task to check
      Returns:
      true if the task is in a final state (ignoring grace period), false otherwise