New Bytes As String Extension

json.go

package utils

import (
	"unsafe"

	jsoniter "github.com/json-iterator/go"
	"github.com/modern-go/reflect2"
)

var JSON = jsoniter.ConfigCompatibleWithStandardLibrary

func init() {
	JSON.RegisterExtension(&BytesAsStringExtension{})
}

var bytesType = reflect2.TypeOfPtr((*[]byte)(nil)).Elem()

type BytesAsStringExtension struct {
	jsoniter.DummyExtension
}

func (extension *BytesAsStringExtension) CreateEncoder(typ reflect2.Type) jsoniter.ValEncoder {
	if typ == bytesType {
		return &bytesAsStringCodec{}
	}
	return nil
}

func (extension *BytesAsStringExtension) CreateDecoder(typ reflect2.Type) jsoniter.ValDecoder {
	if typ == bytesType {
		return &bytesAsStringCodec{}
	}
	return nil
}

type bytesAsStringCodec struct{}

func (codec *bytesAsStringCodec) IsEmpty(ptr unsafe.Pointer) bool {
	return len(*((*[]byte)(ptr))) == 0
}

func (codec *bytesAsStringCodec) Encode(ptr unsafe.Pointer, stream *jsoniter.Stream) {
	stream.WriteString(string(*(*[]byte)(ptr)))
}

func (codec *bytesAsStringCodec) Decode(ptr unsafe.Pointer, iter *jsoniter.Iterator) {
	*(*[]byte)(ptr) = []byte(iter.ReadString())
}