32 lines
584 B
Go
32 lines
584 B
Go
|
package esquery
|
||
|
|
|
||
|
|
import "encoding/json"
|
||
|
|
|
||
|
|
type DisMaxQuery struct {
|
||
|
|
params disMaxParams
|
||
|
|
}
|
||
|
|
|
||
|
|
type disMaxParams struct {
|
||
|
|
Queries []json.Marshaler `json:"queries"`
|
||
|
|
TieBreaker float32 `json:"tie_breaker,omitempty"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func DisMax(queries ...json.Marshaler) *DisMaxQuery {
|
||
|
|
return &DisMaxQuery{
|
||
|
|
params: disMaxParams{
|
||
|
|
Queries: queries,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q *DisMaxQuery) TieBreaker(b float32) *DisMaxQuery {
|
||
|
|
q.params.TieBreaker = b
|
||
|
|
return q
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q DisMaxQuery) MarshalJSON() ([]byte, error) {
|
||
|
|
return json.Marshal(map[string]disMaxParams{
|
||
|
|
"dis_max": q.params,
|
||
|
|
})
|
||
|
|
}
|