use chrono::prelude::*;
use std::sync::atomic::{AtomicUsize, Ordering::SeqCst};
use std::time::Duration;
use woddle::{async_trait, Job, JobConfig, JobRunner, RunnerConfig};
static COUNTER: AtomicUsize = AtomicUsize::new(0);
#[derive(Clone, Debug)]
struct MyJobContext {
pub name: String,
}
#[derive(Clone)]
struct MyJob {
ctx: MyJobContext,
config: JobConfig,
}
#[async_trait]
impl Job for MyJob {
async fn run(&self) {
log::info!(
"starting {} with context: {:?}..",
self.config.name,
self.ctx
);
log::info!(
"job done: time: {}, counter: {}!",
Utc::now(),
COUNTER.fetch_add(1, SeqCst)
);
}
fn get_config(&self) -> &JobConfig {
&self.config
}
}
#[tokio::main]
async fn main() {
pretty_env_logger::init();
let job_cfg = JobConfig::new("my_job", "someSyncKey").cron("0/2 * * * * * *");
let my_job = MyJob {
ctx: MyJobContext {
name: "my context".to_string(),
},
config: job_cfg,
};
for _ in 0..5 {
let config = RunnerConfig::default().check_interval(Duration::from_secs(1));
let job_runner = JobRunner::new(config).add_job(my_job.clone());
tokio::time::sleep(Duration::from_millis(50)).await;
tokio::spawn(async move {
if let Err(e) = job_runner.start().await {
log::error!("error: {}", e);
}
});
}
tokio::time::sleep(Duration::from_secs(11)).await;
assert!(COUNTER.load(SeqCst) >= 5);
log::info!("Success!");
}