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
|
#include "name_service.h"
#include <yql/essentials/core/sql_types/normalize_name.h>
#include <util/charset/utf8.h>
namespace NSQLComplete {
namespace {
void SetPrefix(TString& name, TStringBuf delimeter, const TNamespaced& namespaced) {
if (namespaced.Namespace.empty()) {
return;
}
name.prepend(delimeter);
name.prepend(namespaced.Namespace);
}
void FixPrefix(TString& name, TStringBuf delimeter, const TNamespaced& namespaced) {
if (namespaced.Namespace.empty()) {
return;
}
name.remove(0, namespaced.Namespace.size() + delimeter.size());
}
} // namespace
TGenericName TNameConstraints::Qualified(TGenericName unqualified) const {
return std::visit([&](auto&& name) -> TGenericName {
using T = std::decay_t<decltype(name)>;
if constexpr (std::is_same_v<T, TPragmaName>) {
SetPrefix(name.Identifier, ".", *Pragma);
} else if constexpr (std::is_same_v<T, TFunctionName>) {
SetPrefix(name.Identifier, "::", *Function);
} else if constexpr (std::is_same_v<T, TClusterName>) {
SetPrefix(name.Identifier, ":", *Cluster);
}
return name;
}, std::move(unqualified));
}
TGenericName TNameConstraints::Unqualified(TGenericName qualified) const {
return std::visit([&](auto&& name) -> TGenericName {
using T = std::decay_t<decltype(name)>;
if constexpr (std::is_same_v<T, TPragmaName>) {
FixPrefix(name.Identifier, ".", *Pragma);
} else if constexpr (std::is_same_v<T, TFunctionName>) {
FixPrefix(name.Identifier, "::", *Function);
} else if constexpr (std::is_same_v<T, TClusterName>) {
FixPrefix(name.Identifier, ":", *Cluster);
}
return name;
}, std::move(qualified));
}
TVector<TGenericName> TNameConstraints::Qualified(TVector<TGenericName> unqualified) const {
for (auto& name : unqualified) {
name = Qualified(std::move(name));
}
return unqualified;
}
TVector<TGenericName> TNameConstraints::Unqualified(TVector<TGenericName> qualified) const {
for (auto& name : qualified) {
name = Unqualified(std::move(name));
}
return qualified;
}
TString LowerizeName(TStringBuf name) {
return ToLowerUTF8(name);
}
TString NormalizeName(TStringBuf name) {
TString normalized(name);
TMaybe<NYql::TIssue> error = NYql::NormalizeName(NYql::TPosition(), normalized);
if (!error.Empty()) {
return LowerizeName(name);
}
return normalized;
}
} // namespace NSQLComplete
|
