等待一个goroutine完成

时间:2018-08-27 00:44:21

标签: go goroutine

我正在8个不同的goroutine上运行CPU密集型脚本。这些goroutine中的每一个都至少需要花费几分钟的时间才能完成,我想知道这样的事情是否可行:

for i := 0; i < len(input); i += 1 {
  wait_for_number_of_processes_running_to_be_less_than_8
  go calc_math_and_things(input[i])
}

1 个答案:

答案 0 :(得分:3)

您可以使用缓冲通道返回goroutine的结果并指示终止,并且当您从该通道读取时,您将启动一个新的go例程。像这样:

const maxNRoutine = 8
var routineCtr int
var resultCh = make(chan int, maxNRoutine)

for i := 0; i < len(input); i += 1 {
    if routineCtr < maxNRoutines {
        go calc_math_and_things(resultCh, input[i])
        routineCtr++
        continue
    }
    var result = <- resultCh // go routine is done, log result and start a new one
    println(result)
    go calc_math_and_things(resultCh, intput[i])
}

在您的日常活动中:

func calc_math_and_things(resultCh chan<- int, input byte) {
     // ... do some stuff
     resultCh <- 1
}