7-并发编程之-多路复用select

一 问题引出

在某些场景下我们需要同时从多个通道接收数据。通道在接收数据时，如果没有数据可以接收将会发生阻塞。你也许会写出如下代码使用遍历的方式来实现：

for{
    // 尝试从ch1接收值
    data, ok := <-ch1
    // 尝试从ch2接收值
    data, ok := <-ch2
    …
}

这种方式虽然可以实现从多个通道接收值的需求，但是运行性能会差很多。为了应对这种场景，Go内置了select关键字，可以同时响应多个通道的操作。

二 select多路复用的使用

2.1 基本使用

select的使用类似于switch语句，它有一系列case分支和一个默认的分支。每个case会对应一个通道的通信（接收或发送）过程。select会一直等待，直到某个case的通信操作完成时，就会执行case分支对应的语句。具体格式如下：

    select {
    case <-chan1:
       // 如果chan1成功读到数据，则进行该case处理语句
    case chan2 <- 1:
       // 如果成功向chan2写入数据，则进行该case处理语句
    default:
       // 如果上面都没有成功，则进入default处理流程
    }

2.2 选择一个准备好的信道

select可以同时监听一个或多个channel，直到其中一个channel 数据准备好

package main

import (
	"fmt"
	"time"
)

func task1(ch chan string) {
	time.Sleep(2 * time.Second)
	ch <- "task1 执行完了"
}
func task2(ch chan string) {
	time.Sleep(4 * time.Second)
	ch <- "task2 执行完了"
}
func main() {
	var (
		ch1 = make(chan string)
		ch2 = make(chan string)
	)
	go task1(ch1)
	go task2(ch2)
	select {
	case res := <-ch1:
		fmt.Println(res)
	case res := <-ch2:
		fmt.Println(res)

	}
}

2.3 随机选择一个

如果多个channel同时准备好，则随机选择一个执行

package main

import (
	"fmt"
	"time"
)

func task1(ch chan string) {
	time.Sleep(2 * time.Second)
	ch <- "task1 执行完了"
}
func task2(ch chan string) {
	time.Sleep(2 * time.Second)
	ch <- "task2 执行完了"
}
func main() {
	var (
		ch1 = make(chan string)
		ch2 = make(chan string)
	)
	go task1(ch1)
	go task2(ch2)
	select {
	case res := <-ch1:
		fmt.Println(res)
	case res := <-ch2:
		fmt.Println(res)

	}
}

2.4 判断管道是否存满

package main

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

var wg sync.WaitGroup

// 判断管道有没有存满
func main() {

	output1 := make(chan string, 5)
	// 子协程写数据
	wg.Add(1)
	go write(output1)
	wg.Wait()
}

func write(ch chan string) {
	for {
		select {
		case ch <- "hello":
			fmt.Println("write hello")
		default:
			fmt.Println("channel full")
			wg.Done()
			break
		}
		time.Sleep(time.Millisecond * 500)
	}
}