kern/v4/lev0/result/result.go

package result

import (
	"fmt"
	"reflect"
	"runtime"

	mKh "gitp78su.ipnodns.ru/svi/kern/v4/lev0/helpers"
	mKs "gitp78su.ipnodns.ru/svi/kern/v4/lev0/kspec"
)

// Result — обёртка вокруг результата с возможной ошибкой.
//
// Может быть либо только полезное значение, либо только ошибка.
type Result[T any] struct {
	val     T     // Полезное значение
	isCheck bool  // Если проверено
	err     error // Ошибка
}

// NewOk -- возвращает успешный Result с значением.
func NewOk[T any](result T) mKs.IResult[T] {
	// Для некоторых типов нужна дополнительная проверка через reflect
	v := reflect.ValueOf(result)
	switch v.Kind() {
	case reflect.Pointer, reflect.Slice, reflect.Map, reflect.Chan, reflect.Func, reflect.Interface:
		mKh.Hassert(!v.IsNil(), "NewOk(): result==nil")
	}
	sf := &Result[T]{
		val: result,
	}
	runtime.SetFinalizer(sf, sf.destroy)
	return sf
}

// NewErr -- возвращает Result с ошибкой.
func NewErr[T any](err error) mKs.IResult[T] {
	mKh.Hassert(err != nil, "NewError(): err==nil")
	sf := &Result[T]{
		err: err,
	}
	runtime.SetFinalizer(sf, sf.destroy)
	return sf
}

// WrapErr -- оборачивает существующий Result с ошибкой с новой ошибкой.
func WrapErr[T any](res mKs.IResult[T], err error) mKs.IResult[T] {
	mKh.Hassert(res.IsErr(), "WrapErr(): result not have error")
	mKh.Hassert(err != nil, "WrapErr(): err==nil")
	err0 := res.Err() // Чтобы не было паники при финализации и обернуть ошибку
	err = fmt.Errorf("%v, err=\n\t%w", err0, err)
	sf := NewErr[T](err)
	return sf
}

func (sf *Result[T]) destroy(res *Result[T]) {
	if !res.isCheck {
		mKh.Hassert(false, "Result[T].destroy(): err not checked")
	}
}

// IsOk -- возвращает true, если Result содержит значение.
func (sf *Result[T]) IsOk() bool {
	sf.isCheck = true
	return sf.err == nil
}

// IsErr -- возвращает true, если Result содержит ошибку.
func (sf *Result[T]) IsErr() bool {
	sf.isCheck = true
	return sf.err != nil
}

// Ok -- возвращает значение, если оно есть, иначе паникует.
func (sf *Result[T]) Ok() T {
	mKh.Hassert(sf.err == nil, "Result[T].Ok(): err(%v)!=nil", sf.err.Error())
	sf.isCheck = true
	return sf.val
}

// OkOr -- возвращает значение, если оно есть, или значение по умолчанию.
func (sf *Result[T]) OkOr(defaultValue T) T {
	sf.isCheck = true
	if sf.err != nil {
		sf.val = defaultValue
		sf.err = nil
	}
	return sf.Ok()
}

// OkOrFn -- возвращает значение, если оно есть, или результат выполнения функции.
func (sf *Result[T]) OkOrFn(fn func() T) T {
	mKh.Hassert(fn != nil, "Result[T].OkOrFn(): fn==nil")
	sf.isCheck = true
	if sf.err != nil {
		sf.val = fn()
		sf.err = nil
	}
	return sf.Ok()
}

// Err -- возвращает ошибку, если она есть.
func (sf *Result[T]) Err() error {
	mKh.Hassert(sf.err != nil, "Result[T].Err(): err==nil")
	sf.isCheck = true
	return sf.err
}

// Error -- возвращает строковое представление ошибки, если она есть.
func (sf *Result[T]) Error() string {
	mKh.Hassert(sf.err != nil, "Result[T].Error(): err==nil")
	sf.isCheck = true
	return sf.err.Error()
}

// Hassert -- проверяет, что нет ошибки (с паникой).
func (sf *Result[T]) Hassert(msgFormat string, args ...any) T {
	strErr := ""
	if !sf.isCheck {
		strErr = fmt.Sprintf(", err=\n\t%v", sf.err)
	}
	msg := fmt.Sprintf(msgFormat+strErr, args...)
	mKh.Hassert(sf.err == nil, msg)
	sf.isCheck = true
	return sf.val
}

// Assert -- проверяет, что нет ошибки (с паникой только на локальном стенде).
func (sf *Result[T]) Assert(msgFormat string, args ...any) T {
	strErr := ""
	if !sf.isCheck {
		strErr = fmt.Sprintf(", err=\n\t%v", sf.err)
	}
	msg := fmt.Sprintf(msgFormat+strErr, args...)
	mKh.Assert(sf.err == nil, msg)
	sf.isCheck = true
	return sf.val
}