Add cancellation handler for cancelAndWait()

2020-11-17 12:04:19 +02:00 · 2020-11-17 12:04:19 +02:00 · 879c917242
parent 493cb1dbfd
commit 879c917242
1 changed files with 1 additions and 0 deletions
--- a/chronos/asyncfutures2.nim
+++ b/chronos/asyncfutures2.nim
@ -804,6 +804,7 @@ proc cancelAndWait*[T](fut: Future[T]): Future[void] =
    retFuture.complete()
  else:
    fut.addCallback(continuation)
+    retFuture.cancelCallback = cancellation
    # Initiate cancellation process.
    fut.cancel()
  return retFuture