Golang并发编程之runtime包
runtime包里面定义了一些协程管理相关的api
1.runtime.Gosched()
让出时间片,重新等待安排任务
package main
import (
"fmt"
"runtime"
)
func show(s string) {
for i := 0; i < 2; i++ {
fmt.Println(s)
}
}
func main() {
go show("Python") // 子协程
for i := 0; i < 2; i++ {
runtime.Gosched()
fmt.Println("golang")
}
}
2.runtime.Goexit()
退出当前协程
package main
import (
"fmt"
"runtime"
"time"
)
func show() {
for i := 0; i < 10; i++ {
fmt.Printf("i: %v\n", i)
if i >= 5 {
runtime.Goexit()
}
}
}
func main() {
go show()
time.Sleep(time.Second)
}
3.runtime.GOMAXPROCS
修改运行核心数
package main
import (
"fmt"
"runtime"
"time"
)
func a() {
for i := 0; i < 10; i++ {
fmt.Printf("A: %v\n", i)
}
}
func b() {
for i := 0; i < 10; i++ {
fmt.Printf("B: %v\n", i)
}
}
func main() {
fmt.Printf("runtime.NumCPU(): %v\n", runtime.NumCPU()) // 查看CPU数量
runtime.GOMAXPROCS(1)
go a()
go b()
time.Sleep(time.Second)
}