error.go

package quirk

import (
	"fmt"
)

// Error is a general error that isn't super specific. Just used for when
// there needs to be more context relating to an error.
type Error struct {
	ExtErr   error
	Msg      string
	File     string
	Function string
}

func (e *Error) Error() string {
	if e.ExtErr != nil {
		return fmt.Sprintf("%s:%s: msg[%s] external_err[%v]",
			e.Function, e.File, e.Msg, e.ExtErr,
		)
	}

	return fmt.Sprintf("%s:%s: msg[%s]",
		e.Function, e.File, e.Msg,
	)
}

// Unwrap is used to return any external errors.
// This function was implemented for Go 1.13.
func (e *Error) Unwrap() error {
	if e.ExtErr != nil {
		return e.ExtErr
	}

	return nil
}

// QueryError is used for functions in the query.go file.
type QueryError struct {
	ExtErr   error
	Msg      string
	Function string
	Query    string
}

func (e *QueryError) Error() (res string) {
	switch {
	case e.ExtErr != nil:
		res = fmt.Sprintf("%s:query.go: Query[%s] external_err[%v]",
			e.Function, e.Query, e.ExtErr,
		)
	case e.Msg != "":
		res = fmt.Sprintf("%s:query.go: Msg[%s]",
			e.Function, e.Msg,
		)
	default:
		res = fmt.Sprintf("%s:query.go: Query[%s]",
			e.Function, e.Query,
		)
	}

	return
}

// Unwrap is used to return any external errors.
// This function was implemented for Go 1.13.
func (e *QueryError) Unwrap() error {
	if e.ExtErr != nil {
		return e.ExtErr
	}

	return nil
}

// TransactionError is for when a transaction fails during a mutation.
type TransactionError struct {
	ExtErr   error
	Msg      string
	Function string
	RDF      string
}

func (e *TransactionError) Error() string {
	if e.ExtErr != nil {
		return fmt.Sprintf("%s:mutate_single.go: Msg[%s] RDF[%s] external_err[%v]",
			e.Function, e.Msg, e.RDF, e.ExtErr,
		)
	}

	return fmt.Sprintf("%s:mutate_single.go: Msg[%s] RDF[%s]",
		e.Function, e.Msg, e.RDF,
	)
}

// Unwrap is used to return any external errors.
// This function was implemented for Go 1.13.
func (e *TransactionError) Unwrap() error {
	if e.ExtErr != nil {
		return e.ExtErr
	}

	return nil
}