golang协程的使用-2018-11-06

1. 创建一个协程

一般直接在要执行的函数前,加上 go 就是启用了一个执行该函数的协程。
比如下面的小例子,启动say函数的协程。

package main

import (
    "fmt"
    "time"
)

func say(s string) {
    fmt.Println(s)
}

func main() {
    for i := 0; i < 5; i++ {
        go say("Hello")
    }
}

这样已经是在main函数中启动了一个协程,去执行say函数。但是结果却什么都没打印出来。
后来理解了一下,go启动了一个协程,启动之后,系统不会阻塞,程序可以继续执行后面的流程。这个go协程启动之后,后面直接到了main函数的结束的地方,所以main函数直接结束了,但是协程启动的函数还没来得及执行,所以没有任何信息输出。
可以再main函数结束前,加1s的等待,这样在等待的过程中,协程启动的函数也就执行完输出了结果,最终main函数结束。

func main() {
    for i := 0; i < 5; i++ {
        go say("Hello")
    }
    time.Sleep(time.Second * 1)
}

这也是golang中协程使用需要注意的一个点吧。启动了之后,要注意被启动的函数的正常执行,当所有协程执行完成后,才让main函数退出。

需要配合channel或者waitgroup去使用协程,阻塞主程序的退出。

2. 配合waitGroup启动协程

  • 定义一个WaitGroup对象wg。注意这里是一个全局的变量,使用时都是这个变量的内容,所以不需要定义为指针类型,否则这样声明会是个空指针。
  • wg.Add(n),n是正整数是说有几个协程要处理
  • wg.Done(),他就相当于wg.Add(-1),执行完一次go协程的函数,可以启动一次wg.Done(),n也就-1了。
  • wg.Wait(),是告诉主函数去等一等,等到全部的协程执行完,也就是最终总的n为0的时候,所有的协程都处理完了,main函数结束。
package main

import (
    "fmt"
    "sync"
)

var wg sync.WaitGroup

func say(s string) {
    defer wg.Done()
    fmt.Println(s)
}

func main() {
    for i := 0; i < 5; i++ {
        wg.Add(1)
        go say("Hello")
    }
    wg.Wait()
}

3.配合channel使用协程

上面使用了waitgroup,可以满足所有的协程执行完后,在退出主程序。我们还要学会配合channel去使用协程。
因为我们刚才是直接打印的协程启动的函数里的数据,但是并没有实现数据的传输。channel就是处理协程的数据传输的通道了。
关于channel的用法以后可以再其他文章中补充,这里就直接看怎么能结合goroutine了。

  • 初始化一个channel,有缓冲
  • 启动协程时,在函数中通过channel写入传的参数。
  • 读取channel中的数据
package main

import (
    "fmt"
    "sync"
)

var wg sync.WaitGroup
var ch = make(chan int, 5)

func say(s int) {
    defer wg.Done()
    ch <- s //将s写入ch
    fmt.Println("写入:", s)
}

func main() {
    for i := 0; i < 5; i++ {
        wg.Add(1)
        go say(i)
    }
    wg.Wait() //所有协程执行完了,数据都写入了
    close(ch) //关闭ch
    for v := range ch {
        //遍历出ch中的值
        fmt.Println("v", v)
    }
}

当然,channel的使用还有很多坑,也需要注意,以后遇到记录

你可能感兴趣的:(golang协程的使用-2018-11-06)