go程阻塞

前言

记录一下我自己常用的go程阻塞方法, 确认go程完成

sync.WaitGroup

主要是用来监听go程执行完成

package main

import (
	"fmt"
	"time"
	"sync"
)

func a() {
	fmt.Println("a")
	time.Sleep(1 * time.Second)
}
func b() {
	fmt.Println("b")
	time.Sleep(2 * time.Second)

}
func test() {
	xf := sync.WaitGroup{}
	lis := []string{"a", "b"}
	for _, x := range lis {
		xf.Add(1)
		go func(x string) {
			defer xf.Done()
			switch x {
			case "a":
				a()
			case "b":
				b()
			}
		}(x)
	}
	xf.Wait()
}

func main() {
	test()
}

for循环读取chan

用读取chan的方式来阻塞等待go程结束, 也可以把go程的结果放到chan里

package main

import (
	"fmt"
	"time"
)

func a() {
	fmt.Println("a")
	time.Sleep(1 * time.Second)
}
func b() {
	fmt.Println("b")
	time.Sleep(2 * time.Second)

}
func test() {
	task := make(chan interface{})
	lis := []string{"a", "b"}
	for _, x := range lis {
		go func(x string) {
			switch x {
			case "a":
				a()
			case "b":
				b()
			}
			task <- nil
		}(x)
	}
	for a := 0; a < len(lis); a++ {
		mess := <-task
		fmt.Println(mess)
	}

}

func main() {
	test()
}