中文字幕av专区_日韩电影在线播放_精品国产精品久久一区免费式_av在线免费观看网站

溫馨提示×

golang線程池怎么實現

小億
129
2023-10-26 18:28:28
欄目: 編程語言

在Go語言中,可以使用goroutinechannel來實現線程池。下面是一個簡單的線程池實現示例:

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表示任務隊列,wgsync.WaitGroup用于等待所有任務完成。

NewThreadPool函數用于創建一個新的線程池,Start方法啟動線程池中的工作線程,worker方法用于執行任務,AddTask方法用于往線程池中添加任務,Wait方法用于等待所有任務完成。

main函數中,創建一個線程池并啟動,然后循環添加任務到線程池中,最后調用Wait方法等待所有任務完成。

當運行上述代碼時,你會看到輸出結果中的任務是并發執行的,最多同時執行3個任務,這就是線程池的作用。

0
托克托县| 平陆县| 乌苏市| 石城县| 花垣县| 会昌县| 兰溪市| 建瓯市| 深圳市| 贵南县| 三门峡市| 日照市| 淮南市| 贡觉县| 岑巩县| 昌黎县| 永定县| 花垣县| 祁东县| 河间市| 邻水| 红安县| 淮安市| 迁西县| 马龙县| 巴南区| 景谷| 南华县| 交口县| 桃江县| 新郑市| 和平县| 荃湾区| 巴彦县| 定南县| 客服| 靖宇县| 芦山县| 嘉鱼县| 阜康市| 南平市|