Go语言中的通道

通道(channel)是Go 语言中一种特殊的数据类型,通道本身就是并发安全的,可以通过它在多个 goroutine 之间传递数据。通道是Go 语言编程理念:“Do not communicate by sharing memory; instead, share memory by communicating”(不要通过共享数据来通信,而应该通过通信来共享数据。)的完美实现,在并发编程中经常会遇到它。下面来介绍一下通道的使用方法。

通道的发送和接收

通道包括双向通道和单向通道,这里双向通道只的是支持发送和接收的通道,而单向通道是只能发送或者只能接收的通道。

双向通道

使用make函数声明并初始化一个通道:

ch1 := make(chan string, 3)
  • chan 是表示通道类型的关键字
  • string 表示该通道类型的元素类型
  • 3 表示该通道的容量为3,最多可以缓存3个元素值。

一个通道相当于一个先进先出(FIFO)的队列,使用操作符 <- 进行元素值的发送和接收:

ch1 <- "1"  //向通道ch1发送数据 "1"

接收元素值:

elem1 := <- ch1 // 接收通道中的元素值

首先接收到的元素为先存入通道中的元素值,也就是先进先出:

package main

import "fmt"

func main() {
	str1 := []string{"hello","world", "!"}
	ch1 := make(chan string, len(str1))

	for _, str := range str1 {
		ch1 <- str
	}
	
	for i := 0; i < len(str1); i++ {
		elem := <- ch1
		fmt.Println(elem)
	}
}

执行结果:

hello
world
!

单向通道

单向通道包括只能发送的通道和只能接收的通道:

var WriteChan = make(chan<- interface{}, 1) // 只能发送不能接收的通道
var ReadChan = make(<-chan interface{}, 1) // 只能接收不能发送的通道

单向通道的这种特性可以用来约束函数的输入类型或者输出类型,比如下面的例子约束了只能从通道中接收元素值:

package main

import (
	"fmt"
)

func OnlyReadChan(num int) <-chan int {
	ch := make(chan int, 1)
	ch <- num
	close(ch)
	return ch
}

func main() {

	Chan1 := OnlyReadChan(6)
	num := <- Chan1
	fmt.Println(num)
}

执行结果:

6

通道阻塞

通道操作是并发安全的,在同一时刻,只会执行对同一个通道的任意个发送操作中的某一个,直到这个元素值被完全复制进该通道之后,其他针对该通道的发送操作才可能被执行。接收操作也一样。另外,对于通道中的同一个元素值来说,发送操作和接收操作之间也是互斥的。

发送操作和接收操作是原子操作,也就是说,发送操作绝不会出现只复制了一部分的情况,要么还没有复制,要么已经复制完毕。接收操作在准备好元素值的副本之后,一定会删除掉通道中的原值,绝不会出现通道中仍有残留的情况。在进行发送操作和接收操作时,代码会一直阻塞在那里,完成操作后才会继续执行后面的代码。通道的发送操作和接收操作是很快的,那么什么情况下会出现长时间的阻塞呢?下面介绍几种情况。

缓冲通道的阻塞

缓冲通道是容量大于0的通道,也就是可以缓存数据的通道。

1、发送阻塞

如果缓冲通道已经填满,如果有goroutine继续向该通道发送数据就会阻塞。请看下面的例子:

package main

func main() {
	ch1 := make(chan int, 1)
	ch1 <- 1
	ch1 <- 2
}

执行结果:

fatal error: all goroutines are asleep - deadlock!

goroutine 1 [chan send]:
...........

如果通道可以接收数据(有元素被接收),通道会通知最先等待发送操作的 goroutine再次执行发送操作。

2、接收阻塞

类似的,如果通道已空,如果继续进行接收操作就会被阻塞。

package main

func main() {
	ch1 := make(chan int, 1)
	<- ch1
}

执行结果:

fatal error: all goroutines are asleep - deadlock!

goroutine 1 [chan receive]:
...........

非缓冲通道

非缓冲通道是容量为0的通道,不能缓存数据。

非缓冲通道的数据传递是同步的,发送操作或者接收操作在执行后就会阻塞,需要对应的接收操作或者发送操作执行才会继续传递。由此可以看出缓冲通道使用的是异步方式进行数据传递。

package main

import (
    "fmt"
)

func main() {
    str1 := []string{"hello","world", "!"}
	ch1 := make(chan string, 0)

    go func() {
        for _, str := range str1 {
            ch1 <- str
        }
    }()

	for i := 0; i < len(str1); i++ {
		elem := <- ch1
		fmt.Println(elem)
	}

}

