Anthony Bishopric
9f6da0e77c
The Level() family of log methods defer formatting of log values until it is confident that the log level needs them. This change allows that deferral to happen at the Levelf() and Levelln() levels too. This commit also adds benchmarks to demonstrate the improved performance of these functions
111 lines
2.1 KiB
Go
111 lines
2.1 KiB
Go
package logrus
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func LogAndAssertJSON(t *testing.T, log func(*Logger), assertions func(fields Fields)) {
|
|
var buffer bytes.Buffer
|
|
var fields Fields
|
|
|
|
logger := New()
|
|
logger.Out = &buffer
|
|
logger.Formatter = new(JSONFormatter)
|
|
|
|
log(logger)
|
|
|
|
err := json.Unmarshal(buffer.Bytes(), &fields)
|
|
assert.Nil(t, err)
|
|
|
|
assertions(fields)
|
|
}
|
|
|
|
func TestPrint(t *testing.T) {
|
|
LogAndAssertJSON(t, func(log *Logger) {
|
|
log.Print("test")
|
|
}, func(fields Fields) {
|
|
assert.Equal(t, fields["msg"], "test")
|
|
assert.Equal(t, fields["level"], "info")
|
|
})
|
|
}
|
|
|
|
func TestInfo(t *testing.T) {
|
|
LogAndAssertJSON(t, func(log *Logger) {
|
|
log.Info("test")
|
|
}, func(fields Fields) {
|
|
assert.Equal(t, fields["msg"], "test")
|
|
assert.Equal(t, fields["level"], "info")
|
|
})
|
|
}
|
|
|
|
func TestWarn(t *testing.T) {
|
|
LogAndAssertJSON(t, func(log *Logger) {
|
|
log.Warn("test")
|
|
}, func(fields Fields) {
|
|
assert.Equal(t, fields["msg"], "test")
|
|
assert.Equal(t, fields["level"], "warning")
|
|
})
|
|
}
|
|
|
|
type SlowString string
|
|
|
|
func (s SlowString) String() string {
|
|
time.Sleep(time.Millisecond)
|
|
return string(s)
|
|
}
|
|
|
|
func getLogAtLevel(l Level) *Logger {
|
|
log := New()
|
|
log.Level = l
|
|
log.Out = ioutil.Discard
|
|
return log
|
|
}
|
|
|
|
func BenchmarkLevelDisplayed(b *testing.B) {
|
|
log := getLogAtLevel(Info)
|
|
for i := 0; i < b.N; i++ {
|
|
log.Info(SlowString("foo"))
|
|
}
|
|
}
|
|
|
|
func BenchmarkLevelHidden(b *testing.B) {
|
|
log := getLogAtLevel(Info)
|
|
for i := 0; i < b.N; i++ {
|
|
log.Debug(SlowString("foo"))
|
|
}
|
|
}
|
|
|
|
func BenchmarkLevelfDisplayed(b *testing.B) {
|
|
log := getLogAtLevel(Info)
|
|
for i := 0; i < b.N; i++ {
|
|
log.Infof("%s", SlowString("foo"))
|
|
}
|
|
}
|
|
|
|
func BenchmarkLevelfHidden(b *testing.B) {
|
|
log := getLogAtLevel(Info)
|
|
for i := 0; i < b.N; i++ {
|
|
log.Debugf("%s", SlowString("foo"))
|
|
}
|
|
}
|
|
|
|
func BenchmarkLevellnDisplayed(b *testing.B) {
|
|
log := getLogAtLevel(Info)
|
|
for i := 0; i < b.N; i++ {
|
|
log.Infoln(SlowString("foo"))
|
|
}
|
|
}
|
|
|
|
func BenchmarkLevellnHidden(b *testing.B) {
|
|
log := getLogAtLevel(Info)
|
|
for i := 0; i < b.N; i++ {
|
|
log.Debugln(SlowString("foo"))
|
|
}
|
|
}
|