summaryrefslogtreecommitdiffstats
path: root/yql/essentials/sql/v1/ide/completion/name/service/binding/name_service.cpp
blob: 69cff02fd1aa631cb29cdfee4d25c91cd98015a4 (plain) (blame)
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
#include "name_service.h"

#include <yql/essentials/sql/v1/ide/completion/name/service/static/name_index.h>

namespace NSQLComplete {

namespace {

class TNameService: public INameService {
public:
    explicit TNameService(TVector<TString> names)
        : Index_(BuildNameIndex(std::move(names), NormalizeName))
    {
    }

    NThreading::TFuture<TNameResponse> Lookup(const TNameRequest& request) const override {
        if (request.Constraints.IsEmpty()) {
            return NThreading::MakeFuture<TNameResponse>({});
        }

        TVector<TStringBuf> filtered =
            FilteredByPrefix(request.Prefix, Index_, NormalizeName);
        filtered.crop(request.Limit);

        return NThreading::MakeFuture<TNameResponse>({
            .RankedNames = Transform(filtered),
        });
    }

private:
    static TVector<TGenericName> Transform(TVector<TStringBuf> names) {
        TVector<TGenericName> generic;
        generic.reserve(names.size());
        for (TStringBuf name : names) {
            generic.emplace_back(Transform(name));
        }
        return generic;
    }

    static TGenericName Transform(TStringBuf name) {
        TBindingName unknown;
        unknown.Identifier = name;
        return unknown;
    }

    TNameIndex Index_;
};

} // namespace

INameService::TPtr MakeBindingNameService(TVector<TString> names) {
    return new TNameService(std::move(names));
}

} // namespace NSQLComplete