在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個任務,這就是線程池的作用。