aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/on_disk/chunks/writer.cpp
diff options
context:
space:
mode:
authorDevtools Arcadia <arcadia-devtools@yandex-team.ru>2022-02-07 18:08:42 +0300
committerDevtools Arcadia <arcadia-devtools@mous.vla.yp-c.yandex.net>2022-02-07 18:08:42 +0300
commit1110808a9d39d4b808aef724c861a2e1a38d2a69 (patch)
treee26c9fed0de5d9873cce7e00bc214573dc2195b7 /library/cpp/on_disk/chunks/writer.cpp
downloadydb-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.cpp46
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();
+}