219 lines
4.8 KiB
Go
219 lines
4.8 KiB
Go
// Copyright 2015 The etcd Authors
|
|
//
|
|
// 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 v2store
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"runtime"
|
|
"testing"
|
|
)
|
|
|
|
func BenchmarkStoreSet128Bytes(b *testing.B) {
|
|
benchStoreSet(b, 128, nil)
|
|
}
|
|
|
|
func BenchmarkStoreSet1024Bytes(b *testing.B) {
|
|
benchStoreSet(b, 1024, nil)
|
|
}
|
|
|
|
func BenchmarkStoreSet4096Bytes(b *testing.B) {
|
|
benchStoreSet(b, 4096, nil)
|
|
}
|
|
|
|
func BenchmarkStoreSetWithJson128Bytes(b *testing.B) {
|
|
benchStoreSet(b, 128, json.Marshal)
|
|
}
|
|
|
|
func BenchmarkStoreSetWithJson1024Bytes(b *testing.B) {
|
|
benchStoreSet(b, 1024, json.Marshal)
|
|
}
|
|
|
|
func BenchmarkStoreSetWithJson4096Bytes(b *testing.B) {
|
|
benchStoreSet(b, 4096, json.Marshal)
|
|
}
|
|
|
|
func BenchmarkStoreDelete(b *testing.B) {
|
|
b.StopTimer()
|
|
|
|
s := newStore()
|
|
kvs, _ := generateNRandomKV(b.N, 128)
|
|
|
|
memStats := new(runtime.MemStats)
|
|
runtime.GC()
|
|
runtime.ReadMemStats(memStats)
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
_, err := s.Set(kvs[i][0], false, kvs[i][1], TTLOptionSet{ExpireTime: Permanent})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
setMemStats := new(runtime.MemStats)
|
|
runtime.GC()
|
|
runtime.ReadMemStats(setMemStats)
|
|
|
|
b.StartTimer()
|
|
|
|
for i := range kvs {
|
|
s.Delete(kvs[i][0], false, false)
|
|
}
|
|
|
|
b.StopTimer()
|
|
|
|
// clean up
|
|
e, err := s.Get("/", false, false)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
for _, n := range e.Node.Nodes {
|
|
_, err := s.Delete(n.Key, true, true)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
s.WatcherHub.EventHistory = nil
|
|
|
|
deleteMemStats := new(runtime.MemStats)
|
|
runtime.GC()
|
|
runtime.ReadMemStats(deleteMemStats)
|
|
|
|
fmt.Printf("\nBefore set Alloc: %v; After set Alloc: %v, After delete Alloc: %v\n",
|
|
memStats.Alloc/1000, setMemStats.Alloc/1000, deleteMemStats.Alloc/1000)
|
|
}
|
|
|
|
func BenchmarkWatch(b *testing.B) {
|
|
b.StopTimer()
|
|
s := newStore()
|
|
kvs, _ := generateNRandomKV(b.N, 128)
|
|
b.StartTimer()
|
|
|
|
memStats := new(runtime.MemStats)
|
|
runtime.GC()
|
|
runtime.ReadMemStats(memStats)
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
w, _ := s.Watch(kvs[i][0], false, false, 0)
|
|
|
|
e := newEvent("set", kvs[i][0], uint64(i+1), uint64(i+1))
|
|
s.WatcherHub.notify(e)
|
|
<-w.EventChan()
|
|
s.CurrentIndex++
|
|
}
|
|
|
|
s.WatcherHub.EventHistory = nil
|
|
afterMemStats := new(runtime.MemStats)
|
|
runtime.GC()
|
|
runtime.ReadMemStats(afterMemStats)
|
|
fmt.Printf("\nBefore Alloc: %v; After Alloc: %v\n",
|
|
memStats.Alloc/1000, afterMemStats.Alloc/1000)
|
|
}
|
|
|
|
func BenchmarkWatchWithSet(b *testing.B) {
|
|
b.StopTimer()
|
|
s := newStore()
|
|
kvs, _ := generateNRandomKV(b.N, 128)
|
|
b.StartTimer()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
w, _ := s.Watch(kvs[i][0], false, false, 0)
|
|
|
|
s.Set(kvs[i][0], false, "test", TTLOptionSet{ExpireTime: Permanent})
|
|
<-w.EventChan()
|
|
}
|
|
}
|
|
|
|
func BenchmarkWatchWithSetBatch(b *testing.B) {
|
|
b.StopTimer()
|
|
s := newStore()
|
|
kvs, _ := generateNRandomKV(b.N, 128)
|
|
b.StartTimer()
|
|
|
|
watchers := make([]Watcher, b.N)
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
watchers[i], _ = s.Watch(kvs[i][0], false, false, 0)
|
|
}
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
s.Set(kvs[i][0], false, "test", TTLOptionSet{ExpireTime: Permanent})
|
|
}
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
<-watchers[i].EventChan()
|
|
}
|
|
|
|
}
|
|
|
|
func BenchmarkWatchOneKey(b *testing.B) {
|
|
s := newStore()
|
|
watchers := make([]Watcher, b.N)
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
watchers[i], _ = s.Watch("/foo", false, false, 0)
|
|
}
|
|
|
|
s.Set("/foo", false, "", TTLOptionSet{ExpireTime: Permanent})
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
<-watchers[i].EventChan()
|
|
}
|
|
}
|
|
|
|
func benchStoreSet(b *testing.B, valueSize int, process func(interface{}) ([]byte, error)) {
|
|
s := newStore()
|
|
b.StopTimer()
|
|
kvs, size := generateNRandomKV(b.N, valueSize)
|
|
b.StartTimer()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
resp, err := s.Set(kvs[i][0], false, kvs[i][1], TTLOptionSet{ExpireTime: Permanent})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if process != nil {
|
|
_, err = process(resp)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
b.StopTimer()
|
|
memStats := new(runtime.MemStats)
|
|
runtime.GC()
|
|
runtime.ReadMemStats(memStats)
|
|
fmt.Printf("\nAlloc: %vKB; Data: %vKB; Kvs: %v; Alloc/Data:%v\n",
|
|
memStats.Alloc/1000, size/1000, b.N, memStats.Alloc/size)
|
|
}
|
|
|
|
func generateNRandomKV(n int, valueSize int) ([][]string, uint64) {
|
|
var size uint64
|
|
kvs := make([][]string, n)
|
|
bytes := make([]byte, valueSize)
|
|
|
|
for i := 0; i < n; i++ {
|
|
kvs[i] = make([]string, 2)
|
|
kvs[i][0] = fmt.Sprintf("/%010d/%010d/%010d", n, n, n)
|
|
kvs[i][1] = string(bytes)
|
|
size = size + uint64(len(kvs[i][0])) + uint64(len(kvs[i][1]))
|
|
}
|
|
|
|
return kvs, size
|
|
}
|