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();
+}