slog/json_handler.go - exp - Git at Google

 // Copyright 2022 The Go Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style
 // license that can be found in the LICENSE file.

 package slog

 import (
 	"bytes"
 	"context"
 	"encoding/json"
 	"errors"
 	"fmt"
 	"io"
 	"math"
 	"strconv"
 	"time"
 	"unicode/utf8"

 	"golang.org/x/exp/slog/internal/buffer"
 )

 // JSONHandler is a Handler that writes Records to an io.Writer as
 // line-delimited JSON objects.
 type JSONHandler struct {
 	*commonHandler
 }

 // NewJSONHandler creates a JSONHandler that writes to w,
 // using the default options.
 func NewJSONHandler(w io.Writer) *JSONHandler {
 	return (HandlerOptions{}).NewJSONHandler(w)
 }

 // NewJSONHandler creates a JSONHandler with the given options that writes to w.
 func (opts HandlerOptions) NewJSONHandler(w io.Writer) *JSONHandler {
 	return &JSONHandler{
 		&commonHandler{
 			json: true,
 			w:    w,
 			opts: opts,
 		},
 	}
 }

 // Enabled reports whether the handler handles records at the given level.
 // The handler ignores records whose level is lower.
 func (h *JSONHandler) Enabled(_ context.Context, level Level) bool {
 	return h.commonHandler.enabled(level)
 }

 // WithAttrs returns a new JSONHandler whose attributes consists
 // of h's attributes followed by attrs.
 func (h *JSONHandler) WithAttrs(attrs []Attr) Handler {
 	return &JSONHandler{commonHandler: h.commonHandler.withAttrs(attrs)}
 }

 func (h *JSONHandler) WithGroup(name string) Handler {
 	return &JSONHandler{commonHandler: h.commonHandler.withGroup(name)}
 }

 // Handle formats its argument Record as a JSON object on a single line.
 //
 // If the Record's time is zero, the time is omitted.
 // Otherwise, the key is "time"
 // and the value is output as with json.Marshal.
 //
 // If the Record's level is zero, the level is omitted.
 // Otherwise, the key is "level"
 // and the value of [Level.String] is output.
 //
 // If the AddSource option is set and source information is available,
 // the key is "source"
 // and the value is output as "FILE:LINE".
 //
 // The message's key is "msg".
 //
 // To modify these or other attributes, or remove them from the output, use
 // [HandlerOptions.ReplaceAttr].
 //
 // Values are formatted as with encoding/json.Marshal, with the following
 // exceptions:
 //   - Floating-point NaNs and infinities are formatted as one of the strings
 //     "NaN", "+Inf" or "-Inf".
 //   - Levels are formatted as with Level.String.
 //   - HTML characters are not escaped.
 //
 // Each call to Handle results in a single serialized call to io.Writer.Write.
 func (h *JSONHandler) Handle(_ context.Context, r Record) error {
 	return h.commonHandler.handle(r)
 }

 // Adapted from time.Time.MarshalJSON to avoid allocation.
 func appendJSONTime(s *handleState, t time.Time) {
 	if y := t.Year(); y < 0 || y >= 10000 {
 		// RFC 3339 is clear that years are 4 digits exactly.
 		// See golang.org/issue/4556#c15 for more discussion.
 		s.appendError(errors.New("time.Time year outside of range [0,9999]"))
 	}
 	s.buf.WriteByte('"')
 	*s.buf = t.AppendFormat(*s.buf, time.RFC3339Nano)
 	s.buf.WriteByte('"')
 }

 func appendJSONValue(s *handleState, v Value) error {
 	switch v.Kind() {
 	case KindString:
 		s.appendString(v.str())
 	case KindInt64:
 		*s.buf = strconv.AppendInt(*s.buf, v.Int64(), 10)
 	case KindUint64:
 		*s.buf = strconv.AppendUint(*s.buf, v.Uint64(), 10)
 	case KindFloat64:
 		f := v.Float64()
 		// json.Marshal fails on special floats, so handle them here.
 		switch {
 		case math.IsInf(f, 1):
 			s.buf.WriteString(`"+Inf"`)
 		case math.IsInf(f, -1):
 			s.buf.WriteString(`"-Inf"`)
 		case math.IsNaN(f):
 			s.buf.WriteString(`"NaN"`)
 		default:
 			// json.Marshal is funny about floats; it doesn't
 			// always match strconv.AppendFloat. So just call it.
 			// That's expensive, but floats are rare.
 			if err := appendJSONMarshal(s.buf, f); err != nil {
 				return err
 			}
 		}
 	case KindBool:
 		*s.buf = strconv.AppendBool(*s.buf, v.Bool())
 	case KindDuration:
 		// Do what json.Marshal does.
 		*s.buf = strconv.AppendInt(*s.buf, int64(v.Duration()), 10)
 	case KindTime:
 		s.appendTime(v.Time())
 	case KindAny:
 		a := v.Any()
 		if err, ok := a.(error); ok {
 			s.appendString(err.Error())
 		} else {
 			return appendJSONMarshal(s.buf, a)
 		}
 	default:
 		panic(fmt.Sprintf("bad kind: %d", v.Kind()))
 	}
 	return nil
 }

 func appendJSONMarshal(buf *buffer.Buffer, v any) error {
 	// Use a json.Encoder to avoid escaping HTML.
 	var bb bytes.Buffer
 	enc := json.NewEncoder(&bb)
 	enc.SetEscapeHTML(false)
 	if err := enc.Encode(v); err != nil {
 		return err
 	}
 	bs := bb.Bytes()
 	buf.Write(bs[:len(bs)-1]) // remove final newline
 	return nil
 }

 // appendEscapedJSONString escapes s for JSON and appends it to buf.
 // It does not surround the string in quotation marks.
 //
 // Modified from encoding/json/encode.go:encodeState.string,
 // with escapeHTML set to true.
 //
 // TODO: review whether HTML escaping is necessary.
 func appendEscapedJSONString(buf []byte, s string) []byte {
 	char := func(b byte) { buf = append(buf, b) }
 	str := func(s string) { buf = append(buf, s...) }

 	start := 0
 	for i := 0; i < len(s); {
 		if b := s[i]; b < utf8.RuneSelf {
 			if safeSet[b] {
 				i++
 				continue
 			}
 			if start < i {
 				str(s[start:i])
 			}
 			char('\\')
 			switch b {
 			case '\\', '"':
 				char(b)
 			case '\n':
 				char('n')
 			case '\r':
 				char('r')
 			case '\t':
 				char('t')
 			default:
 				// This encodes bytes < 0x20 except for \t, \n and \r.
 				str(`u00`)
 				char(hex[b>>4])
 				char(hex[b&0xF])
 			}
 			i++
 			start = i
 			continue
 		}
 		c, size := utf8.DecodeRuneInString(s[i:])
 		if c == utf8.RuneError && size == 1 {
 			if start < i {
 				str(s[start:i])
 			}
 			str(`\ufffd`)
 			i += size
 			start = i
 			continue
 		}
 		// U+2028 is LINE SEPARATOR.
 		// U+2029 is PARAGRAPH SEPARATOR.
 		// They are both technically valid characters in JSON strings,
 		// but don't work in JSONP, which has to be evaluated as JavaScript,
 		// and can lead to security holes there. It is valid JSON to
 		// escape them, so we do so unconditionally.
 		// See http://timelessrepo.com/json-isnt-a-javascript-subset for discussion.
 		if c == '\u2028' || c == '\u2029' {
 			if start < i {
 				str(s[start:i])
 			}
 			str(`\u202`)
 			char(hex[c&0xF])
 			i += size
 			start = i
 			continue
 		}
 		i += size
 	}
 	if start < len(s) {
 		str(s[start:])
 	}
 	return buf
 }

 var hex = "0123456789abcdef"

 // Copied from encoding/json/tables.go.
 //
 // safeSet holds the value true if the ASCII character with the given array
 // position can be represented inside a JSON string without any further
 // escaping.
 //
 // All values are true except for the ASCII control characters (0-31), the
 // double quote ("), and the backslash character ("\").
 var safeSet = [utf8.RuneSelf]bool{
 	' ':      true,
 	'!':      true,
 	'"':      false,
 	'#':      true,
 	'$':      true,
 	'%':      true,
 	'&':      true,
 	'\'':     true,
 	'(':      true,
 	')':      true,
 	'*':      true,
 	'+':      true,
 	',':      true,
 	'-':      true,
 	'.':      true,
 	'/':      true,
 	'0':      true,
 	'1':      true,
 	'2':      true,
 	'3':      true,
 	'4':      true,
 	'5':      true,
 	'6':      true,
 	'7':      true,
 	'8':      true,
 	'9':      true,
 	':':      true,
 	';':      true,
 	'<':      true,
 	'=':      true,
 	'>':      true,
 	'?':      true,
 	'@':      true,
 	'A':      true,
 	'B':      true,
 	'C':      true,
 	'D':      true,
 	'E':      true,
 	'F':      true,
 	'G':      true,
 	'H':      true,
 	'I':      true,
 	'J':      true,
 	'K':      true,
 	'L':      true,
 	'M':      true,
 	'N':      true,
 	'O':      true,
 	'P':      true,
 	'Q':      true,
 	'R':      true,
 	'S':      true,
 	'T':      true,
 	'U':      true,
 	'V':      true,
 	'W':      true,
 	'X':      true,
 	'Y':      true,
 	'Z':      true,
 	'[':      true,
 	'\\':     false,
 	']':      true,
 	'^':      true,
 	'_':      true,
 	'`':      true,
 	'a':      true,
 	'b':      true,
 	'c':      true,
 	'd':      true,
 	'e':      true,
 	'f':      true,
 	'g':      true,
 	'h':      true,
 	'i':      true,
 	'j':      true,
 	'k':      true,
 	'l':      true,
 	'm':      true,
 	'n':      true,
 	'o':      true,
 	'p':      true,
 	'q':      true,
 	'r':      true,
 	's':      true,
 	't':      true,
 	'u':      true,
 	'v':      true,
 	'w':      true,
 	'x':      true,
 	'y':      true,
 	'z':      true,
 	'{':      true,
 	'|':      true,
 	'}':      true,
 	'~':      true,
 	'\u007f': true,
 }
	// Copyright 2022 The Go Authors. All rights reserved.
	// Use of this source code is governed by a BSD-style
	// license that can be found in the LICENSE file.

	package slog

	import (
	"bytes"
	"context"
	"encoding/json"
	"errors"
	"fmt"
	"io"
	"math"
	"strconv"
	"time"
	"unicode/utf8"

	"golang.org/x/exp/slog/internal/buffer"
	)

	// JSONHandler is a Handler that writes Records to an io.Writer as
	// line-delimited JSON objects.
	type JSONHandler struct {
	*commonHandler
	}

	// NewJSONHandler creates a JSONHandler that writes to w,
	// using the default options.
	func NewJSONHandler(w io.Writer) *JSONHandler {
	return (HandlerOptions{}).NewJSONHandler(w)
	}

	// NewJSONHandler creates a JSONHandler with the given options that writes to w.
	func (opts HandlerOptions) NewJSONHandler(w io.Writer) *JSONHandler {
	return &JSONHandler{
	&commonHandler{
	json: true,
	w: w,
	opts: opts,
	},
	}
	}

	// Enabled reports whether the handler handles records at the given level.
	// The handler ignores records whose level is lower.
	func (h *JSONHandler) Enabled(_ context.Context, level Level) bool {
	return h.commonHandler.enabled(level)
	}

	// WithAttrs returns a new JSONHandler whose attributes consists
	// of h's attributes followed by attrs.
	func (h *JSONHandler) WithAttrs(attrs []Attr) Handler {
	return &JSONHandler{commonHandler: h.commonHandler.withAttrs(attrs)}
	}

	func (h *JSONHandler) WithGroup(name string) Handler {
	return &JSONHandler{commonHandler: h.commonHandler.withGroup(name)}
	}

	// Handle formats its argument Record as a JSON object on a single line.
	//
	// If the Record's time is zero, the time is omitted.
	// Otherwise, the key is "time"
	// and the value is output as with json.Marshal.
	//
	// If the Record's level is zero, the level is omitted.
	// Otherwise, the key is "level"
	// and the value of [Level.String] is output.
	//
	// If the AddSource option is set and source information is available,
	// the key is "source"
	// and the value is output as "FILE:LINE".
	//
	// The message's key is "msg".
	//
	// To modify these or other attributes, or remove them from the output, use
	// [HandlerOptions.ReplaceAttr].
	//
	// Values are formatted as with encoding/json.Marshal, with the following
	// exceptions:
	// - Floating-point NaNs and infinities are formatted as one of the strings
	// "NaN", "+Inf" or "-Inf".
	// - Levels are formatted as with Level.String.
	// - HTML characters are not escaped.
	//
	// Each call to Handle results in a single serialized call to io.Writer.Write.
	func (h *JSONHandler) Handle(_ context.Context, r Record) error {
	return h.commonHandler.handle(r)
	}

	// Adapted from time.Time.MarshalJSON to avoid allocation.
	func appendJSONTime(s *handleState, t time.Time) {
	if y := t.Year(); y < 0 \|\| y >= 10000 {
	// RFC 3339 is clear that years are 4 digits exactly.
	// See golang.org/issue/4556#c15 for more discussion.
	s.appendError(errors.New("time.Time year outside of range [0,9999]"))
	}
	s.buf.WriteByte('"')
	s.buf = t.AppendFormat(s.buf, time.RFC3339Nano)
	s.buf.WriteByte('"')
	}

	func appendJSONValue(s *handleState, v Value) error {
	switch v.Kind() {
	case KindString:
	s.appendString(v.str())
	case KindInt64:
	s.buf = strconv.AppendInt(s.buf, v.Int64(), 10)
	case KindUint64:
	s.buf = strconv.AppendUint(s.buf, v.Uint64(), 10)
	case KindFloat64:
	f := v.Float64()
	// json.Marshal fails on special floats, so handle them here.
	switch {
	case math.IsInf(f, 1):
	s.buf.WriteString(`"+Inf"`)
	case math.IsInf(f, -1):
	s.buf.WriteString(`"-Inf"`)
	case math.IsNaN(f):
	s.buf.WriteString(`"NaN"`)
	default:
	// json.Marshal is funny about floats; it doesn't
	// always match strconv.AppendFloat. So just call it.
	// That's expensive, but floats are rare.
	if err := appendJSONMarshal(s.buf, f); err != nil {
	return err
	}
	}
	case KindBool:
	s.buf = strconv.AppendBool(s.buf, v.Bool())
	case KindDuration:
	// Do what json.Marshal does.
	s.buf = strconv.AppendInt(s.buf, int64(v.Duration()), 10)
	case KindTime:
	s.appendTime(v.Time())
	case KindAny:
	a := v.Any()
	if err, ok := a.(error); ok {
	s.appendString(err.Error())
	} else {
	return appendJSONMarshal(s.buf, a)
	}
	default:
	panic(fmt.Sprintf("bad kind: %d", v.Kind()))
	}
	return nil
	}

	func appendJSONMarshal(buf *buffer.Buffer, v any) error {
	// Use a json.Encoder to avoid escaping HTML.
	var bb bytes.Buffer
	enc := json.NewEncoder(&bb)
	enc.SetEscapeHTML(false)
	if err := enc.Encode(v); err != nil {
	return err
	}
	bs := bb.Bytes()
	buf.Write(bs[:len(bs)-1]) // remove final newline
	return nil
	}

	// appendEscapedJSONString escapes s for JSON and appends it to buf.
	// It does not surround the string in quotation marks.
	//
	// Modified from encoding/json/encode.go:encodeState.string,
	// with escapeHTML set to true.
	//
	// TODO: review whether HTML escaping is necessary.
	func appendEscapedJSONString(buf []byte, s string) []byte {
	char := func(b byte) { buf = append(buf, b) }
	str := func(s string) { buf = append(buf, s...) }

	start := 0
	for i := 0; i < len(s); {
	if b := s[i]; b < utf8.RuneSelf {
	if safeSet[b] {
	i++
	continue
	}
	if start < i {
	str(s[start:i])
	}
	char('\\')
	switch b {
	case '\\', '"':
	char(b)
	case '\n':
	char('n')
	case '\r':
	char('r')
	case '\t':
	char('t')
	default:
	// This encodes bytes < 0x20 except for \t, \n and \r.
	str(`u00`)
	char(hex[b>>4])
	char(hex[b&0xF])
	}
	i++
	start = i
	continue
	}
	c, size := utf8.DecodeRuneInString(s[i:])
	if c == utf8.RuneError && size == 1 {
	if start < i {
	str(s[start:i])
	}
	str(`\ufffd`)
	i += size
	start = i
	continue
	}
	// U+2028 is LINE SEPARATOR.
	// U+2029 is PARAGRAPH SEPARATOR.
	// They are both technically valid characters in JSON strings,
	// but don't work in JSONP, which has to be evaluated as JavaScript,
	// and can lead to security holes there. It is valid JSON to
	// escape them, so we do so unconditionally.
	// See http://timelessrepo.com/json-isnt-a-javascript-subset for discussion.
	if c == '\u2028' \|\| c == '\u2029' {
	if start < i {
	str(s[start:i])
	}
	str(`\u202`)
	char(hex[c&0xF])
	i += size
	start = i
	continue
	}
	i += size
	}
	if start < len(s) {
	str(s[start:])
	}
	return buf
	}

	var hex = "0123456789abcdef"

	// Copied from encoding/json/tables.go.
	//
	// safeSet holds the value true if the ASCII character with the given array
	// position can be represented inside a JSON string without any further
	// escaping.
	//
	// All values are true except for the ASCII control characters (0-31), the
	// double quote ("), and the backslash character ("\").
	var safeSet = [utf8.RuneSelf]bool{
	' ': true,
	'!': true,
	'"': false,
	'#': true,
	'$': true,
	'%': true,
	'&': true,
	'\'': true,
	'(': true,
	')': true,
	'*': true,
	'+': true,
	',': true,
	'-': true,
	'.': true,
	'/': true,
	'0': true,
	'1': true,
	'2': true,
	'3': true,
	'4': true,
	'5': true,
	'6': true,
	'7': true,
	'8': true,
	'9': true,
	':': true,
	';': true,
	'<': true,
	'=': true,
	'>': true,
	'?': true,
	'@': true,
	'A': true,
	'B': true,
	'C': true,
	'D': true,
	'E': true,
	'F': true,
	'G': true,
	'H': true,
	'I': true,
	'J': true,
	'K': true,
	'L': true,
	'M': true,
	'N': true,
	'O': true,
	'P': true,
	'Q': true,
	'R': true,
	'S': true,
	'T': true,
	'U': true,
	'V': true,
	'W': true,
	'X': true,
	'Y': true,
	'Z': true,
	'[': true,
	'\\': false,
	']': true,
	'^': true,
	'_': true,
	'`': true,
	'a': true,
	'b': true,
	'c': true,
	'd': true,
	'e': true,
	'f': true,
	'g': true,
	'h': true,
	'i': true,
	'j': true,
	'k': true,
	'l': true,
	'm': true,
	'n': true,
	'o': true,
	'p': true,
	'q': true,
	'r': true,
	's': true,
	't': true,
	'u': true,
	'v': true,
	'w': true,
	'x': true,
	'y': true,
	'z': true,
	'{': true,
	'\|': true,
	'}': true,
	'~': true,
	'\u007f': true,
	}