/usr/share/gocode/src/github.com/couchbase/moss/benchmark_get_test.go is in golang-github-couchbase-moss-dev 0.0~git20170914.0.07c86e8-4.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 | // Copyright (c) 2017 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 moss
import (
"fmt"
"io/ioutil"
"os"
"testing"
)
func BenchmarkCollectionSnapshotGets(b *testing.B) {
tmpDir, _ := ioutil.TempDir("", "benchStore")
defer os.RemoveAll(tmpDir)
store, coll, keys := createStoreAndWriteNItems(tmpDir, 10000, 100)
defer store.Close()
defer coll.Close()
b.ResetTimer()
for i := 0; i < b.N; i++ {
ss, err := coll.Snapshot()
if err != nil || ss == nil {
panic("Snapshot() failed!")
}
_, err = ss.Get(keys[i%len(keys)], ReadOptions{})
if err != nil {
panic("Snapshot-Get() failed!")
}
ss.Close()
}
}
func BenchmarkCollectionGets(b *testing.B) {
tmpDir, _ := ioutil.TempDir("", "benchStore")
defer os.RemoveAll(tmpDir)
store, coll, keys := createStoreAndWriteNItems(tmpDir, 10000, 100)
defer store.Close()
defer coll.Close()
b.ResetTimer()
for i := 0; i < b.N; i++ {
_, err := coll.Get(keys[i%len(keys)], ReadOptions{})
if err != nil {
panic("Collection-Get() failed!")
}
}
}
// ---------------------------------------------------------------
func createStoreAndWriteNItems(tmpDir string, items int,
batches int) (s *Store, c Collection, ks [][]byte) {
store, coll, err := OpenStoreCollection(tmpDir,
StoreOptions{},
StorePersistOptions{})
if err != nil || store == nil {
panic("OpenStoreCollection() failed!")
}
keys := make([][]byte, items)
if batches > items {
batches = 1
}
itemsPerBatch := items / batches
itemCount := 0
for i := 0; i < batches; i++ {
if itemsPerBatch > items-itemCount {
itemsPerBatch = items - itemCount
}
if itemsPerBatch <= 0 {
break
}
batch, err := coll.NewBatch(itemsPerBatch, itemsPerBatch*20)
if err != nil {
panic("NewBatch() failed!")
}
for j := 0; j < itemsPerBatch; j++ {
k := []byte(fmt.Sprintf("key%d", i))
v := []byte(fmt.Sprintf("val%d", i))
itemCount++
batch.Set(k, v)
keys[j] = k
}
err = coll.ExecuteBatch(batch, WriteOptions{})
if err != nil {
panic("ExecuteBatch() failed!")
}
}
return store, coll, keys
}
|