通用包
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
utils/pkg/es/search_queries_slice.go

53 lines
1.2 KiB

// Copyright 2012-present 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 es
// SliceQuery allows to partition the documents into several slices.
// It is used e.g. to slice scroll operations in Elasticsearch 5.0 or later.
// See https://www.elastic.co/guide/en/elasticsearch/reference/7.0/search-request-scroll.html#sliced-scroll
// for details.
type SliceQuery struct {
field string
id *int
max *int
}
// NewSliceQuery creates a new SliceQuery.
func NewSliceQuery() *SliceQuery {
return &SliceQuery{}
}
// Field is the name of the field to slice against (_uid by default).
func (s *SliceQuery) Field(field string) *SliceQuery {
s.field = field
return s
}
// Id is the id of the slice.
func (s *SliceQuery) Id(id int) *SliceQuery {
s.id = &id
return s
}
// Max is the maximum number of slices.
func (s *SliceQuery) Max(max int) *SliceQuery {
s.max = &max
return s
}
// Source returns the JSON body.
func (s *SliceQuery) Source() (interface{}, error) {
m := make(map[string]interface{})
if s.field != "" {
m["field"] = s.field
}
if s.id != nil {
m["id"] = *s.id
}
if s.max != nil {
m["max"] = *s.max
}
return m, nil
}