diff options
Diffstat (limited to 'yt/cpp/mapreduce/io/skiff_row_table_reader.cpp')
| -rw-r--r-- | yt/cpp/mapreduce/io/skiff_row_table_reader.cpp | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/yt/cpp/mapreduce/io/skiff_row_table_reader.cpp b/yt/cpp/mapreduce/io/skiff_row_table_reader.cpp index f77f2d71307..6de1d222d53 100644 --- a/yt/cpp/mapreduce/io/skiff_row_table_reader.cpp +++ b/yt/cpp/mapreduce/io/skiff_row_table_reader.cpp @@ -1,5 +1,6 @@ #include "skiff_row_table_reader.h" +#include <yt/cpp/mapreduce/interface/errors.h> #include <yt/cpp/mapreduce/interface/logging/yt_log.h> #include <yt/cpp/mapreduce/interface/skiff_row.h> @@ -100,6 +101,9 @@ void TSkiffRowTableReader::SkipRow() } void TSkiffRowTableReader::CheckValidity() const { + if (IsAborted()) { + ythrow TInputStreamAbortedError() << "Stream was aborted"; + } if (!IsValid()) { ythrow yexception() << "Iterator is not valid"; } @@ -230,6 +234,16 @@ bool TSkiffRowTableReader::IsRawReaderExhausted() const { return Finished_; } +void TSkiffRowTableReader::Abort() +{ + Input_.Abort(); +} + +bool TSkiffRowTableReader::IsAborted() const +{ + return Input_.IsAborted(); +} + //////////////////////////////////////////////////////////////////////////////// } // namespace NYT |