mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-12-15 03:36:37 +00:00
914 lines
26 KiB
Go
914 lines
26 KiB
Go
/*
|
|
Copyright 2021 The logr Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
// Package funcr implements formatting of structured log messages and
|
|
// optionally captures the call site and timestamp.
|
|
//
|
|
// The simplest way to use it is via its implementation of a
|
|
// github.com/go-logr/logr.LogSink with output through an arbitrary
|
|
// "write" function. See New and NewJSON for details.
|
|
//
|
|
// # Custom LogSinks
|
|
//
|
|
// For users who need more control, a funcr.Formatter can be embedded inside
|
|
// your own custom LogSink implementation. This is useful when the LogSink
|
|
// needs to implement additional methods, for example.
|
|
//
|
|
// # Formatting
|
|
//
|
|
// This will respect logr.Marshaler, fmt.Stringer, and error interfaces for
|
|
// values which are being logged. When rendering a struct, funcr will use Go's
|
|
// standard JSON tags (all except "string").
|
|
package funcr
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding"
|
|
"encoding/json"
|
|
"fmt"
|
|
"path/filepath"
|
|
"reflect"
|
|
"runtime"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/go-logr/logr"
|
|
)
|
|
|
|
// New returns a logr.Logger which is implemented by an arbitrary function.
|
|
func New(fn func(prefix, args string), opts Options) logr.Logger {
|
|
return logr.New(newSink(fn, NewFormatter(opts)))
|
|
}
|
|
|
|
// NewJSON returns a logr.Logger which is implemented by an arbitrary function
|
|
// and produces JSON output.
|
|
func NewJSON(fn func(obj string), opts Options) logr.Logger {
|
|
fnWrapper := func(_, obj string) {
|
|
fn(obj)
|
|
}
|
|
return logr.New(newSink(fnWrapper, NewFormatterJSON(opts)))
|
|
}
|
|
|
|
// Underlier exposes access to the underlying logging function. Since
|
|
// callers only have a logr.Logger, they have to know which
|
|
// implementation is in use, so this interface is less of an
|
|
// abstraction and more of a way to test type conversion.
|
|
type Underlier interface {
|
|
GetUnderlying() func(prefix, args string)
|
|
}
|
|
|
|
func newSink(fn func(prefix, args string), formatter Formatter) logr.LogSink {
|
|
l := &fnlogger{
|
|
Formatter: formatter,
|
|
write: fn,
|
|
}
|
|
// For skipping fnlogger.Info and fnlogger.Error.
|
|
l.Formatter.AddCallDepth(1)
|
|
return l
|
|
}
|
|
|
|
// Options carries parameters which influence the way logs are generated.
|
|
type Options struct {
|
|
// LogCaller tells funcr to add a "caller" key to some or all log lines.
|
|
// This has some overhead, so some users might not want it.
|
|
LogCaller MessageClass
|
|
|
|
// LogCallerFunc tells funcr to also log the calling function name. This
|
|
// has no effect if caller logging is not enabled (see Options.LogCaller).
|
|
LogCallerFunc bool
|
|
|
|
// LogTimestamp tells funcr to add a "ts" key to log lines. This has some
|
|
// overhead, so some users might not want it.
|
|
LogTimestamp bool
|
|
|
|
// TimestampFormat tells funcr how to render timestamps when LogTimestamp
|
|
// is enabled. If not specified, a default format will be used. For more
|
|
// details, see docs for Go's time.Layout.
|
|
TimestampFormat string
|
|
|
|
// LogInfoLevel tells funcr what key to use to log the info level.
|
|
// If not specified, the info level will be logged as "level".
|
|
// If this is set to "", the info level will not be logged at all.
|
|
LogInfoLevel *string
|
|
|
|
// Verbosity tells funcr which V logs to produce. Higher values enable
|
|
// more logs. Info logs at or below this level will be written, while logs
|
|
// above this level will be discarded.
|
|
Verbosity int
|
|
|
|
// RenderBuiltinsHook allows users to mutate the list of key-value pairs
|
|
// while a log line is being rendered. The kvList argument follows logr
|
|
// conventions - each pair of slice elements is comprised of a string key
|
|
// and an arbitrary value (verified and sanitized before calling this
|
|
// hook). The value returned must follow the same conventions. This hook
|
|
// can be used to audit or modify logged data. For example, you might want
|
|
// to prefix all of funcr's built-in keys with some string. This hook is
|
|
// only called for built-in (provided by funcr itself) key-value pairs.
|
|
// Equivalent hooks are offered for key-value pairs saved via
|
|
// logr.Logger.WithValues or Formatter.AddValues (see RenderValuesHook) and
|
|
// for user-provided pairs (see RenderArgsHook).
|
|
RenderBuiltinsHook func(kvList []any) []any
|
|
|
|
// RenderValuesHook is the same as RenderBuiltinsHook, except that it is
|
|
// only called for key-value pairs saved via logr.Logger.WithValues. See
|
|
// RenderBuiltinsHook for more details.
|
|
RenderValuesHook func(kvList []any) []any
|
|
|
|
// RenderArgsHook is the same as RenderBuiltinsHook, except that it is only
|
|
// called for key-value pairs passed directly to Info and Error. See
|
|
// RenderBuiltinsHook for more details.
|
|
RenderArgsHook func(kvList []any) []any
|
|
|
|
// MaxLogDepth tells funcr how many levels of nested fields (e.g. a struct
|
|
// that contains a struct, etc.) it may log. Every time it finds a struct,
|
|
// slice, array, or map the depth is increased by one. When the maximum is
|
|
// reached, the value will be converted to a string indicating that the max
|
|
// depth has been exceeded. If this field is not specified, a default
|
|
// value will be used.
|
|
MaxLogDepth int
|
|
}
|
|
|
|
// MessageClass indicates which category or categories of messages to consider.
|
|
type MessageClass int
|
|
|
|
const (
|
|
// None ignores all message classes.
|
|
None MessageClass = iota
|
|
// All considers all message classes.
|
|
All
|
|
// Info only considers info messages.
|
|
Info
|
|
// Error only considers error messages.
|
|
Error
|
|
)
|
|
|
|
// fnlogger inherits some of its LogSink implementation from Formatter
|
|
// and just needs to add some glue code.
|
|
type fnlogger struct {
|
|
Formatter
|
|
write func(prefix, args string)
|
|
}
|
|
|
|
func (l fnlogger) WithName(name string) logr.LogSink {
|
|
l.Formatter.AddName(name)
|
|
return &l
|
|
}
|
|
|
|
func (l fnlogger) WithValues(kvList ...any) logr.LogSink {
|
|
l.Formatter.AddValues(kvList)
|
|
return &l
|
|
}
|
|
|
|
func (l fnlogger) WithCallDepth(depth int) logr.LogSink {
|
|
l.Formatter.AddCallDepth(depth)
|
|
return &l
|
|
}
|
|
|
|
func (l fnlogger) Info(level int, msg string, kvList ...any) {
|
|
prefix, args := l.FormatInfo(level, msg, kvList)
|
|
l.write(prefix, args)
|
|
}
|
|
|
|
func (l fnlogger) Error(err error, msg string, kvList ...any) {
|
|
prefix, args := l.FormatError(err, msg, kvList)
|
|
l.write(prefix, args)
|
|
}
|
|
|
|
func (l fnlogger) GetUnderlying() func(prefix, args string) {
|
|
return l.write
|
|
}
|
|
|
|
// Assert conformance to the interfaces.
|
|
var _ logr.LogSink = &fnlogger{}
|
|
var _ logr.CallDepthLogSink = &fnlogger{}
|
|
var _ Underlier = &fnlogger{}
|
|
|
|
// NewFormatter constructs a Formatter which emits a JSON-like key=value format.
|
|
func NewFormatter(opts Options) Formatter {
|
|
return newFormatter(opts, outputKeyValue)
|
|
}
|
|
|
|
// NewFormatterJSON constructs a Formatter which emits strict JSON.
|
|
func NewFormatterJSON(opts Options) Formatter {
|
|
return newFormatter(opts, outputJSON)
|
|
}
|
|
|
|
// Defaults for Options.
|
|
const defaultTimestampFormat = "2006-01-02 15:04:05.000000"
|
|
const defaultMaxLogDepth = 16
|
|
|
|
func newFormatter(opts Options, outfmt outputFormat) Formatter {
|
|
if opts.TimestampFormat == "" {
|
|
opts.TimestampFormat = defaultTimestampFormat
|
|
}
|
|
if opts.MaxLogDepth == 0 {
|
|
opts.MaxLogDepth = defaultMaxLogDepth
|
|
}
|
|
if opts.LogInfoLevel == nil {
|
|
opts.LogInfoLevel = new(string)
|
|
*opts.LogInfoLevel = "level"
|
|
}
|
|
f := Formatter{
|
|
outputFormat: outfmt,
|
|
prefix: "",
|
|
values: nil,
|
|
depth: 0,
|
|
opts: &opts,
|
|
}
|
|
return f
|
|
}
|
|
|
|
// Formatter is an opaque struct which can be embedded in a LogSink
|
|
// implementation. It should be constructed with NewFormatter. Some of
|
|
// its methods directly implement logr.LogSink.
|
|
type Formatter struct {
|
|
outputFormat outputFormat
|
|
prefix string
|
|
values []any
|
|
valuesStr string
|
|
depth int
|
|
opts *Options
|
|
groupName string // for slog groups
|
|
groups []groupDef
|
|
}
|
|
|
|
// outputFormat indicates which outputFormat to use.
|
|
type outputFormat int
|
|
|
|
const (
|
|
// outputKeyValue emits a JSON-like key=value format, but not strict JSON.
|
|
outputKeyValue outputFormat = iota
|
|
// outputJSON emits strict JSON.
|
|
outputJSON
|
|
)
|
|
|
|
// groupDef represents a saved group. The values may be empty, but we don't
|
|
// know if we need to render the group until the final record is rendered.
|
|
type groupDef struct {
|
|
name string
|
|
values string
|
|
}
|
|
|
|
// PseudoStruct is a list of key-value pairs that gets logged as a struct.
|
|
type PseudoStruct []any
|
|
|
|
// render produces a log line, ready to use.
|
|
func (f Formatter) render(builtins, args []any) string {
|
|
// Empirically bytes.Buffer is faster than strings.Builder for this.
|
|
buf := bytes.NewBuffer(make([]byte, 0, 1024))
|
|
|
|
if f.outputFormat == outputJSON {
|
|
buf.WriteByte('{') // for the whole record
|
|
}
|
|
|
|
// Render builtins
|
|
vals := builtins
|
|
if hook := f.opts.RenderBuiltinsHook; hook != nil {
|
|
vals = hook(f.sanitize(vals))
|
|
}
|
|
f.flatten(buf, vals, false) // keys are ours, no need to escape
|
|
continuing := len(builtins) > 0
|
|
|
|
// Turn the inner-most group into a string
|
|
argsStr := func() string {
|
|
buf := bytes.NewBuffer(make([]byte, 0, 1024))
|
|
|
|
vals = args
|
|
if hook := f.opts.RenderArgsHook; hook != nil {
|
|
vals = hook(f.sanitize(vals))
|
|
}
|
|
f.flatten(buf, vals, true) // escape user-provided keys
|
|
|
|
return buf.String()
|
|
}()
|
|
|
|
// Render the stack of groups from the inside out.
|
|
bodyStr := f.renderGroup(f.groupName, f.valuesStr, argsStr)
|
|
for i := len(f.groups) - 1; i >= 0; i-- {
|
|
grp := &f.groups[i]
|
|
if grp.values == "" && bodyStr == "" {
|
|
// no contents, so we must elide the whole group
|
|
continue
|
|
}
|
|
bodyStr = f.renderGroup(grp.name, grp.values, bodyStr)
|
|
}
|
|
|
|
if bodyStr != "" {
|
|
if continuing {
|
|
buf.WriteByte(f.comma())
|
|
}
|
|
buf.WriteString(bodyStr)
|
|
}
|
|
|
|
if f.outputFormat == outputJSON {
|
|
buf.WriteByte('}') // for the whole record
|
|
}
|
|
|
|
return buf.String()
|
|
}
|
|
|
|
// renderGroup returns a string representation of the named group with rendered
|
|
// values and args. If the name is empty, this will return the values and args,
|
|
// joined. If the name is not empty, this will return a single key-value pair,
|
|
// where the value is a grouping of the values and args. If the values and
|
|
// args are both empty, this will return an empty string, even if the name was
|
|
// specified.
|
|
func (f Formatter) renderGroup(name string, values string, args string) string {
|
|
buf := bytes.NewBuffer(make([]byte, 0, 1024))
|
|
|
|
needClosingBrace := false
|
|
if name != "" && (values != "" || args != "") {
|
|
buf.WriteString(f.quoted(name, true)) // escape user-provided keys
|
|
buf.WriteByte(f.colon())
|
|
buf.WriteByte('{')
|
|
needClosingBrace = true
|
|
}
|
|
|
|
continuing := false
|
|
if values != "" {
|
|
buf.WriteString(values)
|
|
continuing = true
|
|
}
|
|
|
|
if args != "" {
|
|
if continuing {
|
|
buf.WriteByte(f.comma())
|
|
}
|
|
buf.WriteString(args)
|
|
}
|
|
|
|
if needClosingBrace {
|
|
buf.WriteByte('}')
|
|
}
|
|
|
|
return buf.String()
|
|
}
|
|
|
|
// flatten renders a list of key-value pairs into a buffer. If escapeKeys is
|
|
// true, the keys are assumed to have non-JSON-compatible characters in them
|
|
// and must be evaluated for escapes.
|
|
//
|
|
// This function returns a potentially modified version of kvList, which
|
|
// ensures that there is a value for every key (adding a value if needed) and
|
|
// that each key is a string (substituting a key if needed).
|
|
func (f Formatter) flatten(buf *bytes.Buffer, kvList []any, escapeKeys bool) []any {
|
|
// This logic overlaps with sanitize() but saves one type-cast per key,
|
|
// which can be measurable.
|
|
if len(kvList)%2 != 0 {
|
|
kvList = append(kvList, noValue)
|
|
}
|
|
copied := false
|
|
for i := 0; i < len(kvList); i += 2 {
|
|
k, ok := kvList[i].(string)
|
|
if !ok {
|
|
if !copied {
|
|
newList := make([]any, len(kvList))
|
|
copy(newList, kvList)
|
|
kvList = newList
|
|
copied = true
|
|
}
|
|
k = f.nonStringKey(kvList[i])
|
|
kvList[i] = k
|
|
}
|
|
v := kvList[i+1]
|
|
|
|
if i > 0 {
|
|
if f.outputFormat == outputJSON {
|
|
buf.WriteByte(f.comma())
|
|
} else {
|
|
// In theory the format could be something we don't understand. In
|
|
// practice, we control it, so it won't be.
|
|
buf.WriteByte(' ')
|
|
}
|
|
}
|
|
|
|
buf.WriteString(f.quoted(k, escapeKeys))
|
|
buf.WriteByte(f.colon())
|
|
buf.WriteString(f.pretty(v))
|
|
}
|
|
return kvList
|
|
}
|
|
|
|
func (f Formatter) quoted(str string, escape bool) string {
|
|
if escape {
|
|
return prettyString(str)
|
|
}
|
|
// this is faster
|
|
return `"` + str + `"`
|
|
}
|
|
|
|
func (f Formatter) comma() byte {
|
|
if f.outputFormat == outputJSON {
|
|
return ','
|
|
}
|
|
return ' '
|
|
}
|
|
|
|
func (f Formatter) colon() byte {
|
|
if f.outputFormat == outputJSON {
|
|
return ':'
|
|
}
|
|
return '='
|
|
}
|
|
|
|
func (f Formatter) pretty(value any) string {
|
|
return f.prettyWithFlags(value, 0, 0)
|
|
}
|
|
|
|
const (
|
|
flagRawStruct = 0x1 // do not print braces on structs
|
|
)
|
|
|
|
// TODO: This is not fast. Most of the overhead goes here.
|
|
func (f Formatter) prettyWithFlags(value any, flags uint32, depth int) string {
|
|
if depth > f.opts.MaxLogDepth {
|
|
return `"<max-log-depth-exceeded>"`
|
|
}
|
|
|
|
// Handle types that take full control of logging.
|
|
if v, ok := value.(logr.Marshaler); ok {
|
|
// Replace the value with what the type wants to get logged.
|
|
// That then gets handled below via reflection.
|
|
value = invokeMarshaler(v)
|
|
}
|
|
|
|
// Handle types that want to format themselves.
|
|
switch v := value.(type) {
|
|
case fmt.Stringer:
|
|
value = invokeStringer(v)
|
|
case error:
|
|
value = invokeError(v)
|
|
}
|
|
|
|
// Handling the most common types without reflect is a small perf win.
|
|
switch v := value.(type) {
|
|
case bool:
|
|
return strconv.FormatBool(v)
|
|
case string:
|
|
return prettyString(v)
|
|
case int:
|
|
return strconv.FormatInt(int64(v), 10)
|
|
case int8:
|
|
return strconv.FormatInt(int64(v), 10)
|
|
case int16:
|
|
return strconv.FormatInt(int64(v), 10)
|
|
case int32:
|
|
return strconv.FormatInt(int64(v), 10)
|
|
case int64:
|
|
return strconv.FormatInt(int64(v), 10)
|
|
case uint:
|
|
return strconv.FormatUint(uint64(v), 10)
|
|
case uint8:
|
|
return strconv.FormatUint(uint64(v), 10)
|
|
case uint16:
|
|
return strconv.FormatUint(uint64(v), 10)
|
|
case uint32:
|
|
return strconv.FormatUint(uint64(v), 10)
|
|
case uint64:
|
|
return strconv.FormatUint(v, 10)
|
|
case uintptr:
|
|
return strconv.FormatUint(uint64(v), 10)
|
|
case float32:
|
|
return strconv.FormatFloat(float64(v), 'f', -1, 32)
|
|
case float64:
|
|
return strconv.FormatFloat(v, 'f', -1, 64)
|
|
case complex64:
|
|
return `"` + strconv.FormatComplex(complex128(v), 'f', -1, 64) + `"`
|
|
case complex128:
|
|
return `"` + strconv.FormatComplex(v, 'f', -1, 128) + `"`
|
|
case PseudoStruct:
|
|
buf := bytes.NewBuffer(make([]byte, 0, 1024))
|
|
v = f.sanitize(v)
|
|
if flags&flagRawStruct == 0 {
|
|
buf.WriteByte('{')
|
|
}
|
|
for i := 0; i < len(v); i += 2 {
|
|
if i > 0 {
|
|
buf.WriteByte(f.comma())
|
|
}
|
|
k, _ := v[i].(string) // sanitize() above means no need to check success
|
|
// arbitrary keys might need escaping
|
|
buf.WriteString(prettyString(k))
|
|
buf.WriteByte(f.colon())
|
|
buf.WriteString(f.prettyWithFlags(v[i+1], 0, depth+1))
|
|
}
|
|
if flags&flagRawStruct == 0 {
|
|
buf.WriteByte('}')
|
|
}
|
|
return buf.String()
|
|
}
|
|
|
|
buf := bytes.NewBuffer(make([]byte, 0, 256))
|
|
t := reflect.TypeOf(value)
|
|
if t == nil {
|
|
return "null"
|
|
}
|
|
v := reflect.ValueOf(value)
|
|
switch t.Kind() {
|
|
case reflect.Bool:
|
|
return strconv.FormatBool(v.Bool())
|
|
case reflect.String:
|
|
return prettyString(v.String())
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
return strconv.FormatInt(int64(v.Int()), 10)
|
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr:
|
|
return strconv.FormatUint(uint64(v.Uint()), 10)
|
|
case reflect.Float32:
|
|
return strconv.FormatFloat(float64(v.Float()), 'f', -1, 32)
|
|
case reflect.Float64:
|
|
return strconv.FormatFloat(v.Float(), 'f', -1, 64)
|
|
case reflect.Complex64:
|
|
return `"` + strconv.FormatComplex(complex128(v.Complex()), 'f', -1, 64) + `"`
|
|
case reflect.Complex128:
|
|
return `"` + strconv.FormatComplex(v.Complex(), 'f', -1, 128) + `"`
|
|
case reflect.Struct:
|
|
if flags&flagRawStruct == 0 {
|
|
buf.WriteByte('{')
|
|
}
|
|
printComma := false // testing i>0 is not enough because of JSON omitted fields
|
|
for i := 0; i < t.NumField(); i++ {
|
|
fld := t.Field(i)
|
|
if fld.PkgPath != "" {
|
|
// reflect says this field is only defined for non-exported fields.
|
|
continue
|
|
}
|
|
if !v.Field(i).CanInterface() {
|
|
// reflect isn't clear exactly what this means, but we can't use it.
|
|
continue
|
|
}
|
|
name := ""
|
|
omitempty := false
|
|
if tag, found := fld.Tag.Lookup("json"); found {
|
|
if tag == "-" {
|
|
continue
|
|
}
|
|
if comma := strings.Index(tag, ","); comma != -1 {
|
|
if n := tag[:comma]; n != "" {
|
|
name = n
|
|
}
|
|
rest := tag[comma:]
|
|
if strings.Contains(rest, ",omitempty,") || strings.HasSuffix(rest, ",omitempty") {
|
|
omitempty = true
|
|
}
|
|
} else {
|
|
name = tag
|
|
}
|
|
}
|
|
if omitempty && isEmpty(v.Field(i)) {
|
|
continue
|
|
}
|
|
if printComma {
|
|
buf.WriteByte(f.comma())
|
|
}
|
|
printComma = true // if we got here, we are rendering a field
|
|
if fld.Anonymous && fld.Type.Kind() == reflect.Struct && name == "" {
|
|
buf.WriteString(f.prettyWithFlags(v.Field(i).Interface(), flags|flagRawStruct, depth+1))
|
|
continue
|
|
}
|
|
if name == "" {
|
|
name = fld.Name
|
|
}
|
|
// field names can't contain characters which need escaping
|
|
buf.WriteString(f.quoted(name, false))
|
|
buf.WriteByte(f.colon())
|
|
buf.WriteString(f.prettyWithFlags(v.Field(i).Interface(), 0, depth+1))
|
|
}
|
|
if flags&flagRawStruct == 0 {
|
|
buf.WriteByte('}')
|
|
}
|
|
return buf.String()
|
|
case reflect.Slice, reflect.Array:
|
|
// If this is outputing as JSON make sure this isn't really a json.RawMessage.
|
|
// If so just emit "as-is" and don't pretty it as that will just print
|
|
// it as [X,Y,Z,...] which isn't terribly useful vs the string form you really want.
|
|
if f.outputFormat == outputJSON {
|
|
if rm, ok := value.(json.RawMessage); ok {
|
|
// If it's empty make sure we emit an empty value as the array style would below.
|
|
if len(rm) > 0 {
|
|
buf.Write(rm)
|
|
} else {
|
|
buf.WriteString("null")
|
|
}
|
|
return buf.String()
|
|
}
|
|
}
|
|
buf.WriteByte('[')
|
|
for i := 0; i < v.Len(); i++ {
|
|
if i > 0 {
|
|
buf.WriteByte(f.comma())
|
|
}
|
|
e := v.Index(i)
|
|
buf.WriteString(f.prettyWithFlags(e.Interface(), 0, depth+1))
|
|
}
|
|
buf.WriteByte(']')
|
|
return buf.String()
|
|
case reflect.Map:
|
|
buf.WriteByte('{')
|
|
// This does not sort the map keys, for best perf.
|
|
it := v.MapRange()
|
|
i := 0
|
|
for it.Next() {
|
|
if i > 0 {
|
|
buf.WriteByte(f.comma())
|
|
}
|
|
// If a map key supports TextMarshaler, use it.
|
|
keystr := ""
|
|
if m, ok := it.Key().Interface().(encoding.TextMarshaler); ok {
|
|
txt, err := m.MarshalText()
|
|
if err != nil {
|
|
keystr = fmt.Sprintf("<error-MarshalText: %s>", err.Error())
|
|
} else {
|
|
keystr = string(txt)
|
|
}
|
|
keystr = prettyString(keystr)
|
|
} else {
|
|
// prettyWithFlags will produce already-escaped values
|
|
keystr = f.prettyWithFlags(it.Key().Interface(), 0, depth+1)
|
|
if t.Key().Kind() != reflect.String {
|
|
// JSON only does string keys. Unlike Go's standard JSON, we'll
|
|
// convert just about anything to a string.
|
|
keystr = prettyString(keystr)
|
|
}
|
|
}
|
|
buf.WriteString(keystr)
|
|
buf.WriteByte(f.colon())
|
|
buf.WriteString(f.prettyWithFlags(it.Value().Interface(), 0, depth+1))
|
|
i++
|
|
}
|
|
buf.WriteByte('}')
|
|
return buf.String()
|
|
case reflect.Ptr, reflect.Interface:
|
|
if v.IsNil() {
|
|
return "null"
|
|
}
|
|
return f.prettyWithFlags(v.Elem().Interface(), 0, depth)
|
|
}
|
|
return fmt.Sprintf(`"<unhandled-%s>"`, t.Kind().String())
|
|
}
|
|
|
|
func prettyString(s string) string {
|
|
// Avoid escaping (which does allocations) if we can.
|
|
if needsEscape(s) {
|
|
return strconv.Quote(s)
|
|
}
|
|
b := bytes.NewBuffer(make([]byte, 0, 1024))
|
|
b.WriteByte('"')
|
|
b.WriteString(s)
|
|
b.WriteByte('"')
|
|
return b.String()
|
|
}
|
|
|
|
// needsEscape determines whether the input string needs to be escaped or not,
|
|
// without doing any allocations.
|
|
func needsEscape(s string) bool {
|
|
for _, r := range s {
|
|
if !strconv.IsPrint(r) || r == '\\' || r == '"' {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func isEmpty(v reflect.Value) bool {
|
|
switch v.Kind() {
|
|
case reflect.Array, reflect.Map, reflect.Slice, reflect.String:
|
|
return v.Len() == 0
|
|
case reflect.Bool:
|
|
return !v.Bool()
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
return v.Int() == 0
|
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr:
|
|
return v.Uint() == 0
|
|
case reflect.Float32, reflect.Float64:
|
|
return v.Float() == 0
|
|
case reflect.Complex64, reflect.Complex128:
|
|
return v.Complex() == 0
|
|
case reflect.Interface, reflect.Ptr:
|
|
return v.IsNil()
|
|
}
|
|
return false
|
|
}
|
|
|
|
func invokeMarshaler(m logr.Marshaler) (ret any) {
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
ret = fmt.Sprintf("<panic: %s>", r)
|
|
}
|
|
}()
|
|
return m.MarshalLog()
|
|
}
|
|
|
|
func invokeStringer(s fmt.Stringer) (ret string) {
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
ret = fmt.Sprintf("<panic: %s>", r)
|
|
}
|
|
}()
|
|
return s.String()
|
|
}
|
|
|
|
func invokeError(e error) (ret string) {
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
ret = fmt.Sprintf("<panic: %s>", r)
|
|
}
|
|
}()
|
|
return e.Error()
|
|
}
|
|
|
|
// Caller represents the original call site for a log line, after considering
|
|
// logr.Logger.WithCallDepth and logr.Logger.WithCallStackHelper. The File and
|
|
// Line fields will always be provided, while the Func field is optional.
|
|
// Users can set the render hook fields in Options to examine logged key-value
|
|
// pairs, one of which will be {"caller", Caller} if the Options.LogCaller
|
|
// field is enabled for the given MessageClass.
|
|
type Caller struct {
|
|
// File is the basename of the file for this call site.
|
|
File string `json:"file"`
|
|
// Line is the line number in the file for this call site.
|
|
Line int `json:"line"`
|
|
// Func is the function name for this call site, or empty if
|
|
// Options.LogCallerFunc is not enabled.
|
|
Func string `json:"function,omitempty"`
|
|
}
|
|
|
|
func (f Formatter) caller() Caller {
|
|
// +1 for this frame, +1 for Info/Error.
|
|
pc, file, line, ok := runtime.Caller(f.depth + 2)
|
|
if !ok {
|
|
return Caller{"<unknown>", 0, ""}
|
|
}
|
|
fn := ""
|
|
if f.opts.LogCallerFunc {
|
|
if fp := runtime.FuncForPC(pc); fp != nil {
|
|
fn = fp.Name()
|
|
}
|
|
}
|
|
|
|
return Caller{filepath.Base(file), line, fn}
|
|
}
|
|
|
|
const noValue = "<no-value>"
|
|
|
|
func (f Formatter) nonStringKey(v any) string {
|
|
return fmt.Sprintf("<non-string-key: %s>", f.snippet(v))
|
|
}
|
|
|
|
// snippet produces a short snippet string of an arbitrary value.
|
|
func (f Formatter) snippet(v any) string {
|
|
const snipLen = 16
|
|
|
|
snip := f.pretty(v)
|
|
if len(snip) > snipLen {
|
|
snip = snip[:snipLen]
|
|
}
|
|
return snip
|
|
}
|
|
|
|
// sanitize ensures that a list of key-value pairs has a value for every key
|
|
// (adding a value if needed) and that each key is a string (substituting a key
|
|
// if needed).
|
|
func (f Formatter) sanitize(kvList []any) []any {
|
|
if len(kvList)%2 != 0 {
|
|
kvList = append(kvList, noValue)
|
|
}
|
|
for i := 0; i < len(kvList); i += 2 {
|
|
_, ok := kvList[i].(string)
|
|
if !ok {
|
|
kvList[i] = f.nonStringKey(kvList[i])
|
|
}
|
|
}
|
|
return kvList
|
|
}
|
|
|
|
// startGroup opens a new group scope (basically a sub-struct), which locks all
|
|
// the current saved values and starts them anew. This is needed to satisfy
|
|
// slog.
|
|
func (f *Formatter) startGroup(name string) {
|
|
// Unnamed groups are just inlined.
|
|
if name == "" {
|
|
return
|
|
}
|
|
|
|
n := len(f.groups)
|
|
f.groups = append(f.groups[:n:n], groupDef{f.groupName, f.valuesStr})
|
|
|
|
// Start collecting new values.
|
|
f.groupName = name
|
|
f.valuesStr = ""
|
|
f.values = nil
|
|
}
|
|
|
|
// Init configures this Formatter from runtime info, such as the call depth
|
|
// imposed by logr itself.
|
|
// Note that this receiver is a pointer, so depth can be saved.
|
|
func (f *Formatter) Init(info logr.RuntimeInfo) {
|
|
f.depth += info.CallDepth
|
|
}
|
|
|
|
// Enabled checks whether an info message at the given level should be logged.
|
|
func (f Formatter) Enabled(level int) bool {
|
|
return level <= f.opts.Verbosity
|
|
}
|
|
|
|
// GetDepth returns the current depth of this Formatter. This is useful for
|
|
// implementations which do their own caller attribution.
|
|
func (f Formatter) GetDepth() int {
|
|
return f.depth
|
|
}
|
|
|
|
// FormatInfo renders an Info log message into strings. The prefix will be
|
|
// empty when no names were set (via AddNames), or when the output is
|
|
// configured for JSON.
|
|
func (f Formatter) FormatInfo(level int, msg string, kvList []any) (prefix, argsStr string) {
|
|
args := make([]any, 0, 64) // using a constant here impacts perf
|
|
prefix = f.prefix
|
|
if f.outputFormat == outputJSON {
|
|
args = append(args, "logger", prefix)
|
|
prefix = ""
|
|
}
|
|
if f.opts.LogTimestamp {
|
|
args = append(args, "ts", time.Now().Format(f.opts.TimestampFormat))
|
|
}
|
|
if policy := f.opts.LogCaller; policy == All || policy == Info {
|
|
args = append(args, "caller", f.caller())
|
|
}
|
|
if key := *f.opts.LogInfoLevel; key != "" {
|
|
args = append(args, key, level)
|
|
}
|
|
args = append(args, "msg", msg)
|
|
return prefix, f.render(args, kvList)
|
|
}
|
|
|
|
// FormatError renders an Error log message into strings. The prefix will be
|
|
// empty when no names were set (via AddNames), or when the output is
|
|
// configured for JSON.
|
|
func (f Formatter) FormatError(err error, msg string, kvList []any) (prefix, argsStr string) {
|
|
args := make([]any, 0, 64) // using a constant here impacts perf
|
|
prefix = f.prefix
|
|
if f.outputFormat == outputJSON {
|
|
args = append(args, "logger", prefix)
|
|
prefix = ""
|
|
}
|
|
if f.opts.LogTimestamp {
|
|
args = append(args, "ts", time.Now().Format(f.opts.TimestampFormat))
|
|
}
|
|
if policy := f.opts.LogCaller; policy == All || policy == Error {
|
|
args = append(args, "caller", f.caller())
|
|
}
|
|
args = append(args, "msg", msg)
|
|
var loggableErr any
|
|
if err != nil {
|
|
loggableErr = err.Error()
|
|
}
|
|
args = append(args, "error", loggableErr)
|
|
return prefix, f.render(args, kvList)
|
|
}
|
|
|
|
// AddName appends the specified name. funcr uses '/' characters to separate
|
|
// name elements. Callers should not pass '/' in the provided name string, but
|
|
// this library does not actually enforce that.
|
|
func (f *Formatter) AddName(name string) {
|
|
if len(f.prefix) > 0 {
|
|
f.prefix += "/"
|
|
}
|
|
f.prefix += name
|
|
}
|
|
|
|
// AddValues adds key-value pairs to the set of saved values to be logged with
|
|
// each log line.
|
|
func (f *Formatter) AddValues(kvList []any) {
|
|
// Three slice args forces a copy.
|
|
n := len(f.values)
|
|
f.values = append(f.values[:n:n], kvList...)
|
|
|
|
vals := f.values
|
|
if hook := f.opts.RenderValuesHook; hook != nil {
|
|
vals = hook(f.sanitize(vals))
|
|
}
|
|
|
|
// Pre-render values, so we don't have to do it on each Info/Error call.
|
|
buf := bytes.NewBuffer(make([]byte, 0, 1024))
|
|
f.flatten(buf, vals, true) // escape user-provided keys
|
|
f.valuesStr = buf.String()
|
|
}
|
|
|
|
// AddCallDepth increases the number of stack-frames to skip when attributing
|
|
// the log line to a file and line.
|
|
func (f *Formatter) AddCallDepth(depth int) {
|
|
f.depth += depth
|
|
}
|