@@ -291,23 +291,25 @@ func BenchmarkClusterPing(b *testing.B) {
291
291
292
292
ctx := context .Background ()
293
293
cluster := newClusterScenario ()
294
- if err := startCluster (ctx , cluster ); err != nil {
294
+ if err := configureClusterTopology (ctx , cluster ); err != nil {
295
295
b .Fatal (err )
296
296
}
297
297
defer cluster .Close ()
298
298
299
299
client := cluster .newClusterClient (ctx , redisClusterOptions ())
300
300
defer client .Close ()
301
301
302
- b .ResetTimer ()
302
+ b .Run ("cluster ping" , func (b * testing.B ) {
303
+ b .ResetTimer ()
303
304
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
+ }
309
311
}
310
- }
312
+ })
311
313
})
312
314
}
313
315
@@ -318,22 +320,24 @@ func BenchmarkClusterDoInt(b *testing.B) {
318
320
319
321
ctx := context .Background ()
320
322
cluster := newClusterScenario ()
321
- if err := startCluster (ctx , cluster ); err != nil {
323
+ if err := configureClusterTopology (ctx , cluster ); err != nil {
322
324
b .Fatal (err )
323
325
}
324
326
defer cluster .Close ()
325
327
326
328
client := cluster .newClusterClient (ctx , redisClusterOptions ())
327
329
defer client .Close ()
328
330
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
+ }
335
339
}
336
- }
340
+ })
337
341
})
338
342
}
339
343
@@ -344,7 +348,7 @@ func BenchmarkClusterSetString(b *testing.B) {
344
348
345
349
ctx := context .Background ()
346
350
cluster := newClusterScenario ()
347
- if err := startCluster (ctx , cluster ); err != nil {
351
+ if err := configureClusterTopology (ctx , cluster ); err != nil {
348
352
b .Fatal (err )
349
353
}
350
354
defer cluster .Close ()
@@ -354,15 +358,17 @@ func BenchmarkClusterSetString(b *testing.B) {
354
358
355
359
value := string (bytes .Repeat ([]byte {'1' }, 10000 ))
356
360
357
- b .ResetTimer ()
361
+ b .Run ("cluster set string" , func (b * testing.B ) {
362
+ b .ResetTimer ()
358
363
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
+ }
364
370
}
365
- }
371
+ })
366
372
})
367
373
}
368
374
@@ -372,21 +378,6 @@ func BenchmarkExecRingSetAddrsCmd(b *testing.B) {
372
378
ringShard2Name = "ringShardTwo"
373
379
)
374
380
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
-
390
381
ring := redis .NewRing (& redis.RingOptions {
391
382
Addrs : map [string ]string {
392
383
"ringShardOne" : ":" + ringShard1Port ,
0 commit comments