How to shutdown axum::serve with a CancellationToken?

[andinus]

Summary

I have two tasks that I want to shutdown on shutdown_signal(). When I used tokio::select on the second task along with shutdown_signal(), I had to press ctrl + c twice, once to get axum to shutdown and then to shutdown the deletion task.

I figured I could use a CancellationToken and cancel both tasks at once but with this Axum doesn't close connections gracefully. Am I missing something?

Here is the code:

axum version

0.7.4

[andinus]

await on both tasks at the end has fixed the issue. Here is the final code:

I'm not sure I understand why this works and why axum task is killed in original code. Is this the right way to do it?

[ndouglas]

Did you ever get more clarity on this? I'm having a doozy of a time getting Axum to shutdown cleanly.

[jplatte]

Have you seen the graceful-shutdown example in this repo? If yes but it didn't help, could you open a separate discussion explaining your specific problem in more detail?

[ndouglas]

Thank you! I'm just trying to figure out if I can create a minimal reproduction before I harass y'all 🙂