1#include "benchmark/benchmark.h"
2#include <nlohmann/json.hpp>
3#include <fstream>
4#include <test_data.hpp>
5
6using json = nlohmann::json;
7
8//////////////////////////////////////////////////////////////////////////////
9// parse JSON from file
10//////////////////////////////////////////////////////////////////////////////
11
12static void ParseFile(benchmark::State& state, const char* filename)
13{
14    while (state.KeepRunning())
15    {
16        state.PauseTiming();
17        auto* f = new std::ifstream(filename);
18        auto* j = new json();
19        state.ResumeTiming();
20
21        *j = json::parse(*f);
22
23        state.PauseTiming();
24        delete f;
25        delete j;
26        state.ResumeTiming();
27    }
28
29    std::ifstream file(filename, std::ios::binary | std::ios::ate);
30    state.SetBytesProcessed(state.iterations() * file.tellg());
31}
32BENCHMARK_CAPTURE(ParseFile, jeopardy,          TEST_DATA_DIRECTORY "/jeopardy/jeopardy.json");
33BENCHMARK_CAPTURE(ParseFile, canada,            TEST_DATA_DIRECTORY "/nativejson-benchmark/canada.json");
34BENCHMARK_CAPTURE(ParseFile, citm_catalog,      TEST_DATA_DIRECTORY "/nativejson-benchmark/citm_catalog.json");
35BENCHMARK_CAPTURE(ParseFile, twitter,           TEST_DATA_DIRECTORY "/nativejson-benchmark/twitter.json");
36BENCHMARK_CAPTURE(ParseFile, floats,            TEST_DATA_DIRECTORY "/regression/floats.json");
37BENCHMARK_CAPTURE(ParseFile, signed_ints,       TEST_DATA_DIRECTORY "/regression/signed_ints.json");
38BENCHMARK_CAPTURE(ParseFile, unsigned_ints,     TEST_DATA_DIRECTORY "/regression/unsigned_ints.json");
39BENCHMARK_CAPTURE(ParseFile, small_signed_ints, TEST_DATA_DIRECTORY "/regression/small_signed_ints.json");
40
41//////////////////////////////////////////////////////////////////////////////
42// parse JSON from string
43//////////////////////////////////////////////////////////////////////////////
44
45static void ParseString(benchmark::State& state, const char* filename)
46{
47    std::ifstream f(filename);
48    std::string str((std::istreambuf_iterator<char>(f)), std::istreambuf_iterator<char>());
49
50    while (state.KeepRunning())
51    {
52        state.PauseTiming();
53        auto* j = new json();
54        state.ResumeTiming();
55
56        *j = json::parse(str);
57
58        state.PauseTiming();
59        delete j;
60        state.ResumeTiming();
61    }
62
63    state.SetBytesProcessed(state.iterations() * str.size());
64}
65BENCHMARK_CAPTURE(ParseString, jeopardy,          TEST_DATA_DIRECTORY "/jeopardy/jeopardy.json");
66BENCHMARK_CAPTURE(ParseString, canada,            TEST_DATA_DIRECTORY "/nativejson-benchmark/canada.json");
67BENCHMARK_CAPTURE(ParseString, citm_catalog,      TEST_DATA_DIRECTORY "/nativejson-benchmark/citm_catalog.json");
68BENCHMARK_CAPTURE(ParseString, twitter,           TEST_DATA_DIRECTORY "/nativejson-benchmark/twitter.json");
69BENCHMARK_CAPTURE(ParseString, floats,            TEST_DATA_DIRECTORY "/regression/floats.json");
70BENCHMARK_CAPTURE(ParseString, signed_ints,       TEST_DATA_DIRECTORY "/regression/signed_ints.json");
71BENCHMARK_CAPTURE(ParseString, unsigned_ints,     TEST_DATA_DIRECTORY "/regression/unsigned_ints.json");
72BENCHMARK_CAPTURE(ParseString, small_signed_ints, TEST_DATA_DIRECTORY "/regression/small_signed_ints.json");
73
74
75//////////////////////////////////////////////////////////////////////////////
76// serialize JSON
77//////////////////////////////////////////////////////////////////////////////
78
79static void Dump(benchmark::State& state, const char* filename, int indent)
80{
81    std::ifstream f(filename);
82    std::string str((std::istreambuf_iterator<char>(f)), std::istreambuf_iterator<char>());
83    json j = json::parse(str);
84
85    while (state.KeepRunning())
86    {
87        j.dump(indent);
88    }
89
90    state.SetBytesProcessed(state.iterations() * j.dump(indent).size());
91}
92BENCHMARK_CAPTURE(Dump, jeopardy / -,          TEST_DATA_DIRECTORY "/jeopardy/jeopardy.json",                 -1);
93BENCHMARK_CAPTURE(Dump, jeopardy / 4,          TEST_DATA_DIRECTORY "/jeopardy/jeopardy.json",                 4);
94BENCHMARK_CAPTURE(Dump, canada / -,            TEST_DATA_DIRECTORY "/nativejson-benchmark/canada.json",       -1);
95BENCHMARK_CAPTURE(Dump, canada / 4,            TEST_DATA_DIRECTORY "/nativejson-benchmark/canada.json",       4);
96BENCHMARK_CAPTURE(Dump, citm_catalog / -,      TEST_DATA_DIRECTORY "/nativejson-benchmark/citm_catalog.json", -1);
97BENCHMARK_CAPTURE(Dump, citm_catalog / 4,      TEST_DATA_DIRECTORY "/nativejson-benchmark/citm_catalog.json", 4);
98BENCHMARK_CAPTURE(Dump, twitter / -,           TEST_DATA_DIRECTORY "/nativejson-benchmark/twitter.json",      -1);
99BENCHMARK_CAPTURE(Dump, twitter / 4,           TEST_DATA_DIRECTORY "/nativejson-benchmark/twitter.json",      4);
100BENCHMARK_CAPTURE(Dump, floats / -,            TEST_DATA_DIRECTORY "/regression/floats.json",                 -1);
101BENCHMARK_CAPTURE(Dump, floats / 4,            TEST_DATA_DIRECTORY "/regression/floats.json",                 4);
102BENCHMARK_CAPTURE(Dump, signed_ints / -,       TEST_DATA_DIRECTORY "/regression/signed_ints.json",            -1);
103BENCHMARK_CAPTURE(Dump, signed_ints / 4,       TEST_DATA_DIRECTORY "/regression/signed_ints.json",            4);
104BENCHMARK_CAPTURE(Dump, unsigned_ints / -,     TEST_DATA_DIRECTORY "/regression/unsigned_ints.json",          -1);
105BENCHMARK_CAPTURE(Dump, unsigned_ints / 4,     TEST_DATA_DIRECTORY "/regression/unsigned_ints.json",          4);
106BENCHMARK_CAPTURE(Dump, small_signed_ints / -, TEST_DATA_DIRECTORY "/regression/small_signed_ints.json",      -1);
107BENCHMARK_CAPTURE(Dump, small_signed_ints / 4, TEST_DATA_DIRECTORY "/regression/small_signed_ints.json",      4);
108
109
110BENCHMARK_MAIN();
111