Allow to pass exception to the coroutine to be cancelled

1 files changed, 7 insertions, 0 deletions

diff --git a/library/cpp/coroutine/engine/impl.cpp b/library/cpp/coroutine/engine/impl.cpp
index 7ae6f74051..08f358127d 100644
--- a/library/cpp/coroutine/engine/impl.cpp
+++ b/library/cpp/coroutine/engine/impl.cpp
@@ -106,6 +106,13 @@ void TCont::Cancel() noexcept {
     }
 }
 
+void TCont::Cancel(THolder<std::exception> exception) noexcept {
+    if (!Cancelled()) {
+        Exception_ = std::move(exception);
+        Cancel();
+    }
+}
+
 void TCont::ReSchedule() noexcept {
     if (Cancelled()) {
         // Legacy code may expect a Cancelled coroutine to be scheduled without delay.