golang用通道實現(xiàn)信號量，控制并發(fā)個數(shù)

func main() {
    var wg sync.WaitGroup

    sem := make(chan struct{}, 2) // 最多允許2個并發(fā)同時執(zhí)行
    taskNum := 10

    for i := 0; i < taskNum; i++ {
        wg.Add(1)

        go func(id int) {
            defer wg.Done()

            sem <- struct{}{}        // 獲取信號
            defer func() { <-sem }() // 釋放信號

            // do something for task
            time.Sleep(time.Second * 2)
            fmt.Println(id, time.Now())
        }(i)
    }
    wg.Wait()
}