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的使用还有很多坑,也需要注意,以后遇到记录
有疑问加站长微信联系(非本文作者)