Go-select实现超时

Go-select实现超时

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 < 3; i++ {
        ch <- i
    }
    <-quit
    fmt.Println("程序结束")
}