Goのワークプールはどのように実装されていますか。

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メソッドによってすべてのタスクが完了するまで待機されます。

メイン関数においてスレッドプールを作成して開始し、タスクをスレッドプールに追加するループを実施後、`Wait`メソッドの呼び出しで全てのタスクが完了するまで待ちます。

上記のコードを実行すると、出力結果にタスクが最大3つまで同時に実行されている並行処理されているのがわかります。これがスレッドプールの機能です。