go での非同期処理 その２

Go での非同期処理。まちあわせ。

前回 Go での非同期処理 その１でも、channel を使った待ち合わせの方法を実装したが、もっとちゃんとしたやり方がある様子。

WaitGroup を使う

WaitGroup というのを使うみたいだが、使うときと使わないときの比較をしてみよう。

package main

import(
    "fmt"
    "time"
)

func wait(i int) {
    time.Sleep(time.Second)
    fmt.Println(fmt.Sprintf("Wait %d done!", i))
}

func main() {
    for  i := 0; i < 3; i++ {
        go wait(i)
    }

    fmt.Println("Everything Done!")
}

こんなコードを書いてみる。当たり前だが、go routine が動くが、終わる前にメインの処理が終わってしまうので、次のような実行結果になる。

$ go run cmd/wait/*.go
Everything Done!

じゃあ、これをちゃんと、WaitGroup というやつに対応させてみよう。

package main

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

func wait(wg sync.WaitGroup, i int) {
    defer wg.Done()
    time.Sleep(time.Second)
    fmt.Println(fmt.Sprintf("Wait %d done!", i))
}

func main() {
    var wg sync.WaitGroup
    for i := 0; i < 3; i++ {
        wg.Add(1)
        go wait(wg, i)
    }
    wg.Wait()
    fmt.Println("Everything Done!")
}

基本は、WaitGroupを共有して、呼び元で、WaitGroup の Add メソッドを Go Routine の数だけ
呼んであげて、go routine の方で、Done() を呼び出せば、WaitGroup が終了を検知できるしくみになっている。さあ実行

W
io/WaitGo
$ go run cmd/wait/*.go
Wait 1 done!
Wait 0 done!
Wait 2 done!
fatal error: all goroutines are asleep - deadlock!

あかんがな、実行できているけど、Deadlock やがな。調べてみると、WaitGroup を渡すのに、実体を渡していると、関数にわたるときに、コピーされる。つまり、いくら Go Routine のほうで Done() を呼び出したところで、コピーのほうのDone() が呼ばれているだけで、元の、WaitGroup で足した Add の数が減ることがないという現象。じゃあ、ポインタ渡しにかえよう。

package main

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

func wait(wg *sync.WaitGroup, i int) {
    defer wg.Done()
    time.Sleep(time.Second)
    fmt.Println(fmt.Sprintf("Wait %d done!", i))
}

func main() {
    wg := &sync.WaitGroup{}
    for i := 0; i < 3; i++ {
        wg.Add(1)
        go wait(wg, i)
    }
    wg.Wait()
    fmt.Println("Everything Done!")
}

実行結果

$ go run cmd/wait/*.go
Wait 0 done!
Wait 2 done!
Wait 1 done!
Everything Done!

無事待ち合わせ Done!