Merge pull request #75 from foodora/awssub-on-error-func

introduce error handling function
deliveryhero · Sep 18, 2019 · fc00051 · fc00051
2 parents 24ddbf9 + ada3ca6
commit fc00051
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/pubsub/awssub/sub.go b/pubsub/awssub/sub.go
@@ -33,6 +33,9 @@ type (
 		sqsErr error
 
 		Logger pubsub.Logger
+
+		// onErrorFunc is a func is being called when an error occurs
+		onErrorFunc func(error)
 	}
 
 	// SQSMessage is the SQS implementation of `SubscriberMessage`.
@@ -261,6 +264,9 @@ func (s *subscriber) Start() <-chan pubsub.Message {
 				// we've encountered a major error
 				s.Logger.Printf("Error occurred %s", err.Error())
 				s.sqsErr = err
+				if s.onErrorFunc != nil {
+					s.onErrorFunc(err)
+				}
 				time.Sleep(s.cfg.SleepInterval)
 				continue
 			}
@@ -291,6 +297,11 @@ func (s *subscriber) Start() <-chan pubsub.Message {
 	return output
 }
 
+// OnErrorFunc sets subscriber's onErrorFunc field
+func (s *subscriber) SetOnErrorFunc(fn func(error)) {
+	s.onErrorFunc = fn
+}
+
 func (s *subscriber) handleDeletes() {
 	var entriesBuffer []*sqs.DeleteMessageBatchRequestEntry
 	for {