func runBenchmarkLogrus(b *testing.B, run func(), thread bool) {
b.StopTimer()
tempDir, err := ioutil.TempDir("", "protolog")
require.NoError(b, err)
file, err := os.Create(filepath.Join(tempDir, "log.out"))
require.NoError(b, err)
logrus.SetPusherOptions(
logrus.PusherOptions{
Out: file,
Formatter: &stdlogrus.TextFormatter{
ForceColors: true,
},
},
)
b.StartTimer()
if thread {
var wg sync.WaitGroup
wg.Add(b.N)
for i := 0; i < b.N; i++ {
go func() {
run()
wg.Done()
}()
}
wg.Wait()
} else {
for i := 0; i < b.N; i++ {
run()
}
}
_ = protolog.Flush()
b.StopTimer()
_ = os.RemoveAll(tempDir)
b.StartTimer()
}
func init() {
logrus.SetPusherOptions(
logrus.PusherOptions{
Formatter: &stdlogrus.TextFormatter{
ForceColors: true,
},
},
)
}