forked from olivere/elastic
/
search_aggs_percentile_ranks.go
132 lines (115 loc) · 3.32 KB
/
search_aggs_percentile_ranks.go
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
// Copyright 2012-2014 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
// PercentileRanksAggregation
// See: http://www.elasticsearch.org/guide/en/elasticsearch/reference/current/search-aggregations-metrics-percentile-rank-aggregation.html
type PercentileRanksAggregation struct {
field string
script string
lang string
format string
params map[string]interface{}
subAggregations map[string]Aggregation
values []float64
compression *float64
estimator string
}
func NewPercentileRanksAggregation() PercentileRanksAggregation {
a := PercentileRanksAggregation{
params: make(map[string]interface{}),
subAggregations: make(map[string]Aggregation),
values: make([]float64, 0),
}
return a
}
func (a PercentileRanksAggregation) Field(field string) PercentileRanksAggregation {
a.field = field
return a
}
func (a PercentileRanksAggregation) Script(script string) PercentileRanksAggregation {
a.script = script
return a
}
func (a PercentileRanksAggregation) Lang(lang string) PercentileRanksAggregation {
a.lang = lang
return a
}
func (a PercentileRanksAggregation) Format(format string) PercentileRanksAggregation {
a.format = format
return a
}
func (a PercentileRanksAggregation) Param(name string, value interface{}) PercentileRanksAggregation {
a.params[name] = value
return a
}
func (a PercentileRanksAggregation) SubAggregation(name string, subAggregation Aggregation) PercentileRanksAggregation {
a.subAggregations[name] = subAggregation
return a
}
func (a PercentileRanksAggregation) Values(values ...float64) PercentileRanksAggregation {
a.values = make([]float64, 0)
a.values = append(a.values, values...)
return a
}
func (a PercentileRanksAggregation) Compression(compression float64) PercentileRanksAggregation {
a.compression = &compression
return a
}
func (a PercentileRanksAggregation) Estimator(estimator string) PercentileRanksAggregation {
a.estimator = estimator
return a
}
func (a PercentileRanksAggregation) Source() interface{} {
// Example:
// {
// "aggs" : {
// "load_time_outlier" : {
// "percentile_ranks" : {
// "field" : "load_time"
// "values" : [15, 30]
// }
// }
// }
// }
// This method returns only the
// { "percentile_ranks" : { "field" : "load_time", "values" : [15, 30] } }
// part.
source := make(map[string]interface{})
opts := make(map[string]interface{})
source["percentile_ranks"] = opts
// ValuesSourceAggregationBuilder
if a.field != "" {
opts["field"] = a.field
}
if a.script != "" {
opts["script"] = a.script
}
if a.lang != "" {
opts["lang"] = a.lang
}
if a.format != "" {
opts["format"] = a.format
}
if len(a.params) > 0 {
opts["params"] = a.params
}
if len(a.values) > 0 {
opts["values"] = a.values
}
if a.compression != nil {
opts["compression"] = *a.compression
}
if a.estimator != "" {
opts["estimator"] = a.estimator
}
// AggregationBuilder (SubAggregations)
if len(a.subAggregations) > 0 {
aggsMap := make(map[string]interface{})
source["aggregations"] = aggsMap
for name, aggregate := range a.subAggregations {
aggsMap[name] = aggregate.Source()
}
}
return source
}