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

tools/benchmark: support serializable range benchmark

Browse Source
This commit is contained in:
Xiang Li 2016-02-13 07:46:31 -08:00
parent c15b2a5077
commit 3a9a1c7d9b
1 changed files with 18 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -34,12 +34,14 @@ var rangeCmd = &cobra.Command{
} }
var ( var (
rangeTotal int rangeTotal int
rangeConsistency string
) )
func init() { func init() {
RootCmd.AddCommand(rangeCmd) RootCmd.AddCommand(rangeCmd)
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)")
} }
func rangeFunc(cmd *cobra.Command, args []string) { func rangeFunc(cmd *cobra.Command, args []string) {
@ -50,10 +52,19 @@ func rangeFunc(cmd *cobra.Command, args []string) {
k := []byte(args[0]) k := []byte(args[0])
var end []byte var end []byte
if len(args) == 1 { if len(args) == 2 {
end = []byte(args[1]) end = []byte(args[1])
} }
if rangeConsistency == "l" {
fmt.Println("bench with linearizable range")
} else if rangeConsistency == "s" {
fmt.Println("bench with serializable range")
} else {
fmt.Fprintln(os.Stderr, cmd.Usage())
os.Exit(1)
}
results = make(chan result) results = make(chan result)
requests := make(chan etcdserverpb.RangeRequest, totalClients) requests := make(chan etcdserverpb.RangeRequest, totalClients)
bar = pb.New(rangeTotal) bar = pb.New(rangeTotal)
@ -72,9 +83,11 @@ func rangeFunc(cmd *cobra.Command, args []string) {
go func() { go func() {
for i := 0; i < rangeTotal; i++ { for i := 0; i < rangeTotal; i++ {
requests <- etcdserverpb.RangeRequest{ r := etcdserverpb.RangeRequest{Key: k, RangeEnd: end}
Key: k, if rangeConsistency == "s" {
RangeEnd: end} r.Serializable = true
}
requests <- r
} }
close(requests) close(requests)
}() }()