yt/cpp/mapreduce/io/job_reader.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

#include "job_reader.h"

#include <yt/cpp/mapreduce/interface/logging/yt_log.h>

namespace NYT {

////////////////////////////////////////////////////////////////////////////////

TJobReader::TJobReader(int fd)
    : TJobReader(Duplicate(fd))
{ }

TJobReader::TJobReader(const TFile& file)
    : FdFile_(file)
    , FdInput_(FdFile_)
    , BufferedInput_(&FdInput_, BUFFER_SIZE)
{ }

bool TJobReader::Retry(
    const TMaybe<ui32>& /*rangeIndex*/,
    const TMaybe<ui64>& /*rowIndex*/,
    const std::exception_ptr& /*error*/)
{
    return false;
}

void TJobReader::ResetRetries()
{ }

bool TJobReader::HasRangeIndices() const
{
    return true;
}

size_t TJobReader::DoRead(void* buf, size_t len)
{
    return BufferedInput_.Read(buf, len);
}

////////////////////////////////////////////////////////////////////////////////

TRawTableReaderPtr CreateRawJobReader(int fd)
{
    return ::MakeIntrusive<TJobReader>(fd);
}

////////////////////////////////////////////////////////////////////////////////

} // namespace NYT