123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169 |
- // Package errwrap implements methods to formalize error wrapping in Go.
- //
- // All of the top-level functions that take an `error` are built to be able
- // to take any error, not just wrapped errors. This allows you to use errwrap
- // without having to type-check and type-cast everywhere.
- package errwrap
- import (
- "errors"
- "reflect"
- "strings"
- )
- // WalkFunc is the callback called for Walk.
- type WalkFunc func(error)
- // Wrapper is an interface that can be implemented by custom types to
- // have all the Contains, Get, etc. functions in errwrap work.
- //
- // When Walk reaches a Wrapper, it will call the callback for every
- // wrapped error in addition to the wrapper itself. Since all the top-level
- // functions in errwrap use Walk, this means that all those functions work
- // with your custom type.
- type Wrapper interface {
- WrappedErrors() []error
- }
- // Wrap defines that outer wraps inner, returning an error type that
- // can be cleanly used with the other methods in this package, such as
- // Contains, GetAll, etc.
- //
- // This function won't modify the error message at all (the outer message
- // will be used).
- func Wrap(outer, inner error) error {
- return &wrappedError{
- Outer: outer,
- Inner: inner,
- }
- }
- // Wrapf wraps an error with a formatting message. This is similar to using
- // `fmt.Errorf` to wrap an error. If you're using `fmt.Errorf` to wrap
- // errors, you should replace it with this.
- //
- // format is the format of the error message. The string '{{err}}' will
- // be replaced with the original error message.
- func Wrapf(format string, err error) error {
- outerMsg := "<nil>"
- if err != nil {
- outerMsg = err.Error()
- }
- outer := errors.New(strings.Replace(
- format, "{{err}}", outerMsg, -1))
- return Wrap(outer, err)
- }
- // Contains checks if the given error contains an error with the
- // message msg. If err is not a wrapped error, this will always return
- // false unless the error itself happens to match this msg.
- func Contains(err error, msg string) bool {
- return len(GetAll(err, msg)) > 0
- }
- // ContainsType checks if the given error contains an error with
- // the same concrete type as v. If err is not a wrapped error, this will
- // check the err itself.
- func ContainsType(err error, v interface{}) bool {
- return len(GetAllType(err, v)) > 0
- }
- // Get is the same as GetAll but returns the deepest matching error.
- func Get(err error, msg string) error {
- es := GetAll(err, msg)
- if len(es) > 0 {
- return es[len(es)-1]
- }
- return nil
- }
- // GetType is the same as GetAllType but returns the deepest matching error.
- func GetType(err error, v interface{}) error {
- es := GetAllType(err, v)
- if len(es) > 0 {
- return es[len(es)-1]
- }
- return nil
- }
- // GetAll gets all the errors that might be wrapped in err with the
- // given message. The order of the errors is such that the outermost
- // matching error (the most recent wrap) is index zero, and so on.
- func GetAll(err error, msg string) []error {
- var result []error
- Walk(err, func(err error) {
- if err.Error() == msg {
- result = append(result, err)
- }
- })
- return result
- }
- // GetAllType gets all the errors that are the same type as v.
- //
- // The order of the return value is the same as described in GetAll.
- func GetAllType(err error, v interface{}) []error {
- var result []error
- var search string
- if v != nil {
- search = reflect.TypeOf(v).String()
- }
- Walk(err, func(err error) {
- var needle string
- if err != nil {
- needle = reflect.TypeOf(err).String()
- }
- if needle == search {
- result = append(result, err)
- }
- })
- return result
- }
- // Walk walks all the wrapped errors in err and calls the callback. If
- // err isn't a wrapped error, this will be called once for err. If err
- // is a wrapped error, the callback will be called for both the wrapper
- // that implements error as well as the wrapped error itself.
- func Walk(err error, cb WalkFunc) {
- if err == nil {
- return
- }
- switch e := err.(type) {
- case *wrappedError:
- cb(e.Outer)
- Walk(e.Inner, cb)
- case Wrapper:
- cb(err)
- for _, err := range e.WrappedErrors() {
- Walk(err, cb)
- }
- default:
- cb(err)
- }
- }
- // wrappedError is an implementation of error that has both the
- // outer and inner errors.
- type wrappedError struct {
- Outer error
- Inner error
- }
- func (w *wrappedError) Error() string {
- return w.Outer.Error()
- }
- func (w *wrappedError) WrappedErrors() []error {
- return []error{w.Outer, w.Inner}
- }
|