Switch from Godep to go vendoring

vendor/github.com/blevesearch/bleve/search/collectors/collector_top_score.go

@@ -0,0 +1,135 @@
+//  Copyright (c) 2014 Couchbase, Inc.
+//  Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
+//  except in compliance with the License. You may obtain a copy of the License at
+//    http://www.apache.org/licenses/LICENSE-2.0
+//  Unless required by applicable law or agreed to in writing, software distributed under the
+//  License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
+//  either express or implied. See the License for the specific language governing permissions
+//  and limitations under the License.
+
+package collectors
+
+import (
+	"container/list"
+	"time"
+
+	"github.com/blevesearch/bleve/search"
+)
+
+type TopScoreCollector struct {
+	k             int
+	skip          int
+	results       *list.List
+	took          time.Duration
+	maxScore      float64
+	total         uint64
+	facetsBuilder *search.FacetsBuilder
+}
+
+func NewTopScorerCollector(k int) *TopScoreCollector {
+	return &TopScoreCollector{
+		k:       k,
+		skip:    0,
+		results: list.New(),
+	}
+}
+
+func NewTopScorerSkipCollector(k, skip int) *TopScoreCollector {
+	return &TopScoreCollector{
+		k:       k,
+		skip:    skip,
+		results: list.New(),
+	}
+}
+
+func (tksc *TopScoreCollector) Total() uint64 {
+	return tksc.total
+}
+
+func (tksc *TopScoreCollector) MaxScore() float64 {
+	return tksc.maxScore
+}
+
+func (tksc *TopScoreCollector) Took() time.Duration {
+	return tksc.took
+}
+
+func (tksc *TopScoreCollector) Collect(searcher search.Searcher) error {
+	startTime := time.Now()
+	next, err := searcher.Next()
+	for err == nil && next != nil {
+		tksc.collectSingle(next)
+		if tksc.facetsBuilder != nil {
+			err = tksc.facetsBuilder.Update(next)
+			if err != nil {
+				break
+			}
+		}
+		next, err = searcher.Next()
+	}
+	// compute search duration
+	tksc.took = time.Since(startTime)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (tksc *TopScoreCollector) collectSingle(dm *search.DocumentMatch) {
+	// increment total hits
+	tksc.total++
+
+	// update max score
+	if dm.Score > tksc.maxScore {
+		tksc.maxScore = dm.Score
+	}
+
+	for e := tksc.results.Front(); e != nil; e = e.Next() {
+		curr := e.Value.(*search.DocumentMatch)
+		if dm.Score < curr.Score {
+
+			tksc.results.InsertBefore(dm, e)
+			// if we just made the list too long
+			if tksc.results.Len() > (tksc.k + tksc.skip) {
+				// remove the head
+				tksc.results.Remove(tksc.results.Front())
+			}
+			return
+		}
+	}
+	// if we got to the end, we still have to add it
+	tksc.results.PushBack(dm)
+	if tksc.results.Len() > (tksc.k + tksc.skip) {
+		// remove the head
+		tksc.results.Remove(tksc.results.Front())
+	}
+}
+
+func (tksc *TopScoreCollector) Results() search.DocumentMatchCollection {
+	if tksc.results.Len()-tksc.skip > 0 {
+		rv := make(search.DocumentMatchCollection, tksc.results.Len()-tksc.skip)
+		i := 0
+		skipped := 0
+		for e := tksc.results.Back(); e != nil; e = e.Prev() {
+			if skipped < tksc.skip {
+				skipped++
+				continue
+			}
+			rv[i] = e.Value.(*search.DocumentMatch)
+			i++
+		}
+		return rv
+	}
+	return search.DocumentMatchCollection{}
+}
+
+func (tksc *TopScoreCollector) SetFacetsBuilder(facetsBuilder *search.FacetsBuilder) {
+	tksc.facetsBuilder = facetsBuilder
+}
+
+func (tksc *TopScoreCollector) FacetResults() search.FacetResults {
+	if tksc.facetsBuilder != nil {
+		return tksc.facetsBuilder.Results()
+	}
+	return search.FacetResults{}
+}

vendor/github.com/blevesearch/bleve/search/collectors/collector_top_score_test.go

@@ -0,0 +1,249 @@
+//  Copyright (c) 2014 Couchbase, Inc.
+//  Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
+//  except in compliance with the License. You may obtain a copy of the License at
+//    http://www.apache.org/licenses/LICENSE-2.0
+//  Unless required by applicable law or agreed to in writing, software distributed under the
+//  License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
+//  either express or implied. See the License for the specific language governing permissions
+//  and limitations under the License.
+
+package collectors
+
+import (
+	"math/rand"
+	"strconv"
+	"testing"
+
+	"github.com/blevesearch/bleve/search"
+)
+
+func TestTop10Scores(t *testing.T) {
+
+	// a stub search with more than 10 matches
+	// the top-10 scores are > 10
+	// everything else is less than 10
+	searcher := &stubSearcher{
+		matches: search.DocumentMatchCollection{
+			&search.DocumentMatch{
+				ID:    "a",
+				Score: 11,
+			},
+			&search.DocumentMatch{
+				ID:    "b",
+				Score: 9,
+			},
+			&search.DocumentMatch{
+				ID:    "c",
+				Score: 11,
+			},
+			&search.DocumentMatch{
+				ID:    "d",
+				Score: 9,
+			},
+			&search.DocumentMatch{
+				ID:    "e",
+				Score: 11,
+			},
+			&search.DocumentMatch{
+				ID:    "f",
+				Score: 9,
+			},
+			&search.DocumentMatch{
+				ID:    "g",
+				Score: 11,
+			},
+			&search.DocumentMatch{
+				ID:    "h",
+				Score: 9,
+			},
+			&search.DocumentMatch{
+				ID:    "i",
+				Score: 11,
+			},
+			&search.DocumentMatch{
+				ID:    "j",
+				Score: 11,
+			},
+			&search.DocumentMatch{
+				ID:    "k",
+				Score: 11,
+			},
+			&search.DocumentMatch{
+				ID:    "l",
+				Score: 99,
+			},
+			&search.DocumentMatch{
+				ID:    "m",
+				Score: 11,
+			},
+			&search.DocumentMatch{
+				ID:    "n",
+				Score: 11,
+			},
+		},
+	}
+
+	collector := NewTopScorerCollector(10)
+	err := collector.Collect(searcher)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	maxScore := collector.MaxScore()
+	if maxScore != 99.0 {
+		t.Errorf("expected max score 99.0, got %f", maxScore)
+	}
+
+	total := collector.Total()
+	if total != 14 {
+		t.Errorf("expected 14 total results, got %d", total)
+	}
+
+	results := collector.Results()
+
+	if len(results) != 10 {
+		t.Fatalf("expected 10 results, got %d", len(results))
+	}
+
+	if results[0].ID != "l" {
+		t.Errorf("expected first result to have ID 'l', got %s", results[0].ID)
+	}
+
+	if results[0].Score != 99.0 {
+		t.Errorf("expected highest score to be 99.0, got %f", results[0].Score)
+	}
+
+	minScore := 1000.0
+	for _, result := range results {
+		if result.Score < minScore {
+			minScore = result.Score
+		}
+	}
+
+	if minScore < 10 {
+		t.Errorf("expected minimum score to be higher than 10, got %f", minScore)
+	}
+}
+
+func TestTop10ScoresSkip10(t *testing.T) {
+
+	// a stub search with more than 10 matches
+	// the top-10 scores are > 10
+	// everything else is less than 10
+	searcher := &stubSearcher{
+		matches: search.DocumentMatchCollection{
+			&search.DocumentMatch{
+				ID:    "a",
+				Score: 11,
+			},
+			&search.DocumentMatch{
+				ID:    "b",
+				Score: 9.5,
+			},
+			&search.DocumentMatch{
+				ID:    "c",
+				Score: 11,
+			},
+			&search.DocumentMatch{
+				ID:    "d",
+				Score: 9,
+			},
+			&search.DocumentMatch{
+				ID:    "e",
+				Score: 11,
+			},
+			&search.DocumentMatch{
+				ID:    "f",
+				Score: 9,
+			},
+			&search.DocumentMatch{
+				ID:    "g",
+				Score: 11,
+			},
+			&search.DocumentMatch{
+				ID:    "h",
+				Score: 9,
+			},
+			&search.DocumentMatch{
+				ID:    "i",
+				Score: 11,
+			},
+			&search.DocumentMatch{
+				ID:    "j",
+				Score: 11,
+			},
+			&search.DocumentMatch{
+				ID:    "k",
+				Score: 11,
+			},
+			&search.DocumentMatch{
+				ID:    "l",
+				Score: 99,
+			},
+			&search.DocumentMatch{
+				ID:    "m",
+				Score: 11,
+			},
+			&search.DocumentMatch{
+				ID:    "n",
+				Score: 11,
+			},
+		},
+	}
+
+	collector := NewTopScorerSkipCollector(10, 10)
+	err := collector.Collect(searcher)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	maxScore := collector.MaxScore()
+	if maxScore != 99.0 {
+		t.Errorf("expected max score 99.0, got %f", maxScore)
+	}
+
+	total := collector.Total()
+	if total != 14 {
+		t.Errorf("expected 14 total results, got %d", total)
+	}
+
+	results := collector.Results()
+
+	if len(results) != 4 {
+		t.Fatalf("expected 4 results, got %d", len(results))
+	}
+
+	if results[0].ID != "b" {
+		t.Errorf("expected first result to have ID 'b', got %s", results[0].ID)
+	}
+
+	if results[0].Score != 9.5 {
+		t.Errorf("expected highest score to be 9.5ß, got %f", results[0].Score)
+	}
+}
+
+func BenchmarkTop10of100000Scores(b *testing.B) {
+
+	matches := make(search.DocumentMatchCollection, 0, 100000)
+	for i := 0; i < 100000; i++ {
+		matches = append(matches, &search.DocumentMatch{
+			ID:    strconv.Itoa(i),
+			Score: rand.Float64(),
+		})
+	}
+	searcher := &stubSearcher{
+		matches: matches,
+	}
+
+	collector := NewTopScorerCollector(10)
+	b.ResetTimer()
+
+	err := collector.Collect(searcher)
+	if err != nil {
+		b.Fatal(err)
+	}
+	res := collector.Results()
+	for _, dm := range res {
+		b.Logf("%s - %f\n", dm.ID, dm.Score)
+	}
+}

vendor/github.com/blevesearch/bleve/search/collectors/search_test.go

@@ -0,0 +1,60 @@
+//  Copyright (c) 2014 Couchbase, Inc.
+//  Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
+//  except in compliance with the License. You may obtain a copy of the License at
+//    http://www.apache.org/licenses/LICENSE-2.0
+//  Unless required by applicable law or agreed to in writing, software distributed under the
+//  License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
+//  either express or implied. See the License for the specific language governing permissions
+//  and limitations under the License.
+
+package collectors
+
+import (
+	"github.com/blevesearch/bleve/search"
+)
+
+type stubSearcher struct {
+	index   int
+	matches search.DocumentMatchCollection
+}
+
+func (ss *stubSearcher) Next() (*search.DocumentMatch, error) {
+	if ss.index < len(ss.matches) {
+		rv := ss.matches[ss.index]
+		ss.index++
+		return rv, nil
+	}
+	return nil, nil
+}
+
+func (ss *stubSearcher) Advance(ID string) (*search.DocumentMatch, error) {
+
+	for ss.index < len(ss.matches) && ss.matches[ss.index].ID < ID {
+		ss.index++
+	}
+	if ss.index < len(ss.matches) {
+		rv := ss.matches[ss.index]
+		ss.index++
+		return rv, nil
+	}
+	return nil, nil
+}
+
+func (ss *stubSearcher) Close() error {
+	return nil
+}
+
+func (ss *stubSearcher) Weight() float64 {
+	return 0.0
+}
+
+func (ss *stubSearcher) SetQueryNorm(float64) {
+}
+
+func (ss *stubSearcher) Count() uint64 {
+	return uint64(len(ss.matches))
+}
+
+func (ss *stubSearcher) Min() int {
+	return 0
+}