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

Merge pull request #7141 from heyitsanthony/rate-limit-range

Browse Source 
benchmark: option to rate limit range benchmark
This commit is contained in:
Anthony Romano 2017-01-11 15:44:33 -08:00 committed by GitHub
commit 0df543dbb3
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tools/benchmark/cmd/range.go
View File

@ -16,6 +16,7 @@ package cmd
import ( import (
"fmt" "fmt"
"math"
"os" "os"
"time" "time"
@ -24,6 +25,7 @@ import (
"github.com/spf13/cobra" "github.com/spf13/cobra"
"golang.org/x/net/context" "golang.org/x/net/context"
"golang.org/x/time/rate"
"gopkg.in/cheggaaa/pb.v1" "gopkg.in/cheggaaa/pb.v1"
) )
@ -36,12 +38,14 @@ var rangeCmd = &cobra.Command{
} }
var ( var (
rangeRate int
rangeTotal int rangeTotal int
rangeConsistency string rangeConsistency string
) )
func init() { func init() {
RootCmd.AddCommand(rangeCmd) RootCmd.AddCommand(rangeCmd)
rangeCmd.Flags().IntVar(&rangeRate, "rate", 0, "Maximum range requests per second (0 is no limit)")
rangeCmd.Flags().IntVar(&rangeTotal, "total", 10000, "Total number of range requests") rangeCmd.Flags().IntVar(&rangeTotal, "total", 10000, "Total number of range requests")
rangeCmd.Flags().StringVar(&rangeConsistency, "consistency", "l", "Linearizable(l) or Serializable(s)") rangeCmd.Flags().StringVar(&rangeConsistency, "consistency", "l", "Linearizable(l) or Serializable(s)")
} }
@ -67,6 +71,11 @@ func rangeFunc(cmd *cobra.Command, args []string) {
os.Exit(1) os.Exit(1)
} }
if rangeRate == 0 {
rangeRate = math.MaxInt32
}
limit := rate.NewLimiter(rate.Limit(rangeRate), 1)
requests := make(chan v3.Op, totalClients) requests := make(chan v3.Op, totalClients)
clients := mustCreateClients(totalClients, totalConns) clients := mustCreateClients(totalClients, totalConns)
@ -80,6 +89,8 @@ func rangeFunc(cmd *cobra.Command, args []string) {
go func(c *v3.Client) { go func(c *v3.Client) {
defer wg.Done() defer wg.Done()
for op := range requests { for op := range requests {
limit.Wait(context.Background())
st := time.Now() st := time.Now()
_, err := c.Do(context.Background(), op) _, err := c.Do(context.Background(), op)
r.Results() <- report.Result{Err: err, Start: st, End: time.Now()} r.Results() <- report.Result{Err: err, Start: st, End: time.Now()}