fix(bench): initial commit
parent
75c02ed0da
commit
cc88215b46
|
@ -1,6 +1,7 @@
|
||||||
src/
|
src/
|
||||||
pkg/
|
pkg/
|
||||||
/etcd
|
/etcd
|
||||||
|
/etcdbench
|
||||||
/server/release_version.go
|
/server/release_version.go
|
||||||
/go-bindata
|
/go-bindata
|
||||||
/machine*
|
/machine*
|
||||||
|
|
|
@ -0,0 +1,64 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"flag"
|
||||||
|
"log"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/coreos/go-etcd/etcd"
|
||||||
|
)
|
||||||
|
|
||||||
|
func doWrite(client *etcd.Client, key string, c chan int) {
|
||||||
|
client.Set(key, key, 0)
|
||||||
|
c <- 1
|
||||||
|
}
|
||||||
|
|
||||||
|
func write(client *etcd.Client, requests int, end chan int) {
|
||||||
|
c := make(chan int)
|
||||||
|
|
||||||
|
for i := 0; i < requests; i++ {
|
||||||
|
key := strconv.Itoa(i)
|
||||||
|
go doWrite(client, key, c)
|
||||||
|
<-c
|
||||||
|
}
|
||||||
|
end <- 1
|
||||||
|
}
|
||||||
|
|
||||||
|
func watch(client *etcd.Client, key string) {
|
||||||
|
receiver := make(chan *etcd.Response)
|
||||||
|
go client.Watch(key, 0, true, receiver, nil)
|
||||||
|
|
||||||
|
log.Printf("watching: %s", key)
|
||||||
|
|
||||||
|
received := 0
|
||||||
|
for {
|
||||||
|
<-receiver
|
||||||
|
received++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
rWrites := flag.Int("write-requests", 50000, "number of writes")
|
||||||
|
cWrites := flag.Int("concurrent-writes", 500, "number of concurrent writes")
|
||||||
|
|
||||||
|
watches := flag.Int("watches", 500, "number of writes")
|
||||||
|
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
|
client := etcd.NewClient(nil)
|
||||||
|
|
||||||
|
for i := 0; i < *watches; i++ {
|
||||||
|
key := strconv.Itoa(i)
|
||||||
|
go watch(client, key)
|
||||||
|
}
|
||||||
|
|
||||||
|
wChan := make(chan int, *cWrites)
|
||||||
|
for i := 0; i < *cWrites; i++ {
|
||||||
|
go write(client, (*rWrites / *cWrites), wChan)
|
||||||
|
}
|
||||||
|
|
||||||
|
for i := 0; i < *cWrites; i++ {
|
||||||
|
<-wChan
|
||||||
|
log.Printf("Completed %d writes", (*rWrites / *cWrites))
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue