library/cpp/on_disk/chunks/writer.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

#include <util/ysaveload.h> 
 
#include "writer.h" 
 
static inline void WriteAux(IOutputStream* out, const TVector<ui64>& data) {
    ::SavePodArray(out, data.data(), data.size());
} 
 
/*************************** TBuffersWriter ***************************/ 
 
TChunkedDataWriter::TChunkedDataWriter(IOutputStream& slave)
    : Slave(slave) 
    , Offset(0) 
{ 
} 
 
TChunkedDataWriter::~TChunkedDataWriter() {
} 
 
void TChunkedDataWriter::NewBlock() { 
    if (Offsets.size()) { 
        Lengths.push_back(Offset - Offsets.back()); 
    } 
 
    Pad(16); 
    Offsets.push_back(Offset); 
} 
 
void TChunkedDataWriter::WriteFooter() { 
    Lengths.push_back(Offset - Offsets.back()); 
    WriteAux(this, Lengths); 
    WriteAux(this, Offsets); 
    WriteBinary<ui64>(Offsets.size()); 
    WriteBinary<ui64>(Version); 
    WriteBinary<ui64>(0); 
} 
 
size_t TChunkedDataWriter::GetCurrentBlockOffset() const { 
    Y_ASSERT(!Offsets.empty());
    Y_ASSERT(Offset >= Offsets.back());
    return Offset - Offsets.back(); 
} 
 
size_t TChunkedDataWriter::GetBlockCount() const { 
    return Offsets.size(); 
}