You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.2 KiB
52 lines
1.2 KiB
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) |
|
} |
|
}
|
|
|