9 #include "../index/Serialization.h"
10 #include "../index/dex/Dex.h"
11 #include "benchmark/benchmark.h"
12 #include "llvm/ADT/SmallVector.h"
13 #include "llvm/ADT/StringRef.h"
14 #include "llvm/Support/Path.h"
15 #include "llvm/Support/Regex.h"
27 std::unique_ptr<SymbolIndex> buildMem() {
31 std::unique_ptr<SymbolIndex> buildDex() {
36 std::vector<FuzzyFindRequest> extractQueriesFromLogs() {
38 std::string Log((std::istreambuf_iterator<char>(InputStream)),
39 std::istreambuf_iterator<char>());
41 std::vector<FuzzyFindRequest> Requests;
42 auto JSONArray = llvm::json::parse(Log);
46 llvm::errs() <<
"Error when parsing JSON requests file: "
50 if (!JSONArray->getAsArray()) {
51 llvm::errs() <<
"Error: top-level value is not a JSON array: " << Log
56 for (
const auto &Item : *JSONArray->getAsArray()) {
57 FuzzyFindRequest Request;
60 llvm::errs() <<
"Error when deserializing request: " << Item <<
'\n';
63 Requests.push_back(Request);
68 static void MemQueries(benchmark::State &State) {
69 const auto Mem = buildMem();
70 const auto Requests = extractQueriesFromLogs();
72 for (
const auto &Request : Requests)
73 Mem->fuzzyFind(Request, [](
const Symbol &S) {});
75 BENCHMARK(MemQueries);
77 static void DexQueries(benchmark::State &State) {
78 const auto Dex = buildDex();
79 const auto Requests = extractQueriesFromLogs();
81 for (
const auto &Request : Requests)
82 Dex->fuzzyFind(Request, [](
const Symbol &S) {});
84 BENCHMARK(DexQueries);
86 static void DexBuild(benchmark::State &State) {
100 int main(
int argc,
char *argv[]) {
102 llvm::errs() <<
"Usage: " << argv[0]
103 <<
" global-symbol-index.yaml requests.json "
104 "BENCHMARK_OPTIONS...\n";
114 ::benchmark::Initialize(&argc, argv);
115 ::benchmark::RunSpecifiedBenchmarks();