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/skiff_table_reader.cpp | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

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

diff --git a/yt/cpp/mapreduce/io/skiff_table_reader.cpp b/yt/cpp/mapreduce/io/skiff_table_reader.cpp
index c96adade41e..94662741b45 100644
--- a/yt/cpp/mapreduce/io/skiff_table_reader.cpp
+++ b/yt/cpp/mapreduce/io/skiff_table_reader.cpp
@@ -1,5 +1,6 @@
 #include "skiff_table_reader.h"
 
+#include <yt/cpp/mapreduce/interface/errors.h>
 #include <yt/cpp/mapreduce/interface/logging/yt_log.h>
 
 #include <library/cpp/yson/node/node_io.h>
@@ -155,6 +156,16 @@ bool TSkiffTableReader::IsRawReaderExhausted() const
     return Finished_;
 }
 
+void TSkiffTableReader::Abort()
+{
+    Input_.Abort();
+}
+
+bool TSkiffTableReader::IsAborted() const
+{
+    return Input_.IsAborted();
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 
 TVector<TSkiffTableReader::TSkiffTableSchema> TSkiffTableReader::CreateSkiffTableSchemas(
@@ -288,6 +299,9 @@ void TSkiffTableReader::ReadRow()
 
 void TSkiffTableReader::EnsureValidity() const
 {
+    if (IsAborted()) {
+        ythrow TInputStreamAbortedError() << "Stream was aborted";
+    }
     Y_ENSURE(Valid_, "Iterator is not valid");
 }
 
-- 
cgit v1.3