WrapNil

Author: nikandfor

errors2.go

@@ -32,16 +32,24 @@ func Wrap(err error, format string, args ...any) error {
 		panic("wrapping nil error")
 	}
 
-	return &wrap{
+	return wrap{
 		err: err,
 		msg: fmt.Sprintf(format, args...),
 	}
 }
 
-func (w *wrap) Error() string {
+func WrapNil(err error, format string, args ...any) error {
+	if err == nil {
+		return nil
+	}
+
+	return Wrap(err, format, args...)
+}
+
+func (w wrap) Error() string {
 	return fmt.Sprintf("%v: %v", w.msg, w.err)
 }
 
-func (w *wrap) Unwrap() error {
+func (w wrap) Unwrap() error {
 	return w.err
 }

errors2_test.go

@@ -80,6 +80,25 @@ func TestErrors(tb *testing.T) {
 
 		_ = errors.Wrap(nil, "wrap nil")
 	}()
+
+	if e1, e2 := errors.Wrap(err1, "wrap"), errors.Wrap(err1, "wrap"); e1 != e2 {
+		tb.Errorf("expected the same result\n%v\n%v", e1, e2)
+	}
+
+	if e1, e2 := errors.Wrap(err1, "wrap"), errors.WrapNil(err1, "wrap"); e1 != e2 {
+		tb.Errorf("expected the same result\n%v\n%v", e1, e2)
+	}
+
+	func() {
+		defer func() {
+			p := recover()
+			if p != nil {
+				tb.Errorf("panic: %v", p)
+			}
+		}()
+
+		_ = errors.WrapNil(nil, "wrap nil")
+	}()
 }
 
 func (e *testErr) Error() string { return fmt.Sprintf("testErr(%v)", e.val) }