golang select 用法

序文

select 是用来配合channel使用的

空 select

• 没有内容的select 会阻塞
• 没有内容是指，没有case，也没有default
• 如果没有其它的任务指执行，将会触发死锁

package main

import (
    "fmt"
    "time"
)

/**
* 没有内容的select 会阻塞
* 没有内容是指，没有case，也没有default
* 如果没有其它的任务指执行，将会触发死锁
*/
func main() {
    go func() {
        time.Sleep(2 * time.Second)
        fmt.Println("没有内容的select 会阻塞")
    }()
    select {}
}

只有default的 select

• 只有default的select 和串行化没有区别

package main

import "fmt"


/**
 只有default的select 和串行化没有区别
 */
func main() {
	go func() {
		fmt.Println("quick")
	}()
	select {
	default:
		fmt.Println("end")
	}
}

带 case 的 select

有case，有 default

• 如果能匹配到case 就 执行 case
• 匹配不到case，就执行default
• 有 default，就代表了不会阻塞

package main

import (
	"fmt"
)

func main() {
	ch1 := make(chan int, 2)
	ch2 := make(chan int, 2)

	select {
	case v1 := <-ch1:
		fmt.Println(v1)
	case v2 := <-ch2:
		fmt.Println(v2)
	default:
		fmt.Println(22)
	}
}

package main

import (
	"fmt"
	"time"
)

func main() {
	ch1 := make(chan int, 2)
	ch2 := make(chan int, 2)

	go func() {
		ch1 <- 1
	}()

	time.Sleep(1 * time.Second)

	select {
	case v1 := <-ch1:
		fmt.Printf("get v1 chan value %d", v1)
	case v2 := <-ch2:
		fmt.Printf("get v1 chan value %d", v2)
	default:
		fmt.Println(22)
	}
}

有case，无default

• 会阻塞 一直等到case匹配上

package main

import (
	"fmt"
	"time"
)

func main() {
	ch1 := make(chan int, 2)
	ch2 := make(chan int, 2)

	fmt.Printf("start unix:%d \n", time.Now().Unix())
	go func() {
		time.Sleep(3 * time.Second)
		ch1 <- 1
	}()

	select {
	case v1 := <-ch1:
		fmt.Printf("case unix: %d \n", time.Now().Unix())
		fmt.Printf("get v1 chan value %d \n", v1)
	case v2 := <-ch2:
		fmt.Printf("get v1 chan value %d \n", v2)
	}

	fmt.Println("end")
}

select 只匹配一次，如果要进行 n > 1 的 匹配，使用 for + select

package main

import (
	"fmt"
	"time"
)

func main() {
	ch1 := make(chan int, 2)
	ch2 := make(chan int, 2)

	fmt.Printf("start unix:%d \n", time.Now().Unix())
	go func() {
		for {
			time.Sleep(1 * time.Second)
			ch1 <- 1
		}
	}()

	for {
		select {
		case v1 := <-ch1:
			fmt.Printf("case unix: %d \n", time.Now().Unix())
			fmt.Printf("get v1 chan value %d \n", v1)
		case v2 := <-ch2:
			fmt.Printf("get v1 chan value %d \n", v2)
		}
	}

}

匹配是无序的

package main

import (
	"fmt"
	"time"
)

func test() {
	ch1 := make(chan int)
	ch2 := make(chan int)

	go func() {
		ch1 <- 1
		close(ch1)
	}()

	go func() {
		time.Sleep(1 * time.Second)
		ch2 <- 1
		close(ch2)
	}()

	time.Sleep(2 * time.Second)

	// 如果有顺序，那么因该每次都是v1
	select {
	case v1 := <-ch1:
		fmt.Printf("case unix: %d \n", time.Now().Unix())
		fmt.Printf("get v1 chan value %d \n", v1)
	case v2 := <-ch2:
		fmt.Printf("case unix: %d \n", time.Now().Unix())
		fmt.Printf("get v2 chan value %d \n", v2)
	}
}

func main() {
	for i := 0; i < 10; i++ {
		test()
	}
}