Go语言进门13之runtime包案例讲解

// 定义两个函数，Test01和Test02
// Test02在Test01运行结束之后运行
// 并且Test02在i = 3的时候结束运行
//Test01
func Test01(wg *sync.WaitGroup) {
	defer func() {
		fmt.Println("Test01结束!")
		wg.Done()
	}()
	fmt.Println("协程01循环输出：")
	for i := 0; i < 5; i++ {
		fmt.Println("协程01 -", i)
	}
}
// Test02
func Test02(wg *sync.WaitGroup) {
	defer func() {
		fmt.Println("Test02结束!")
		wg.Done()
	}()
    // 等待Test01运行结束
	runtime.Gosched()
	fmt.Println("协程02循环输出：")
	for i := 0; i < 5; i++ {
		if i == 3 {
			fmt.Println("i = 3 退出!")
            // 当i = 3时执行defer里的语句然后退出
			runtime.Goexit()
		}
		fmt.Println("协程02 -", i)
	}
}
// main函数调用
func main() {
	var wg sync.WaitGroup
	wg.Add(2)
	go Test01(&wg)
	go Test02(&wg)
	wg.Wait()
	fmt.Println("This is main function!")
}
/*输出：
协程01循环输出：
协程01 - 0
协程01 - 1
协程01 - 2
协程01 - 3
协程01 - 4
Test01结束!
协程02循环输出：
协程02 - 0
协程02 - 1
协程02 - 2
i = 3 退出!
Test02结束!
This is main function!
*/