40 lines
1.1 KiB
Go
40 lines
1.1 KiB
Go
package esquery
|
|
|
|
import "github.com/fatih/structs"
|
|
|
|
// DisMaxQuery represents a compound query of type "dis_max", as described in
|
|
// https://www.elastic.co/guide/en/elasticsearch/reference/current/query-dsl-dis-max-query.html
|
|
type DisMaxQuery struct {
|
|
queries []Mappable
|
|
tieBreaker float32
|
|
}
|
|
|
|
// DisMax creates a new compound query of type "dis_max" with the provided
|
|
// queries.
|
|
func DisMax(queries ...Mappable) *DisMaxQuery {
|
|
return &DisMaxQuery{
|
|
queries: queries,
|
|
}
|
|
}
|
|
|
|
// TieBreaker sets the "tie_breaker" value for the query.
|
|
func (q *DisMaxQuery) TieBreaker(b float32) *DisMaxQuery {
|
|
q.tieBreaker = b
|
|
return q
|
|
}
|
|
|
|
// Map returns a map representation of the dis_max query, thus implementing
|
|
// the Mappable interface.
|
|
func (q *DisMaxQuery) Map() map[string]interface{} {
|
|
inner := make([]map[string]interface{}, len(q.queries))
|
|
for i, iq := range q.queries {
|
|
inner[i] = iq.Map()
|
|
}
|
|
return map[string]interface{}{
|
|
"dis_max": structs.Map(struct {
|
|
Queries []map[string]interface{} `structs:"queries"`
|
|
TieBreaker float32 `structs:"tie_breaker,omitempty"`
|
|
}{inner, q.tieBreaker}),
|
|
}
|
|
}
|