mart-w/kanidm
1
0
Fork 0
mirror of https://github.com/kanidm/kanidm.git synced 2025-02-23 12:37:00 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Improve tasks daemon shutdown (#1806)

Browse source
This commit is contained in:
Firstyear 2023-07-04 15:53:48 +10:00 committed by GitHub
parent 83e4d3a85e
commit 12121bae37
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
unix_integration/src/tasks_daemon.rs
View file

@ -330,6 +330,7 @@ async fn main() -> ExitCode {
debug!("Attempting to use {} ...", task_sock_path);
let (broadcast_tx, mut broadcast_rx) = broadcast::channel(4);
let mut d_broadcast_rx = broadcast_tx.subscribe();
let server = tokio::spawn(async move {
loop {
@ -345,7 +346,14 @@ async fn main() -> ExitCode {
info!("Found kanidm_unixd, waiting for tasks ...");
// Yep! Now let the main handler do it's job.
// If it returns (dc, etc, then we loop and try again).
handle_tasks(stream, &cfg).await;
tokio::select! {
_ = d_broadcast_rx.recv() => {
break;
}
_ = handle_tasks(stream, &cfg) => {
continue;
}
}
}
Err(e) => {
debug!("\\---> {:?}", e);