Golang中的Channel-蒲公英云

Golang中的Channel

goroutine和goroutine之间通过channel通信

channel的创建

1 使用make创建

channel创建可以用make方法，如：

c := make(chan int)

此外，如果make方法第二个参数多写一个数字，则表示对channel作buffer缓冲：

// 表示channel中每接收3个后再给接收者
c := make(chan int, 3)

2 使用var创建

不用make的话，也可以用var：

var c chan int  // 此时c为nil

如果创建channel数组，则写法如下：

var channels [10]chan int

channel的传与收

往创建好的channel中传数据使用<-表示，如：

c <- 1
c <- 2

注意：往channel传数据后必须要有接收者，否则程序运行会报deadlock

较完整的示例片段如下：

// 创建一个channel
c := make(chan int)
// 起一个goroutine，用来接收channel中的数据并打印
go func() { 
    for { 
        n := <-c
        fmt.Println(n)
    }
}()
// 向channel中传数据
c <- 1
c <- 2
// 这里加个sleep，是为了避免channel中的数据还没打印，main函数就退出了
time.Sleep(time.Millisecond)

channel可作参数和返回值

channel可以作为参数传递，也可以作返回值。

需要注意的是，channel作为收数据还是发数据，写法是不一样的。
chan<- int 和<-chan int这两个是有区别的：

chan<- int表示send only，只能往里发
<-chan int表示revice only，只能从里收

channel的close

发送方可以使用close©方法来关闭channel，以告诉接收方没有新数据了。
channel关闭后，接收者还能接收数据不会报错，不过接收到的是对应类型的零值

如何判断channel是否关闭？
方法(1): 通过if判断

n, ok := <-c 
if !ok { 
    fmt.Println("channel 没有数据了")
    break
}

方法(2): 通过range

for n := range c { 
    ...
}

使用sync.WaitGroup等待goroutine任务结束

之前使用time.Sleep(time.Millisecond)的方式来让main函数等待任务完成，但是不太好。
golang中提供了sync.WaitGroup方法来实现这个功能

例：

package main
import (
    "fmt"
    "sync"
)
func doWork(id int,
    w worker) { 
    for n := range w.in { 
        fmt.Printf("Worker %d received %c\n",
            id, n)
        w.done()
    }
}
type worker struct { 
    in   chan int
    done func()
}
func createWorker(
    id int, wg *sync.WaitGroup) worker { 
    w := worker{ 
        in: make(chan int),
        done: func() { 
            wg.Done()
        },
    }
    go doWork(id, w)
    return w
}
func chanDemo() { 
    var wg sync.WaitGroup
    var workers [10]worker
    for i := 0; i < 10; i++ { 
        workers[i] = createWorker(i, &wg)
    }
    wg.Add(20)
    for i, worker := range workers { 
        worker.in <- 'a' + i
    }
    for i, worker := range workers { 
        worker.in <- 'A' + i
    }
    wg.Wait()
}
func main() { 
    chanDemo()
}

select的使用

例:

var c1, c2 chan int
select { 
case n := <-c1:
    fmt.Println("received from c1:", n)
case n := <-c2:
    fmt.Println("received from c2:", n)
default:
    fmt.Println("no value received")
}