blob: 984d947c07920b40e118676c141ef1e03540a471 (
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
#include "replicating.h"
#include <yql/essentials/sql/v1/ide/completion/name/object/simple/static/schema.h>
#include <library/cpp/case_insensitive_string/case_insensitive_string.h>
#include <util/generic/hash.h>
namespace NSQLComplete {
namespace {
class TNameService: public INameService {
public:
TNameService(INameService::TPtr origin, IRanking::TPtr ranking)
: Origin_(std::move(origin))
, Ranking_(std::move(ranking))
{
}
NThreading::TFuture<TNameResponse> Lookup(const TNameRequest& request) const final {
NThreading::TFuture<TNameResponse> response = Origin_->Lookup(request);
if (!request.Constraints.Column || request.Constraints.Column->TableAlias) {
return response;
}
// TODO(YQL-20095): Explore real problem to fix this.
// NOLINTNEXTLINE(bugprone-exception-escape)
return std::move(response).Apply([request, ranking = Ranking_](auto f) -> TNameResponse {
TNameResponse response = f.ExtractValue();
TVector<TGenericName> replicatable = ReplicatableColumns(response.RankedNames, request.Prefix);
std::ranges::move(std::move(replicatable), std::back_inserter(response.RankedNames));
ranking->CropToSortedPrefix(response.RankedNames, request.Constraints, request.Limit);
return response;
});
}
private:
static TVector<TGenericName> ReplicatableColumns(const TVector<TGenericName>& names, TStringBuf prefix) {
THashMap<TString, size_t> references;
for (const TGenericName& name : names) {
if (!std::holds_alternative<TColumnName>(name)) {
continue;
}
const TColumnName& column = std::get<TColumnName>(name);
if (column.TableAlias.empty()) {
continue;
}
references[column.Identifier] += 1;
}
TVector<TGenericName> replicatable;
for (auto& [column, count] : references) {
if (count != 1) {
continue;
}
// TODO(YQL-19747): introduce a single source of truth of filtration policy
if (!TCaseInsensitiveStringBuf(column).StartsWith(prefix)) {
continue;
}
TColumnName name;
name.Identifier = column;
replicatable.emplace_back(std::move(name));
}
return replicatable;
}
INameService::TPtr Origin_;
IRanking::TPtr Ranking_;
};
} // namespace
INameService::TPtr MakeColumnReplicatingService(INameService::TPtr origin, IRanking::TPtr ranking) {
return new TNameService(std::move(origin), std::move(ranking));
}
} // namespace NSQLComplete
|
