Add unordered version of LargeRandom ondemand benchmark

jkeiser · jkeiser · commit 98666e84ea37 · 2020-12-23T09:14:45.000-08:00
diff --git a/benchmark/bench_ondemand.cpp b/benchmark/bench_ondemand.cpp
@@ -12,6 +12,7 @@ SIMDJSON_POP_DISABLE_WARNINGS
 #include "partial_tweets/dom.h"
 
 #include "largerandom/ondemand.h"
+#include "largerandom/ondemand_unordered.h"
 // #include "largerandom/iter.h"
 #include "largerandom/dom.h"
 
diff --git a/benchmark/largerandom/CMakeLists.txt b/benchmark/largerandom/CMakeLists.txt
@@ -1,4 +1,5 @@
 if (TARGET benchmark::benchmark)
   link_libraries(benchmark::benchmark)
   add_executable(bench_ondemand_largerandom bench_ondemand_largerandom.cpp)
+  add_executable(bench_ondemand_unordered_largerandom bench_ondemand_unordered_largerandom.cpp)
 endif()
diff --git a/benchmark/largerandom/bench_ondemand_unordered_largerandom.cpp b/benchmark/largerandom/bench_ondemand_unordered_largerandom.cpp
@@ -0,0 +1,14 @@
+#include "simdjson.h"
+#include <iostream>
+#include <sstream>
+#include <random>
+#include <vector>
+SIMDJSON_PUSH_DISABLE_ALL_WARNINGS
+#include <benchmark/benchmark.h>
+SIMDJSON_POP_DISABLE_WARNINGS
+
+#define BENCHMARK_NO_DOM
+
+#include "largerandom/ondemand_unordered.h"
+
+BENCHMARK_MAIN();
diff --git a/benchmark/largerandom/ondemand_unordered.h b/benchmark/largerandom/ondemand_unordered.h
@@ -0,0 +1,38 @@
+#pragma once
+
+#if SIMDJSON_EXCEPTIONS
+
+#include "largerandom.h"
+
+namespace largerandom {
+
+using namespace simdjson;
+using namespace simdjson::builtin;
+
+class OnDemandUnordered {
+public:
+  simdjson_really_inline bool Run(const padded_string &json);
+  simdjson_really_inline const std::vector<my_point> &Result() { return container; }
+  simdjson_really_inline size_t ItemCount() { return container.size(); }
+
+private:
+  ondemand::parser parser{};
+  std::vector<my_point> container{};
+};
+
+simdjson_really_inline bool OnDemandUnordered::Run(const padded_string &json) {
+  container.clear();
+
+  auto doc = parser.iterate(json);
+  for (ondemand::object coord : doc) {
+    container.emplace_back(my_point{coord["x"], coord["y"], coord["z"]});
+  }
+
+  return true;
+}
+
+BENCHMARK_TEMPLATE(LargeRandom, OnDemandUnordered);
+
+} // namespace largerandom
+
+#endif // SIMDJSON_EXCEPTIONS
