decode_string.go

package gojay

import (
	"unsafe"
)

// DecodeString reads the next JSON-encoded value from the decoder's input (io.Reader) and stores it in the string pointed to by v.
//
// See the documentation for Unmarshal for details about the conversion of JSON into a Go value.
func (dec *Decoder) DecodeString(v *string) error {
	if dec.isPooled == 1 {
		panic(InvalidUsagePooledDecoderError("Invalid usage of pooled decoder"))
	}
	return dec.decodeString(v)
}
func (dec *Decoder) decodeString(v *string) error {
	for ; dec.cursor < dec.length || dec.read(); dec.cursor++ {
		switch dec.data[dec.cursor] {
		case ' ', '\n', '\t', '\r', ',':
			// is string
			continue
		case '"':
			dec.cursor++
			start, end, err := dec.getString()
			if err != nil {
				return err
			}
			// we do minus one to remove the last quote
			d := dec.data[start : end-1]
			*v = *(*string)(unsafe.Pointer(&d))
			dec.cursor = end
			return nil
		// is nil
		case 'n':
			dec.cursor++
			err := dec.assertNull()
			if err != nil {
				return err
			}
			return nil
		default:
			dec.err = dec.makeInvalidUnmarshalErr(v)
			err := dec.skipData()
			if err != nil {
				return err
			}
			return nil
		}
	}
	return nil
}

func (dec *Decoder) decodeStringNull(v **string) error {
	for ; dec.cursor < dec.length || dec.read(); dec.cursor++ {
		switch dec.data[dec.cursor] {
		case ' ', '\n', '\t', '\r', ',':
			// is string
			continue
		case '"':
			dec.cursor++
			start, end, err := dec.getString()

			if err != nil {
				return err
			}
			if *v == nil {
				*v = new(string)
			}
			// we do minus one to remove the last quote
			d := dec.data[start : end-1]
			**v = *(*string)(unsafe.Pointer(&d))
			dec.cursor = end
			return nil
		// is nil
		case 'n':
			dec.cursor++
			err := dec.assertNull()
			if err != nil {
				return err
			}
			return nil
		default:
			dec.err = dec.makeInvalidUnmarshalErr(v)
			err := dec.skipData()
			if err != nil {
				return err
			}
			return nil
		}
	}
	return nil
}

func (dec *Decoder) decodeStringBytesOrNull() (val []byte, isNull bool, err error) {
	for ; dec.cursor < dec.length || dec.read(); dec.cursor++ {
		switch dec.data[dec.cursor] {
		case ' ', '\n', '\t', '\r', ',':
			// is string
			continue
		case '"':
			dec.cursor++
			var start, end int
			if start, end, err = dec.getString(); err == nil {
				// it is wrong to return slice of json because we using ButBeuffer pool at decode.go:UnmarshalJSONObjectWithPool()
				// so we return the slice of that ByteBuffer, store it on caller's side,
				// then another thread will take that ByteBuffer from pool and write to it
				// whereas our thread will read that memory on encodeBuffer()
				// val = dec.data[start : end-1]

				val = make([]byte, end-start-1)
				// we do minus one to remove the last quote
				copy(val, dec.data[start:end-1])
				dec.cursor = end
			}
			return
		// is nil
		case 'n':
			dec.cursor++
			if err = dec.assertNull(); err == nil {
				isNull = true
			}
			return
		default:
			dec.err = dec.makeInvalidUnmarshalErr("")
			err = dec.skipData()
			return
		}
	}
	return
}

func (dec *Decoder) parseEscapedString() error {
	if dec.cursor >= dec.length && !dec.read() {
		return dec.raiseInvalidJSONErr(dec.cursor)
	}
	switch dec.data[dec.cursor] {
	case '"':
		dec.data[dec.cursor] = '"'
	case '\\':
		dec.data[dec.cursor] = '\\'
	case '/':
		dec.data[dec.cursor] = '/'
	case 'b':
		dec.data[dec.cursor] = '\b'
	case 'f':
		dec.data[dec.cursor] = '\f'
	case 'n':
		dec.data[dec.cursor] = '\n'
	case 'r':
		dec.data[dec.cursor] = '\r'
	case 't':
		dec.data[dec.cursor] = '\t'
	case 'u':
		start := dec.cursor
		dec.cursor++
		str, err := dec.parseUnicode()
		if err != nil {
			return err
		}
		diff := dec.cursor - start
		dec.data = append(append(dec.data[:start-1], str...), dec.data[dec.cursor:]...)
		dec.length = len(dec.data)
		dec.cursor += len(str) - diff - 1

		return nil
	default:
		return dec.raiseInvalidJSONErr(dec.cursor)
	}

	dec.data = append(dec.data[:dec.cursor-1], dec.data[dec.cursor:]...)
	dec.length--

	// Since we've lost a character, our dec.cursor offset is now
	// 1 past the escaped character which is precisely where we
	// want it.

	return nil
}

