intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 56 insertions, 0 deletions

diff --git a/util/stream/multi.cpp b/util/stream/multi.cpp
new file mode 100644
index 0000000000..b2354298a0
--- /dev/null
+++ b/util/stream/multi.cpp
@@ -0,0 +1,56 @@
+#include "null.h"
+#include "multi.h"
+
+TMultiInput::TMultiInput(IInputStream* f, IInputStream* s) noexcept
+    : C_(f)
+    , N_(s)
+{
+}
+
+TMultiInput::~TMultiInput() = default;
+
+size_t TMultiInput::DoRead(void* buf, size_t len) {
+    const size_t ret = C_->Read(buf, len);
+
+    if (ret) {
+        return ret;
+    }
+
+    C_ = N_;
+    N_ = &Cnull;
+
+    return C_->Read(buf, len);
+}
+
+size_t TMultiInput::DoReadTo(TString& st, char ch) {
+    size_t ret = C_->ReadTo(st, ch);
+    if (ret == st.size() + 1) { // found a symbol, not eof
+        return ret;
+    }
+
+    C_ = N_;
+    N_ = &Cnull;
+
+    if (ret == 0) {
+        ret += C_->ReadTo(st, ch);
+    } else {
+        TString tmp;
+        ret += C_->ReadTo(tmp, ch);
+        st += tmp;
+    }
+
+    return ret;
+}
+
+size_t TMultiInput::DoSkip(size_t len) {
+    const size_t ret = C_->Skip(len);
+
+    if (ret) {
+        return ret;
+    }
+
+    C_ = N_;
+    N_ = &Cnull;
+
+    return C_->Skip(len);
+}