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 --count-only for range

Browse Source 
Signed-off-by: Wei Fu <fuweid89@gmail.com>
This commit is contained in:
Wei Fu 2023-07-07 15:03:27 +08:00
parent ac88260aaf
commit 70c8f04a18
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -42,6 +42,7 @@ var (
rangeTotal int
rangeConsistency string
rangeLimit int64
rangeCountOnly bool
)
func init() {
@ -50,6 +51,7 @@ func init() {
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)")
rangeCmd.Flags().BoolVar(&rangeCountOnly, "count-only", false, "Only returns the count of keys")
}
func rangeFunc(cmd *cobra.Command, args []string) {
@ -103,6 +105,9 @@ func rangeFunc(cmd *cobra.Command, args []string) {
go func() {
for i := 0; i < rangeTotal; i++ {
opts := []v3.OpOption{v3.WithRange(end), v3.WithLimit(rangeLimit)}
if rangeCountOnly {
opts = append(opts, v3.WithCountOnly())
}
if rangeConsistency == "s" {
opts = append(opts, v3.WithSerializable())
}