Minor improvements on queued worker

Author: raynaudoe

workers/workQueue/dispatcher.go

@@ -2,7 +2,6 @@ package WorkQueue
 
 import (
 	"fmt"
-	"github.com/Zondax/zindexer/connections"
 	"go.uber.org/zap"
 	"time"
 )
@@ -34,10 +33,10 @@ func NewJobDispatcher(cfg DispatcherConfig) JobDispatcher {
 	return d
 }
 
-func (j JobDispatcher) BuildWorkers(count int, dataSource connections.DataSource, constructor WorkerConstructor) {
+func (j JobDispatcher) BuildWorkers(count int, constructor WorkerConstructor) {
 	for i := 0; i < count; i++ {
 		workerId := fmt.Sprintf("worker.%d", i)
-		worker := constructor(workerId, dataSource, j.workerChannel)
+		worker := constructor(workerId, j.workerChannel)
 		worker.Worker.Start()
 	}
 }
@@ -57,6 +56,10 @@ func (j JobDispatcher) Start() {
 	}
 }
 
+func (j JobDispatcher) EnqueueWork(w Work) {
+	j.jobPool.EnqueueJob(w)
+}
+
 func (j JobDispatcher) dispatch() {
 	go func() {
 		for {

workers/workQueue/worker_queued.go

@@ -1,7 +1,5 @@
 package WorkQueue
 
-import cn "github.com/Zondax/zindexer/connections"
-
 type IQueuedWorker interface {
 	Start()
 	DoWork(Work)
@@ -11,7 +9,7 @@ type QueuedWorker struct {
 	Worker IQueuedWorker
 }
 
-type WorkerConstructor func(string, cn.DataSource, chan chan Work) QueuedWorker
+type WorkerConstructor func(string, chan chan Work) QueuedWorker
 
 type Work struct {
 	JobId  int64