benchmark: correctly count number of watchers

release-3.0
Xiang Li 2016-06-20 10:37:17 -07:00
parent 6f28b43806
commit 2a2dd1075f
1 changed files with 4 additions and 2 deletions

View File

@ -39,6 +39,7 @@ changing the value of the watched keys with concurrent put
requests. requests.
During the test, each watcher watches (--total/--watchers) keys During the test, each watcher watches (--total/--watchers) keys
(a watcher might watch on the same key multiple times if (a watcher might watch on the same key multiple times if
--watched-key-total is small). --watched-key-total is small).
@ -97,7 +98,6 @@ func watchFunc(cmd *cobra.Command, args []string) {
binary.PutVarint(k, int64(rand.Intn(watchKeySpaceSize))) binary.PutVarint(k, int64(rand.Intn(watchKeySpaceSize)))
} }
watched[i] = string(k) watched[i] = string(k)
numWatchers[watched[i]] = numWatchers[watched[i]] + 1
} }
requests := make(chan string, totalClients) requests := make(chan string, totalClients)
@ -126,7 +126,9 @@ func watchFunc(cmd *cobra.Command, args []string) {
go func() { go func() {
for i := 0; i < watchTotal; i++ { for i := 0; i < watchTotal; i++ {
requests <- watched[i%len(watched)] key := watched[i%len(watched)]
requests <- key
numWatchers[key]++
} }
close(requests) close(requests)
}() }()