- 基本使用
package main import ( "fmt" "sync" ) var wg sync.WaitGroup func hello() { fmt.Println("hello func...") wg.Done() // 通知计数器减1 } func main() { wg.Add(4) // 计数器，4个并发任务 go hello() go hello() go hello() go hello() fmt.Println("main func!") wg.Wait() // 等待所有任务执行完成 }
- 改造一下，开启10000个goroutine
package main import ( "fmt" "sync" ) var wg sync.WaitGroup func hello(i int) { fmt.Println("hello func...", i) wg.Done() } func main() { // wg.Add(10000) for i := 0; i < 10000; i++ { wg.Add(1) go hello(i) } fmt.Println("main func!") wg.Wait() }
- 将上一个例子改造成匿名函数
package main import ( "fmt" "sync" ) var wg sync.WaitGroup func main() { // wg.Add(10000) for i := 0; i < 10000; i++ { go func(i int) { fmt.Println("hello...", i) }(i) } fmt.Println("main func!") wg.Wait() }
- 指定占用CPU核心数
package main import ( "fmt" "runtime" "sync" ) var wg sync.WaitGroup func test1() { for i := 0; i < 10; i++ { fmt.Println("func test1...", i) } } func test2() { for i := 0; i < 10; i++ { fmt.Println("func test2...", i) } } func main() { runtime.GOMAXPROCS(1) // 只占用1个CPU核心 wg.Add(2) go test1() go test2() wg.Wait() }
- 带缓冲区的通道，类似于异步的操作
package main import "fmt" func main() { ch1 := make(chan int, 1) // 只能存放1个值的缓冲区通道 ch1 <- 10 // 发送 x := <-ch1 // 接收 fmt.Println(x) close(ch1) }
- 无缓冲区的通道，又称为同步通道
package main import "fmt" func main() { ch1 := make(chan int) // 无缓冲区通道，又称为同步通道， ch1 <- 10 // 此时这里会处于阻塞的状态，除非有另外一个goroutine去取值，它才会发送 x := <-ch1 fmt.Println(x) close(ch1) }
- 获取通道的容量和通道里的元素数量
package main import "fmt" func main() { ch1 := make(chan int, 10) ch1 <- 89 ch1 <- 70 fmt.Println(len(ch1)) // 获取通道中元素的数量 fmt.Println(cap(ch1)) // 获取通道的容量 close(ch1) }
- 通道和goroutine的小栗子
package main import ( "fmt" "sync" ) type myinterface interface{} var ch1 = make(chan myinterface, 1) var wg sync.WaitGroup func sendData(i myinterface) { fmt.Printf("向通道发送 %v 成功\n", i) ch1 <- i wg.Done() } func readData() { v := <-ch1 fmt.Println("从通道获取的值：", v) wg.Done() } func main() { nameArray := []string{"ttr", "tantianran"} wg.Add(2) go sendData(nameArray) go readData() wg.Wait() }
- 通道+goroutine，实现协同干活例子2
package main import "fmt" func producer(ch chan int) { for i := 0; i < 10; i++ { ch <- i } close(ch) } func consumer(ch1 chan int, ch2 chan int) { for { v, ok := <-ch1 if !ok { break } ch2 <- v * 2 } close(ch2) } func main() { ch1 := make(chan int, 100) ch2 := make(chan int, 200) go producer(ch1) go consumer(ch1, ch2) for i := range ch2 { fmt.Println(i) } }
