YT: Add NodeFromYsonStreamNonGreedy; use it in TNode::Load

В рамках
3547980204d51d6eba4c3b56989a916379526673

1 files changed, 57 insertions, 0 deletions

diff --git a/library/cpp/yson/node/benchmark/saveload.cpp b/library/cpp/yson/node/benchmark/saveload.cpp
new file mode 100644
index 0000000000..838075f2e4
--- /dev/null
+++ b/library/cpp/yson/node/benchmark/saveload.cpp
@@ -0,0 +1,57 @@
+#include <benchmark/benchmark.h>
+
+#include <library/cpp/yson/node/node_io.h>
+
+using namespace NYT;
+
+namespace {
+
+static NYT::TNode GenerateList(size_t size)
+{
+    NYT::TNode result = NYT::TNode::CreateList();
+
+    for (size_t i = 0; i < size; ++i) {
+        result.AsList().emplace_back(NYT::TNode("val"));
+    }
+
+    return result;
+}
+
+} // namespace
+
+static void BM_SaveLoadGreedy(benchmark::State& state, size_t size)
+{
+    auto list = GenerateList(size);
+
+    TString bytes;
+    TStringOutput outputStream{bytes};
+    NodeToYsonStream(list, &outputStream, ::NYson::EYsonFormat::Binary);
+
+    for (const auto& _ : state) {
+        TStringInput inputStream{bytes};
+        NodeFromYsonStream(&inputStream);
+    }
+}
+
+static void BM_SaveLoadNonGreedy(benchmark::State& state, size_t size)
+{
+    auto list = GenerateList(size);
+
+    TString bytes;
+    TStringOutput outputStream{bytes};
+    NodeToYsonStream(list, &outputStream, ::NYson::EYsonFormat::Binary);
+
+    for (const auto& _ : state) {
+        TStringInput inputStream{bytes};
+        NodeFromYsonStreamNonGreedy(&inputStream);
+    }
+}
+
+BENCHMARK_CAPTURE(BM_SaveLoadGreedy, greedy_10, 10ul);
+BENCHMARK_CAPTURE(BM_SaveLoadNonGreedy, non_greedy_10, 10ul);
+BENCHMARK_CAPTURE(BM_SaveLoadGreedy, greedy_100, 100ul);
+BENCHMARK_CAPTURE(BM_SaveLoadNonGreedy, non_greedy_100, 100ul);
+BENCHMARK_CAPTURE(BM_SaveLoadGreedy, greedy_1000, 1000ul);
+BENCHMARK_CAPTURE(BM_SaveLoadNonGreedy, non_greedy_1000, 1000ul);
+BENCHMARK_CAPTURE(BM_SaveLoadGreedy, greedy_10000, 10000ul);
+BENCHMARK_CAPTURE(BM_SaveLoadNonGreedy, non_greedy_10000, 10000ul);