Implement basic runner structure

This commit is contained in:
Joscha 2023-08-09 16:00:43 +02:00
parent 23ae5613c7
commit bf08d00922
2 changed files with 97 additions and 2 deletions

View file

@ -1,9 +1,15 @@
mod coordinator; mod coordinator;
mod server;
use tracing::error; use std::sync::{Arc, Mutex};
use tokio::task::JoinSet;
use tracing::{debug, error};
use crate::config::Config; use crate::config::Config;
use self::{coordinator::Coordinator, server::Server};
pub struct Runner { pub struct Runner {
config: &'static Config, config: &'static Config,
} }
@ -14,6 +20,26 @@ impl Runner {
} }
pub async fn run(&self) { pub async fn run(&self) {
error!("Runner not yet implemented"); if self.config.runner_servers.is_empty() {
error!("No servers specified in config");
return;
}
let names = self.config.runner_servers.keys().cloned().collect();
let coordinator = Arc::new(Mutex::new(Coordinator::new(names)));
let mut tasks = JoinSet::new();
for (name, config) in self.config.runner_servers.iter() {
debug!("Launching task for server {name}");
let mut server = Server::new(
name.clone(),
config,
self.config.runner_ping_delay,
coordinator.clone(),
);
tasks.spawn(async move { server.run().await });
}
while tasks.join_next().await.is_some() {}
} }
} }

69
src/runner/server.rs Normal file
View file

@ -0,0 +1,69 @@
use std::{
sync::{Arc, Mutex},
time::Duration,
};
use time::OffsetDateTime;
use tracing::{debug, info_span, warn, Instrument};
use crate::{config::RunnerServerConfig, somehow};
use super::coordinator::Coordinator;
enum RunState {
Preparing,
Running,
Finished, // TODO Include run results here
}
struct Run {
id: String,
hash: String,
start: OffsetDateTime,
state: RunState,
}
pub struct Server {
name: String,
config: &'static RunnerServerConfig,
ping_delay: Duration,
coordinator: Arc<Mutex<Coordinator>>,
run: Option<Arc<Mutex<Run>>>,
}
impl Server {
pub fn new(
name: String,
config: &'static RunnerServerConfig,
ping_delay: Duration,
coordinator: Arc<Mutex<Coordinator>>,
) -> Self {
Self {
name,
config,
ping_delay,
coordinator,
run: None,
}
}
pub async fn run(&mut self) {
let name = self.name.clone();
async {
loop {
match self.ping().await {
Ok(()) => {}
Err(e) => warn!("Error talking to server:\n{e:?}"),
}
tokio::time::sleep(self.ping_delay).await;
}
}
.instrument(info_span!("runner", name))
.await;
}
async fn ping(&mut self) -> somehow::Result<()> {
debug!("Pinging");
Ok(())
}
}