Go言語で非同期タスクキューを実装する方法

Goでゴルーチンとチャネルを使って非同期タスクキューを実装できます。例を次に示します。

package main

import (
	"fmt"
	"sync"
	"time"
)

func main() {
	// 创建一个无缓冲的channel，用于任务队列
	taskQueue := make(chan func())
	// 使用sync.WaitGroup来等待所有任务完成
	var wg sync.WaitGroup

	// 启动消费者goroutine，不断从任务队列中取出任务并执行
	go func() {
		for task := range taskQueue {
			task()
			wg.Done()
		}
	}()

	// 添加一些任务到任务队列
	for i := 0; i < 10; i++ {
		// 增加等待的任务数量
		wg.Add(1)
		// 创建一个任务，并将其添加到任务队列
		task := func() {
			fmt.Println("执行任务", i)
			time.Sleep(1 * time.Second) // 模拟任务执行时间
		}
		taskQueue <- task
	}

	// 等待所有任务完成
	wg.Wait()
	// 关闭任务队列
	close(taskQueue)

	fmt.Println("所有任务已完成")
}

上記のサンプルコードでは、タスクキューとしてバッファなしのチャネルtaskQueueを作成し、taskQueue <- taskによってタスクキューにタスクを追加する。その後、タスクキューが閉じられるまでタスクキューからタスクを取り出して実行し続けるコンシューマーGoroutineを開始する。

sync.WaitGroup を用いて全てのタスク処理が完了するまで待機することで、全てのタスクが終了するまでプログラムが終了しないようにします。タスクが全て完了した時点でタスクキューをクローズし、プログラムを終了します。

これにより、単純な非同期タスクキューを実装しました。