summaryrefslogtreecommitdiff
path: root/vendor/github.com/goccy/go-json/encode.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/goccy/go-json/encode.go')
-rw-r--r--vendor/github.com/goccy/go-json/encode.go326
1 files changed, 326 insertions, 0 deletions
diff --git a/vendor/github.com/goccy/go-json/encode.go b/vendor/github.com/goccy/go-json/encode.go
new file mode 100644
index 000000000..4bd899f38
--- /dev/null
+++ b/vendor/github.com/goccy/go-json/encode.go
@@ -0,0 +1,326 @@
+package json
+
+import (
+ "context"
+ "io"
+ "os"
+ "unsafe"
+
+ "github.com/goccy/go-json/internal/encoder"
+ "github.com/goccy/go-json/internal/encoder/vm"
+ "github.com/goccy/go-json/internal/encoder/vm_color"
+ "github.com/goccy/go-json/internal/encoder/vm_color_indent"
+ "github.com/goccy/go-json/internal/encoder/vm_indent"
+)
+
+// An Encoder writes JSON values to an output stream.
+type Encoder struct {
+ w io.Writer
+ enabledIndent bool
+ enabledHTMLEscape bool
+ prefix string
+ indentStr string
+}
+
+// NewEncoder returns a new encoder that writes to w.
+func NewEncoder(w io.Writer) *Encoder {
+ return &Encoder{w: w, enabledHTMLEscape: true}
+}
+
+// Encode writes the JSON encoding of v to the stream, followed by a newline character.
+//
+// See the documentation for Marshal for details about the conversion of Go values to JSON.
+func (e *Encoder) Encode(v interface{}) error {
+ return e.EncodeWithOption(v)
+}
+
+// EncodeWithOption call Encode with EncodeOption.
+func (e *Encoder) EncodeWithOption(v interface{}, optFuncs ...EncodeOptionFunc) error {
+ ctx := encoder.TakeRuntimeContext()
+ ctx.Option.Flag = 0
+
+ err := e.encodeWithOption(ctx, v, optFuncs...)
+
+ encoder.ReleaseRuntimeContext(ctx)
+ return err
+}
+
+// EncodeContext call Encode with context.Context and EncodeOption.
+func (e *Encoder) EncodeContext(ctx context.Context, v interface{}, optFuncs ...EncodeOptionFunc) error {
+ rctx := encoder.TakeRuntimeContext()
+ rctx.Option.Flag = 0
+ rctx.Option.Flag |= encoder.ContextOption
+ rctx.Option.Context = ctx
+
+ err := e.encodeWithOption(rctx, v, optFuncs...)
+
+ encoder.ReleaseRuntimeContext(rctx)
+ return err
+}
+
+func (e *Encoder) encodeWithOption(ctx *encoder.RuntimeContext, v interface{}, optFuncs ...EncodeOptionFunc) error {
+ if e.enabledHTMLEscape {
+ ctx.Option.Flag |= encoder.HTMLEscapeOption
+ }
+ ctx.Option.Flag |= encoder.NormalizeUTF8Option
+ ctx.Option.DebugOut = os.Stdout
+ for _, optFunc := range optFuncs {
+ optFunc(ctx.Option)
+ }
+ var (
+ buf []byte
+ err error
+ )
+ if e.enabledIndent {
+ buf, err = encodeIndent(ctx, v, e.prefix, e.indentStr)
+ } else {
+ buf, err = encode(ctx, v)
+ }
+ if err != nil {
+ return err
+ }
+ if e.enabledIndent {
+ buf = buf[:len(buf)-2]
+ } else {
+ buf = buf[:len(buf)-1]
+ }
+ buf = append(buf, '\n')
+ if _, err := e.w.Write(buf); err != nil {
+ return err
+ }
+ return nil
+}
+
+// SetEscapeHTML specifies whether problematic HTML characters should be escaped inside JSON quoted strings.
+// The default behavior is to escape &, <, and > to \u0026, \u003c, and \u003e to avoid certain safety problems that can arise when embedding JSON in HTML.
+//
+// In non-HTML settings where the escaping interferes with the readability of the output, SetEscapeHTML(false) disables this behavior.
+func (e *Encoder) SetEscapeHTML(on bool) {
+ e.enabledHTMLEscape = on
+}
+
+// SetIndent instructs the encoder to format each subsequent encoded value as if indented by the package-level function Indent(dst, src, prefix, indent).
+// Calling SetIndent("", "") disables indentation.
+func (e *Encoder) SetIndent(prefix, indent string) {
+ if prefix == "" && indent == "" {
+ e.enabledIndent = false
+ return
+ }
+ e.prefix = prefix
+ e.indentStr = indent
+ e.enabledIndent = true
+}
+
+func marshalContext(ctx context.Context, v interface{}, optFuncs ...EncodeOptionFunc) ([]byte, error) {
+ rctx := encoder.TakeRuntimeContext()
+ rctx.Option.Flag = 0
+ rctx.Option.Flag = encoder.HTMLEscapeOption | encoder.NormalizeUTF8Option | encoder.ContextOption
+ rctx.Option.Context = ctx
+ for _, optFunc := range optFuncs {
+ optFunc(rctx.Option)
+ }
+
+ buf, err := encode(rctx, v)
+ if err != nil {
+ encoder.ReleaseRuntimeContext(rctx)
+ return nil, err
+ }
+
+ // this line exists to escape call of `runtime.makeslicecopy` .
+ // if use `make([]byte, len(buf)-1)` and `copy(copied, buf)`,
+ // dst buffer size and src buffer size are differrent.
+ // in this case, compiler uses `runtime.makeslicecopy`, but it is slow.
+ buf = buf[:len(buf)-1]
+ copied := make([]byte, len(buf))
+ copy(copied, buf)
+
+ encoder.ReleaseRuntimeContext(rctx)
+ return copied, nil
+}
+
+func marshal(v interface{}, optFuncs ...EncodeOptionFunc) ([]byte, error) {
+ ctx := encoder.TakeRuntimeContext()
+
+ ctx.Option.Flag = 0
+ ctx.Option.Flag |= (encoder.HTMLEscapeOption | encoder.NormalizeUTF8Option)
+ for _, optFunc := range optFuncs {
+ optFunc(ctx.Option)
+ }
+
+ buf, err := encode(ctx, v)
+ if err != nil {
+ encoder.ReleaseRuntimeContext(ctx)
+ return nil, err
+ }
+
+ // this line exists to escape call of `runtime.makeslicecopy` .
+ // if use `make([]byte, len(buf)-1)` and `copy(copied, buf)`,
+ // dst buffer size and src buffer size are differrent.
+ // in this case, compiler uses `runtime.makeslicecopy`, but it is slow.
+ buf = buf[:len(buf)-1]
+ copied := make([]byte, len(buf))
+ copy(copied, buf)
+
+ encoder.ReleaseRuntimeContext(ctx)
+ return copied, nil
+}
+
+func marshalNoEscape(v interface{}) ([]byte, error) {
+ ctx := encoder.TakeRuntimeContext()
+
+ ctx.Option.Flag = 0
+ ctx.Option.Flag |= (encoder.HTMLEscapeOption | encoder.NormalizeUTF8Option)
+
+ buf, err := encodeNoEscape(ctx, v)
+ if err != nil {
+ encoder.ReleaseRuntimeContext(ctx)
+ return nil, err
+ }
+
+ // this line exists to escape call of `runtime.makeslicecopy` .
+ // if use `make([]byte, len(buf)-1)` and `copy(copied, buf)`,
+ // dst buffer size and src buffer size are differrent.
+ // in this case, compiler uses `runtime.makeslicecopy`, but it is slow.
+ buf = buf[:len(buf)-1]
+ copied := make([]byte, len(buf))
+ copy(copied, buf)
+
+ encoder.ReleaseRuntimeContext(ctx)
+ return copied, nil
+}
+
+func marshalIndent(v interface{}, prefix, indent string, optFuncs ...EncodeOptionFunc) ([]byte, error) {
+ ctx := encoder.TakeRuntimeContext()
+
+ ctx.Option.Flag = 0
+ ctx.Option.Flag |= (encoder.HTMLEscapeOption | encoder.NormalizeUTF8Option | encoder.IndentOption)
+ for _, optFunc := range optFuncs {
+ optFunc(ctx.Option)
+ }
+
+ buf, err := encodeIndent(ctx, v, prefix, indent)
+ if err != nil {
+ encoder.ReleaseRuntimeContext(ctx)
+ return nil, err
+ }
+
+ buf = buf[:len(buf)-2]
+ copied := make([]byte, len(buf))
+ copy(copied, buf)
+
+ encoder.ReleaseRuntimeContext(ctx)
+ return copied, nil
+}
+
+func encode(ctx *encoder.RuntimeContext, v interface{}) ([]byte, error) {
+ b := ctx.Buf[:0]
+ if v == nil {
+ b = encoder.AppendNull(ctx, b)
+ b = encoder.AppendComma(ctx, b)
+ return b, nil
+ }
+ header := (*emptyInterface)(unsafe.Pointer(&v))
+ typ := header.typ
+
+ typeptr := uintptr(unsafe.Pointer(typ))
+ codeSet, err := encoder.CompileToGetCodeSet(ctx, typeptr)
+ if err != nil {
+ return nil, err
+ }
+
+ p := uintptr(header.ptr)
+ ctx.Init(p, codeSet.CodeLength)
+ ctx.KeepRefs = append(ctx.KeepRefs, header.ptr)
+
+ buf, err := encodeRunCode(ctx, b, codeSet)
+ if err != nil {
+ return nil, err
+ }
+ ctx.Buf = buf
+ return buf, nil
+}
+
+func encodeNoEscape(ctx *encoder.RuntimeContext, v interface{}) ([]byte, error) {
+ b := ctx.Buf[:0]
+ if v == nil {
+ b = encoder.AppendNull(ctx, b)
+ b = encoder.AppendComma(ctx, b)
+ return b, nil
+ }
+ header := (*emptyInterface)(unsafe.Pointer(&v))
+ typ := header.typ
+
+ typeptr := uintptr(unsafe.Pointer(typ))
+ codeSet, err := encoder.CompileToGetCodeSet(ctx, typeptr)
+ if err != nil {
+ return nil, err
+ }
+
+ p := uintptr(header.ptr)
+ ctx.Init(p, codeSet.CodeLength)
+ buf, err := encodeRunCode(ctx, b, codeSet)
+ if err != nil {
+ return nil, err
+ }
+
+ ctx.Buf = buf
+ return buf, nil
+}
+
+func encodeIndent(ctx *encoder.RuntimeContext, v interface{}, prefix, indent string) ([]byte, error) {
+ b := ctx.Buf[:0]
+ if v == nil {
+ b = encoder.AppendNull(ctx, b)
+ b = encoder.AppendCommaIndent(ctx, b)
+ return b, nil
+ }
+ header := (*emptyInterface)(unsafe.Pointer(&v))
+ typ := header.typ
+
+ typeptr := uintptr(unsafe.Pointer(typ))
+ codeSet, err := encoder.CompileToGetCodeSet(ctx, typeptr)
+ if err != nil {
+ return nil, err
+ }
+
+ p := uintptr(header.ptr)
+ ctx.Init(p, codeSet.CodeLength)
+ buf, err := encodeRunIndentCode(ctx, b, codeSet, prefix, indent)
+
+ ctx.KeepRefs = append(ctx.KeepRefs, header.ptr)
+
+ if err != nil {
+ return nil, err
+ }
+
+ ctx.Buf = buf
+ return buf, nil
+}
+
+func encodeRunCode(ctx *encoder.RuntimeContext, b []byte, codeSet *encoder.OpcodeSet) ([]byte, error) {
+ if (ctx.Option.Flag & encoder.DebugOption) != 0 {
+ if (ctx.Option.Flag & encoder.ColorizeOption) != 0 {
+ return vm_color.DebugRun(ctx, b, codeSet)
+ }
+ return vm.DebugRun(ctx, b, codeSet)
+ }
+ if (ctx.Option.Flag & encoder.ColorizeOption) != 0 {
+ return vm_color.Run(ctx, b, codeSet)
+ }
+ return vm.Run(ctx, b, codeSet)
+}
+
+func encodeRunIndentCode(ctx *encoder.RuntimeContext, b []byte, codeSet *encoder.OpcodeSet, prefix, indent string) ([]byte, error) {
+ ctx.Prefix = []byte(prefix)
+ ctx.IndentStr = []byte(indent)
+ if (ctx.Option.Flag & encoder.DebugOption) != 0 {
+ if (ctx.Option.Flag & encoder.ColorizeOption) != 0 {
+ return vm_color_indent.DebugRun(ctx, b, codeSet)
+ }
+ return vm_indent.DebugRun(ctx, b, codeSet)
+ }
+ if (ctx.Option.Flag & encoder.ColorizeOption) != 0 {
+ return vm_color_indent.Run(ctx, b, codeSet)
+ }
+ return vm_indent.Run(ctx, b, codeSet)
+}