Channel使用技巧

前言

Go协程一般使用channel（通道）通信从而协调/同步他们的工作。合理利用Go协程和channel能帮助我们大大提高程序的性能。本文将介绍一些使用channel的场景及技巧

场景一，使用channel返回运算结果

计算斐波那契数列，在学习递归时候这是个经典问题。现在我们不用递归实现，而是用channel返回计算得出的斐波那契数列。 计算前40个斐波那契数列的值，看下效率

package main

import (
    "fmt"
    "time"
)
//计算斐波那契数列并写到ch中
func fibonacci(n int, ch chan<- int) {
    first, second := 1, 1
    for i := 0; i < n; i++ {
        ch <- first
        first, second = second, first+second
    }
    close(ch)
}

func main() {
    ch := make(chan int, 40)
    i := 0
    start := time.Now()
    go fibonacci(cap(ch), ch)
    for result := range ch {
        fmt.Printf("fibonacci(%d) is: %d\n", i, result)
        i++
    }
    end := time.Now()
    delta := end.Sub(start)
    fmt.Printf("took the time: %s\n", delta)
}

只花了7ms，效率是递归实现的100倍（主要是算法效率问题）

fibonacci(33) is: 5702887
fibonacci(34) is: 9227465
fibonacci(35) is: 14930352
fibonacci(36) is: 24157817
fibonacci(37) is: 39088169
fibonacci(38) is: 63245986
fibonacci(39) is: 102334155
took the time: 8.0004ms

使用for-range读取channel返回的结果十分便利。当channel关闭且没有数据时，for循环会自动退出，无需主动监测channel是否关