51 lines
1.3 KiB
Rust
51 lines
1.3 KiB
Rust
use diesel::r2d2;
|
|
use dotenvy::dotenv;
|
|
use fang::PgConnection;
|
|
use fang::Queue;
|
|
use fang::Queueable;
|
|
use fang::RetentionMode;
|
|
use fang::WorkerPool;
|
|
use simple_worker::MyFailingTask;
|
|
use simple_worker::MyTask;
|
|
use std::env;
|
|
use std::thread::sleep;
|
|
use std::time::Duration;
|
|
|
|
pub fn connection_pool(pool_size: u32) -> r2d2::Pool<r2d2::ConnectionManager<PgConnection>> {
|
|
let database_url = env::var("DATABASE_URL").expect("DATABASE_URL must be set");
|
|
|
|
let manager = r2d2::ConnectionManager::<PgConnection>::new(database_url);
|
|
|
|
r2d2::Pool::builder()
|
|
.max_size(pool_size)
|
|
.build(manager)
|
|
.unwrap()
|
|
}
|
|
|
|
fn main() {
|
|
dotenv().ok();
|
|
|
|
env_logger::init();
|
|
|
|
let queue = Queue::builder().connection_pool(connection_pool(3)).build();
|
|
|
|
let mut worker_pool = WorkerPool::<Queue>::builder()
|
|
.queue(queue)
|
|
.retention_mode(RetentionMode::KeepAll)
|
|
.number_of_workers(3_u32)
|
|
.task_type("worker_pool_test".to_string())
|
|
.build();
|
|
|
|
worker_pool.queue.insert_task(&MyTask::new(1)).unwrap();
|
|
worker_pool.queue.insert_task(&MyTask::new(1000)).unwrap();
|
|
|
|
worker_pool
|
|
.queue
|
|
.insert_task(&MyFailingTask::new(5000))
|
|
.unwrap();
|
|
|
|
worker_pool.start().unwrap();
|
|
|
|
sleep(Duration::from_secs(100))
|
|
}
|