diff options
| -rw-r--r-- | main.go | 12 | ||||
| -rw-r--r-- | workers.go | 15 |
2 files changed, 20 insertions, 7 deletions
@@ -188,6 +188,7 @@ func main() { // === concurrent === queue := make(chan *Utterance, len(utterances)) geshaft := make(chan *Utterance, len(utterances)) + done := make(chan bool, 1) for _, ut := range utterances { if _, err := os.Stat(ut.OutPath); os.IsNotExist(err) { @@ -204,23 +205,28 @@ func main() { workers := 100 for i := 0; i < workers; i++ { - go worker(queue, i, geshaft) + go worker(queue, done, i, geshaft) } for { if len(queue) == 0 { + fmt.Println("empty queue: sending true to done") + done <- true // give time for workers to finish - time.Sleep(3 * time.Second) + fmt.Println("sleeping for few seconds") + time.Sleep(5 * time.Second) + fmt.Println("closing queue") close(queue) + fmt.Println("breaking from the loop") break } time.Sleep(1 * time.Second) } + close(geshaft) for ut := range geshaft { filteredUtterances = append(filteredUtterances, ut) } - close(geshaft) newMeta := utterancesToFileTextMap(filteredUtterances) writeCSV(mapToCSV(newMeta)) } @@ -2,15 +2,22 @@ package main import "fmt" -func worker(queue chan *Utterance, worknumber int, geshaft chan *Utterance) { +func worker(queue chan *Utterance, done chan bool, worknumber int, geshaft chan *Utterance) { for { if len(queue) == 0 { fmt.Println("empty queue, number", worknumber) return } - ut := <-queue - if err := cutoutClipAndTranscode(ut); err == nil { - geshaft <- ut + select { + + case ut := <-queue: + if err := cutoutClipAndTranscode(ut); err == nil { + geshaft <- ut + } + + case <-done: + fmt.Println("done signal, number", worknumber) + return } } } |