redis/go-redis
1
0
Fork 0
mirror of https://github.com/redis/go-redis.git synced 2025-07-26 19:21:03 +03:00
Code Activity

More benchmarks.

Browse Source
This commit is contained in:
Vladimir Mihailenco
2016-03-10 11:10:47 +02:00
parent eb78eedafe
commit b90cea8e8d
6 changed files with 321 additions and 224 deletions
Download Patch File Download Diff File Expand all files Collapse all files

199
redis_test.go
View File

@ -1,9 +1,7 @@
package redis_test
import (
"bytes"
"net"
"testing"
"time"
. "github.com/onsi/ginkgo"
@ -192,200 +190,3 @@ var _ = Describe("Client", func() {
Expect(cn.UsedAt.Equal(future)).To(BeTrue())
})
})
//------------------------------------------------------------------------------
func benchRedisClient() *redis.Client {
client := redis.NewClient(&redis.Options{
Addr: ":6379",
})
if err := client.FlushDb().Err(); err != nil {
panic(err)
}
return client
}
func BenchmarkRedisPing(b *testing.B) {
client := benchRedisClient()
defer client.Close()
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
if err := client.Ping().Err(); err != nil {
b.Fatal(err)
}
}
})
}
func BenchmarkRedisSet(b *testing.B) {
client := benchRedisClient()
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)
}
}
})
}
func BenchmarkRedisGetNil(b *testing.B) {
client := benchRedisClient()
defer client.Close()
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
if err := client.Get("key").Err(); err != redis.Nil {
b.Fatal(err)
}
}
})
}
func benchmarkRedisSetGet(b *testing.B, size int) {
client := benchRedisClient()
defer client.Close()
value := string(bytes.Repeat([]byte{'1'}, size))
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
if err := client.Set("key", value, 0).Err(); err != nil {
b.Fatal(err)
}
got, err := client.Get("key").Result()
if err != nil {
b.Fatal(err)
}
if got != value {
b.Fatalf("got != value")
}
}
})
}
func BenchmarkRedisSetGet64Bytes(b *testing.B) {
benchmarkRedisSetGet(b, 64)
}
func BenchmarkRedisSetGet1KB(b *testing.B) {
benchmarkRedisSetGet(b, 1024)
}
func BenchmarkRedisSetGet10KB(b *testing.B) {
benchmarkRedisSetGet(b, 10*1024)
}
func BenchmarkRedisSetGet1MB(b *testing.B) {
benchmarkRedisSetGet(b, 1024*1024)
}
func BenchmarkRedisSetGetBytes(b *testing.B) {
client := benchRedisClient()
defer client.Close()
value := 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)
}
got, err := client.Get("key").Bytes()
if err != nil {
b.Fatal(err)
}
if !bytes.Equal(got, value) {
b.Fatalf("got != value")
}
}
})
}
func BenchmarkRedisMGet(b *testing.B) {
client := benchRedisClient()
defer client.Close()
if err := client.MSet("key1", "hello1", "key2", "hello2").Err(); err != nil {
b.Fatal(err)
}
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
if err := client.MGet("key1", "key2").Err(); err != nil {
b.Fatal(err)
}
}
})
}
func BenchmarkSetExpire(b *testing.B) {
client := benchRedisClient()
defer client.Close()
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
if err := client.Set("key", "hello", 0).Err(); err != nil {
b.Fatal(err)
}
if err := client.Expire("key", time.Second).Err(); err != nil {
b.Fatal(err)
}
}
})
}
func BenchmarkPipeline(b *testing.B) {
client := benchRedisClient()
defer client.Close()
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
_, err := client.Pipelined(func(pipe *redis.Pipeline) error {
pipe.Set("key", "hello", 0)
pipe.Expire("key", time.Second)
return nil
})
if err != nil {
b.Fatal(err)
}
}
})
}
func BenchmarkZAdd(b *testing.B) {
client := benchRedisClient()
defer client.Close()
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
if err := client.ZAdd("key", redis.Z{float64(1), "hello"}).Err(); err != nil {
b.Fatal(err)
}
}
})
}