add sethandler

Signed-off-by: Pavel Okhlopkov <pavel.okhlopkov@flant.com>

Author: ldmonster

pkg/task/queue.go

@@ -11,6 +11,7 @@ type TaskQueue interface {
 	SetStatus(status string)
 	GetName() string
 	GetHandler() func(ctx context.Context, t Task) TaskResult
+	SetHandler(handler func(ctx context.Context, t Task) TaskResult)
 	Start(ctx context.Context)
 	Stop()
 

pkg/task/queue/task_queue_list.go

@@ -292,6 +292,12 @@ func (q *TaskQueue) GetHandler() func(ctx context.Context, t task.Task) task.Tas
 	return q.handler
 }
 
+func (q *TaskQueue) SetHandler(handler func(ctx context.Context, t task.Task) task.TaskResult) {
+	q.m.RLock()
+	defer q.m.RUnlock()
+	q.handler = handler
+}
+
 func (q *TaskQueue) IsEmpty() bool {
 	defer q.MeasureActionTime("IsEmpty")()
 	q.m.RLock()