channelを使って、goroutineと関数の間で、データをやり取りすることができます。
次の例では、スライスの中身を足し合わせた結果を、channelを使って返却します。
func useChannel(s []int, c chan int) {
var sum int
for _, v := range s {
sum += v
}
c <- sum // sumをchannelに入れて返却
}
func main() {
s := []int{2, 3, 7, 8}
c := make(chan int) // channelはint型を指定（サイズを指定しないunbuffered channel。デフォルトでサイズは0）
go useChannelS(s, c)
x := <-c // 返ってきたchannelを受け取る
fmt.Println(x) // 20
}
goroutineの処理の後に記述した、channelの中身を変数に格納する処理
x := <-c
は、後続の処理をブロッキングするため、sync.WaitGroup.Wait()などは必要ありません。
上記例では、最終的な合計結果（sum）をchannelに渡していますが、
close(channel)と,channelからの送信の際にfor文を用いると、
途中経過を出力することができます。
func useChannel(s []int, c chan int) {
var sum int
for _, v := range s {
sum += v
c <- sum // 毎回sumを受信
}
close(c) // forループから抜けたらcloseすることで、cをrangeで回す際の終点を知らせる
}
func main() {
s := []int{2, 3, 8, 9, 12}
c := make(chan int, len(s)) // 第2引数を指定して、あらかじめ容量を確保
go useChannel(s, c)
for i := range c {
fmt.Println(i) // 受信の都度出力
}
}
もちろん、channelの型はint以外にも対応します。
stringの例です。
func useChannelS(s string, c chan string) {
c <- s + "を受け取りましたので返します。"
}
func main() {
c2 := make(chan string)
go useChannelS("A", c2)
xx := <-c2
fmt.Println(xx) // Aを受け取りましたので返します。
}
channelをinterface型にしてswitch typeを使えば、任意の型を指定することができます。
func interfaceChannel(i interface{}, c chan interface{}) {
var rv interface{}
switch ii := i.(type) {
case string:
rv = ii + "ですね"
case int:
rv = ii + 100
default:
rv = "undefined"
}
c <- rv
}
func main() {
// crypto/randパッケージを使用した乱数生成
seed, _ := crand.Int(crand.Reader, big.NewInt(math.MaxInt64))
rand.Seed(seed.Int64())
rn := rand.Int63() % 2
var i interface{}
if rn == 0 {
i = 13
} else {
i = "あかさたな"
}
c3 := make(chan interface{})
go interfaceChannel(i, c3)
xxx := <-c3
fmt.Println(xxx)
}
channelに関する、実に様々なパターンが記述されています。
hori-ryota.com
