Skip to content
Navigation Menu
{{ message }}
-
Notifications
You must be signed in to change notification settings - Fork 73
Expand file tree
/
Copy pathattribute.cc
More file actions
330 lines (255 loc) · 9.15 KB
/
Copy pathattribute.cc
File metadata and controls
330 lines (255 loc) · 9.15 KB
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
// Copyright 2022 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include "base/attribute.h"
#include <cstddef>
#include <cstdint>
#include <string>
#include "absl/base/macros.h"
#include "absl/base/nullability.h"
#include "absl/status/status.h"
#include "absl/status/statusor.h"
#include "absl/strings/str_cat.h"
#include "absl/types/variant.h"
#include "base/kind.h"
#include "internal/status_macros.h"
namespace cel {
namespace {
// Visitor for appending string representation for different qualifier kinds.
class AttributeStringPrinter {
public:
// String representation for the given qualifier is appended to output.
// output must be non-null.
explicit AttributeStringPrinter(std::string* output, Kind type)
: output_(*output), type_(type) {}
absl::Status operator()(const Kind& ignored) const {
// Attributes are represented as a variant, with illegal attribute
// qualifiers represented with their type as the first alternative.
return absl::InvalidArgumentError(
absl::StrCat("Unsupported attribute qualifier ", KindToString(type_)));
}
absl::Status operator()(int64_t index) {
absl::StrAppend(&output_, "[", index, "]");
return absl::OkStatus();
}
absl::Status operator()(uint64_t index) {
absl::StrAppend(&output_, "[", index, "]");
return absl::OkStatus();
}
absl::Status operator()(bool bool_key) {
absl::StrAppend(&output_, "[", (bool_key) ? "true" : "false", "]");
return absl::OkStatus();
}
absl::Status operator()(const std::string& field) {
absl::StrAppend(&output_, ".", field);
return absl::OkStatus();
}
private:
std::string& output_;
Kind type_;
};
// Visitor for appending string representation for different qualifier kinds.
class AttributeQualifierStringPrinter {
public:
// String representation for the given qualifier is appended to output.
explicit AttributeQualifierStringPrinter(std::string* absl_nonnull output,
Kind type)
: output_(*output), type_(type) {}
absl::Status operator()(const Kind& ignored) const {
// Attributes are represented as a variant, with illegal attribute
// qualifiers represented with their type as the first alternative.
return absl::InvalidArgumentError(
absl::StrCat("Unsupported attribute qualifier ", KindToString(type_)));
}
absl::Status operator()(int64_t index) {
absl::StrAppend(&output_, index);
return absl::OkStatus();
}
absl::Status operator()(uint64_t index) {
absl::StrAppend(&output_, index);
return absl::OkStatus();
}
absl::Status operator()(bool bool_key) {
absl::StrAppend(&output_, (bool_key) ? "true" : "false");
return absl::OkStatus();
}
absl::Status operator()(const std::string& field) {
absl::StrAppend(&output_, field);
return absl::OkStatus();
}
private:
std::string& output_;
Kind type_;
};
struct AttributeQualifierTypeVisitor final {
Kind operator()(const Kind& type) const { return type; }
Kind operator()(int64_t ignored) const {
static_cast<void>(ignored);
return Kind::kInt64;
}
Kind operator()(uint64_t ignored) const {
static_cast<void>(ignored);
return Kind::kUint64;
}
Kind operator()(const std::string& ignored) const {
static_cast<void>(ignored);
return Kind::kString;
}
Kind operator()(bool ignored) const {
static_cast<void>(ignored);
return Kind::kBool;
}
};
struct AttributeQualifierTypeComparator final {
const Kind lhs;
bool operator()(const Kind& rhs) const {
return static_cast<int>(lhs) < static_cast<int>(rhs);
}
bool operator()(int64_t) const { return false; }
bool operator()(uint64_t other) const { return false; }
bool operator()(const std::string&) const { return false; }
bool operator()(bool other) const { return false; }
};
struct AttributeQualifierIntComparator final {
const int64_t lhs;
bool operator()(const Kind&) const { return true; }
bool operator()(int64_t rhs) const { return lhs < rhs; }
bool operator()(uint64_t) const { return true; }
bool operator()(const std::string&) const { return true; }
bool operator()(bool) const { return false; }
};
struct AttributeQualifierUintComparator final {
const uint64_t lhs;
bool operator()(const Kind&) const { return true; }
bool operator()(int64_t) const { return false; }
bool operator()(uint64_t rhs) const { return lhs < rhs; }
bool operator()(const std::string&) const { return true; }
bool operator()(bool) const { return false; }
};
struct AttributeQualifierStringComparator final {
const std::string& lhs;
bool operator()(const Kind&) const { return true; }
bool operator()(int64_t) const { return false; }
bool operator()(uint64_t) const { return false; }
bool operator()(const std::string& rhs) const { return lhs < rhs; }
bool operator()(bool) const { return false; }
};
struct AttributeQualifierBoolComparator final {
const bool lhs;
bool operator()(const Kind&) const { return true; }
bool operator()(int64_t) const { return true; }
bool operator()(uint64_t) const { return true; }
bool operator()(const std::string&) const { return true; }
bool operator()(bool rhs) const { return lhs < rhs; }
};
} // namespace
struct AttributeQualifier::ComparatorVisitor final {
const AttributeQualifier::Variant& rhs;
bool operator()(const Kind& lhs) const {
return absl::visit(AttributeQualifierTypeComparator{lhs}, rhs);
}
bool operator()(int64_t lhs) const {
return absl::visit(AttributeQualifierIntComparator{lhs}, rhs);
}
bool operator()(uint64_t lhs) const {
return absl::visit(AttributeQualifierUintComparator{lhs}, rhs);
}
bool operator()(const std::string& lhs) const {
return absl::visit(AttributeQualifierStringComparator{lhs}, rhs);
}
bool operator()(bool lhs) const {
return absl::visit(AttributeQualifierBoolComparator{lhs}, rhs);
}
};
Kind AttributeQualifier::kind() const {
return absl::visit(AttributeQualifierTypeVisitor{}, value_);
}
bool AttributeQualifier::operator<(const AttributeQualifier& other) const {
// The order is not publicly documented because it is subject to change.
// Currently we sort in the following order, with each type being sorted
// against itself: bool, int, uint, string, type.
return absl::visit(ComparatorVisitor{other.value_}, value_);
}
bool Attribute::operator==(const Attribute& other) const {
// We cannot check pointer equality as a short circuit because we have to
// treat all invalid AttributeQualifier as not equal to each other.
// TODO(issues/41) we only support Ident-rooted attributes at the moment.
if (variable_name() != other.variable_name()) {
return false;
}
if (qualifier_path().size() != other.qualifier_path().size()) {
return false;
}
for (size_t i = 0; i < qualifier_path().size(); i++) {
if (!(qualifier_path()[i] == other.qualifier_path()[i])) {
return false;
}
}
return true;
}
bool Attribute::operator<(const Attribute& other) const {
if (impl_.get() == other.impl_.get()) {
return false;
}
auto lhs_begin = qualifier_path().begin();
auto lhs_end = qualifier_path().end();
auto rhs_begin = other.qualifier_path().begin();
auto rhs_end = other.qualifier_path().end();
while (lhs_begin != lhs_end && rhs_begin != rhs_end) {
if (*lhs_begin < *rhs_begin) {
return true;
}
if (!(*lhs_begin == *rhs_begin)) {
return false;
}
lhs_begin++;
rhs_begin++;
}
if (lhs_begin == lhs_end && rhs_begin == rhs_end) {
// Neither has any elements left, they are equal. Compare variable names.
return variable_name() < other.variable_name();
}
if (lhs_begin == lhs_end) {
// Left has no more elements. Right is greater.
return true;
}
// Right has no more elements. Left is greater.
ABSL_ASSERT(rhs_begin == rhs_end);
return false;
}
const absl::StatusOr<std::string> Attribute::AsString() const {
if (variable_name().empty()) {
return absl::InvalidArgumentError(
"Only ident rooted attributes are supported.");
}
std::string result = std::string(variable_name());
for (const auto& qualifier : qualifier_path()) {
CEL_RETURN_IF_ERROR(absl::visit(
AttributeStringPrinter(&result, qualifier.kind()), qualifier.value_));
}
return result;
}
bool AttributeQualifier::IsMatch(const AttributeQualifier& other) const {
if (absl::holds_alternative<Kind>(value_) ||
absl::holds_alternative<Kind>(other.value_)) {
return false;
}
return value_ == other.value_;
}
absl::StatusOr<std::string> AttributeQualifier::AsString() const {
std::string result;
CEL_RETURN_IF_ERROR(
absl::visit(AttributeQualifierStringPrinter(&result, kind()), value_));
return result;
}
} // namespace cel
You can’t perform that action at this time.