Struct InMemoryTaskStore

pub struct InMemoryTaskStore { /* private fields */ }

Implementations

impl InMemoryTaskStore

pub fn new() -> Self

Creates a new instance of InMemoryTaskStore.

Examples found in repository

examples/basic.rs (line 17)

async fn main() {
    tracing_subscriber::fmt()
        .with_max_level(tracing::Level::INFO)
        .init();

    let task_store = InMemoryTaskStore::new();
    let context = TaskContext::new(task_store)
        .register::<MyTask1>()
        .register::<MyTask2>()
        .set_concurrency("default", 10)
        .start()
        .await;
    let mut tasks = Vec::new();
    for _ in 0..100 {
        tasks.push(TaskAndDelay {
            inner: MyTask1::new("name1".to_string(), 32),
            delay_seconds: None,
        });
    }

    tokio::spawn(async move {
        context.add_tasks(tasks).await.unwrap();
    });
    tokio::time::sleep(Duration::from_secs(20)).await;
}

Trait Implementations

impl Clone for InMemoryTaskStore

fn clone(&self) -> InMemoryTaskStore

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Default for InMemoryTaskStore

fn default() -> InMemoryTaskStore

Returns the “default value” for a type.

impl TaskStore for InMemoryTaskStore

type Error = InMemoryTaskStoreError

fn restore_tasks<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<(), Self::Error>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Restores task states by cleaning up all tasks in a running state and handling their next run times.

fn get<'life0, 'life1, 'async_trait>( &'life0 self, task_id: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<Option<TaskMeta>, Self::Error>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Retrieves task metadata based on the task ID.

fn list<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<Vec<TaskMeta>, Self::Error>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Lists all task metadata.

fn store_task<'life0, 'async_trait>( &'life0 self, task: TaskMeta, ) -> Pin<Box<dyn Future<Output = Result<(), Self::Error>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Stores task metadata.

fn store_tasks<'life0, 'async_trait>( &'life0 self, tasks: Vec<TaskMeta>, ) -> Pin<Box<dyn Future<Output = Result<(), Self::Error>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Stores tasks metadata.

fn fetch_pending_tasks<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<Vec<TaskMeta>, Self::Error>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Fetches all pending tasks from the store.

fn update_task_execution_status<'life0, 'life1, 'async_trait>( &'life0 self, task_id: &'life1 str, is_success: bool, last_error: Option<String>, next_run: Option<i64>, ) -> Pin<Box<dyn Future<Output = Result<(), Self::Error>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Updates the execution status of a task.

fn heartbeat<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, task_id: &'life1 str, runner_id: &'life2 str, ) -> Pin<Box<dyn Future<Output = Result<(), Self::Error>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait,

Updates the heartbeat for a task.

fn set_task_stopped<'life0, 'life1, 'async_trait>( &'life0 self, task_id: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<(), Self::Error>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Marks a task as stopped.

fn set_task_removed<'life0, 'life1, 'async_trait>( &'life0 self, task_id: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<(), Self::Error>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Marks a task as removed.

fn cleanup<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<(), Self::Error>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Cleans up the task store by removing tasks marked as removed.