1.1 channel介绍
Go并发是源自CSP模型,通过channel来实现协程的同步,Go并发不通过共享内存来通信,而是通过通信来共享内存,Go内建channel实现了go协程之间数据的读写相关操作,通道(channel)是一种特殊的类型,在任何时候,同时只能有一个 goroutine 访问通道进行发送和获取数据。channel可以看作一个消息队列,遵循先进先出的原则,从而保证了收发数据的顺序性。
channel有如下特点:
- channel本身是一个队列,先进先出
- 线程安全,不需要枷锁,但是如果传递数组指针或者其他非线程安全的指针或引用,需要额外做好保护
- channel是引用类型,必须make之后才能使用,一旦初始化容量,就不会改变了
- 当写满时,不可以写,取空时,不可以取,如果容量满则发送阻塞直到通道数据被取走,如果通道数据为空,则接收阻塞,可通过select方式非阻塞读写
- 通道一次只能接收一个数据元素
- 每个 channel 都有一个特殊的类型,也就是 channels 可发送数据的类型,比如chan int
Go语言提倡使用channel的方法代替共享内存,当一个资源需要在 goroutine 之间共享时,通道在 goroutine 之间架起了一个channel,并提供了确保同步交换数据的机制,声明channel时,需要指定将要被共享的数据的类型,可以通过通道共享内置类型、命名类型、结构类型和引用类型的值或者指针等。
channel与goroutine交互示意图如下:
1.2 channel使用
1.2.1 channel声明和初始化
var 通道变量 chan 通道类型
var chan1 chan int //通道传递int类型的数据
通过此声明chan1为nil,需要配合make进行初始化后,才能使用
var chan1 chan int
chan1 = make(chan int) //无缓冲
chan2 := make(chan int,3) //缓冲大小为3的通道
type as struct{
data []byte
name string
}
a := &as{
make([]byte,0,1024),"buf"
}
chan3 := make(chan *as,3) //缓冲大小为3,传输类型为as的指针
a <- a
1.2.2 channel数据发送与接收
通道创建好后,根据make时是否创建缓冲分为有缓冲chan和无缓冲的chan
1.2.2.1 无缓冲chan的数据传输
如果创建时缓冲大小设置为0,或者未设置,此时把数据往通道中发送时,如果接收方一直都没有接收,那么发送操作将持续阻塞,如果接收时一直未有数据发送,则也一直阻塞,实例如下:
package main
import "fmt"
func main() {
chan1 := make(chan int)
go gorun1(chan1)
var a int = 3
fmt.Println(a)
chan1 <- a
a++
fmt.Println(a)
chan1 <- a
a++
fmt.Println(a)
}
//传递的时chan引用
func gorun1(c chan int) {
v := <-c
fmt.Println("gorun1 receive:", v)
}
/*打印结果
3
4
gorun1 receive: 3
*/
gorun1是在go程中运行,第一次发送数据3时,gorun1读取了数据3并打印,第二次发送数据4时,由于gorun1已经退出,无人接收,所以一直阻塞在chan1<-a
无缓冲通道的chan比较常用于等待go程退出的场景,实例如下:
package main
import "fmt"
var done chan any
func main() {
done = make(chan any)
chan1 := make(chan int)
go gorun1(chan1)
go gorun2(chan1)
a := <-done //阻塞,等待gorun1退出时执行close(done)
fmt.Println(a)
}
func gorun2(c chan int) {
c <- 3
c <- 4
c <- 5
c <- 0
}
//传递的时chan引用
func gorun1(c chan int) {
defer close(done) //close chan时会触发向chan发送一个nil
for {
select {
case v := <-c:
fmt.Println("gorun1 receive:", v)
if v == 0 {
return
}
}
}
}
从例子中可以看出,直到gorun2向chan1发送0时,gorun1退出,整个进程才推出,实例还使用了select,实现了无缓冲非阻塞接收。这里done实际上实现了wait-set的功能。
chan数据接收时,可通过如下方式判断是否关闭,当无缓冲时,其阻塞,直到关闭时ok为false,v为nil
v,ok := <-chan1 //经过测试无缓冲时,无数据写入时,其是阻塞的,直到close或者写入数据时,才会被触发
数据接收时,还可使用for range方式循环遍历接收,实例如下:
package main
import (
"fmt"
"time"
)
var done chan any
func main() {
done = make(chan any)
chan1 := make(chan int)
go gorun1(chan1)
go gorun2(chan1)
a, ok := <-done //等待gorun1退出
fmt.Println(a, ok)
}
func gorun2(c chan int) {
c <- 3
time.Sleep(time.Duration(2) * time.Second)
c <- 4
time.Sleep(time.Duration(2) * time.Second)
c <- 5
time.Sleep(time.Duration(2) * time.Second)
c <- 0
time.Sleep(time.Duration(2) * time.Second)
}
//传递的时chan引用
func gorun1(c chan int) {
defer close(done) //close chan时会触发向chan发送一个nil
for v := range c { //循环接收,如果无数据则阻塞直到数据到来
fmt.Println("gorun1 receive:", v)
if v == 0 {
return
}
}
}
1.2.2.2 有缓冲chan数据传输
为了满足并发数据处理的需求,chan可以声明为带有缓冲区的通道,此时只有通道未满,数据都可写入缓冲器,接收端只要缓冲区不为空都可以接收数据,但是chan不支持批量读写,每次只能写入一个数据,每次只能读取一个数据,当缓冲区满时,发送端阻塞;当缓冲区为空时,接收端阻塞;可以通过select方式来避免阻塞,实例如下:
package main
import (
"fmt"
)
func main() {
chan1 := make(chan int, 3) //缓冲大小为3
chan1 <- 1 //放入第一个数据
fmt.Println(1)
chan1 <- 2 //放入第二个数据
fmt.Println(2)
chan1 <- 3 //放入第三个数据
fmt.Println(3)
fmt.Println(<-chan1) //读取数据
fmt.Println(<-chan1) //读取数据
fmt.Println(<-chan1) //读取数据
fmt.Println(<-chan1) //读取数据
}
实例声明了缓冲区大小为3的chan,在写入数据时,连续写入3个并未阻塞,连续接收三个也未阻塞,第四次读取时由于无数据,阻塞
有关阻塞/非阻塞/无缓冲/缓冲,可参照此篇文章:
https://blog.csdn.net/aggie4628/article/details/124646319
1.3 channel的应用场景
1.3.1 通过channel实现信号量功能
前面已经介绍了此种场景,通过无缓冲的chan,接收阻塞直到另一个并发go程处理完程,并发送close(chan)或者发送任意值后,触发当前现场继续执行,实例如下:
package main
import (
"fmt"
"time"
)
var done chan any
func main() {
done = make(chan any)
go task1()
<-done//等待task1执行完成
fmt.Println("task2")
}
func task1() {
fmt.Println("task1")
time.Sleep(time.Duration(5) * time.Second)
done <- nil //执行完成后发送一个nil触发done解除阻塞
}
1.3.2 通过channel实现广播通知功能
可利用已关闭的channel读取数据时总是非阻塞的特性,可以实现在一个协程中向其他多个协程广播某个事件发生的通知,例如定义一个exitp chan any来控制所有go程退出,实例如下:
package main
import (
"fmt"
"time"
)
var exitp chan any
func main() {
exitp = make(chan any)
go task1()
go task2()
fmt.Println("task3")
time.Sleep(time.Duration(3) * time.Second)
close(exitp)
time.Sleep(time.Duration(1) * time.Second)
}
func task1() {
fmt.Println("task1")
time.Sleep(time.Duration(1) * time.Second)
<-exitp
fmt.Println("exit task1")
}
func task2() {
fmt.Println("task2")
time.Sleep(time.Duration(1) * time.Second)
<-exitp
fmt.Println("exit task2")
}
通过close(chan)的特性,只能通知一次,不能携带参数,如果需要携带参数通知,则需要建立一个总线channel的方式,分发到子chan中,例如
package main
import (
"fmt"
"time"
)
type aa struct {
message chan any
name string
}
func (p *aa) Run() {
defer close(p.message)
for {
select {
case ss := <-p.message:
fmt.Println(p.name, ss)
case <-Exitp:
fmt.Println("exit :", p.name)
return
}
}
}
var Eventbus chan any
var Exitp chan any
var Reicevermap map[string]*aa
func main() {
Exitp = make(chan any) //控制go程退出
Eventbus = make(chan any, 1)
Reicevermap = make(map[string]*aa)
a := &aa{make(chan any, 3), "aa"} //aa接收者
go a.Run() //aa接收广播线程
Reicevermap[a.name] = a
b := &aa{make(chan any, 3), "bb"} //bb接收者
go b.Run() //bb接收广播线程
Reicevermap[b.name] = b
fmt.Println("111")
go broadcast()
time.Sleep(time.Duration(1) * time.Second)
Eventbus <- "第一个广播"
time.Sleep(time.Duration(3) * time.Second)
close(Exitp)
time.Sleep(time.Duration(3) * time.Second)
}
func broadcast() {
for event := range Eventbus {//不建议此种方式接收chan数据,建议用select
for _, v := range Reicevermap {
v.message <- event
}
}
}
1.3.3 通过channel互斥量
通过设置容量未1的chan来实现互斥的功能,这种用法较少,因为互斥量本来就可以用锁来实现,没必要滥用chan,这里就不做举例
1.3.4 channel多写多读控制
原则上尽量一个channel只有一个go程写,一个go程读取,这样最好控制,尽量避免多写多读的情况,此种情况如果无法避免,可以引入额外的chan来做分发,就如上述带参数的广播的例子,这里也不做举例
1.4 channel使用注意事项
- channel不用时最好close关闭,虽然说gc会回收,但是还是有序关闭为好
- channel不能重复close否则会报恐慌错误
- 涉及到大数据传输时,传输指针或者引用,来避免大内存拷贝
- channel灵活使用,可让多进程并发通信变得很简单,但是也不能滥用