feat: add ability to call a function after the job function runs (#125)

Author: pcfreak30

options.go

@@ -73,13 +73,21 @@ func WithFn(fn func(context.Context, core.QueuedMessage) error) Option {
 	})
 }
 
+// WithAfterFn set callback function after job done
+func WithAfterFn(afterFn func()) Option {
+	return OptionFunc(func(q *Options) {
+		q.afterFn = afterFn
+	})
+}
+
 // Options for custom args in Queue
 type Options struct {
 	workerCount int
 	logger      Logger
 	queueSize   int
 	worker      core.Worker
 	fn          func(context.Context, core.QueuedMessage) error
+	afterFn     func()
 	metric      Metric
 }
 

queue.go

@@ -29,6 +29,7 @@ type (
 		worker       core.Worker
 		stopOnce     sync.Once
 		stopFlag     int32
+		afterFn      func()
 	}
 )
 
@@ -46,6 +47,7 @@ func NewQueue(opts ...Option) (*Queue, error) {
 		logger:       o.logger,
 		worker:       o.worker,
 		metric:       &metric{},
+		afterFn:      o.afterFn,
 	}
 
 	if q.worker == nil {
@@ -163,6 +165,9 @@ func (q *Queue) work(task core.QueuedMessage) {
 		} else {
 			q.metric.IncFailureTask()
 		}
+		if q.afterFn != nil {
+			q.afterFn()
+		}
 	}()
 
 	if err = q.run(task); err != nil {