[WIP]: broker ErrorHandler option (#1296)

* broker ErrorHandler option Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org> * rewrite Event interface, add error Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org> * implement new interface Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org> * change ErrorHandler func to broker.Handler Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org> * fix Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
2020-03-07 00:25:16 +03:00
parent 11be2c68b9
commit 8ee5607254
8 changed files with 94 additions and 14 deletions
--- a/broker/memory/memory.go
+++ b/broker/memory/memory.go
@@ -27,6 +27,7 @@ type memoryBroker struct {
 type memoryEvent struct {
 	opts    broker.Options
 	topic   string
+	err     error
 	message interface{}
 }

@@ -120,6 +121,11 @@ func (m *memoryBroker) Publish(topic string, msg *broker.Message, opts ...broker

 	for _, sub := range subs {
 		if err := sub.handler(p); err != nil {
+			p.err = err
+			if eh := m.opts.ErrorHandler; eh != nil {
+				eh(p)
+				continue
+			}
 			return err
 		}
 	}
@@ -197,6 +203,10 @@ func (m *memoryEvent) Ack() error {
 	return nil
 }

+func (m *memoryEvent) Error() error {
+	return m.err
+}
+
 func (m *memorySubscriber) Options() broker.SubscribeOptions {
 	return m.opts
 }