File: search_queries_match.go

package info (click to toggle)
golang-gopkg-olivere-elastic.v2 2.0.12-2
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, bullseye, buster, sid, trixie
  • size: 1,964 kB
  • sloc: makefile: 17
file content (198 lines) | stat: -rw-r--r-- 5,048 bytes parent folder | download | duplicates (2)
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
// Copyright 2012-2015 Oliver Eilhard. All rights reserved.
// Use of this source code is governed by a MIT-license.
// See http://olivere.mit-license.org/license.txt for details.

package elastic

// MatchQuery is a family of queries that accept text/numerics/dates,
// analyzes it, and constructs a query out of it. For more details,
// see http://www.elastic.co/guide/en/elasticsearch/reference/current/query-dsl-match-query.html.
//
// To create a new MatchQuery, use NewMatchQuery. To create specific types
// of queries, e.g. a match_phrase query, use NewMatchQuery(...).Type("phrase"),
// or use one of the shortcuts like NewMatchPhraseQuery(...).
type MatchQuery struct {
	Query
	name                string
	value               interface{}
	matchQueryType      string // boolean, phrase, phrase_prefix
	operator            string // or / and
	analyzer            string
	boost               *float32
	slop                *int
	fuzziness           string
	prefixLength        *int
	maxExpansions       *int
	minimumShouldMatch  string
	rewrite             string
	fuzzyRewrite        string
	lenient             *bool
	fuzzyTranspositions *bool
	zeroTermsQuery      string
	cutoffFrequency     *float32
	queryName           string
}

// NewMatchQuery creates a new MatchQuery.
func NewMatchQuery(name string, value interface{}) MatchQuery {
	q := MatchQuery{name: name, value: value}
	return q
}

// NewMatchPhraseQuery creates a new MatchQuery with type phrase.
func NewMatchPhraseQuery(name string, value interface{}) MatchQuery {
	q := MatchQuery{name: name, value: value, matchQueryType: "phrase"}
	return q
}

// NewMatchPhrasePrefixQuery creates a new MatchQuery with type phrase_prefix.
func NewMatchPhrasePrefixQuery(name string, value interface{}) MatchQuery {
	q := MatchQuery{name: name, value: value, matchQueryType: "phrase_prefix"}
	return q
}

// Type can be "boolean", "phrase", or "phrase_prefix".
func (q MatchQuery) Type(matchQueryType string) MatchQuery {
	q.matchQueryType = matchQueryType
	return q
}

func (q MatchQuery) Operator(operator string) MatchQuery {
	q.operator = operator
	return q
}

func (q MatchQuery) Analyzer(analyzer string) MatchQuery {
	q.analyzer = analyzer
	return q
}

func (q MatchQuery) Boost(boost float32) MatchQuery {
	q.boost = &boost
	return q
}

func (q MatchQuery) Slop(slop int) MatchQuery {
	q.slop = &slop
	return q
}

func (q MatchQuery) Fuzziness(fuzziness string) MatchQuery {
	q.fuzziness = fuzziness
	return q
}

func (q MatchQuery) PrefixLength(prefixLength int) MatchQuery {
	q.prefixLength = &prefixLength
	return q
}

func (q MatchQuery) MaxExpansions(maxExpansions int) MatchQuery {
	q.maxExpansions = &maxExpansions
	return q
}

func (q MatchQuery) MinimumShouldMatch(minimumShouldMatch string) MatchQuery {
	q.minimumShouldMatch = minimumShouldMatch
	return q
}

func (q MatchQuery) Rewrite(rewrite string) MatchQuery {
	q.rewrite = rewrite
	return q
}

func (q MatchQuery) FuzzyRewrite(fuzzyRewrite string) MatchQuery {
	q.fuzzyRewrite = fuzzyRewrite
	return q
}

func (q MatchQuery) Lenient(lenient bool) MatchQuery {
	q.lenient = &lenient
	return q
}

func (q MatchQuery) FuzzyTranspositions(fuzzyTranspositions bool) MatchQuery {
	q.fuzzyTranspositions = &fuzzyTranspositions
	return q
}

// ZeroTermsQuery can be "all" or "none".
func (q MatchQuery) ZeroTermsQuery(zeroTermsQuery string) MatchQuery {
	q.zeroTermsQuery = zeroTermsQuery
	return q
}

func (q MatchQuery) CutoffFrequency(cutoff float32) MatchQuery {
	q.cutoffFrequency = &cutoff
	return q
}

func (q MatchQuery) QueryName(queryName string) MatchQuery {
	q.queryName = queryName
	return q
}

func (q MatchQuery) Source() interface{} {
	// {"match":{"name":{"query":"value","type":"boolean/phrase"}}}
	source := make(map[string]interface{})

	match := make(map[string]interface{})
	source["match"] = match

	query := make(map[string]interface{})
	match[q.name] = query

	query["query"] = q.value

	if q.matchQueryType != "" {
		query["type"] = q.matchQueryType
	}
	if q.operator != "" {
		query["operator"] = q.operator
	}
	if q.analyzer != "" {
		query["analyzer"] = q.analyzer
	}
	if q.boost != nil {
		query["boost"] = *q.boost
	}
	if q.slop != nil {
		query["slop"] = *q.slop
	}
	if q.fuzziness != "" {
		query["fuzziness"] = q.fuzziness
	}
	if q.prefixLength != nil {
		query["prefix_length"] = *q.prefixLength
	}
	if q.maxExpansions != nil {
		query["max_expansions"] = *q.maxExpansions
	}
	if q.minimumShouldMatch != "" {
		query["minimum_should_match"] = q.minimumShouldMatch
	}
	if q.rewrite != "" {
		query["rewrite"] = q.rewrite
	}
	if q.fuzzyRewrite != "" {
		query["fuzzy_rewrite"] = q.fuzzyRewrite
	}
	if q.lenient != nil {
		query["lenient"] = *q.lenient
	}
	if q.fuzzyTranspositions != nil {
		query["fuzzy_transpositions"] = *q.fuzzyTranspositions
	}
	if q.zeroTermsQuery != "" {
		query["zero_terms_query"] = q.zeroTermsQuery
	}
	if q.cutoffFrequency != nil {
		query["cutoff_frequency"] = q.cutoffFrequency
	}
	if q.queryName != "" {
		query["_name"] = q.queryName
	}

	return source
}