AdGuardDNS/internal/dnsserver/forward/error.go

package forward

import (
	"fmt"

	"github.com/AdguardTeam/golibs/errors"
)

// Common Errors

// Error is the forwarding error.
type Error struct {
	Err      error
	Main     Upstream
	Fallback Upstream
}

// type check
var _ error = (*Error)(nil)

// Error implements the error interface for *Error.
func (err *Error) Error() (msg string) {
	if err.Fallback == nil {
		return fmt.Sprintf("forwarding to %s: %s", err.Main, err.Err)
	} else if err.Main == nil {
		return fmt.Sprintf("forwarding to fallback %s: %s", err.Fallback, err.Err)
	}

	return fmt.Sprintf(
		"forwarding to %s with fallback %s: %s",
		err.Main,
		err.Fallback,
		err.Err,
	)
}

// type check
var _ errors.Wrapper = (*Error)(nil)

// Unwrap implements the errors.Wrapper interface for *Error.
func (err *Error) Unwrap() (unwrapped error) {
	return err.Err
}

// annotate is a deferrable helper for forwarding errors.
func annotate(err error, ups, fallbackUps Upstream) (wrapped error) {
	if err == nil {
		return nil
	}

	return &Error{
		Err:      err,
		Main:     ups,
		Fallback: fallbackUps,
	}
}