2023-03-10 22:41:34 +00:00
|
|
|
use backie::{PgTaskStore, RetentionMode, WorkerPool};
|
2023-03-04 18:07:17 +00:00
|
|
|
use diesel_async::pg::AsyncPgConnection;
|
|
|
|
use diesel_async::pooled_connection::{bb8::Pool, AsyncDieselConnectionManager};
|
2023-03-10 22:41:34 +00:00
|
|
|
use simple_worker::MyApplicationContext;
|
|
|
|
use simple_worker::MyFailingTask;
|
|
|
|
use simple_worker::MyTask;
|
2022-07-31 13:32:37 +00:00
|
|
|
|
|
|
|
#[tokio::main]
|
|
|
|
async fn main() {
|
|
|
|
env_logger::init();
|
|
|
|
|
2023-03-09 15:59:45 +00:00
|
|
|
let connection_url = "postgres://postgres:password@localhost/backie";
|
2023-03-04 18:07:17 +00:00
|
|
|
|
2022-07-31 13:32:37 +00:00
|
|
|
log::info!("Starting...");
|
2022-08-27 15:58:38 +00:00
|
|
|
let max_pool_size: u32 = 3;
|
2023-03-04 18:07:17 +00:00
|
|
|
let manager = AsyncDieselConnectionManager::<AsyncPgConnection>::new(connection_url);
|
|
|
|
let pool = Pool::builder()
|
|
|
|
.max_size(max_pool_size)
|
|
|
|
.min_idle(Some(1))
|
|
|
|
.build(manager)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
2023-03-09 15:59:45 +00:00
|
|
|
log::info!("Pool created ...");
|
2023-03-04 18:07:17 +00:00
|
|
|
|
2023-03-10 22:41:34 +00:00
|
|
|
let task_store = PgTaskStore::new(pool);
|
2022-07-31 13:32:37 +00:00
|
|
|
|
2023-03-09 15:59:45 +00:00
|
|
|
let (tx, mut rx) = tokio::sync::watch::channel(false);
|
2022-07-31 13:32:37 +00:00
|
|
|
|
2023-03-10 22:41:34 +00:00
|
|
|
// Some global application context I want to pass to my background tasks
|
|
|
|
let my_app_context = MyApplicationContext::new("Backie Example App");
|
2022-07-31 13:32:37 +00:00
|
|
|
|
2023-03-10 22:41:34 +00:00
|
|
|
// Register the task types I want to use and start the worker pool
|
|
|
|
let (join_handle, queue) = WorkerPool::new(task_store, move |_| my_app_context.clone())
|
|
|
|
.register_task_type::<MyTask>(1, RetentionMode::RemoveDone)
|
|
|
|
.register_task_type::<MyFailingTask>(1, RetentionMode::RemoveDone)
|
|
|
|
.start(async move {
|
|
|
|
let _ = rx.changed().await;
|
|
|
|
})
|
2022-07-31 15:15:00 +00:00
|
|
|
.await
|
|
|
|
.unwrap();
|
2022-08-27 15:58:38 +00:00
|
|
|
|
2023-03-10 22:41:34 +00:00
|
|
|
log::info!("Workers started ...");
|
2022-07-31 13:32:37 +00:00
|
|
|
|
2023-03-10 22:41:34 +00:00
|
|
|
let task1 = MyTask::new(0);
|
|
|
|
let task2 = MyTask::new(20_000);
|
|
|
|
let task3 = MyFailingTask::new(50_000);
|
2022-08-04 15:22:53 +00:00
|
|
|
|
2023-03-10 22:41:34 +00:00
|
|
|
queue.enqueue(task1).await.unwrap();
|
|
|
|
queue.enqueue(task2).await.unwrap();
|
|
|
|
queue.enqueue(task3).await.unwrap();
|
2023-03-09 15:59:45 +00:00
|
|
|
log::info!("Tasks created ...");
|
2023-03-10 22:41:34 +00:00
|
|
|
|
|
|
|
// Wait for Ctrl+C
|
|
|
|
let _ = tokio::signal::ctrl_c().await;
|
2023-03-09 15:59:45 +00:00
|
|
|
log::info!("Stopping ...");
|
|
|
|
tx.send(true).unwrap();
|
2023-03-10 22:41:34 +00:00
|
|
|
join_handle.await.unwrap();
|
|
|
|
log::info!("Workers Stopped!");
|
2022-07-31 13:32:37 +00:00
|
|
|
}
|