golang怎样控制并发数
在Go中,可使用goroutine和channel来控制并发数。
首先,可使用make
函数创建一个带有指定并发数的channel。例如,如果要控制并发数为10,可以创建一个大小为10的无缓冲channel:
concurrency := 10
ch := make(chan struct{}, concurrency)
然后,在需要控制并发的地方,在履行每一个任务之前,可以向channel发送一个空的struct{}值来占用一个并发槽,表示一个并发任务的启动。当任务完成后,从channel接收一个值来释放一个并发槽。
ch <- struct{}{} // 占用一个并发槽
go func() {
defer func() { <-ch }() // 释放一个并发槽
// 履行并发任务的代码
}()
这样就可以够通过控制channel的发送和接收来控制并发数了。
完全示例代码以下:
package main
import (
"fmt"
"time"
)
func main() {
concurrency := 10
ch := make(chan struct{}, concurrency)
tasks := []string{"task1", "task2", "task3", "task4", "task5", "task6", "task7", "task8", "task9", "task10"}
for _, task := range tasks {
ch <- struct{}{} // 占用一个并发槽
go func(t string) {
defer func() { <-ch }() // 释放一个并发槽
fmt.Printf("开始履行任务:%s
", t)
time.Sleep(time.Second) // 摹拟任务的耗时
fmt.Printf("任务:%s 履行完成
", t)
}(task)
}
// 等待所有任务完成
for i := 0; i < concurrency; i++ {
ch <- struct{}{}
}
}
在上面的示例中,我们创建了一个包括10个并发槽的channel,然后使用goroutine来履行任务。在每一个任务开始和结束时,我们输出相应的日志。最后,我们使用一个for循环来等待所有任务完成。
注意,为了确保所有的goroutine都能够履行终了,我们在最后的for循环中再次向channel发送了10个空的struct{}值,以确保释放所有的并发槽。
这样就可以够通过channel来控制并发数了。
TOP