diff options
author | Devtools Arcadia <arcadia-devtools@yandex-team.ru> | 2022-02-07 18:08:42 +0300 |
---|---|---|
committer | Devtools Arcadia <arcadia-devtools@mous.vla.yp-c.yandex.net> | 2022-02-07 18:08:42 +0300 |
commit | 1110808a9d39d4b808aef724c861a2e1a38d2a69 (patch) | |
tree | e26c9fed0de5d9873cce7e00bc214573dc2195b7 /library/cpp/on_disk/chunks/writer.cpp | |
download | ydb-1110808a9d39d4b808aef724c861a2e1a38d2a69.tar.gz |
intermediate changes
ref:cde9a383711a11544ce7e107a78147fb96cc4029
Diffstat (limited to 'library/cpp/on_disk/chunks/writer.cpp')
-rw-r--r-- | library/cpp/on_disk/chunks/writer.cpp | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/library/cpp/on_disk/chunks/writer.cpp b/library/cpp/on_disk/chunks/writer.cpp new file mode 100644 index 0000000000..6dc7397f09 --- /dev/null +++ b/library/cpp/on_disk/chunks/writer.cpp @@ -0,0 +1,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(); +} |