File tree Expand file tree Collapse file tree 1 file changed +12
-10
lines changed Expand file tree Collapse file tree 1 file changed +12
-10
lines changed Original file line number Diff line number Diff line change @@ -280,30 +280,32 @@ func BenchmarkRBLifeCycle(b *testing.B) {
280
280
}
281
281
282
282
func BenchmarkRBPut (b * testing.B ) {
283
- rbs := make ([]* RingBuffer , 0 , b .N )
284
-
285
- for i := 0 ; i < b .N ; i ++ {
286
- rbs = append (rbs , NewRingBuffer (2 ))
287
- }
283
+ rb := NewRingBuffer (uint64 (b .N ))
288
284
289
285
b .ResetTimer ()
290
286
291
287
for i := 0 ; i < b .N ; i ++ {
292
- rbs [i ].Put (i )
288
+ ok , err := rb .Offer (i )
289
+ if ! ok {
290
+ b .Fail ()
291
+ }
292
+ if err != nil {
293
+ b .Log (err )
294
+ b .Fail ()
295
+ }
293
296
}
294
297
}
295
298
296
299
func BenchmarkRBGet (b * testing.B ) {
297
- rbs := make ([] * RingBuffer , 0 , b .N )
300
+ rb := NewRingBuffer ( uint64 ( b .N ) )
298
301
299
302
for i := 0 ; i < b .N ; i ++ {
300
- rbs = append (rbs , NewRingBuffer (2 ))
301
- rbs [i ].Put (i )
303
+ rb .Offer (i )
302
304
}
303
305
304
306
b .ResetTimer ()
305
307
306
308
for i := 0 ; i < b .N ; i ++ {
307
- rbs [ i ] .Get ()
309
+ rb .Get ()
308
310
}
309
311
}
You can’t perform that action at this time.
0 commit comments