Function benchmarkRedisSetGetBuffer

bench_test.go:222–248

benchmarkRedisSetGetBuffer round-trips a payload through SetFromBuffer and GetToBuffer, exercising both zero-copy paths on a shared buffer.

(b *testing.B, size int)

// benchmarkRedisSetGetBuffer round-trips a payload through SetFromBuffer
// and GetToBuffer, exercising both zero-copy paths on a shared buffer.
func benchmarkRedisSetGetBuffer(b *testing.B, size int) {
	ctx := context.Background()
	client := benchmarkRedisClient(ctx, 10)
	defer client.Close()

	value := bytes.Repeat([]byte{'x'}, size)

	b.SetBytes(int64(size))
	b.ReportAllocs()
	b.ResetTimer()

	b.RunParallel(func(pb *testing.PB) {
		buf := make([]byte, size)
		for pb.Next() {
			if err := client.SetFromBuffer(ctx, "key", value).Err(); err != nil {
				b.Fatal(err)
			}
			n, err := client.GetToBuffer(ctx, "key", buf).Result()
			if err != nil {
				b.Fatal(err)
			}
			if n != size {
				b.Fatalf("n = %d, want %d", n, size)
			}
		}
	})
}

// benchmarkRedisSet runs Set with a []byte value, the closest existing
// peer of SetFromBuffer for direct comparison.