Go Select使用

Go Select使用

Go中的select和channel配合使用，通过select可以监听多个channel的I/O读写事件，当 IO操作发生时，触发相应的动作。

基本用法

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

使用规则

1.如果没有default分支,select会阻塞在多个channel上，对多个channel的读/写事件进行监控。
2.如果有一个或多个IO操作可以完成，则Go运行时系统会随机的选择一个执行，否则的话，如果有default分支，则执行default分支语句，如果连default都没有，则select语句会一直阻塞，直到至少有一个IO操作可以进行。　　　

快速返回

同时监听不同的channel，做同一件工作，可以最快的返回结果。

package main

import (
	"fmt"
	"github.com/kirinlabs/HttpRequest"
)

func main() {
	ch1 := make(chan int)
	ch2 := make(chan int)
	ch3 := make(chan int)
	go Getdata("https://www.baidu.com",ch1)
	go Getdata("https://www.baidu.com",ch2)
	go Getdata("https://www.baidu.com",ch3)
	select{
		case v:=<- ch1:
			fmt.Println(v)
		case v:=<- ch2:
			fmt.Println(v)
		case v:=<- ch3:
			fmt.Println(v)
	}
}

func Getdata(url string,ch chan int){
	req,err := HttpRequest.Get(url)
	if err != nil{

	}else{
		ch <- req.StatusCode()
	}
}

随机返回

同时监控不同的channel，配上default，select也不会阻塞。

package main

import (
	"fmt"
	"github.com/kirinlabs/HttpRequest"
)

func main() {
	ch1 := make(chan int)
	ch2 := make(chan int)
	ch3 := make(chan int)
	go func(){
		for {
			Getdata("https://www.baidu.com", ch1)
			Getdata("https://cn.bing.com", ch2)
			Getdata("https://cn.bing.com", ch3)
		}
	}()
	go func(){
		for {
			select {
				case v := <-ch1:
					fmt.Println("信道1的结果：",v)
				case v := <-ch2:
					fmt.Println("信道2的结果：",v)
				case v := <-ch3:
					fmt.Println("信道3的结果：",v)
				default:
					continue
			}
		}
	}()
	select{}
}

func Getdata(url string,ch chan int){
	req,err := HttpRequest.Get(url)
	if err != nil{

	}else{
		ch <- req.StatusCode()
	}
}

通过select来检测channel的关闭事件

func TestSelect1() {
    start := time.Now()
    c := make(chan interface{})

    go func() {
        time.Sleep(2*time.Second)
        close(c)
    }()

    fmt.Println("Blocking on read...")
    select {
    case <-c:
        fmt.Printf("Unblocked %v later.\n", time.Since(start))
    }
}

注意：当close channel时，读取channel的一方会从channel中读取到value,false，此时的value一般情况下为nil。
该例子也可以用来通知当不使用channel时，关闭channel的情况。　

通过channel通知，从而退出死循环

func TestExitLoop() {
    done := make(chan interface{})

    go func() {
        time.Sleep(2*time.Second)
        close(done)
    }()

    workCounter := 0
loop:
    for {
        select {
        case <-done:
            break loop
        default:
        }

        // Simulate work
        workCounter++
        time.Sleep(1*time.Second)
    }

    fmt.Printf("在通知退出循环时，执行了%d次.\n", workCounter)
}

启动一个goroutine，该goroutine在2s后，关闭channel。此时，主协程会在select中的case <-done分支中得到通知，跳出死循环。而在此之前，会执行default分支的代码，这里是什么都不做。

超时机制

package main
import (
	"fmt"
	"time"
)
func main() {
	ch := make(chan int)
	quit := make(chan bool)
	//新开一个协程
	go func() {
		for {
			select {
			case num := <-ch:
				fmt.Println("num = ", num)
			case <-time.After(3 * time.Second):
				fmt.Println("超时")
				quit <- true
			}
		}
	}()
	for i := 0; i < 5; i++ {
		ch <- i
		time.Sleep(time.Second)
	}
	<-quit
	fmt.Println("程序结束")
}

死锁与默认情况

package main

func main() {
    ch := make(chan string)
    select {
        case <-ch:
    }
}

上面的程序中，我们在第 4 行创建了一个信道 ch。我们在 select 内部（第 6 行），试图读取信道 ch。由于没有 Go 协程向该信道写入数据，因此 select 语句会一直阻塞，导致死锁。该程序会触发运行时 panic，报错信息如下：

fatal error: all goroutines are asleep - deadlock!

goroutine 1 [chan receive]:  
main.main()  
    /tmp/sandbox416567824/main.go:6 +0x80

如果存在默认情况，就不会发生死锁，因为在没有其他 case 准备就绪时，会执行默认情况。我们用默认情况重写后，程序如下：

package main

import "fmt"

func main() {
    ch := make(chan string)
    select {
        case <-ch:
        default:
            fmt.Println("default case executed")
    }
}　　

以上程序会输出：

default case executed

如果 select 只含有值为 nil 的信道，也同样会执行默认情况。

package main

import "fmt"

func main() {
    var ch chan string
    select {
        case v := <-ch:
            fmt.Println("received value", v)
        default:
            fmt.Println("default case executed")

    }
}

在线运行程序

在上面程序中，ch 等于 nil，而我们试图在 select 中读取 ch（第 8 行）。如果没有默认情况，select 会一直阻塞，导致死锁。由于我们在 select内部加入了默认情况，程序会执行它，并输出：

default case executed

空 select

package main
func main() {
    select {}
}

我们已经知道，除非有 case 执行，select 语句就会一直阻塞着。在这里，select 语句没有任何 case，因此它会一直阻塞，导致死锁。该程序会触发 panic，输出如下：

fatal error: all goroutines are asleep - deadlock!

goroutine 1 [select (no cases)]:  
main.main()  
    /tmp/sandbox299546399/main.go:4 +0x20