Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

benchmark: make number of watcher streams configurable in watch-get

Browse Source 
Each stream uses a client goroutine and a grpc stream; the setup causes
considerable client-side latency on the first get requests.
This commit is contained in:
Anthony Romano 2016-05-11 10:00:19 -07:00
parent 2aa4dd52cc
commit f5165a0149
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tools/benchmark/cmd/watch_get.go
View File

@ -35,14 +35,16 @@ var watchGetCmd = &cobra.Command{
}
var (
watchGetTotalStreams int
watchEvents int
firstWatch sync.Once
watchGetTotalWatchers int
watchGetTotalStreams int
watchEvents int
firstWatch sync.Once
)
func init() {
RootCmd.AddCommand(watchGetCmd)
watchGetCmd.Flags().IntVar(&watchGetTotalStreams, "watchers", 10000, "Total number of watchers")
watchGetCmd.Flags().IntVar(&watchGetTotalWatchers, "watchers", 10000, "Total number of watchers")
watchGetCmd.Flags().IntVar(&watchGetTotalStreams, "streams", 1, "Total number of watcher streams")
watchGetCmd.Flags().IntVar(&watchEvents, "events", 8, "Number of events per watcher")
}
@ -71,18 +73,18 @@ func watchGetFunc(cmd *cobra.Command, args []string) {
// results from trying to do serialized gets with concurrent watchers
results = make(chan result)
bar = pb.New(watchGetTotalStreams * watchEvents)
bar = pb.New(watchGetTotalWatchers * watchEvents)
bar.Format("Bom !")
bar.Start()
pdoneC := printReport(results)
wg.Add(len(streams))
wg.Add(watchGetTotalWatchers)
ctx, cancel := context.WithCancel(context.TODO())
f := func() {
doSerializedGet(ctx, getClient[0], results)
}
for i := range streams {
go doUnsyncWatch(streams[i], watchRev, f)
for i := 0; i < watchGetTotalWatchers; i++ {
go doUnsyncWatch(streams[i%len(streams)], watchRev, f)
}
wg.Wait()
cancel()