github.com/zerosnake0/jzon@v0.0.9-0.20230801092939-1b135cb83f7f/streamer.go

package jzon

import (
	"io"
)

// Streamer is a chained class for encoding object to json
type Streamer struct {
	cfg *EncoderConfig

	writer io.Writer
	buffer []byte

	Error error
	poped bool

	// TODO: 1. type of context?
	// TODO: 2. should context be reset as well?
	Context interface{} // custom stream context

	// runtime
	escapeHTML bool
	safeSet    []string
	// prefix  string
	// indent  string
}

// NewStreamer returns a new streamer
func NewStreamer() *Streamer {
	return DefaultEncoderConfig.NewStreamer()
}

// Release the streamer, the streamer should not be used after call
func (s *Streamer) Release() {
	s.cfg.returnStreamer(s)
}

func (s *Streamer) reset() {
	s.writer = nil
	s.Error = nil
	s.poped = false
	s.buffer = s.buffer[:0]
	s.Context = nil
}

// Reset resets the streamer with a new writer
func (s *Streamer) Reset(w io.Writer) {
	s.reset()
	s.writer = w
}

// EscapeHTML set if the string should be html-escaped
func (s *Streamer) EscapeHTML(on bool) {
	s.escapeHTML = on
	if on {
		s.safeSet = htmlSafeSet[:]
	} else {
		s.safeSet = safeSet[:]
	}
}

// func (s *Streamer) SetIndent(prefix, indent string) {
// 	s.prefix = prefix
// 	s.indent = indent
// }

// Flush flushes from buffer to the writer
func (s *Streamer) Flush() error {
	if s.Error != nil {
		return s.Error
	}
	if s.writer == nil {
		return ErrNoAttachedWriter
	}
	l := len(s.buffer)
	// see comment of io.Writer
	n, err := s.writer.Write(s.buffer)
	if n < l {
		copy(s.buffer, s.buffer[n:])
		s.buffer = s.buffer[:l-n]
	} else {
		s.buffer = s.buffer[:0]
	}
	return err
}

func (s *Streamer) onVal() {
	if s.poped {
		s.buffer = append(s.buffer, ',')
	} else {
		s.poped = true
	}
}

// RawString writes a raw object (in string)
func (s *Streamer) RawString(raw string) *Streamer {
	if s.Error != nil {
		return s
	}
	s.onVal()
	s.buffer = append(s.buffer, raw...)
	return s
}

// Raw writes a raw object (in byte slice)
func (s *Streamer) Raw(raw []byte) *Streamer {
	if s.Error != nil {
		return s
	}
	s.onVal()
	s.buffer = append(s.buffer, raw...)
	return s
}

func (s *Streamer) null() {
	s.onVal()
	s.buffer = append(s.buffer, 'n', 'u', 'l', 'l')
}

// Null writes a `null`
func (s *Streamer) Null() *Streamer {
	if s.Error != nil {
		return s
	}
	s.null()
	return s
}

// True writes a `true`
func (s *Streamer) True() *Streamer {
	if s.Error != nil {
		return s
	}
	s.onVal()
	s.buffer = append(s.buffer, 't', 'r', 'u', 'e')
	return s
}

// False writes a `false`
func (s *Streamer) False() *Streamer {
	if s.Error != nil {
		return s
	}
	s.onVal()
	s.buffer = append(s.buffer, 'f', 'a', 'l', 's', 'e')
	return s
}

// Bool writes a boolean value
func (s *Streamer) Bool(b bool) *Streamer {
	if b {
		return s.True()
	}
	return s.False()
}

// ObjectStart starts to write an object
func (s *Streamer) ObjectStart() *Streamer {
	if s.Error != nil {
		return s
	}
	s.onVal()
	s.buffer = append(s.buffer, '{')
	s.poped = false
	return s
}

// Field writes an object field
func (s *Streamer) Field(field string) *Streamer {
	if s.Error != nil {
		return s
	}
	s.onVal()
	s.buffer = encodeString(s.buffer, field, s.safeSet)
	s.buffer = append(s.buffer, ':')
	s.poped = false
	return s
}

// RawField writes an object field (in raw byte slice)
func (s *Streamer) RawField(b []byte) *Streamer {
	if s.Error != nil {
		return s
	}
	s.onVal()
	s.buffer = append(s.buffer, b...)
	s.buffer = append(s.buffer, ':')
	s.poped = false
	return s
}

// ObjectEnd ends the object writing
func (s *Streamer) ObjectEnd() *Streamer {
	if s.Error != nil {
		return s
	}
	s.buffer = append(s.buffer, '}')
	s.poped = true
	return s
}

// ArrayStart starts to write an array
func (s *Streamer) ArrayStart() *Streamer {
	if s.Error != nil {
		return s
	}
	s.onVal()
	s.buffer = append(s.buffer, '[')
	s.poped = false
	return s
}

// ArrayEnd ends the array writing
func (s *Streamer) ArrayEnd() *Streamer {
	if s.Error != nil {
		return s
	}
	s.buffer = append(s.buffer, ']')
	s.poped = true
	return s
}