| 148 | } |
| 149 | |
| 150 | func Benchmark(b *testing.B) { |
| 151 | log.SetOutput(io.Discard) |
| 152 | defer log.SetOutput(os.Stderr) |
| 153 | benchs := map[string]time.Duration{ |
| 154 | "Waiter": 0, |
| 155 | "Pooler": 10 * time.Millisecond, |
| 156 | } |
| 157 | for name, interval := range benchs { |
| 158 | b.Run(name, func(b *testing.B) { |
| 159 | w := diode.NewWriter(io.Discard, 100000, interval, nil) |
| 160 | log := zerolog.New(w) |
| 161 | defer w.Close() |
| 162 | |
| 163 | b.SetParallelism(1000) |
| 164 | b.RunParallel(func(pb *testing.PB) { |
| 165 | for pb.Next() { |
| 166 | log.Print("test") |
| 167 | } |
| 168 | }) |
| 169 | }) |
| 170 | } |
| 171 | } |