added instant-init; signal handler also waits for server

Cargo.lock

@@ -616,7 +616,7 @@ dependencies = [
 
 [[package]]
 name = "markov"
-version = "0.7.2"
+version = "0.7.3"
 dependencies = [
  "async-compression",
  "bzip2-sys",

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "markov"
-version = "0.7.2"
+version = "0.7.3"
 description = "Generate string of text from Markov chain fed by stdin"
 authors = ["Avril <flanchan@cumallover.me>"]
 edition = "2018"

src/main.rs

@@ -269,7 +269,7 @@ async fn main() {
 				      }) {
 		    Ok((addr, server)) => {
 			info!("Server bound on {:?}", addr);
-			(server, s2.into_inner().into_save_initialiser())
+			(server, s2.into_inner().into_initialiser())
 		    },
 		    Err(err) => {
 			error!("Failed to bind server: {}", err);

src/msg.rs

@@ -69,7 +69,7 @@ impl Initialiser
     {
 	if !*self.rx.borrow() {
 	    while !self.rx.recv().await.ok_or_else(|| InitWaitError)? {
-		tokio::task::yield_now().await;
+		//tokio::task::yield_now().await;
 	    }
 	    Ok(())
 	} else {

src/save.rs

@@ -77,8 +77,9 @@ pub async fn host(mut state: Box<State>)
 {
     let to = state.config().file.to_owned();
     let interval = state.config().save_interval();
-    if state.on_init_save().await.is_ok() {
-	trace!("Init get");
+    trace!("Setup oke. Waiting on init");
+    if state.on_init().await.is_ok() {
+	debug!("Begin save handler");
 	while Arc::strong_count(state.when()) > 1 {
 	    {
 		let chain = state.chain().read().await;
@@ -102,7 +103,7 @@ pub async fn host(mut state: Box<State>)
 	    }
 	}
     } else {
-	trace!("Shutdown called before init completed");
+	debug!("Shutdown called before init");
     }
     trace!("Saver exiting");
 }

src/signals.rs

@@ -15,6 +15,9 @@ pub async fn handle(mut state: State)
     let mut usr2 = unix::signal(SignalKind::user_defined2()).expect("Failed to hook SIGUSR2");
     let mut quit = unix::signal(SignalKind::quit()).expect("Failed to hook SIGQUIT");
 
+    trace!("Setup oke. Waiting on init");
+    if state.on_init().await.is_ok() {
+	debug!("Begin signal handler");
 	loop {
 	    tokio::select! {
 		_ = state.on_shutdown() => {
@@ -55,5 +58,8 @@ pub async fn handle(mut state: State)
 		},
 	    }
 	}
-    trace!("Graceful shutdown");
+    } else {
+	debug!("Shutdown called before init()");
+    }
+    trace!("Exiting");
 }

src/state.rs

@@ -28,7 +28,7 @@ pub struct State
     exclude: Arc<(sanitise::filter::Filter, sanitise::filter::Filter)>,
     chain: Arc<RwLock<Chain<String>>>,
     save: Arc<Notify>,
-    save_begin: Initialiser,
+    begin: Initialiser,
 
     shutdown: Arc<watch::Sender<bool>>,
     shutdown_recv: watch::Receiver<bool>,
@@ -37,30 +37,34 @@ pub struct State
 impl State
 {
     /// Consume this `state` into its initialiser
-    pub fn into_save_initialiser(self) -> Initialiser
+    pub fn into_initialiser(self) -> Initialiser
     {
-	self.save_begin
+	self.begin
     }
 
     /// Allow the saver task to start work
-    pub fn init_save(self) -> Result<(), msg::InitError>
+    pub fn init(self) -> Result<(), msg::InitError>
     {
-	self.save_begin.set()
+	self.begin.set()
     }
 
-    /// Has `init_save` been called?
-    pub fn is_init_save(&self) -> bool
+    /// Has `init` been called?
+    pub fn is_init(&self) -> bool
     {
-	self.save_begin.is_set()
+	self.begin.is_set()
     }
 
-    /// A future that completes either when `init_save` is called, or `shutdown`.
-    pub async fn on_init_save(&mut self) -> Result<(), ShutdownError>
+    /// A future that completes either when `init` is called, or `shutdown`.
+    pub async fn on_init(&mut self) -> Result<(), ShutdownError>
     {
-	tokio::select!{
-	    Ok(()) = self.save_begin.wait() => Ok(()),
-	    Some(true) = self.shutdown_recv.recv() => {
-		debug!("on_init_save(): shutdown received");
+	if self.has_shutdown() {
+	    return Err(ShutdownError);
+	}
+	let mut begin = self.begin.clone();
+	tokio::select!{ //fuck
+	    Ok(()) = begin.wait() => Ok(()),
+	    _ = self.on_shutdown() => {
+		debug!("on_init(): shutdown received");
 		Err(ShutdownError)
 	    }
 	    else => Err(ShutdownError)
@@ -85,7 +89,7 @@ impl State
 	    config: Arc::new(config),
 	    chain,
 	    save,
-	    save_begin: Initialiser::new(),
+	    begin: Initialiser::new(),
 	    shutdown: Arc::new(shutdown),
 	    shutdown_recv,
 	}