From a6423bf330c10089d7a0e2aaa8d971d0dcbb3fcf Mon Sep 17 00:00:00 2001
From: maybenotilya <maybenotilya@yandex-team.com>
Date: Fri, 17 Apr 2026 11:57:44 +0300
Subject: YT-26179: Add Abort for readers

* Changelog entry
  Type: feature
  Component: cpp-sdk

Add Abort for table readers
commit_hash:be5026883c90d18333f20468946f1e5a5b71839b
---
 yt/cpp/mapreduce/io/node_table_reader.cpp | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'yt/cpp/mapreduce/io/node_table_reader.cpp')

diff --git a/yt/cpp/mapreduce/io/node_table_reader.cpp b/yt/cpp/mapreduce/io/node_table_reader.cpp
index 558c42b30ee..01eaf5d9462 100644
--- a/yt/cpp/mapreduce/io/node_table_reader.cpp
+++ b/yt/cpp/mapreduce/io/node_table_reader.cpp
@@ -3,6 +3,7 @@
 #include <yt/cpp/mapreduce/common/node_builder.h>
 #include <yt/cpp/mapreduce/common/wait_proxy.h>
 
+#include <yt/cpp/mapreduce/interface/errors.h>
 #include <yt/cpp/mapreduce/interface/logging/yt_log.h>
 
 #include <library/cpp/yson/parser.h>
@@ -341,6 +342,16 @@ bool TNodeTableReader::IsRawReaderExhausted() const
     return Finished_;
 }
 
+void TNodeTableReader::Abort()
+{
+    Input_.Abort();
+}
+
+bool TNodeTableReader::IsAborted() const
+{
+    return Input_.IsAborted();
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 
 void TNodeTableReader::PrepareParsing()
@@ -368,6 +379,9 @@ void TNodeTableReader::OnStreamError(std::exception_ptr exception, TString error
 
 void TNodeTableReader::CheckValidity() const
 {
+    if (IsAborted()) {
+        ythrow TInputStreamAbortedError() << "Stream was aborted";
+    }
     if (!Valid_) {
         ythrow yexception() << "Iterator is not valid";
     }
-- 
cgit v1.3