feat(errorx): add withstack helper (#230)

errorsx/errors.go

@@ -28,6 +28,16 @@ func Cause(err error) error {
  return err
 }
 
+// WithStack mirror pkg/errors.WithStack but does not wrap existing stack
+// traces.
+func WithStack(err error) error {
+ if _, ok := err.(StackTracer); ok {
+ return err
+ }
+
+ return errors.WithStack(err)
+}
+
 // StatusCodeCarrier can be implemented by an error to support setting status codes in the error itself.
 type StatusCodeCarrier interface {
  // StatusCode returns the status code of this error.

errorsx/errors_test.go

@@ -0,0 +1,18 @@
+package errorsx
+
+import (
+ "testing"
+
+ "github.com/pkg/errors"
+ "github.com/stretchr/testify/assert"
+)
+
+func TestWithStack(t *testing.T) {
+ t.Run("case=wrap", func(t *testing.T) {
+ orig := errors.New("hi")
+ wrap := WithStack(orig)
+
+ assert.EqualValues(t, orig.(StackTracer).StackTrace(), wrap.(StackTracer).StackTrace())
+ assert.EqualValues(t, orig.(StackTracer).StackTrace(), WithStack(wrap).(StackTracer).StackTrace())
+ })
+}