1 files changed, 143 insertions, 0 deletions
diff --git a/vendor/gopkg.in/olivere/elastic.v5/search_aggs_metrics_top_hits.go b/vendor/gopkg.in/olivere/elastic.v5/search_aggs_metrics_top_hits.go
new file mode 100644
index 000000000..43dd36cdb
--- /dev/null
+++ b/vendor/gopkg.in/olivere/elastic.v5/search_aggs_metrics_top_hits.go
@@ -0,0 +1,143 @@
+// 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
+
+// TopHitsAggregation keeps track of the most relevant document
+// being aggregated. This aggregator is intended to be used as a
+// sub aggregator, so that the top matching documents
+// can be aggregated per bucket.
+//
+// It can effectively be used to group result sets by certain fields via
+// a bucket aggregator. One or more bucket aggregators determines by
+// which properties a result set get sliced into.
+//
+// See: https://www.elastic.co/guide/en/elasticsearch/reference/5.2/search-aggregations-metrics-top-hits-aggregation.html
+type TopHitsAggregation struct {
+	searchSource *SearchSource
+}
+
+func NewTopHitsAggregation() *TopHitsAggregation {
+	return &TopHitsAggregation{
+		searchSource: NewSearchSource(),
+	}
+}
+
+func (a *TopHitsAggregation) From(from int) *TopHitsAggregation {
+	a.searchSource = a.searchSource.From(from)
+	return a
+}
+
+func (a *TopHitsAggregation) Size(size int) *TopHitsAggregation {
+	a.searchSource = a.searchSource.Size(size)
+	return a
+}
+
+func (a *TopHitsAggregation) TrackScores(trackScores bool) *TopHitsAggregation {
+	a.searchSource = a.searchSource.TrackScores(trackScores)
+	return a
+}
+
+func (a *TopHitsAggregation) Explain(explain bool) *TopHitsAggregation {
+	a.searchSource = a.searchSource.Explain(explain)
+	return a
+}
+
+func (a *TopHitsAggregation) Version(version bool) *TopHitsAggregation {
+	a.searchSource = a.searchSource.Version(version)
+	return a
+}
+
+func (a *TopHitsAggregation) NoStoredFields() *TopHitsAggregation {
+	a.searchSource = a.searchSource.NoStoredFields()
+	return a
+}
+
+func (a *TopHitsAggregation) FetchSource(fetchSource bool) *TopHitsAggregation {
+	a.searchSource = a.searchSource.FetchSource(fetchSource)
+	return a
+}
+
+func (a *TopHitsAggregation) FetchSourceContext(fetchSourceContext *FetchSourceContext) *TopHitsAggregation {
+	a.searchSource = a.searchSource.FetchSourceContext(fetchSourceContext)
+	return a
+}
+
+func (a *TopHitsAggregation) DocvalueFields(docvalueFields ...string) *TopHitsAggregation {
+	a.searchSource = a.searchSource.DocvalueFields(docvalueFields...)
+	return a
+}
+
+func (a *TopHitsAggregation) DocvalueField(docvalueField string) *TopHitsAggregation {
+	a.searchSource = a.searchSource.DocvalueField(docvalueField)
+	return a
+}
+
+func (a *TopHitsAggregation) ScriptFields(scriptFields ...*ScriptField) *TopHitsAggregation {
+	a.searchSource = a.searchSource.ScriptFields(scriptFields...)
+	return a
+}
+
+func (a *TopHitsAggregation) ScriptField(scriptField *ScriptField) *TopHitsAggregation {
+	a.searchSource = a.searchSource.ScriptField(scriptField)
+	return a
+}
+
+func (a *TopHitsAggregation) Sort(field string, ascending bool) *TopHitsAggregation {
+	a.searchSource = a.searchSource.Sort(field, ascending)
+	return a
+}
+
+func (a *TopHitsAggregation) SortWithInfo(info SortInfo) *TopHitsAggregation {
+	a.searchSource = a.searchSource.SortWithInfo(info)
+	return a
+}
+
+func (a *TopHitsAggregation) SortBy(sorter ...Sorter) *TopHitsAggregation {
+	a.searchSource = a.searchSource.SortBy(sorter...)
+	return a
+}
+
+func (a *TopHitsAggregation) Highlight(highlight *Highlight) *TopHitsAggregation {
+	a.searchSource = a.searchSource.Highlight(highlight)
+	return a
+}
+
+func (a *TopHitsAggregation) Highlighter() *Highlight {
+	return a.searchSource.Highlighter()
+}
+
+func (a *TopHitsAggregation) Source() (interface{}, error) {
+	// Example:
+	// {
+	//   "aggs": {
+	//       "top_tag_hits": {
+	//           "top_hits": {
+	//               "sort": [
+	//                   {
+	//                       "last_activity_date": {
+	//                           "order": "desc"
+	//                       }
+	//                   }
+	//               ],
+	//               "_source": {
+	//                   "include": [
+	//                       "title"
+	//                   ]
+	//               },
+	//               "size" : 1
+	//           }
+	//       }
+	//   }
+	// }
+	// This method returns only the { "top_hits" : { ... } } part.
+
+	source := make(map[string]interface{})
+	src, err := a.searchSource.Source()
+	if err != nil {
+		return nil, err
+	}
+	source["top_hits"] = src
+	return source, nil
+}