nullfloat64.go

package nulltype

import (
	"bytes"
	"database/sql"
	"database/sql/driver"
	"fmt"
	"strconv"
	"unsafe"

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

/* SQL and JSon null.Float64 */

type Float64 sql.NullFloat64

func NewFloat64(f float64) Float64 {
	nf := Float64{}
	nf.Valid = true
	nf.Float64 = f
	return nf
}

func (nf *Float64) Encode(ptr unsafe.Pointer, stream *jsoniter.Stream) {
	val := (*Float64)(ptr)

	if val.Valid {
		stream.WriteVal(val.Float64)
	} else {
		stream.WriteVal(nil)
	}
}

// IsEmpty detect whether primitive.ObjectID is empty.
func (nf *Float64) IsEmpty(ptr unsafe.Pointer) bool {
	val := (*Float64)(ptr)
	return !val.Valid
}

func (nf *Float64) UnmarshalCSV(b string) error {
	var err error
	nf.Float64, err = strconv.ParseFloat(b, 64)
	return err
}

// MarshalCSV marshals CSV
func (nf Float64) MarshalCSV() (string, error) {
	if nf.Valid {
		return fmt.Sprintf("%f", nf.Float64), nil
	}
	return "", nil
}

func (nf *Float64) UnmarshalJSON(b []byte) error {
	var i float64
	if err := json.Unmarshal(b, &i); err != nil {
		return err
	}
	if bytes.Compare(b, []byte("null")) == 0 {
		nf.Valid = false
		return nil
	}
	nf.Float64 = i
	nf.Valid = true
	return nil
}

func (nf Float64) MarshalJSON() ([]byte, error) {
	if nf.Valid {
		return json.Marshal(nf.Float64)
	}
	return json.Marshal(nil)
}

func (nf *Float64) Scan(value any) error {
	if value == nil {
		nf.Float64, nf.Valid = 0, false
		return nil
	}
	nf.Valid = true
	return convertAssignRows(&nf.Float64, value)
}

func (nf Float64) Value() (driver.Value, error) {
	if !nf.Valid {
		return nil, nil
	}
	return nf.Float64, nil
}