高度な技：GoにおけるWaitGroupとゴー程のスケジューリング

2年 ago
結衣, 春花
1 minute

GoにおけるWaitGroupとgoroutine調度は高度なテクニックとして非常に重要であり、同時実行タスクやgoroutineの同期において非常に役立ちます。

WaitGroupは、一連のgoroutineがすべて実行を終えるのを待つカウンタです。すべてのgoroutineが完了するまでメインgoroutineをブロックします。WaitGroupを使用すると、メインgoroutineが終了する前にすべてのgoroutineが完了するよう確実にできます。

WaitGroupを使用したサンプルコードを以下に示します。

package main
import (
"fmt"
"sync"
)
func worker(id int, wg *sync.WaitGroup) {
defer wg.Done()
fmt.Printf("Worker %d starting\n", id)
// 模拟耗时操作
for i := 0; i < 100000000; i++ {
}
fmt.Printf("Worker %d done\n", id)
}
func main() {
var wg sync.WaitGroup
for i := 1; i <= 5; i++ {
wg.Add(1)
go worker(i, &wg)
}
wg.Wait()
fmt.Println("All workers done")
}

上の例では、5つのワーカーゴルーチンを作成し、それらが完了するまでWaitGroupを使用して待ちます。各ワーカーゴルーチンの開始と終了時には、WaitGroupのDone()メソッドを使用してカウンタを減らします。メインのゴルーチンでは、Wait()メソッドを使用して、カウンタが0になるまでブロックします。

ゴーランタイムシステムは Goroutine Scheduler と呼ばれるコンポーネントを使用して、複数のゴルーチン間でどのようにスケジュールと切り替えを行うかを管理しています。スケジューラは、いつゴルーチンを切り替えて、どのスレッドの執行にどのゴルーチンを割り当てるかを決定する一連のルールに従っています。

コルーチンスケジューリングにおける高度なテクニックを以下に示します。

  1. GOMAXPROCS環境変数は同時に実行できる最大OSスレッド数を設定する。 デフォルト値はマシン上のCPUコア数。GOMAXPROCSの値を設定することでコンカレントスケジューラは振る舞いを変更することができる。
  2. runtime.Gosched():この関数は、ゴルーチンが現在のスレッドの実行権を自主的に放棄し、他のゴルーチンが実行できるようにします。場合によっては、これを使用してゴルーチンが長い間スレッドを占有するのを防ぐことができます。
  3. runtime.LockOSThread()とruntime.UnlockOSThread()：この2つの関数は、それぞれ現在のコルーチンの現在のスレッドへのロック、または現在のスレッドへのロックの解除に使用されます。特定の状況では、コルーチンを特定のスレッドにロックして実行する必要がある場合があります。

高度な技法を使用することで、コルーチン実行の管理や制御を向上させ、同時並行処理の性能や効率を向上させることが期待できます。

bannerAds