refactoring: barbench_test
Vladimir Bauer
2 years ago
| 1 | 1 | |
| 2 | 2 | import ( |
| 3 | 3 | "io" |
| 4 | "sync" | |
| 5 | 4 | "testing" |
| 6 | 5 | |
| 7 | 6 | "github.com/vbauerster/mpb/v8" |
| 9 | 8 | |
| 10 | 9 | const total = 1000 |
| 11 | 10 | |
| 12 | func BenchmarkNopStyle1Bar(b *testing.B) { | |
| 11 | func BenchmarkNopStyleB1(b *testing.B) { | |
| 13 | 12 | bench(b, mpb.NopStyle(), false, 1) |
| 14 | 13 | } |
| 15 | 14 | |
| 16 | func BenchmarkNopStyle1BarWithAutoRefresh(b *testing.B) { | |
| 15 | func BenchmarkNopStyleWithAutoRefreshB1(b *testing.B) { | |
| 17 | 16 | bench(b, mpb.NopStyle(), true, 1) |
| 18 | 17 | } |
| 19 | 18 | |
| 20 | func BenchmarkNopStyle2Bars(b *testing.B) { | |
| 19 | func BenchmarkNopStylesB2(b *testing.B) { | |
| 21 | 20 | bench(b, mpb.NopStyle(), false, 2) |
| 22 | 21 | } |
| 23 | 22 | |
| 24 | func BenchmarkNopStyle2BarsWithAutoRefresh(b *testing.B) { | |
| 23 | func BenchmarkNopStylesWithAutoRefreshB2(b *testing.B) { | |
| 25 | 24 | bench(b, mpb.NopStyle(), true, 2) |
| 26 | 25 | } |
| 27 | 26 | |
| 28 | func BenchmarkNopStyle3Bars(b *testing.B) { | |
| 27 | func BenchmarkNopStylesB3(b *testing.B) { | |
| 29 | 28 | bench(b, mpb.NopStyle(), false, 3) |
| 30 | 29 | } |
| 31 | 30 | |
| 32 | func BenchmarkNopStyle3BarsWithAutoRefresh(b *testing.B) { | |
| 31 | func BenchmarkNopStylesWithAutoRefreshB3(b *testing.B) { | |
| 33 | 32 | bench(b, mpb.NopStyle(), true, 3) |
| 34 | 33 | } |
| 35 | 34 | |
| 36 | func BenchmarkBarStyle1Bar(b *testing.B) { | |
| 35 | func BenchmarkBarStyleB1(b *testing.B) { | |
| 37 | 36 | bench(b, mpb.BarStyle(), false, 1) |
| 38 | 37 | } |
| 39 | 38 | |
| 40 | func BenchmarkBarStyle1BarWithAutoRefresh(b *testing.B) { | |
| 39 | func BenchmarkBarStyleWithAutoRefreshB1(b *testing.B) { | |
| 41 | 40 | bench(b, mpb.BarStyle(), true, 1) |
| 42 | 41 | } |
| 43 | 42 | |
| 44 | func BenchmarkBarStyle2Bars(b *testing.B) { | |
| 43 | func BenchmarkBarStylesB2(b *testing.B) { | |
| 45 | 44 | bench(b, mpb.BarStyle(), false, 2) |
| 46 | 45 | } |
| 47 | 46 | |
| 48 | func BenchmarkBarStyle2BarsWithAutoRefresh(b *testing.B) { | |
| 47 | func BenchmarkBarStylesWithAutoRefreshB2(b *testing.B) { | |
| 49 | 48 | bench(b, mpb.BarStyle(), true, 2) |
| 50 | 49 | } |
| 51 | 50 | |
| 52 | func BenchmarkBarStyle3Bars(b *testing.B) { | |
| 51 | func BenchmarkBarStylesB3(b *testing.B) { | |
| 53 | 52 | bench(b, mpb.BarStyle(), false, 3) |
| 54 | 53 | } |
| 55 | 54 | |
| 56 | func BenchmarkBarStyle3BarsWithAutoRefresh(b *testing.B) { | |
| 55 | func BenchmarkBarStylesWithAutoRefreshB3(b *testing.B) { | |
| 57 | 56 | bench(b, mpb.BarStyle(), true, 3) |
| 58 | 57 | } |
| 59 | 58 | |
| 60 | 59 | func bench(b *testing.B, builder mpb.BarFillerBuilder, autoRefresh bool, n int) { |
| 61 | var wg sync.WaitGroup | |
| 62 | 60 | p := mpb.New( |
| 63 | 61 | mpb.WithWidth(100), |
| 64 | 62 | mpb.WithOutput(io.Discard), |
| 65 | 63 | mpb.ContainerOptional(mpb.WithAutoRefresh(), autoRefresh), |
| 66 | 64 | ) |
| 65 | defer p.Wait() | |
| 67 | 66 | b.ResetTimer() |
| 68 | 67 | for i := 0; i < b.N; i++ { |
| 68 | var bars []*mpb.Bar | |
| 69 | 69 | for j := 0; j < n; j++ { |
| 70 | bar := p.New(total, builder) | |
| 70 | bars = append(bars, p.New(total, builder)) | |
| 71 | 71 | switch j { |
| 72 | 72 | case n - 1: |
| 73 | complete(b, bar) | |
| 73 | complete(bars[j]) | |
| 74 | 74 | default: |
| 75 | wg.Add(1) | |
| 76 | go func() { | |
| 77 | complete(b, bar) | |
| 78 | wg.Done() | |
| 79 | }() | |
| 75 | go complete(bars[j]) | |
| 80 | 76 | } |
| 81 | 77 | } |
| 82 | wg.Wait() | |
| 78 | for _, bar := range bars { | |
| 79 | bar.Wait() | |
| 80 | } | |
| 83 | 81 | } |
| 84 | p.Wait() | |
| 85 | 82 | } |
| 86 | 83 | |
| 87 | func complete(b *testing.B, bar *mpb.Bar) { | |
| 84 | func complete(bar *mpb.Bar) { | |
| 88 | 85 | for i := 0; i < total; i++ { |
| 89 | 86 | bar.Increment() |
| 90 | 87 | } |
| 91 | bar.Wait() | |
| 92 | 88 | } |