diff --git a/pkg/pool/pool.go b/pkg/pool/pool.go index 45e1df6..1e8fb78 100644 --- a/pkg/pool/pool.go +++ b/pkg/pool/pool.go @@ -59,7 +59,7 @@ func (tp *TaskPool) WaitForTask(taskID int) { <-waitChan } -func (tp *TaskPool) MarkTaskComplete(taskID int) { +func (tp *TaskPool) markTaskComplete(taskID int) { tp.lck.Lock() defer tp.lck.Unlock() diff --git a/pkg/pool/worker.go b/pkg/pool/worker.go index de5f55a..8cd5473 100644 --- a/pkg/pool/worker.go +++ b/pkg/pool/worker.go @@ -19,6 +19,6 @@ func (w *Worker) Start(wg *sync.WaitGroup) { for task := range w.queue { task.f() - w.pool.MarkTaskComplete(task.id) + w.pool.markTaskComplete(task.id) } }