snapshot of https://github.com/ThreeDotsLabs/watermill.git (master, 2026-09-15; blobs > 131072B omitted)

message/router/middleware/ignore_errors.go

@@ -0,0 +1,38 @@
+package middleware
+
+import (
+	"github.com/ThreeDotsLabs/watermill/message"
+	"github.com/pkg/errors"
+)
+
+// IgnoreErrors provides a middleware that makes the handler ignore some explicitly whitelisted errors.
+type IgnoreErrors struct {
+	ignoredErrors map[string]struct{}
+}
+
+// NewIgnoreErrors creates a new IgnoreErrors middleware.
+func NewIgnoreErrors(errs []error) IgnoreErrors {
+	errsMap := make(map[string]struct{}, len(errs))
+
+	for _, err := range errs {
+		errsMap[err.Error()] = struct{}{}
+	}
+
+	return IgnoreErrors{errsMap}
+}
+
+// Middleware returns the IgnoreErrors middleware.
+func (i IgnoreErrors) Middleware(h message.HandlerFunc) message.HandlerFunc {
+	return func(msg *message.Message) ([]*message.Message, error) {
+		events, err := h(msg)
+		if err != nil {
+			if _, ok := i.ignoredErrors[errors.Cause(err).Error()]; ok {
+				return events, nil
+			}
+
+			return events, err
+		}
+
+		return events, nil
+	}
+}