Improve error handling: add counters for internal, other and because of ban errors/fails, provide source of errors and reasons of coroutine cancellations

1 files changed, 13 insertions, 6 deletions

diff --git a/library/cpp/coroutine/engine/impl.h b/library/cpp/coroutine/engine/impl.h
index ca523cdaf3..ec4af17fcf 100644
--- a/library/cpp/coroutine/engine/impl.h
+++ b/library/cpp/coroutine/engine/impl.h
@@ -30,6 +30,12 @@ namespace NCoro::NStack {
 }
 
 class TCont : private TIntrusiveListItem<TCont> {
+    friend class TContExecutor;
+    friend class TIntrusiveListItem<TCont>;
+    friend class NCoro::TEventWaitQueue;
+    friend class NCoro::TTrampoline;
+
+public:
     struct TJoinWait: public TIntrusiveListItem<TJoinWait> {
         TJoinWait(TCont& c) noexcept;
 
@@ -39,11 +45,6 @@ class TCont : private TIntrusiveListItem<TCont> {
         TCont& Cont_;
     };
 
-    friend class TContExecutor;
-    friend class TIntrusiveListItem<TCont>;
-    friend class NCoro::TEventWaitQueue;
-    friend class NCoro::TTrampoline;
-
 private:
     TCont(
         NCoro::NStack::IAllocator& allocator,
@@ -99,7 +100,9 @@ public:
     /// \param this корутина, которая будет ждать
     /// \param c корутина, которую будем ждать
     /// \param deadLine максимальное время ожидания
-    bool Join(TCont* c, TInstant deadLine = TInstant::Max()) noexcept;
+    /// \param forceStop кастомный обработчик ситуации, когда завершается время ожидания или отменяется ожидающая корутина (this)
+    /// дефолтное поведение - отменить ожидаемую корутину (c->Cancel())
+    bool Join(TCont* c, TInstant deadLine = TInstant::Max(), std::function<void(TJoinWait&, TCont*)> forceStop = {}) noexcept;
 
     void ReSchedule() noexcept;
 
@@ -113,6 +116,10 @@ public:
         return std::move(Exception_);
     }
 
+    void SetException(THolder<std::exception> exception) noexcept {
+        Exception_ = std::move(exception);
+    }
+
 private:
     void Terminate();