46 lines
907 B
Go
46 lines
907 B
Go
|
package opensearch_query
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
)
|
||
|
|
|
||
|
|
type bucketAggregationRequest map[string]*aggregationFunction
|
||
|
|
|
||
|
|
func (b bucketAggregationRequest) addAggregation(name, aggType, field string) error {
|
||
|
|
switch aggType {
|
||
|
|
case "terms":
|
||
|
|
default:
|
||
|
|
return fmt.Errorf("aggregation function %q not supported", aggType)
|
||
|
|
}
|
||
|
|
|
||
|
|
b[name] = &aggregationFunction{
|
||
|
|
aggType: aggType,
|
||
|
|
field: field,
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b bucketAggregationRequest) addNestedAggregation(name string, a aggregationRequest) {
|
||
|
|
b[name].nested = a
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b bucketAggregationRequest) bucketSize(name string, size int) error {
|
||
|
|
if size <= 0 {
|
||
|
|
return errors.New("invalid size; must be integer value > 0")
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, ok := b[name]; !ok {
|
||
|
|
return fmt.Errorf("aggregation %q not found", name)
|
||
|
|
}
|
||
|
|
|
||
|
|
b[name].setSize(size)
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b bucketAggregationRequest) missing(name, missing string) {
|
||
|
|
b[name].setMissing(missing)
|
||
|
|
}
|