执行结果:

hello
world
!  

上面的代码中3个goroutine向通道写了三次数据,必须有三次接收,不然会阻塞。

对值为nil的通道进行发送操作和接收操作也会发生阻塞:

var ch1 chan int
ch1 <- 1 // 阻塞
<-ch1 // 阻塞

通道关闭

可以使用close()方法来关闭通道,通道关闭后,不能再对通道进行发送操作,可以进行接收操作。

package main

import "fmt"

func main() {
	ch1 := make(chan int, 1)
	ch1 <- 1
	close(ch1)
    
    ele := <-ch1
	fmt.Println(ele)  
    
	ch1 <- 2
}

执行结果:

1
panic: send on closed channel

goroutine 1 [running]:
.....

如果通道关闭时,里面还有元素,进行接收操作时,返回的通道关闭标志仍然为true:

package main

import "fmt"

func main() {
	ch1 := make(chan int, 1)
	ch1 <- 1
	close(ch1)

	ele1, statu1 := <-ch1
	fmt.Println(ele1, statu1)
	ele2, statu2 := <-ch1
	fmt.Println(ele2, statu2)
}

执行结果:

1 true
0 false

由于通道的这种特性,可以让发送方来关闭通道。前面的例子可以这样写:

package main

import (
    "fmt"
)

func main() {
    str1 := []string{"hello","world", "!"}
	ch1 := make(chan string, 0)

    go func() {
        for _, str := range str1 {
            ch1 <- str
        }
        close(ch1)
    }()

	for i := 0; i < len(str1); i++ {
		elem := <- ch1
		fmt.Println(elem)
	}

}

另外,不能对关闭的通道再次关闭:

package main

// import "fmt"

func main() {
	ch1 := make(chan int, 1)
	ch1 <- 1
	close(ch1)
	close(ch1)

}

执行结果:

panic: close of closed channel

select语句与通道

select语句通常与通道联用,它是专为通道而设计的。select语句执行时,一般只有一个case表达式或者default语句会被运行。

package main

import "fmt"

func main() {
    ch1 := make(chan int, 1)
	num := 2
	
	select {
		case data := <-ch1:
			fmt.Println("Read data: ", data)
		case ch1 <- num:
			fmt.Println("Write data: ", num)	
		default:
			fmt.Println("No candidate case is selected!")
	}
}

执行结果:

Write data:  2

需要注意的是,如果没有default默认分支,case表达式都没有满足条件,那么select语句就会被阻塞,直到至少有一个case表达式满足条件为止。

如果同时有多个分支满足条件,会随机选择一个分支执行

for语句与select语句联用时,分支中的break语句只能结束当前select语句的执行,而不会退出for循环。下面的代码永远不会退出循环:

package main

import "fmt"

func main() {
    ch1 := make(chan int, 1)
	for {
		select {
		case ch1 <- 6:
			fmt.Println("Write data: 6")
		case data := <-ch1:
			fmt.Println(data)
			break
		}
	}
}

解决方案是使用goto语句和标签。

方法1:

package main

import "fmt"

func main() {
    ch1 := make(chan int, 1)
	num := 6
	for {
		select {
		case ch1 <- num:
			fmt.Println("Write data: ", num)
		case data := <-ch1:
			fmt.Println("Read data: ", data)
			goto loop
		}
	}	
	loop:
	fmt.Println(ch1)
}

执行结果:

Write data:  6
Read data:  6
0xc00000e0e0

方法2:

package main

import "fmt"

func main() {
	ch1 := make(chan int, 1)
	num := 6
	loop:
	for {
		select {
		case ch1 <- num:
			fmt.Println("Write data: ", num)
		case data := <-ch1:
			fmt.Println("Read data: ", data)
			break loop
		}
	}
	fmt.Println(ch1)
}

执行结果:

Write data:  6
Read data:  6
0xc0000e4000

总结

本文主要介绍了通道的基本操作:初始化、发送、接收和关闭,要注意在什么情况下会引起通道阻塞。select语句通常与通道联用,介绍了分支的选择规则以及for语句与select语句联用时如何退出循环。

通道是 Go 语言并发编程的重要实现基础,还是有必要掌握的。

--THE END--
posted @ 2021-09-14 21:49  测试开发小记  阅读(238)  评论(0编辑  收藏  举报