333 lines
10 KiB
Go
333 lines
10 KiB
Go
package zerolog
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"time"
|
|
)
|
|
|
|
// Context configures a new sub-logger with contextual fields.
|
|
type Context struct {
|
|
l Logger
|
|
}
|
|
|
|
// Logger returns the logger with the context previously set.
|
|
func (c Context) Logger() Logger {
|
|
return c.l
|
|
}
|
|
|
|
// Fields is a helper function to use a map to set fields using type assertion.
|
|
func (c Context) Fields(fields map[string]interface{}) Context {
|
|
c.l.context = appendFields(c.l.context, fields)
|
|
return c
|
|
}
|
|
|
|
// Dict adds the field key with the dict to the logger context.
|
|
func (c Context) Dict(key string, dict *Event) Context {
|
|
dict.buf = appendEndMarker(dict.buf)
|
|
c.l.context = append(appendKey(c.l.context, key), dict.buf...)
|
|
eventPool.Put(dict)
|
|
return c
|
|
}
|
|
|
|
// Array adds the field key with an array to the event context.
|
|
// Use zerolog.Arr() to create the array or pass a type that
|
|
// implement the LogArrayMarshaler interface.
|
|
func (c Context) Array(key string, arr LogArrayMarshaler) Context {
|
|
c.l.context = appendKey(c.l.context, key)
|
|
if arr, ok := arr.(*Array); ok {
|
|
c.l.context = arr.write(c.l.context)
|
|
return c
|
|
}
|
|
var a *Array
|
|
if aa, ok := arr.(*Array); ok {
|
|
a = aa
|
|
} else {
|
|
a = Arr()
|
|
arr.MarshalZerologArray(a)
|
|
}
|
|
c.l.context = a.write(c.l.context)
|
|
return c
|
|
}
|
|
|
|
// Object marshals an object that implement the LogObjectMarshaler interface.
|
|
func (c Context) Object(key string, obj LogObjectMarshaler) Context {
|
|
e := newEvent(levelWriterAdapter{ioutil.Discard}, 0, true)
|
|
e.Object(key, obj)
|
|
c.l.context = appendObjectData(c.l.context, e.buf)
|
|
eventPool.Put(e)
|
|
return c
|
|
}
|
|
|
|
// Str adds the field key with val as a string to the logger context.
|
|
func (c Context) Str(key, val string) Context {
|
|
c.l.context = appendString(appendKey(c.l.context, key), val)
|
|
return c
|
|
}
|
|
|
|
// Strs adds the field key with val as a string to the logger context.
|
|
func (c Context) Strs(key string, vals []string) Context {
|
|
c.l.context = appendStrings(appendKey(c.l.context, key), vals)
|
|
return c
|
|
}
|
|
|
|
// Bytes adds the field key with val as a []byte to the logger context.
|
|
func (c Context) Bytes(key string, val []byte) Context {
|
|
c.l.context = appendBytes(appendKey(c.l.context, key), val)
|
|
return c
|
|
}
|
|
|
|
// Hex adds the field key with val as a hex string to the logger context.
|
|
func (c Context) Hex(key string, val []byte) Context {
|
|
c.l.context = appendHex(appendKey(c.l.context, key), val)
|
|
return c
|
|
}
|
|
|
|
// RawJSON adds already encoded JSON to context.
|
|
//
|
|
// No sanity check is performed on b; it must not contain carriage returns and
|
|
// be valid JSON.
|
|
func (c Context) RawJSON(key string, b []byte) Context {
|
|
c.l.context = appendJSON(appendKey(c.l.context, key), b)
|
|
return c
|
|
}
|
|
|
|
// AnErr adds the field key with err as a string to the logger context.
|
|
func (c Context) AnErr(key string, err error) Context {
|
|
if err != nil {
|
|
c.l.context = appendError(appendKey(c.l.context, key), err)
|
|
}
|
|
return c
|
|
}
|
|
|
|
// Errs adds the field key with errs as an array of strings to the logger context.
|
|
func (c Context) Errs(key string, errs []error) Context {
|
|
c.l.context = appendErrors(appendKey(c.l.context, key), errs)
|
|
return c
|
|
}
|
|
|
|
// Err adds the field "error" with err as a string to the logger context.
|
|
// To customize the key name, change zerolog.ErrorFieldName.
|
|
func (c Context) Err(err error) Context {
|
|
if err != nil {
|
|
c.l.context = appendError(appendKey(c.l.context, ErrorFieldName), err)
|
|
}
|
|
return c
|
|
}
|
|
|
|
// Bool adds the field key with val as a bool to the logger context.
|
|
func (c Context) Bool(key string, b bool) Context {
|
|
c.l.context = appendBool(appendKey(c.l.context, key), b)
|
|
return c
|
|
}
|
|
|
|
// Bools adds the field key with val as a []bool to the logger context.
|
|
func (c Context) Bools(key string, b []bool) Context {
|
|
c.l.context = appendBools(appendKey(c.l.context, key), b)
|
|
return c
|
|
}
|
|
|
|
// Int adds the field key with i as a int to the logger context.
|
|
func (c Context) Int(key string, i int) Context {
|
|
c.l.context = appendInt(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Ints adds the field key with i as a []int to the logger context.
|
|
func (c Context) Ints(key string, i []int) Context {
|
|
c.l.context = appendInts(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Int8 adds the field key with i as a int8 to the logger context.
|
|
func (c Context) Int8(key string, i int8) Context {
|
|
c.l.context = appendInt8(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Ints8 adds the field key with i as a []int8 to the logger context.
|
|
func (c Context) Ints8(key string, i []int8) Context {
|
|
c.l.context = appendInts8(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Int16 adds the field key with i as a int16 to the logger context.
|
|
func (c Context) Int16(key string, i int16) Context {
|
|
c.l.context = appendInt16(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Ints16 adds the field key with i as a []int16 to the logger context.
|
|
func (c Context) Ints16(key string, i []int16) Context {
|
|
c.l.context = appendInts16(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Int32 adds the field key with i as a int32 to the logger context.
|
|
func (c Context) Int32(key string, i int32) Context {
|
|
c.l.context = appendInt32(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Ints32 adds the field key with i as a []int32 to the logger context.
|
|
func (c Context) Ints32(key string, i []int32) Context {
|
|
c.l.context = appendInts32(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Int64 adds the field key with i as a int64 to the logger context.
|
|
func (c Context) Int64(key string, i int64) Context {
|
|
c.l.context = appendInt64(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Ints64 adds the field key with i as a []int64 to the logger context.
|
|
func (c Context) Ints64(key string, i []int64) Context {
|
|
c.l.context = appendInts64(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Uint adds the field key with i as a uint to the logger context.
|
|
func (c Context) Uint(key string, i uint) Context {
|
|
c.l.context = appendUint(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Uints adds the field key with i as a []uint to the logger context.
|
|
func (c Context) Uints(key string, i []uint) Context {
|
|
c.l.context = appendUints(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Uint8 adds the field key with i as a uint8 to the logger context.
|
|
func (c Context) Uint8(key string, i uint8) Context {
|
|
c.l.context = appendUint8(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Uints8 adds the field key with i as a []uint8 to the logger context.
|
|
func (c Context) Uints8(key string, i []uint8) Context {
|
|
c.l.context = appendUints8(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Uint16 adds the field key with i as a uint16 to the logger context.
|
|
func (c Context) Uint16(key string, i uint16) Context {
|
|
c.l.context = appendUint16(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Uints16 adds the field key with i as a []uint16 to the logger context.
|
|
func (c Context) Uints16(key string, i []uint16) Context {
|
|
c.l.context = appendUints16(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Uint32 adds the field key with i as a uint32 to the logger context.
|
|
func (c Context) Uint32(key string, i uint32) Context {
|
|
c.l.context = appendUint32(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Uints32 adds the field key with i as a []uint32 to the logger context.
|
|
func (c Context) Uints32(key string, i []uint32) Context {
|
|
c.l.context = appendUints32(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Uint64 adds the field key with i as a uint64 to the logger context.
|
|
func (c Context) Uint64(key string, i uint64) Context {
|
|
c.l.context = appendUint64(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Uints64 adds the field key with i as a []uint64 to the logger context.
|
|
func (c Context) Uints64(key string, i []uint64) Context {
|
|
c.l.context = appendUints64(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
// Float32 adds the field key with f as a float32 to the logger context.
|
|
func (c Context) Float32(key string, f float32) Context {
|
|
c.l.context = appendFloat32(appendKey(c.l.context, key), f)
|
|
return c
|
|
}
|
|
|
|
// Floats32 adds the field key with f as a []float32 to the logger context.
|
|
func (c Context) Floats32(key string, f []float32) Context {
|
|
c.l.context = appendFloats32(appendKey(c.l.context, key), f)
|
|
return c
|
|
}
|
|
|
|
// Float64 adds the field key with f as a float64 to the logger context.
|
|
func (c Context) Float64(key string, f float64) Context {
|
|
c.l.context = appendFloat64(appendKey(c.l.context, key), f)
|
|
return c
|
|
}
|
|
|
|
// Floats64 adds the field key with f as a []float64 to the logger context.
|
|
func (c Context) Floats64(key string, f []float64) Context {
|
|
c.l.context = appendFloats64(appendKey(c.l.context, key), f)
|
|
return c
|
|
}
|
|
|
|
type timestampHook struct{}
|
|
|
|
func (ts timestampHook) Run(e *Event, level Level, msg string) {
|
|
e.Timestamp()
|
|
}
|
|
|
|
var th = timestampHook{}
|
|
|
|
// Timestamp adds the current local time as UNIX timestamp to the logger context with the "time" key.
|
|
// To customize the key name, change zerolog.TimestampFieldName.
|
|
//
|
|
// NOTE: It won't dedupe the "time" key if the *Context has one already.
|
|
func (c Context) Timestamp() Context {
|
|
c.l = c.l.Hook(th)
|
|
return c
|
|
}
|
|
|
|
// Time adds the field key with t formated as string using zerolog.TimeFieldFormat.
|
|
func (c Context) Time(key string, t time.Time) Context {
|
|
c.l.context = appendTime(appendKey(c.l.context, key), t, TimeFieldFormat)
|
|
return c
|
|
}
|
|
|
|
// Times adds the field key with t formated as string using zerolog.TimeFieldFormat.
|
|
func (c Context) Times(key string, t []time.Time) Context {
|
|
c.l.context = appendTimes(appendKey(c.l.context, key), t, TimeFieldFormat)
|
|
return c
|
|
}
|
|
|
|
// Dur adds the fields key with d divided by unit and stored as a float.
|
|
func (c Context) Dur(key string, d time.Duration) Context {
|
|
c.l.context = appendDuration(appendKey(c.l.context, key), d, DurationFieldUnit, DurationFieldInteger)
|
|
return c
|
|
}
|
|
|
|
// Durs adds the fields key with d divided by unit and stored as a float.
|
|
func (c Context) Durs(key string, d []time.Duration) Context {
|
|
c.l.context = appendDurations(appendKey(c.l.context, key), d, DurationFieldUnit, DurationFieldInteger)
|
|
return c
|
|
}
|
|
|
|
// Interface adds the field key with obj marshaled using reflection.
|
|
func (c Context) Interface(key string, i interface{}) Context {
|
|
c.l.context = appendInterface(appendKey(c.l.context, key), i)
|
|
return c
|
|
}
|
|
|
|
type callerHook struct{}
|
|
|
|
func (ch callerHook) Run(e *Event, level Level, msg string) {
|
|
e.caller(4)
|
|
}
|
|
|
|
var ch = callerHook{}
|
|
|
|
// Caller adds the file:line of the caller with the zerolog.CallerFieldName key.
|
|
func (c Context) Caller() Context {
|
|
c.l = c.l.Hook(ch)
|
|
return c
|
|
}
|