golang线程池怎么实现
短信预约 -IT技能 免费直播动态提醒
在Go语言中,可以使用`goroutine`和`channel`来实现线程池。下面是一个简单的线程池实现示例:
package main
import (
"fmt"
"sync"
)
type ThreadPool struct {
workerNum int // 工作线程数量
taskQueue chan func() // 任务队列
wg sync.WaitGroup // 等待所有任务完成
}
func NewThreadPool(workerNum int) *ThreadPool {
return &ThreadPool{
workerNum: workerNum,
taskQueue: make(chan func()),
}
}
func (tp *ThreadPool) Start() {
for i := 0; i < tp.workerNum; i++ {
go tp.worker()
}
}
func (tp *ThreadPool) worker() {
for task := range tp.taskQueue {
task()
tp.wg.Done()
}
}
func (tp *ThreadPool) AddTask(task func()) {
tp.wg.Add(1)
tp.taskQueue <- task
}
func (tp *ThreadPool) Wait() {
tp.wg.Wait()
}
func main() {
// 创建线程池
pool := NewThreadPool(3)
// 启动线程池
pool.Start()
// 添加任务到线程池
for i := 0; i < 10; i++ {
taskNum := i
pool.AddTask(func() {
fmt.Printf("Task %d is running\n", taskNum)
})
}
// 等待所有任务完成
pool.Wait()
}
在上面的示例中,`ThreadPool`结构体表示线程池,包含`workerNum`表示工作线程的数量,`taskQueue`表示任务队列,`wg`为`sync.WaitGroup`用于等待所有任务完成。
`NewThreadPool`函数用于创建一个新的线程池,`Start`方法启动线程池中的工作线程,`worker`方法用于执行任务,`AddTask`方法用于往线程池中添加任务,`Wait`方法用于等待所有任务完成。
在`main`函数中,创建一个线程池并启动,然后循环添加任务到线程池中,最后调用`Wait`方法等待所有任务完成。
当运行上述代码时,你会看到输出结果中的任务是并发执行的,最多同时执行3个任务,这就是线程池的作用。
免责声明:
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341