mirror of
				https://github.com/superseriousbusiness/gotosocial.git
				synced 2025-10-31 14:32:24 -05:00 
			
		
		
		
	
		
			
				
	
	
		
			911 lines
		
	
	
	
		
			26 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			911 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
 | |
| 	parentValuesStr string
 | |
| 	depth           int
 | |
| 	opts            *Options
 | |
| 	group           string // for slog groups
 | |
| 	groupDepth      int
 | |
| }
 | |
| 
 | |
| // 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
 | |
| )
 | |
| 
 | |
| // 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 line
 | |
| 	}
 | |
| 
 | |
| 	vals := builtins
 | |
| 	if hook := f.opts.RenderBuiltinsHook; hook != nil {
 | |
| 		vals = hook(f.sanitize(vals))
 | |
| 	}
 | |
| 	f.flatten(buf, vals, false, false) // keys are ours, no need to escape
 | |
| 	continuing := len(builtins) > 0
 | |
| 
 | |
| 	if f.parentValuesStr != "" {
 | |
| 		if continuing {
 | |
| 			buf.WriteByte(f.comma())
 | |
| 		}
 | |
| 		buf.WriteString(f.parentValuesStr)
 | |
| 		continuing = true
 | |
| 	}
 | |
| 
 | |
| 	groupDepth := f.groupDepth
 | |
| 	if f.group != "" {
 | |
| 		if f.valuesStr != "" || len(args) != 0 {
 | |
| 			if continuing {
 | |
| 				buf.WriteByte(f.comma())
 | |
| 			}
 | |
| 			buf.WriteString(f.quoted(f.group, true)) // escape user-provided keys
 | |
| 			buf.WriteByte(f.colon())
 | |
| 			buf.WriteByte('{') // for the group
 | |
| 			continuing = false
 | |
| 		} else {
 | |
| 			// The group was empty
 | |
| 			groupDepth--
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	if f.valuesStr != "" {
 | |
| 		if continuing {
 | |
| 			buf.WriteByte(f.comma())
 | |
| 		}
 | |
| 		buf.WriteString(f.valuesStr)
 | |
| 		continuing = true
 | |
| 	}
 | |
| 
 | |
| 	vals = args
 | |
| 	if hook := f.opts.RenderArgsHook; hook != nil {
 | |
| 		vals = hook(f.sanitize(vals))
 | |
| 	}
 | |
| 	f.flatten(buf, vals, continuing, true) // escape user-provided keys
 | |
| 
 | |
| 	for i := 0; i < groupDepth; i++ {
 | |
| 		buf.WriteByte('}') // for the groups
 | |
| 	}
 | |
| 
 | |
| 	if f.outputFormat == outputJSON {
 | |
| 		buf.WriteByte('}') // for the whole line
 | |
| 	}
 | |
| 
 | |
| 	return buf.String()
 | |
| }
 | |
| 
 | |
| // flatten renders a list of key-value pairs into a buffer.  If continuing is
 | |
| // true, it assumes that the buffer has previous values and will emit a
 | |
| // separator (which depends on the output format) before the first pair it
 | |
| // writes.  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, continuing bool, 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 || continuing {
 | |
| 			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(group string) {
 | |
| 	// Unnamed groups are just inlined.
 | |
| 	if group == "" {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	// Any saved values can no longer be changed.
 | |
| 	buf := bytes.NewBuffer(make([]byte, 0, 1024))
 | |
| 	continuing := false
 | |
| 
 | |
| 	if f.parentValuesStr != "" {
 | |
| 		buf.WriteString(f.parentValuesStr)
 | |
| 		continuing = true
 | |
| 	}
 | |
| 
 | |
| 	if f.group != "" && f.valuesStr != "" {
 | |
| 		if continuing {
 | |
| 			buf.WriteByte(f.comma())
 | |
| 		}
 | |
| 		buf.WriteString(f.quoted(f.group, true)) // escape user-provided keys
 | |
| 		buf.WriteByte(f.colon())
 | |
| 		buf.WriteByte('{') // for the group
 | |
| 		continuing = false
 | |
| 	}
 | |
| 
 | |
| 	if f.valuesStr != "" {
 | |
| 		if continuing {
 | |
| 			buf.WriteByte(f.comma())
 | |
| 		}
 | |
| 		buf.WriteString(f.valuesStr)
 | |
| 	}
 | |
| 
 | |
| 	// NOTE: We don't close the scope here - that's done later, when a log line
 | |
| 	// is actually rendered (because we have N scopes to close).
 | |
| 
 | |
| 	f.parentValuesStr = buf.String()
 | |
| 
 | |
| 	// Start collecting new values.
 | |
| 	f.group = group
 | |
| 	f.groupDepth++
 | |
| 	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, false, 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
 | |
| }
 |