You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
122 lines
2.7 KiB
Go
122 lines
2.7 KiB
Go
package logx
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"strconv"
|
|
"sync"
|
|
"sync/atomic"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestAddGlobalFields(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
writer := NewWriter(&buf)
|
|
old := Reset()
|
|
SetWriter(writer)
|
|
defer SetWriter(old)
|
|
|
|
Info("hello")
|
|
buf.Reset()
|
|
|
|
AddGlobalFields(Field("a", "1"), Field("b", "2"))
|
|
AddGlobalFields(Field("c", "3"))
|
|
Info("world")
|
|
var m map[string]any
|
|
assert.NoError(t, json.Unmarshal(buf.Bytes(), &m))
|
|
assert.Equal(t, "1", m["a"])
|
|
assert.Equal(t, "2", m["b"])
|
|
assert.Equal(t, "3", m["c"])
|
|
}
|
|
|
|
func TestContextWithFields(t *testing.T) {
|
|
ctx := ContextWithFields(context.Background(), Field("a", 1), Field("b", 2))
|
|
vals := ctx.Value(fieldsContextKey)
|
|
assert.NotNil(t, vals)
|
|
fields, ok := vals.([]LogField)
|
|
assert.True(t, ok)
|
|
assert.EqualValues(t, []LogField{Field("a", 1), Field("b", 2)}, fields)
|
|
}
|
|
|
|
func TestWithFields(t *testing.T) {
|
|
ctx := WithFields(context.Background(), Field("a", 1), Field("b", 2))
|
|
vals := ctx.Value(fieldsContextKey)
|
|
assert.NotNil(t, vals)
|
|
fields, ok := vals.([]LogField)
|
|
assert.True(t, ok)
|
|
assert.EqualValues(t, []LogField{Field("a", 1), Field("b", 2)}, fields)
|
|
}
|
|
|
|
func TestWithFieldsAppend(t *testing.T) {
|
|
var dummyKey struct{}
|
|
ctx := context.WithValue(context.Background(), dummyKey, "dummy")
|
|
ctx = ContextWithFields(ctx, Field("a", 1), Field("b", 2))
|
|
ctx = ContextWithFields(ctx, Field("c", 3), Field("d", 4))
|
|
vals := ctx.Value(fieldsContextKey)
|
|
assert.NotNil(t, vals)
|
|
fields, ok := vals.([]LogField)
|
|
assert.True(t, ok)
|
|
assert.Equal(t, "dummy", ctx.Value(dummyKey))
|
|
assert.EqualValues(t, []LogField{
|
|
Field("a", 1),
|
|
Field("b", 2),
|
|
Field("c", 3),
|
|
Field("d", 4),
|
|
}, fields)
|
|
}
|
|
|
|
func TestWithFieldsAppendCopy(t *testing.T) {
|
|
const count = 10
|
|
ctx := context.Background()
|
|
for i := 0; i < count; i++ {
|
|
ctx = ContextWithFields(ctx, Field(strconv.Itoa(i), 1))
|
|
}
|
|
|
|
af := Field("foo", 1)
|
|
bf := Field("bar", 2)
|
|
ctxa := ContextWithFields(ctx, af)
|
|
ctxb := ContextWithFields(ctx, bf)
|
|
|
|
assert.EqualValues(t, af, ctxa.Value(fieldsContextKey).([]LogField)[count])
|
|
assert.EqualValues(t, bf, ctxb.Value(fieldsContextKey).([]LogField)[count])
|
|
}
|
|
|
|
func BenchmarkAtomicValue(b *testing.B) {
|
|
b.ReportAllocs()
|
|
|
|
var container atomic.Value
|
|
vals := []LogField{
|
|
Field("a", "b"),
|
|
Field("c", "d"),
|
|
Field("e", "f"),
|
|
}
|
|
container.Store(&vals)
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
val := container.Load()
|
|
if val != nil {
|
|
_ = *val.(*[]LogField)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkRWMutex(b *testing.B) {
|
|
b.ReportAllocs()
|
|
|
|
var lock sync.RWMutex
|
|
vals := []LogField{
|
|
Field("a", "b"),
|
|
Field("c", "d"),
|
|
Field("e", "f"),
|
|
}
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
lock.RLock()
|
|
_ = vals
|
|
lock.RUnlock()
|
|
}
|
|
}
|