151 lines
4.4 KiB
Go
151 lines
4.4 KiB
Go
// Copyright 2012-2015 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: http://www.elasticsearch.org/guide/en/elasticsearch/reference/current/search-aggregations-metrics-top-hits-aggregation.html
|
|
type TopHitsAggregation struct {
|
|
searchSource *SearchSource
|
|
}
|
|
|
|
func NewTopHitsAggregation() TopHitsAggregation {
|
|
a := TopHitsAggregation{
|
|
searchSource: NewSearchSource(),
|
|
}
|
|
return a
|
|
}
|
|
|
|
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) NoFields() TopHitsAggregation {
|
|
a.searchSource = a.searchSource.NoFields()
|
|
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) FieldDataFields(fieldDataFields ...string) TopHitsAggregation {
|
|
a.searchSource = a.searchSource.FieldDataFields(fieldDataFields...)
|
|
return a
|
|
}
|
|
|
|
func (a TopHitsAggregation) FieldDataField(fieldDataField string) TopHitsAggregation {
|
|
a.searchSource = a.searchSource.FieldDataField(fieldDataField)
|
|
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) PartialFields(partialFields ...*PartialField) TopHitsAggregation {
|
|
a.searchSource = a.searchSource.PartialFields(partialFields...)
|
|
return a
|
|
}
|
|
|
|
func (a TopHitsAggregation) PartialField(partialField *PartialField) TopHitsAggregation {
|
|
a.searchSource = a.searchSource.PartialField(partialField)
|
|
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{} {
|
|
// 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{})
|
|
source["top_hits"] = a.searchSource.Source()
|
|
return source
|
|
}
|