Skip to content

Commit

Permalink
Add cluster benchmark
Browse files Browse the repository at this point in the history
  • Loading branch information
vmihailenco committed Mar 24, 2017
1 parent 14f2598 commit 5a99b80
Showing 1 changed file with 34 additions and 0 deletions.
34 changes: 34 additions & 0 deletions cluster_test.go
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
package redis_test

import (
"bytes"
"fmt"
"net"
"strconv"
Expand Down Expand Up @@ -704,3 +705,36 @@ func BenchmarkRedisClusterPing(b *testing.B) {
}
})
}

func BenchmarkRedisClusterSetString(b *testing.B) {
if testing.Short() {
b.Skip("skipping in short mode")
}

cluster := &clusterScenario{
ports: []string{"8220", "8221", "8222", "8223", "8224", "8225"},
nodeIds: make([]string, 6),
processes: make(map[string]*redisProcess, 6),
clients: make(map[string]*redis.Client, 6),
}

if err := startCluster(cluster); err != nil {
b.Fatal(err)
}
defer stopCluster(cluster)

client := cluster.clusterClient(redisClusterOptions())
defer client.Close()

value := string(bytes.Repeat([]byte{'1'}, 10000))

b.ResetTimer()

b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
if err := client.Set("key", value, 0).Err(); err != nil {
b.Fatal(err)
}
}
})
}

0 comments on commit 5a99b80

Please sign in to comment.