golang并发编程之runtime包

runtime包里面定义了一些协程管理相关的api

runtime.Gosched()

让出CPU时间片，重新等待安排任务(时间片比较短)

package main

import (
	"fmt"
)

func show(s string) {
	for i := 0; i < 3; i++ {
		fmt.Println(s)
	}
}

func main() {
	go show("JAVA")
	go show("PHP")
	//主协程
	for i := 0; i < 2; i++ {
		//runtime.Gosched() //注释前后的运行结果不同
		fmt.Println("golang:", i)
	}
}

注释后（也有可能全部执行完）

golang: 0
golang: 1
JAVA

不注释 （大概率子程序全部完）

PHP
PHP
PHP
golang: 0
JAVA
JAVA
JAVA
golang: 1

runtime.Goexit()

退出当前协程

package main

import (
	"fmt"
	"runtime"
	"time"
)

func show() {
	for i := 0; i < 10; i++ {
		if i > 5 {
			runtime.Goexit() //使用前后
		}
		fmt.Printf("i: %v\n", i)
	}
}

func main() {
	go show()

	time.Sleep(time.Second)
}

未使用运行结果

i: 0
i: 1
i: 2
i: 3
i: 4
i: 5
i: 6
i: 7
i: 8
i: 9

使用后

i: 0
i: 1
i: 2
i: 3
i: 4
i: 5

runtime.GOMAXPROCS

使用最大cpu数

package main

import (
	"fmt"
	"runtime"
	"time"
)

func a() {
	for i := 0; i < 10; i++ {
		fmt.Println("a:", i)
	}
}

func b() {
	for i := 0; i < 10; i++ {
		fmt.Println("b:", i)
	}
}

func main() {
	fmt.Printf("runtime.NumCPU(): %v\n", runtime.NumCPU())
	runtime.GOMAXPROCS(2) //修改为1时，不会交替执行，大于1时，交替执行
	go a()
	go b()
	time.Sleep(time.Second)
}