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
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once
#include <vespa/vespalib/stllike/string.h>
#include <memory>
#include <vector>
namespace search::query {
/**
* Represents a predicate query, with features and range features.
*/
class PredicateQueryTerm {
static const uint64_t ALL_SUB_QUERIES = 0xffffffffffffffffULL;
template <typename ValueType>
class Entry {
vespalib::string _key;
ValueType _value;
uint64_t _sub_query_bitmap;
public:
Entry(vespalib::stringref key, const ValueType &value,
uint64_t sub_query_bitmap = ALL_SUB_QUERIES) noexcept
: _key(key), _value(value), _sub_query_bitmap(sub_query_bitmap) {}
const vespalib::string & getKey() const { return _key; }
const ValueType & getValue() const { return _value; }
uint64_t getSubQueryBitmap() const { return _sub_query_bitmap; }
bool operator==(const Entry<ValueType> &other) const {
return _key == other._key
&& _value == other._value
&& _sub_query_bitmap == other._sub_query_bitmap;
}
};
std::vector<Entry<vespalib::string>> _features;
std::vector<Entry<uint64_t>> _range_features;
public:
using UP = std::unique_ptr<PredicateQueryTerm>;
PredicateQueryTerm() noexcept : _features(), _range_features() {}
void addFeature(vespalib::stringref key, vespalib::stringref value,
uint64_t sub_query_bitmask = ALL_SUB_QUERIES) {
_features.emplace_back(key, value, sub_query_bitmask);
}
void addRangeFeature(vespalib::stringref key, uint64_t value,
uint64_t sub_query_bitmask = ALL_SUB_QUERIES) {
_range_features.emplace_back(key, value, sub_query_bitmask);
}
const std::vector<Entry<vespalib::string>> &getFeatures() const
{ return _features; }
const std::vector<Entry<uint64_t>> &getRangeFeatures() const
{ return _range_features; }
bool operator==(const PredicateQueryTerm &other) const {
return _features == other._features
&& _range_features == other._range_features;
}
};
}
|
