indexed_iter_test.go

Documentation: github.com/syndtr/goleveldb/leveldb/iterator

     1  // Copyright (c) 2014, Suryandaru Triandana <syndtr@gmail.com>
     2  // All rights reserved.
     3  //
     4  // Use of this source code is governed by a BSD-style license that can be
     5  // found in the LICENSE file.
     6  
     7  package iterator_test
     8  
     9  import (
    10  	"sort"
    11  
    12  	. "github.com/onsi/ginkgo"
    13  
    14  	"github.com/syndtr/goleveldb/leveldb/comparer"
    15  	. "github.com/syndtr/goleveldb/leveldb/iterator"
    16  	"github.com/syndtr/goleveldb/leveldb/testutil"
    17  )
    18  
    19  type keyValue struct {
    20  	key []byte
    21  	testutil.KeyValue
    22  }
    23  
    24  type keyValueIndex []keyValue
    25  
    26  func (x keyValueIndex) Search(key []byte) int {
    27  	return sort.Search(x.Len(), func(i int) bool {
    28  		return comparer.DefaultComparer.Compare(x[i].key, key) >= 0
    29  	})
    30  }
    31  
    32  func (x keyValueIndex) Len() int                        { return len(x) }
    33  func (x keyValueIndex) Index(i int) (key, value []byte) { return x[i].key, nil }
    34  func (x keyValueIndex) Get(i int) Iterator              { return NewArrayIterator(x[i]) }
    35  
    36  var _ = testutil.Defer(func() {
    37  	Describe("Indexed iterator", func() {
    38  		Test := func(n ...int) func() {
    39  			if len(n) == 0 {
    40  				rnd := testutil.NewRand()
    41  				n = make([]int, rnd.Intn(17)+3)
    42  				for i := range n {
    43  					n[i] = rnd.Intn(19) + 1
    44  				}
    45  			}
    46  
    47  			return func() {
    48  				It("Should iterates and seeks correctly", func(done Done) {
    49  					// Build key/value.
    50  					index := make(keyValueIndex, len(n))
    51  					sum := 0
    52  					for _, x := range n {
    53  						sum += x
    54  					}
    55  					kv := testutil.KeyValue_Generate(nil, sum, 1, 1, 10, 4, 4)
    56  					for i, j := 0, 0; i < len(n); i++ {
    57  						for x := n[i]; x > 0; x-- {
    58  							key, value := kv.Index(j)
    59  							index[i].key = key
    60  							index[i].Put(key, value)
    61  							j++
    62  						}
    63  					}
    64  
    65  					// Test the iterator.
    66  					t := testutil.IteratorTesting{
    67  						KeyValue: kv.Clone(),
    68  						Iter:     NewIndexedIterator(NewArrayIndexer(index), true),
    69  					}
    70  					testutil.DoIteratorTesting(&t)
    71  					done <- true
    72  				}, 15.0)
    73  			}
    74  		}
    75  
    76  		Describe("with 100 keys", Test(100))
    77  		Describe("with 50-50 keys", Test(50, 50))
    78  		Describe("with 50-1 keys", Test(50, 1))
    79  		Describe("with 50-1-50 keys", Test(50, 1, 50))
    80  		Describe("with 1-50 keys", Test(1, 50))
    81  		Describe("with random N-keys", Test())
    82  	})
    83  })
    84
View as plain text