fix init bug

Cargo.lock

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

Cargo.toml

@@ -43,6 +43,9 @@ hog-buffer = []
 # Enable the /api/ route
 api = []
 
+# Do not wait 2 seconds before starting worker tasks after server
+instant-init = []
+
 [profile.release]
 opt-level = 3
 lto = "fat"

src/main.rs

@@ -281,7 +281,14 @@ async fn main() {
 	    tokio::join![
 		server,
 		async move {
-		    trace!("Init set");
+		    cfg_if! {
+			if #[cfg(feature="instant-init")] {
+			    trace!("Setting init");
+			} else {
+			    trace!("Setting init in 2 seconds for good measure.");
+			    tokio::time::delay_for(tokio::time::Duration::from_secs(2)).await;
+			}
+		    }
 		    init.set().expect("Failed to initialise saver")	
 		},
 	    ];
@@ -290,6 +297,7 @@ async fn main() {
 	// Cleanup
 	async move {
 	    trace!("Cleanup");
+	    debug!("Waiting on {} tasks now", tasks.len());
 	    
 	    join_all(tasks).await;
 	}

src/msg.rs

@@ -68,9 +68,10 @@ impl Initialiser
     pub async fn wait(&mut self) -> Result<(), InitWaitError>
     {
 	if !*self.rx.borrow() {
-	    self.rx.recv().await
+	    while !self.rx.recv().await.ok_or_else(|| InitWaitError)? {
-		.ok_or_else(|| InitWaitError)
+		tokio::task::yield_now().await;
-		.and_then(|x| if x {Ok(())} else {Err(InitWaitError)})
+	    }
+	    Ok(())
 	} else {
 	    Ok(())
 	}
@@ -95,14 +96,8 @@ impl Future for Initialiser
 {
     type Output = Result<(), InitWaitError>;
     fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
-	
+	let uhh = self.wait();
-	if !*self.rx.borrow() {
+	tokio::pin!(uhh);
-	    let rx = self.rx.recv();
+	uhh.poll(cx)
-	    tokio::pin!(rx);
-	    rx.poll(cx).map(|x| x.ok_or_else(|| InitWaitError)
-			    .and_then(|x| if x {Ok(())} else {Err(InitWaitError)}))
-	} else {
-	    Poll::Ready(Ok(()))
-	}
     }
 }

src/state.rs

@@ -58,9 +58,7 @@ impl State
     pub async fn on_init_save(&mut self) -> Result<(), ShutdownError>
     {
 	tokio::select!{
-	    Ok(()) = self.save_begin.wait() => {
+	    Ok(()) = self.save_begin.wait() => Ok(()),
-		Ok(())
-	    }
 	    Some(true) = self.shutdown_recv.recv() => {
 		debug!("on_init_save(): shutdown received");
 		Err(ShutdownError)