yql/essentials/sql/v1/complete/bench/main.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

#include <benchmark/benchmark.h>

#include <yql/essentials/sql/v1/complete/name/static/name_service.h>
#include <yql/essentials/sql/v1/complete/name/static/ranking.h>
#include <yql/essentials/sql/v1/complete/sql_complete.h>

#include <yql/essentials/sql/v1/lexer/antlr4_pure/lexer.h>
#include <yql/essentials/sql/v1/lexer/antlr4_pure_ansi/lexer.h>

#include <util/generic/xrange.h>
#include <util/system/compiler.h>

namespace NSQLComplete {

    NSQLComplete::TLexerSupplier MakePureLexerSupplier() {
        NSQLTranslationV1::TLexers lexers;
        lexers.Antlr4Pure = NSQLTranslationV1::MakeAntlr4PureLexerFactory();
        lexers.Antlr4PureAnsi = NSQLTranslationV1::MakeAntlr4PureAnsiLexerFactory();
        return [lexers = std::move(lexers)](bool ansi) {
            return NSQLTranslationV1::MakeLexer(
                lexers, ansi, /* antlr4 = */ true,
                NSQLTranslationV1::ELexerFlavor::Pure);
        };
    }

    void BenchmarkComplete(benchmark::State& state) {
        auto names = NSQLComplete::MakeDefaultNameSet();
        auto ranking = NSQLComplete::MakeDefaultRanking();
        auto service = MakeStaticNameService(std::move(names), std::move(ranking));
        auto engine = MakeSqlCompletionEngine(MakePureLexerSupplier(), std::move(service));

        for (const auto _ : state) {
            auto completion = engine->Complete({"SELECT "});
            benchmark::DoNotOptimize(completion);
        }
    }

} // namespace NSQLComplete

BENCHMARK(NSQLComplete::BenchmarkComplete);