diff options
Diffstat (limited to 'yt/cpp/mapreduce/io/stream_raw_reader.cpp')
| -rw-r--r-- | yt/cpp/mapreduce/io/stream_raw_reader.cpp | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/yt/cpp/mapreduce/io/stream_raw_reader.cpp b/yt/cpp/mapreduce/io/stream_raw_reader.cpp index ec19b67d0b0..f69eb9f034f 100644 --- a/yt/cpp/mapreduce/io/stream_raw_reader.cpp +++ b/yt/cpp/mapreduce/io/stream_raw_reader.cpp @@ -5,6 +5,10 @@ #include "skiff_table_reader.h" #include "yamr_table_reader.h" +#include <yt/cpp/mapreduce/common/abortable_stream.h> + +#include <yt/yt/core/concurrency/async_stream_helpers.h> + #include <util/system/env.h> #include <util/string/type.h> @@ -33,6 +37,12 @@ namespace NDetail { //////////////////////////////////////////////////////////////////////////////// +TInputStreamProxy::TInputStreamProxy(IInputStream* stream) + : Stream_(CreateAbortableInputStreamAdapter(NConcurrency::CreateAsyncAdapter(stream))) +{ } + +//////////////////////////////////////////////////////////////////////////////// + ::TIntrusivePtr<IProtoReaderImpl> CreateProtoReader( IInputStream* stream, const TTableReaderOptions& /* options */, |