Struct TaskRegistry 

pub struct TaskRegistry { /* private fields */ }

A struct for managing a registry of task handlers.

Implementations

impl TaskRegistry

pub fn new() -> Self

Creates a new TaskRegistry.

Examples found in repository

examples/consumer.rs (line 42)

async fn main() {
    let args: Vec<String> = env::args().collect();
    if args.len() != 2 {
        eprintln!("Usage: consumer <num_workers>");
        std::process::exit(1);
    }

    let num_workers = args[1].parse().expect("Invalid number of tasks");
    let database_url = "postgresql://postgres:postgres@localhost/queue";

    let pool = postgres_queue::connect(database_url)
        .await
        .expect("Failed to connect to the database");

    initialize_database(&pool)
        .await
        .expect("Failed to initialize database");

    let mut registry = TaskRegistry::new();
    registry.register_task("send_email".to_string(), send_email_handler);

    let pool_arc = std::sync::Arc::new(pool);

    // Run the task processor
    let tasks = registry
        .run(&pool_arc, num_workers)
        .await
        .expect("Failed to run tasks");

    println!("Running {} tasks", tasks.len());

    // Wait for all tasks to complete
    for task in tasks {
        task.await.expect("Task failed");
    }

    println!("All tasks completed.");
}

pub fn register_task<F, Fut>(&mut self, name: String, handler: F)

where F: Fn(i32, TaskData) -> Fut + Send + Sync + 'static, Fut: Future<Output = Result<(), TaskError>> + Send + Sync + 'static,

Registers a task handler with the provided name.

Examples found in repository

examples/consumer.rs (line 43)

async fn main() {
    let args: Vec<String> = env::args().collect();
    if args.len() != 2 {
        eprintln!("Usage: consumer <num_workers>");
        std::process::exit(1);
    }

    let num_workers = args[1].parse().expect("Invalid number of tasks");
    let database_url = "postgresql://postgres:postgres@localhost/queue";

    let pool = postgres_queue::connect(database_url)
        .await
        .expect("Failed to connect to the database");

    initialize_database(&pool)
        .await
        .expect("Failed to initialize database");

    let mut registry = TaskRegistry::new();
    registry.register_task("send_email".to_string(), send_email_handler);

    let pool_arc = std::sync::Arc::new(pool);

    // Run the task processor
    let tasks = registry
        .run(&pool_arc, num_workers)
        .await
        .expect("Failed to run tasks");

    println!("Running {} tasks", tasks.len());

    // Wait for all tasks to complete
    for task in tasks {
        task.await.expect("Task failed");
    }

    println!("All tasks completed.");
}

pub fn handlers(&self) -> &Arc<HashMap<String, TaskHandler>>

Returns a reference to the task handlers.

pub async fn run( &self, pool: &Pool, num_workers: usize, ) -> Result<Vec<JoinHandle<()>>, TaskError>

Runs the task handlers with the provided number of workers.

Examples found in repository

examples/consumer.rs (line 49)

async fn main() {
    let args: Vec<String> = env::args().collect();
    if args.len() != 2 {
        eprintln!("Usage: consumer <num_workers>");
        std::process::exit(1);
    }

    let num_workers = args[1].parse().expect("Invalid number of tasks");
    let database_url = "postgresql://postgres:postgres@localhost/queue";

    let pool = postgres_queue::connect(database_url)
        .await
        .expect("Failed to connect to the database");

    initialize_database(&pool)
        .await
        .expect("Failed to initialize database");

    let mut registry = TaskRegistry::new();
    registry.register_task("send_email".to_string(), send_email_handler);

    let pool_arc = std::sync::Arc::new(pool);

    // Run the task processor
    let tasks = registry
        .run(&pool_arc, num_workers)
        .await
        .expect("Failed to run tasks");

    println!("Running {} tasks", tasks.len());

    // Wait for all tasks to complete
    for task in tasks {
        task.await.expect("Task failed");
    }

    println!("All tasks completed.");
}

Trait Implementations

impl Default for TaskRegistry

fn default() -> Self

Returns the “default value” for a type.