OBM/worker.go

package main

import (
	"sync"
)

// a basic implementation of a concurrent worker
func worker(jobs <-chan job, results chan result, WG *sync.WaitGroup) {
	defer WG.Done()
	for job := range jobs {
		var successful, failed uint = 0, 0

		// the order is: Replace->Retrieve->Remove (if all 3 options are enabled)
		if job.replacementImagePath != "" {
			s, f := job.beatmap.ReplaceBackgrounds(job.replacementImagePath)
			successful += s
			failed += f
		}
		if job.retrievementPath != "" {
			s, f := job.beatmap.RetrieveBackgrounds(job.retrievementPath)
			successful += s
			failed += f
		}
		if job.remove == true {
			s, f := job.beatmap.RemoveBackgrounds()
			successful += s
			failed += f
		}

		results <- result{
			beatmapName:   job.beatmap.Name,
			numberOfDiffs: uint(len(job.beatmap.Diffs)),
			successful:    successful,
			failed:        failed,
		}
	}

}

// the `starter` that `glues` workers and jobs together
func workerPool(jobs chan job, results chan result, numOfWorkers int, WG *sync.WaitGroup) {
	// check if there are less jobs than workers
	if numOfWorkers > len(jobs) {
		numOfWorkers = len(jobs)
	}

	// replacing backgrounds for each beatmap concurrently
	for i := 0; i < numOfWorkers; i++ {
		WG.Add(1)
		go worker(jobs, results, WG)
	}
}