站长资源脚本专栏
Golang Goroutine的使用
什么是 Goroutine
goroutine 是 Go 并行设计的核心。goroutine 说到底其实就是协程,它比线程更小,十几个 goroutine 可能体现在底层就是五六个线程,Go 语言内部帮你实现了这些 goroutine 之间的内存共享。
执行 goroutine 只需极少的栈内存(大概是4~5KB),当然会根据相应的数据伸缩。也正因为如此,可同时运行成千上万个并发任务。goroutine 比 thread 更易用、更高效、更轻便。
一般情况下,一个普通计算机跑几十个线程就有点负载过大了,但是同样的机器却可以轻松地让成百上千个 goroutine 进行资源竞争。
Goroutine 的创建
只需在函数调"htmlcode">
package main import ( "fmt" "time" ) func foo() { i := 0 for true { i++ fmt.Println("new goroutine: i = ", i) time.Sleep(time.Second) } } func main() { // 创建一个 goroutine, 启动另外一个任务 go foo() i := 0 for true { i++ fmt.Println("main goroutine: i = ", i) time.Sleep(time.Second) } }
结果:
main goroutine: i = 1
new goroutine: i = 1
new goroutine: i = 2
main goroutine: i = 2
main goroutine: i = 3
new goroutine: i = 3
...
Goroutine 特性
主go程 退出后,其它的 子go程 也会自动退出:
package main import ( "fmt" "time" ) func foo() { i := 0 for true { i++ fmt.Println("new goroutine: i = ", i) time.Sleep(time.Second) } } func main() { // 创建一个 goroutine, 启动另外一个任务 go foo() time.Sleep(time.Second * 3) fmt.Println("main goroutine exit") }
运行结果:
new goroutine: i = 1
new goroutine: i = 2
new goroutine: i = 3
main goroutine exit
runtime 包
Gosched
runtime.Gosched() 用于出让当前 go程 所占用的 CPU 时间片,让出当前 goroutine 的执行权限,调度器安排其他等待的任务运行,并在下次再获得 cpu 时间轮片的时候,从该出让 cpu 的位置恢复执行。
有点像跑接力赛,A 跑了一会碰到代码 runtime.Gosched() 就把接力棒交给 B 了,A 歇着了,B 继续跑。
例如:
package main import ( "fmt" "runtime" "time" ) func main() { // 创建一个 goroutine go func(s string) { for i := 0; i < 2; i++ { fmt.Println(s) } }("world") for i := 0; i < 2; i++ { runtime.Gosched() fmt.Println("hello") } time.Sleep(time.Second * 3) }
运行结果:
world
world
hello
hello
如果没有 runtime.Gosched() 则运行结果如下:
hello
hello
world
world
注意: runtime.Gosched() 只是出让一次机会,看下面的代码,注意运行结果:
package main import ( "fmt" "runtime" "time" ) func main() { // 创建一个 goroutine go func(s string) { for i := 0; i < 2; i++ { fmt.Println(s) time.Sleep(time.Second) } }("world") for i := 0; i < 2; i++ { runtime.Gosched() fmt.Println("hello") } }
运行结果:
world
hello
hello
为什么 world 只有一次呢?因为之前我们说过,主 goroutine 退出后,其它的工作 goroutine 也会自动退出。
Goexit
调用 runtime.Goexit() 将立即终止当前 goroutine 执"htmlcode">
package main import ( "fmt" "runtime" "time" ) func main() { go func() { defer fmt.Println("A.defer") func() { defer fmt.Println("B.defer") runtime.Goexit() // 终止当前 goroutine fmt.Println("B") // 不会执行 }() fmt.Println("A") // 不会执行 }() // 不要忘记 () time.Sleep(time.Second * 3) }
运行结果:
B.defer
A.defer
GOMAXPROCS
调用 runtime.GOMAXPROCS() 用来设置可以并行计算的 CPU 核数的最大值,并返回 上一次(没有则是电脑默认的) 设置的值。
package main import ( "fmt" "runtime" ) func main() { runtime.GOMAXPROCS(1) // 将 cpu 设置为单核 for true { go fmt.Print(0) // 子 go 程 fmt.Print(1) // 主 go 程 } }
运行结果:
111111 ... 1000000 ... 0111 ...
在执行 runtime.GOMAXPROCS(1) 时,最多同时只能有一个 goroutine 被执行。所以会打印很多 1。过了一段时间后,GO 调度器会将其置为休眠,并唤醒另一个 goroutine,这时候就开始打印很多 0 了,在打印的时候,goroutine 是被调度到操作系统线程上的。
package main import ( "fmt" "runtime" ) func main() { runtime.GOMAXPROCS(2) for true { go fmt.Print(0) fmt.Print(1) } }
运行结果:
111111111111111000000000000000111111111111111110000000000000000011111111100000...
在执行 runtime.GOMAXPROCS(2)
时, 我们使用了两个 CPU,所以两个 goroutine 可以一起被执行,以同样的频率交替打印 0 和 1。
runtime 包中的其它函数
中文文档在这里:https://studygolang.com/pkgdoc
这里就简单列举一下一些函数以及功能。
func GOROOT() string
GOROOT 返回 Go 的根目录。如果存在 GOROOT 环境变量,返回该变量的值;否则,返回创建 Go 时的根目录。
func Version() string
返回 Go 的版本字符串。它要么是递交的 hash 和创建时的日期;要么是发行标签如 "go1.3"。
func NumCPU() int
NumCPU返回本地机器的逻辑CPU个数(真 · 八核)。
func GC()
GC执行一次垃圾回收。(如果你迫切的希望做一次垃圾回收,可以调用此函数)