Add withTaskCancellationOrGracefulShutdownHandler
#176
Conversation
| } | ||
| } | ||
|
|
||
| func testWithTaskCancellationOrGracefulShutdownHandler_TaskCancellation() async throws { |
There was a problem hiding this comment.
It was quite awkward to construct a test that actually tested cancellation as AsyncStream.next() exits immediately if it is cancelled.
There was a problem hiding this comment.
I don't think you need any of the streams here. Just create a LockedValueBox<CheckedContinuation<Void, Never>> and then in the operation closure you create the continuation and store it into the lock and in the handler closure you just resume it.
There was a problem hiding this comment.
I need the first stream to ensure the withTaskCancellationOrGracefulShutdownHandler operation actually gets run.
The locked value box would have to be a LockedValueBox<CheckedContinuation<Void, Never>?> and then if I was to be correct I would have to deal with the situation where the cancellation handler gets called before the operation and optional CheckedContinuation is still nil
Execute an operation with a graceful shutdown or task cancellation handler that’s immediately invoked if the current task is shutting down gracefully or has been cancelled.