go并发(一)

前几天被小伙伴发现半年前写的代码里面有一个并发的bug。回想一下,golang并发这边的知识确实忘得差不多了,打算写两篇笔记记录一下。golang里的并发主要有两种,一种是使用goroutine(类似于线程)并通过channel实现线程间通信;另一种是通过shared variables来实现。我们先来看看第一种

goroutine

首先我们来看一个goroutine的例子

func main (){
 go func() {
 for i := 0; i < 10; i++ {
 fmt.Println("foo", i)
 }
 }()
 fmt.Println("bar")
}

上面的go foo()开了一个新的goroutine,但是它并不会跑完(可能能跑一两个循环),因为main先退出了,整个程序也就结束了。为了让它能够跑完,我们可以让main等一等它。

var wg sync.WaitGroup
func main (){
 wg.Add(1)
 go func() {
 defer wg.Done()
 for i := 0; i < 10; i++ {
 fmt.Println("foo", i)
 }
 }()
 fmt.Println("bar")
 wg.Wait()
}

为了显示他们是并发的,我们多开几个goroutine,看他们的输出是否是乱序的。

var wg sync.WaitGroup
func foo(s string){
 defer wg.Done()
 for i := 0; i < 10; i++ {
 fmt.Println(s, i)
 }
}
func main (){
 wg.Add(3)
 go foo("foo")
 go foo("oof")
 go foo("bar")
 wg.Wait()
}

有兴趣的同学可以看看net/http是如何使用goroutine的

Channels

"If goroutines are the activities of a concurrent go program, channels are the connections between them"
这句话是对channel最好的解释了。channel用于在goroutine之间传递值,是一种"线程"间通信方式。

ch := make(chan int) //创建一个unbuffered channel
ch := make(chan int, 3) //创建一个buffered channel,容量为3
ch <- x //发送值到channel
x = <- ch //从channel中取出值
<- ch //取值并丢弃
close(ch) //关闭channel

语法太简单了!这年头连编程语言都要拟物化了!

unbuffered channel

发送数据到unbuffered channel的操作会阻塞发送的goroutine,直到有某个线程接收了这个channel的值。而如果一个goroutine试图去一个没有值的channel上取值,它也会被阻塞,直到这个channel上被发送了值。

func main() {
 c := make(chan int)
 go func(c chan int) {
 time.Sleep(3 * time.Second)
 fmt.Println("before received")
 fmt.Println(<- c)
 }(c)
 c <- 1
 fmt.Println("after received")
}

运行上面代码我们明显看到,在channel中的值被读取之前,main被阻塞了,直到channel被读取,main才继续运行。

func main() {
 c := make(chan int)
 go func(c chan int) {
 fmt.Println(<- c)
 fmt.Println("after received")
 }(c)
 time.Sleep(3 * time.Second)
 fmt.Println("before sending")
 c <- 1
 time.Sleep(1 * time.Second)
}

上面的代码中,新开的goroutine首先去读channel,可是由于channel中没有值,所以它被阻塞了,直到main中向channel发送值,goroutine才拿到它想要的值并继续运行。

close channel

发送完成后,可以关闭channel,关闭后所有对这个channel的写操作都会panic,而读操作依旧可以进行,当所有值都读完后,继续读该channel会得到zero value

func main() {
 naturals := make(chan int)
 squares := make(chan int)
 go func() {
 for x := 0; x < 100 ;x++ {
 naturals <- x
 }
 close(naturals)
 }()
 go func() {
 for x := range naturals{
 squares <- x * x
 }
 close(squares)
 }()
 for x := range squares{
 fmt.Println(x)
 }
}

buffered channels

buffered channels是一个有限长队列,下面我们来创建一个容量为3的buffered channels

bc := make(chan int, 3)

如果bc里面已经有了三个没有被读取的值,继续往里面发送的话程序就会出错。我们可以用cap函数和len函数来检查bc的容量以及里面现在有几个值。

本文来自:简书

感谢作者:bigtom

查看原文:go并发(一)

• 编辑
• 预览