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 limit flag for range

Browse Source 
With limit flag, we can verify the performance for the pagination.

Signed-off-by: Wei Fu <fuweid89@gmail.com>
This commit is contained in:
Wei Fu 2023-05-25 12:55:06 +08:00
parent 04346e870f
commit b365f3cda4
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -41,6 +41,7 @@ var (
rangeRate int
rangeTotal int
rangeConsistency string
rangeLimit int64
)
func init() {
@ -48,6 +49,7 @@ func init() {
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().StringVar(&rangeConsistency, "consistency", "l", "Linearizable(l) or Serializable(s)")
rangeCmd.Flags().Int64Var(&rangeLimit, "limit", 0, "Maximum number of results to return from range request (0 is no limit)")
}
func rangeFunc(cmd *cobra.Command, args []string) {
@ -100,7 +102,7 @@ func rangeFunc(cmd *cobra.Command, args []string) {
go func() {
for i := 0; i < rangeTotal; i++ {
opts := []v3.OpOption{v3.WithRange(end)}
opts := []v3.OpOption{v3.WithRange(end), v3.WithLimit(rangeLimit)}
if rangeConsistency == "s" {
opts = append(opts, v3.WithSerializable())
}