From 622998add12734a6c2b5d79918338a4d6dca7ce6 Mon Sep 17 00:00:00 2001
From: George Goldberg <george@gberg.me>
Date: Tue, 9 May 2017 14:13:24 +0100
Subject: PLT-6398: Add dependency on go elastic search library. (#6340)

---
 .../olivere/elastic.v5/search_collapse_builder.go  | 68 ++++++++++++++++++++++
 1 file changed, 68 insertions(+)
 create mode 100644 vendor/gopkg.in/olivere/elastic.v5/search_collapse_builder.go

(limited to 'vendor/gopkg.in/olivere/elastic.v5/search_collapse_builder.go')

diff --git a/vendor/gopkg.in/olivere/elastic.v5/search_collapse_builder.go b/vendor/gopkg.in/olivere/elastic.v5/search_collapse_builder.go
new file mode 100644
index 000000000..48ed17acb
--- /dev/null
+++ b/vendor/gopkg.in/olivere/elastic.v5/search_collapse_builder.go
@@ -0,0 +1,68 @@
+// 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
+
+// CollapseBuilder enables field collapsing on a search request.
+// See https://www.elastic.co/guide/en/elasticsearch/reference/5.3/search-request-collapse.html
+// for details.
+type CollapseBuilder struct {
+	field                      string
+	innerHit                   *InnerHit
+	maxConcurrentGroupRequests *int
+}
+
+// NewCollapseBuilder creates a new CollapseBuilder.
+func NewCollapseBuilder(field string) *CollapseBuilder {
+	return &CollapseBuilder{field: field}
+}
+
+// Field to collapse.
+func (b *CollapseBuilder) Field(field string) *CollapseBuilder {
+	b.field = field
+	return b
+}
+
+// InnerHit option to expand the collapsed results.
+func (b *CollapseBuilder) InnerHit(innerHit *InnerHit) *CollapseBuilder {
+	b.innerHit = innerHit
+	return b
+}
+
+// MaxConcurrentGroupRequests is the maximum number of group requests that are
+// allowed to be ran concurrently in the inner_hits phase.
+func (b *CollapseBuilder) MaxConcurrentGroupRequests(max int) *CollapseBuilder {
+	b.maxConcurrentGroupRequests = &max
+	return b
+}
+
+// Source generates the JSON serializable fragment for the CollapseBuilder.
+func (b *CollapseBuilder) Source() (interface{}, error) {
+	// {
+	//   "field": "user",
+	//   "inner_hits": {
+	//     "name": "last_tweets",
+	//     "size": 5,
+	//     "sort": [{ "date": "asc" }]
+	//   },
+	//   "max_concurrent_group_searches": 4
+	// }
+	src := map[string]interface{}{
+		"field": b.field,
+	}
+
+	if b.innerHit != nil {
+		hits, err := b.innerHit.Source()
+		if err != nil {
+			return nil, err
+		}
+		src["inner_hits"] = hits
+	}
+
+	if b.maxConcurrentGroupRequests != nil {
+		src["max_concurrent_group_searches"] = *b.maxConcurrentGroupRequests
+	}
+
+	return src, nil
+}
-- 
cgit v1.2.3-1-g7c22