golang線程池怎么實(shí)現(xiàn)

在Go語(yǔ)言中，可以使用goroutine和channel來實(shí)現(xiàn)線程池。下面是一個(gè)簡(jiǎn)單的線程池實(shí)現(xiàn)示例：

package main

import (
	"fmt"
	"sync"
)

type ThreadPool struct {
	workerNum int           // 工作線程數(shù)量
	taskQueue chan func()  // 任務(wù)隊(duì)列
	wg        sync.WaitGroup // 等待所有任務(wù)完成
}

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() {
	// 創(chuàng)建線程池
	pool := NewThreadPool(3)
	// 啟動(dòng)線程池
	pool.Start()

	// 添加任務(wù)到線程池
	for i := 0; i < 10; i++ {
		taskNum := i
		pool.AddTask(func() {
			fmt.Printf("Task %d is running\n", taskNum)
		})
	}

	// 等待所有任務(wù)完成
	pool.Wait()
}

在上面的示例中，ThreadPool結(jié)構(gòu)體表示線程池，包含workerNum表示工作線程的數(shù)量，taskQueue表示任務(wù)隊(duì)列，wg為sync.WaitGroup用于等待所有任務(wù)完成。

NewThreadPool函數(shù)用于創(chuàng)建一個(gè)新的線程池，Start方法啟動(dòng)線程池中的工作線程，worker方法用于執(zhí)行任務(wù)，AddTask方法用于往線程池中添加任務(wù)，Wait方法用于等待所有任務(wù)完成。

在main函數(shù)中，創(chuàng)建一個(gè)線程池并啟動(dòng)，然后循環(huán)添加任務(wù)到線程池中，最后調(diào)用Wait方法等待所有任務(wù)完成。

當(dāng)運(yùn)行上述代碼時(shí)，你會(huì)看到輸出結(jié)果中的任務(wù)是并發(fā)執(zhí)行的，最多同時(shí)執(zhí)行3個(gè)任務(wù)，這就是線程池的作用。