feat: allow to get all tags in wrapped errors

Southclaws · Sep 15, 2023 · 568e1b7 · 568e1b7
1 parent f02308e
commit 568e1b7
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/ftag/ftag.go b/ftag/ftag.go
@@ -57,6 +57,23 @@ func Get(err error) Kind {
 	return Internal
 }
 
+func GetAll(err error) []Kind {
+	if err == nil {
+		return nil
+	}
+
+	var ks []Kind
+	for err != nil {
+		if f, ok := err.(*withKind); ok {
+			ks = append(ks, f.tag)
+		}
+
+		err = errors.Unwrap(err)
+	}
+
+	return ks
+}
+
 // Common kinds of error:
 
 const (

diff --git a/tests/ftag_test.go b/tests/ftag_test.go
@@ -26,3 +26,12 @@ func TestWrapWithKindChanging(t *testing.T) {
 
 	assert.Equal(t, ftag.NotFound, out, "Should always pick the most recent kind from an error chain.")
 }
+
+func TestMultipleWrappedKind(t *testing.T) {
+	err := ftag.Wrap(errors.New("a problem"), ftag.Internal)
+	err = ftag.Wrap(err, ftag.InvalidArgument)
+	err = ftag.Wrap(err, ftag.NotFound)
+	out := ftag.GetAll(err)
+
+	assert.Equal(t, []ftag.Kind{ftag.NotFound, ftag.InvalidArgument, ftag.Internal}, out)
+}