intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 35 insertions, 0 deletions

diff --git a/library/cpp/yson_pull/range.h b/library/cpp/yson_pull/range.h
new file mode 100644
index 0000000000..f4fcf3f206
--- /dev/null
+++ b/library/cpp/yson_pull/range.h
@@ -0,0 +1,35 @@
+#pragma once
+
+#include "reader.h"
+
+#include <util/generic/iterator.h>
+
+namespace NYsonPull {
+    class TStreamEventsRange: public TInputRangeAdaptor<TStreamEventsRange> {
+        TReader Reader_;
+        bool AtEnd;
+
+    public:
+        TStreamEventsRange(THolder<NInput::IStream> stream, EStreamType mode)
+            : Reader_{std::move(stream), mode}
+            , AtEnd(false)
+        {
+        }
+
+        const TEvent* Last() const noexcept {
+            return &Reader_.LastEvent();
+        }
+
+        const TEvent* Next() {
+            if (Y_UNLIKELY(AtEnd)) {
+                return nullptr;
+            }
+
+            auto* event = &Reader_.NextEvent();
+            if (event->Type() == EEventType::EndStream) {
+                AtEnd = true;
+            }
+            return event;
+        }
+    };
+}