1 files changed, 60 insertions, 0 deletions
diff --git a/library/cpp/yson/node/benchmark/reserve.cpp b/library/cpp/yson/node/benchmark/reserve.cpp
new file mode 100644
index 0000000000..d9600e1224
--- /dev/null
+++ b/library/cpp/yson/node/benchmark/reserve.cpp
@@ -0,0 +1,60 @@
+#include <benchmark/benchmark.h>
+
+#include <library/cpp/yson/node/node_builder.h>
+
+using namespace NYT;
+
+static void BM_Map(benchmark::State& state, const std::tuple<int, bool>& input) {
+    for (auto _ : state) {
+        TNode node;
+        TNodeBuilder builder(&node);
+        if (std::get<1>(input)) {
+            builder.OnBeginMap(std::get<0>(input));
+        } else {
+            builder.OnBeginMap();
+        }
+        for (auto i = 0; i < std::get<0>(input); ++i) {
+            builder.OnKeyedItem(ToString(i));
+            builder.OnUint64Scalar(i);
+        }
+        builder.OnEndMap();
+    }
+}
+
+static void BM_List(benchmark::State& state, const std::tuple<int, bool>& input) {
+    for (auto _ : state) {
+        TNode node;
+        TNodeBuilder builder(&node);
+        if (std::get<1>(input)) {
+            builder.OnBeginList(std::get<0>(input));
+        } else {
+            builder.OnBeginList();
+        }
+        for (auto i = 0; i < std::get<0>(input); ++i) {
+            builder.OnListItem();
+            builder.OnUint64Scalar(i);
+        }
+        builder.OnEndList();
+    }
+}
+
+BENCHMARK_CAPTURE(BM_Map, map_10, std::make_tuple(10u, false));
+BENCHMARK_CAPTURE(BM_Map, map_100, std::make_tuple(100u, false));
+BENCHMARK_CAPTURE(BM_Map, map_300, std::make_tuple(300u, false));
+BENCHMARK_CAPTURE(BM_Map, map_500, std::make_tuple(500u, false));
+BENCHMARK_CAPTURE(BM_Map, map_1000, std::make_tuple(1000u, false));
+BENCHMARK_CAPTURE(BM_Map, map_reserve_10, std::make_tuple(10u, true));
+BENCHMARK_CAPTURE(BM_Map, map_reserve_100, std::make_tuple(100u, true));
+BENCHMARK_CAPTURE(BM_Map, map_reserve_300, std::make_tuple(300u, true));
+BENCHMARK_CAPTURE(BM_Map, map_reserve_500, std::make_tuple(500u, true));
+BENCHMARK_CAPTURE(BM_Map, map_reserve_1000, std::make_tuple(1000u, true));
+BENCHMARK_CAPTURE(BM_List, list_10, std::make_tuple(10u, false));
+BENCHMARK_CAPTURE(BM_List, list_100, std::make_tuple(100u, false));
+BENCHMARK_CAPTURE(BM_List, list_300, std::make_tuple(300u, false));
+BENCHMARK_CAPTURE(BM_List, list_500, std::make_tuple(500u, false));
+BENCHMARK_CAPTURE(BM_List, list_1000, std::make_tuple(1000u, false));
+BENCHMARK_CAPTURE(BM_List, list_reserve_10, std::make_tuple(10u, true));
+BENCHMARK_CAPTURE(BM_List, list_reserve_100, std::make_tuple(100u, true));
+BENCHMARK_CAPTURE(BM_List, list_reserve_300, std::make_tuple(300u, true));
+BENCHMARK_CAPTURE(BM_List, list_reserve_500, std::make_tuple(500u, true));
+BENCHMARK_CAPTURE(BM_List, list_reserve_1000, std::make_tuple(1000u, true));