wip: wait for flakehub pushes before shutting down?

This commit is contained in:
Cole Helbling 2024-04-24 14:23:22 -07:00
parent 1c83c37f4e
commit f06ed35360

View file

@ -45,16 +45,16 @@ async fn workflow_finish(
gha_cache.shutdown().await?; gha_cache.shutdown().await?;
} }
// Wait for the Attic push workers to finish.
if let Some(attic_state) = state.flakehub_state.write().await.take() {
tracing::info!("Waiting for FlakeHub cache uploads to finish");
attic_state.push_session.wait().await?;
}
if let Some(sender) = state.shutdown_sender.lock().await.take() { if let Some(sender) = state.shutdown_sender.lock().await.take() {
sender sender
.send(()) .send(())
.map_err(|_| Error::Internal("Sending shutdown server message".to_owned()))?; .map_err(|_| Error::Internal("Sending shutdown server message".to_owned()))?;
// Wait for the Attic push workers to finish.
if let Some(attic_state) = state.flakehub_state.write().await.take() {
tracing::info!("Waiting for FlakeHub cache uploads to finish");
attic_state.push_session.wait().await?;
}
} }
// NOTE(cole-h): see `init_logging` // NOTE(cole-h): see `init_logging`