sqlite3/json.go

package sqlite3

import (
	"encoding/json"
	"strconv"
	"time"
	"unsafe"
)

// JSON returns:
// a [json.Marshaler] that can be used as an argument to
// [database/sql.DB.Exec] and similar methods to
// store value as JSON; and
// a [database/sql.Scanner] that can be used as an argument to
// [database/sql.Row.Scan] and similar methods to
// decode JSON into value.
func JSON(value any) any {
	return jsonValue{value}
}

type jsonValue struct{ any }

func (j jsonValue) MarshalJSON() ([]byte, error) {
	return json.Marshal(j.any)
}

func (j jsonValue) UnmarshalJSON(data []byte) error {
	return json.Unmarshal(data, j.any)
}

func (j jsonValue) Scan(value any) error {
	var mem [40]byte
	buf := mem[:0]

	switch v := value.(type) {
	case []byte:
		buf = v
	case string:
		buf = unsafe.Slice(unsafe.StringData(v), len(v))
	case int64:
		buf = strconv.AppendInt(nil, v, 10)
	case float64:
		buf = strconv.AppendFloat(nil, v, 'g', -1, 64)
	case time.Time:
		buf = append(buf, '"')
		buf = v.AppendFormat(buf, time.RFC3339Nano)
		buf = append(buf, '"')
	case nil:
		buf = append(buf, "null"...)
	}

	return j.UnmarshalJSON(buf)
}