mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-02-03 00:38:40 +08:00
59c0013cd1
* feat: add logc package * feat: add logc, add AddGlobalFields for both logc and logx * chore: add benchmarks * chore: add more tests * chore: simplify globalFields in logx * chore: remove outdated comments
105 lines
2.2 KiB
Go
105 lines
2.2 KiB
Go
package logx
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"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]interface{}
|
|
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 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()
|
|
}
|
|
}
|