blob: 36a77e7618bdff7a9b4bb9d0a95a545383ad460d (
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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once
#include <vespa/searchlib/fef/matchdatalayout.h>
#include "queryenvironment.h"
namespace search {
namespace fef {
namespace test {
class QueryEnvironmentBuilder {
public:
/**
* Constructs a new query environment builder.
*
* @param queryEnv The query environment to build in.
* @param layout The layout of match data to simultaneously update.
*/
QueryEnvironmentBuilder(QueryEnvironment &queryEnv, MatchDataLayout &layout);
~QueryEnvironmentBuilder();
/**
* Add a term node searching all known fields to this query
* environment. This will update both the environment and the
* match data layout.
*
* @return Reference to the corresponding term data.
*/
SimpleTermData &addAllFields();
/**
* Add a term node searching in the given fields to this query
* environment. This will update both the environment and the
* match data layout. All fields are required to be of type INDEX.
*
* @return Pointer to the corresponding term data or NULL if one of the fields does not exists.
*/
SimpleTermData *addIndexNode(const std::vector<vespalib::string> &fieldNames);
/**
* Add an attribute node searching in the given attribute to this query environment.
* This will update both the environment and the match data layout.
*
* @return Pointer to the corresponding term data or NULL if attribute does not exists.
*/
SimpleTermData *addAttributeNode(const vespalib::string & attrName);
/**
* Add a term node searching in the given virtual field.
*/
SimpleTermData *add_virtual_node(const vespalib::string &virtual_field);
/** Returns a reference to the query environment of this. */
QueryEnvironment &getQueryEnv() { return _queryEnv; }
/** Returns a const reference to the query environment of this. */
const QueryEnvironment &getQueryEnv() const { return _queryEnv; }
/** Returns a reference to the match data layout of this. */
MatchDataLayout &getLayout() { return _layout; }
/** Returns a const reference to the match data layout of this. */
const MatchDataLayout &getLayout() const { return _layout; }
QueryEnvironmentBuilder& set_avg_field_length(const vespalib::string& field_name, double avg_field_length);
private:
QueryEnvironmentBuilder(const QueryEnvironmentBuilder &); // hide
QueryEnvironmentBuilder & operator=(const QueryEnvironmentBuilder &); // hide
SimpleTermData *add_node(const FieldInfo &info);
private:
QueryEnvironment &_queryEnv;
MatchDataLayout &_layout;
};
} // namespace test
} // namespace fef
} // namespace search
|