intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 46 insertions, 0 deletions

diff --git a/library/cpp/blockcodecs/core/stream.h b/library/cpp/blockcodecs/core/stream.h
new file mode 100644
index 0000000000..fd44ef88f2
--- /dev/null
+++ b/library/cpp/blockcodecs/core/stream.h
@@ -0,0 +1,46 @@
+#pragma once
+
+#include <util/stream/walk.h>
+#include <util/stream/input.h>
+#include <util/stream/output.h>
+#include <util/stream/zerocopy.h>
+#include <util/generic/buffer.h>
+
+namespace NBlockCodecs {
+    struct ICodec;
+
+    class TCodedOutput: public IOutputStream {
+    public:
+        TCodedOutput(IOutputStream* out, const ICodec* c, size_t bufLen);
+        ~TCodedOutput() override;
+
+    private:
+        void DoWrite(const void* buf, size_t len) override;
+        void DoFlush() override;
+        void DoFinish() override;
+
+        bool FlushImpl();
+
+    private:
+        const ICodec* C_;
+        TBuffer D_;
+        TBuffer O_;
+        IOutputStream* S_;
+    };
+
+    class TDecodedInput: public IWalkInput {
+    public:
+        TDecodedInput(IInputStream* in);
+        TDecodedInput(IInputStream* in, const ICodec* codec);
+
+        ~TDecodedInput() override;
+
+    private:
+        size_t DoUnboundedNext(const void** ptr) override;
+
+    private:
+        TBuffer D_;
+        IInputStream* S_;
+        const ICodec* C_;
+    };
+}