diff --git a/src/main.rs b/src/main.rs index 322181d..9b68e20 100644 --- a/src/main.rs +++ b/src/main.rs @@ -2,7 +2,7 @@ use std::time::Duration; use tokio::time::timeout; mod config; -mod world; +mod scheduler; mod feed; mod worker; mod trends; @@ -18,9 +18,9 @@ async fn main() { .expect("Call with config.yaml") ); - let mut world = world::World::new(); + let mut scheduler = scheduler::Scheduler::new(); for host in config.hosts.into_iter() { - world.introduce(host); + scheduler.introduce(host); } let client = reqwest::Client::builder() @@ -43,9 +43,9 @@ async fn main() { let mut workers_active = 0usize; let (message_tx, mut message_rx) = tokio::sync::mpsc::unbounded_channel(); loop { - println!("{} workers active, queued {} of {}", workers_active, world.queue_len(), world.size()); + println!("{} workers active, queued {} of {}", workers_active, scheduler.queue_len(), scheduler.size()); let next_task = if workers_active < config.max_workers { - world.dequeue() + scheduler.dequeue() } else { Err(Duration::from_secs(5)) }; @@ -56,15 +56,15 @@ async fn main() { match message { Message::Fetched { host, next_interval, latest_timestamp } => { workers_active -= 1; - world.enqueue(host, next_interval, latest_timestamp); + scheduler.enqueue(host, next_interval, latest_timestamp); } Message::Error { host } => { workers_active -= 1; - world.enqueue(host, Duration::from_secs(config.interval_after_error), None); + scheduler.enqueue(host, Duration::from_secs(config.interval_after_error), None); } Message::IntroduceHosts { hosts } => { for host in hosts.into_iter() { - world.introduce(host); + scheduler.introduce(host); } } } diff --git a/src/world.rs b/src/scheduler.rs similarity index 97% rename from src/world.rs rename to src/scheduler.rs index 7ec1c67..16ce50a 100644 --- a/src/world.rs +++ b/src/scheduler.rs @@ -11,14 +11,14 @@ pub struct Instance { } /// Scheduler -pub struct World { +pub struct Scheduler { instances: HashMap, queue: BTreeMap, } -impl World { +impl Scheduler { pub fn new() -> Self { - World { + Scheduler { instances: HashMap::new(), queue: BTreeMap::new(), }