use diesel::r2d2;
use dotenv::dotenv;
use fang::PgConnection;
use fang::Queue;
use fang::Queueable;
use fang::RetentionMode;
use fang::WorkerPool;
use simple_cron_worker::MyCronTask;
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(2)).build();

    let mut worker_pool = WorkerPool::<Queue>::builder()
        .queue(queue)
        .retention_mode(RetentionMode::KeepAll)
        .number_of_workers(2_u32)
        .task_type("cron_test".to_string())
        .build();

    worker_pool.queue.schedule_task(&MyCronTask {}).unwrap();

    worker_pool.start().unwrap();

    sleep(Duration::from_secs(100))
}