@@ -291,23 +291,25 @@ func BenchmarkClusterPing(b *testing.B) {
291291
292292 ctx := context .Background ()
293293 cluster := newClusterScenario ()
294- if err := startCluster (ctx , cluster ); err != nil {
294+ if err := configureClusterTopology (ctx , cluster ); err != nil {
295295 b .Fatal (err )
296296 }
297297 defer cluster .Close ()
298298
299299 client := cluster .newClusterClient (ctx , redisClusterOptions ())
300300 defer client .Close ()
301301
302- b .ResetTimer ()
302+ b .Run ("cluster ping" , func (b * testing.B ) {
303+ b .ResetTimer ()
303304
304- b .RunParallel (func (pb * testing.PB ) {
305- for pb .Next () {
306- err := client .Ping (ctx ).Err ()
307- if err != nil {
308- b .Fatal (err )
305+ b .RunParallel (func (pb * testing.PB ) {
306+ for pb .Next () {
307+ err := client .Ping (ctx ).Err ()
308+ if err != nil {
309+ b .Fatal (err )
310+ }
309311 }
310- }
312+ })
311313 })
312314}
313315
@@ -318,22 +320,24 @@ func BenchmarkClusterDoInt(b *testing.B) {
318320
319321 ctx := context .Background ()
320322 cluster := newClusterScenario ()
321- if err := startCluster (ctx , cluster ); err != nil {
323+ if err := configureClusterTopology (ctx , cluster ); err != nil {
322324 b .Fatal (err )
323325 }
324326 defer cluster .Close ()
325327
326328 client := cluster .newClusterClient (ctx , redisClusterOptions ())
327329 defer client .Close ()
328330
329- b .ResetTimer ()
330- b .RunParallel (func (pb * testing.PB ) {
331- for pb .Next () {
332- err := client .Do (ctx , "SET" , 10 , 10 ).Err ()
333- if err != nil {
334- b .Fatal (err )
331+ b .Run ("cluster do set int" , func (b * testing.B ) {
332+ b .ResetTimer ()
333+ b .RunParallel (func (pb * testing.PB ) {
334+ for pb .Next () {
335+ err := client .Do (ctx , "SET" , 10 , 10 ).Err ()
336+ if err != nil {
337+ b .Fatal (err )
338+ }
335339 }
336- }
340+ })
337341 })
338342}
339343
@@ -344,7 +348,7 @@ func BenchmarkClusterSetString(b *testing.B) {
344348
345349 ctx := context .Background ()
346350 cluster := newClusterScenario ()
347- if err := startCluster (ctx , cluster ); err != nil {
351+ if err := configureClusterTopology (ctx , cluster ); err != nil {
348352 b .Fatal (err )
349353 }
350354 defer cluster .Close ()
@@ -354,15 +358,17 @@ func BenchmarkClusterSetString(b *testing.B) {
354358
355359 value := string (bytes .Repeat ([]byte {'1' }, 10000 ))
356360
357- b .ResetTimer ()
361+ b .Run ("cluster set string" , func (b * testing.B ) {
362+ b .ResetTimer ()
358363
359- b .RunParallel (func (pb * testing.PB ) {
360- for pb .Next () {
361- err := client .Set (ctx , "key" , value , 0 ).Err ()
362- if err != nil {
363- b .Fatal (err )
364+ b .RunParallel (func (pb * testing.PB ) {
365+ for pb .Next () {
366+ err := client .Set (ctx , "key" , value , 0 ).Err ()
367+ if err != nil {
368+ b .Fatal (err )
369+ }
364370 }
365- }
371+ })
366372 })
367373}
368374
@@ -372,21 +378,6 @@ func BenchmarkExecRingSetAddrsCmd(b *testing.B) {
372378 ringShard2Name = "ringShardTwo"
373379 )
374380
375- for _ , port := range []string {ringShard1Port , ringShard2Port } {
376- if _ , err := startRedis (port ); err != nil {
377- b .Fatal (err )
378- }
379- }
380-
381- b .Cleanup (func () {
382- for _ , p := range processes {
383- if err := p .Close (); err != nil {
384- b .Errorf ("Failed to stop redis process: %v" , err )
385- }
386- }
387- processes = nil
388- })
389-
390381 ring := redis .NewRing (& redis.RingOptions {
391382 Addrs : map [string ]string {
392383 "ringShardOne" : ":" + ringShard1Port ,
0 commit comments