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
|
// 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
// FuzzyQuery uses similarity based on Levenshtein edit distance for
// string fields, and a +/- margin on numeric and date fields.
// For more details, see
// http://www.elasticsearch.org/guide/en/elasticsearch/reference/current/query-dsl-fuzzy-query.html
type FuzzyQuery struct {
Query
name string
value interface{}
boost float32
fuzziness interface{}
prefixLength *int
maxExpansions *int
transpositions *bool
queryName string
}
// NewFuzzyQuery creates a new fuzzy query.
func NewFuzzyQuery() FuzzyQuery {
q := FuzzyQuery{
boost: -1.0,
}
return q
}
func (q FuzzyQuery) Name(name string) FuzzyQuery {
q.name = name
return q
}
func (q FuzzyQuery) Value(value interface{}) FuzzyQuery {
q.value = value
return q
}
func (q FuzzyQuery) Boost(boost float32) FuzzyQuery {
q.boost = boost
return q
}
// Fuzziness can be an integer/long like 0, 1 or 2 as well as strings like "auto",
// "0..1", "1..4" or "0.0..1.0".
func (q FuzzyQuery) Fuzziness(fuzziness interface{}) FuzzyQuery {
q.fuzziness = fuzziness
return q
}
func (q FuzzyQuery) PrefixLength(prefixLength int) FuzzyQuery {
q.prefixLength = &prefixLength
return q
}
func (q FuzzyQuery) MaxExpansions(maxExpansions int) FuzzyQuery {
q.maxExpansions = &maxExpansions
return q
}
func (q FuzzyQuery) Transpositions(transpositions bool) FuzzyQuery {
q.transpositions = &transpositions
return q
}
func (q FuzzyQuery) QueryName(queryName string) FuzzyQuery {
q.queryName = queryName
return q
}
// Creates the query source for the ids query.
func (q FuzzyQuery) Source() interface{} {
// {
// "fuzzy" : {
// "user" : {
// "value" : "ki",
// "boost" : 1.0,
// "fuzziness" : 2,
// "prefix_length" : 0,
// "max_expansions" : 100
// }
// }
source := make(map[string]interface{})
query := make(map[string]interface{})
source["fuzzy"] = query
fq := make(map[string]interface{})
query[q.name] = fq
fq["value"] = q.value
if q.boost != -1.0 {
fq["boost"] = q.boost
}
if q.transpositions != nil {
fq["transpositions"] = *q.transpositions
}
if q.fuzziness != nil {
fq["fuzziness"] = q.fuzziness
}
if q.prefixLength != nil {
fq["prefix_length"] = *q.prefixLength
}
if q.maxExpansions != nil {
fq["max_expansions"] = *q.maxExpansions
}
if q.queryName != "" {
fq["_name"] = q.queryName
}
return source
}
|
