nilable.go

package ctypes

import "encoding/json"

type Nilable[T any] struct {
	val   *T
	isSet bool
}

func NewNilable[T any](v T) Nilable[T] {
	return Nilable[T]{
		val:   &v,
		isSet: true,
	}
}

func (n *Nilable[T]) UnmarshalJSON(in []byte) error {
	var v T
	if err := json.Unmarshal(in, &v); err != nil {
		return err
	}

	n.val = &v
	n.isSet = true

	return nil
}

func (n *Nilable[T]) MarshalJSON() ([]byte, error) {
	if n.val == nil {
		return nil, nil
	}

	return json.Marshal(n.val)
}

func (n *Nilable[T]) Get() *T {
	return n.val
}

func (n *Nilable[T]) GetOrDefault() T {
	if n.val == nil {
		var v T
		return v
	}

	return *n.val
}

func (n *Nilable[T]) Check() (*T, bool) {
	return n.val, n.isSet
}

func (n *Nilable[T]) Set(v T) {
	n.val = &v
	n.isSet = true
}

func (n *Nilable[T]) Unset() {
	n.val = nil
	n.isSet = false
}