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的容量以及里面现在有几个值。

推荐阅读更多精彩内容