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.
111 lines
2.6 KiB
111 lines
2.6 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 elastic |
|
|
|
import ( |
|
"context" |
|
"encoding/json" |
|
"fmt" |
|
"net/url" |
|
"strings" |
|
) |
|
|
|
// MultiSearch executes one or more searches in one roundtrip. |
|
type MultiSearchService struct { |
|
client *Client |
|
requests []*SearchRequest |
|
indices []string |
|
pretty bool |
|
maxConcurrentRequests *int |
|
preFilterShardSize *int |
|
} |
|
|
|
func NewMultiSearchService(client *Client) *MultiSearchService { |
|
builder := &MultiSearchService{ |
|
client: client, |
|
} |
|
return builder |
|
} |
|
|
|
func (s *MultiSearchService) Add(requests ...*SearchRequest) *MultiSearchService { |
|
s.requests = append(s.requests, requests...) |
|
return s |
|
} |
|
|
|
func (s *MultiSearchService) Index(indices ...string) *MultiSearchService { |
|
s.indices = append(s.indices, indices...) |
|
return s |
|
} |
|
|
|
func (s *MultiSearchService) Pretty(pretty bool) *MultiSearchService { |
|
s.pretty = pretty |
|
return s |
|
} |
|
|
|
func (s *MultiSearchService) MaxConcurrentSearches(max int) *MultiSearchService { |
|
s.maxConcurrentRequests = &max |
|
return s |
|
} |
|
|
|
func (s *MultiSearchService) PreFilterShardSize(size int) *MultiSearchService { |
|
s.preFilterShardSize = &size |
|
return s |
|
} |
|
|
|
func (s *MultiSearchService) Do(ctx context.Context) (*MultiSearchResult, error) { |
|
// Build url |
|
path := "/_msearch" |
|
|
|
// Parameters |
|
params := make(url.Values) |
|
if s.pretty { |
|
params.Set("pretty", fmt.Sprintf("%v", s.pretty)) |
|
} |
|
if v := s.maxConcurrentRequests; v != nil { |
|
params.Set("max_concurrent_searches", fmt.Sprintf("%v", *v)) |
|
} |
|
if v := s.preFilterShardSize; v != nil { |
|
params.Set("pre_filter_shard_size", fmt.Sprintf("%v", *v)) |
|
} |
|
|
|
// Set body |
|
var lines []string |
|
for _, sr := range s.requests { |
|
// Set default indices if not specified in the request |
|
if !sr.HasIndices() && len(s.indices) > 0 { |
|
sr = sr.Index(s.indices...) |
|
} |
|
|
|
header, err := json.Marshal(sr.header()) |
|
if err != nil { |
|
return nil, err |
|
} |
|
body, err := sr.Body() |
|
if err != nil { |
|
return nil, err |
|
} |
|
lines = append(lines, string(header)) |
|
lines = append(lines, body) |
|
} |
|
body := strings.Join(lines, "\n") + "\n" // add trailing \n |
|
|
|
// Get response |
|
res, err := s.client.PerformRequest(ctx, "GET", path, params, body) |
|
if err != nil { |
|
return nil, err |
|
} |
|
|
|
// Return result |
|
ret := new(MultiSearchResult) |
|
if err := s.client.decoder.Decode(res.Body, ret); err != nil { |
|
return nil, err |
|
} |
|
return ret, nil |
|
} |
|
|
|
// MultiSearchResult is the outcome of running a multi-search operation. |
|
type MultiSearchResult struct { |
|
Responses []*SearchResult `json:"responses,omitempty"` |
|
}
|
|
|