func (dec *Decoder) getString() (int, int, error) {
	// extract key
	var keyStart = dec.cursor
	// var str *Builder
	for dec.cursor < dec.length || dec.read() {
		switch dec.data[dec.cursor] {
		// string found
		case '"':
			dec.cursor = dec.cursor + 1
			return keyStart, dec.cursor, nil
		// slash found
		case '\\':
			dec.cursor = dec.cursor + 1
			err := dec.parseEscapedString()
			if err != nil {
				return 0, 0, err
			}
		default:
			dec.cursor = dec.cursor + 1
			continue
		}
	}
	return 0, 0, dec.raiseInvalidJSONErr(dec.cursor)
}

func (dec *Decoder) skipEscapedString() error {
	start := dec.cursor
	for ; dec.cursor < dec.length || dec.read(); dec.cursor++ {
		if dec.data[dec.cursor] != '\\' {
			d := dec.data[dec.cursor]
			dec.cursor = dec.cursor + 1
			nSlash := dec.cursor - start
			switch d {
			case '"':
				// nSlash must be odd
				if nSlash&1 != 1 {
					return dec.raiseInvalidJSONErr(dec.cursor)
				}
				return nil
			case 'u': // is unicode, we skip the following characters and place the cursor one one byte backward to avoid it breaking when returning to skipString
				if err := dec.skipString(); err != nil {
					return err
				}
				dec.cursor--
				return nil
			case 'n', 'r', 't', '/', 'f', 'b':
				return nil
			default:
				// nSlash must be even
				if nSlash&1 == 1 {
					return dec.raiseInvalidJSONErr(dec.cursor)
				}
				return nil
			}
		}
	}
	return dec.raiseInvalidJSONErr(dec.cursor)
}

func (dec *Decoder) skipString() error {
	for dec.cursor < dec.length || dec.read() {
		switch dec.data[dec.cursor] {
		// found the closing quote
		// let's return
		case '"':
			dec.cursor = dec.cursor + 1
			return nil
		// solidus found start parsing an escaped string
		case '\\':
			dec.cursor = dec.cursor + 1
			err := dec.skipEscapedString()
			if err != nil {
				return err
			}
		default:
			dec.cursor = dec.cursor + 1
			continue
		}
	}
	return dec.raiseInvalidJSONErr(len(dec.data) - 1)
}

// Add Values functions

// AddString decodes the JSON value within an object or an array to a *string.
// If next key is not a JSON string nor null, InvalidUnmarshalError will be returned.
func (dec *Decoder) AddString(v *string) error {
	return dec.String(v)
}

// AddStringNull decodes the JSON value within an object or an array to a *string.
// If next key is not a JSON string nor null, InvalidUnmarshalError will be returned.
// If a `null` is encountered, gojay does not change the value of the pointer.
func (dec *Decoder) AddStringNull(v **string) error {
	return dec.StringNull(v)
}

// String decodes the JSON value within an object or an array to a *string.
// If next key is not a JSON string nor null, InvalidUnmarshalError will be returned.
func (dec *Decoder) String(v *string) error {
	err := dec.decodeString(v)
	if err != nil {
		return err
	}
	dec.called |= 1
	return nil
}

// StringNull decodes the JSON value within an object or an array to a **string.
// If next key is not a JSON string nor null, InvalidUnmarshalError will be returned.
// If a `null` is encountered, gojay does not change the value of the pointer.
func (dec *Decoder) StringNull(v **string) error {
	err := dec.decodeStringNull(v)
	if err != nil {
		return err
	}
	dec.called |= 1
	return nil
}

// StringBytesOrNull is alloc-free version of StringNull(). Returns a string as bytes
func (dec *Decoder) StringBytesOrNull() (val []byte, isNull bool, err error) {
	if val, isNull, err = dec.decodeStringBytesOrNull(); err == nil {
		if err = dec.err; err == nil {
			dec.called |= 1
		}
	}
	return
}