go語言中如何使用select的實現示例

Posted on by

在golang語言中，select語句 就是用來監聽和channel有關的IO操作，當IO操作發生時，觸發相應的case動作。

有瞭 select語句，可以實現 main主線程 與 goroutine線程 之間的互動。

1.基本語法

select {
    case <-ch1 :     // 檢測有沒有數據可讀
        // 一旦成功讀取到數據，則進行該case處理語句
    case ch2 <- 1 :  // 檢測有沒有數據可寫
        // 一旦成功向ch2寫入數據，則進行該case處理語句
    default:
        // 如果以上都沒有符合條件，那麼進入default處理流程
}

註意事項

  • select語句 隻能用於channel信道的IO操作，每個case都必須是一個信道。
  • 如果不設置 default條件，當沒有IO操作發生時，select語句就會一直阻塞；
  • 如果有一個或多個IO操作發生時，Go運行時會隨機選擇一個case執行，但此時將無法保證執行順序；
  • 對於case語句，如果存在信道值為nil的讀寫操作，則該分支將被忽略，可以理解為相當於從select語句中刪除瞭這個case；
  • 對於空的 select語句，會引起死鎖；
  • 對於在 for中的select語句，不能添加 default，否則會引起cpu占用過高的問題；

（1）多個IO操作發生時，case語句是隨機執行的

func main()  {
    ch1 := make(chan int, 1)   // 創建 一個長度帶緩沖的整型通道
    ch1 <- 1                   // 向通道中寫入數據

    ch2 := make(chan int, 1)
    ch2 <- 2

    select {
        case <- ch1:
            fmt.Println("ch1 read")
        case <- ch2:
            fmt.Println("ch2 read")
    }
}

多次執行後，會隨機打印 “ch1 read” 或 “ch2 read”

（2）空select語句

func main()  {
    select {
    
    }
}

執行後，引發死鎖，打印如下：

fatal error: all goroutines are asleep – deadlock!

goroutine 1 [select (no cases)]:
main.main()
        xxx/test.go:4 +0x27
exit status 2

（3）for中的select 引起CPU資源消耗過高

func main()  {
    quit := make(chan bool)
    go func() {
        for {
            select {
            case <-quit:
                fmt.Println("quit")
                // 使用 return 就會退出整個goroutine線程；如果使用 break，程序仍然在for循環中執行
                return
            default:
                fmt.Println("default")
            }
        }
    }()

    time.Sleep(3 * time.Second)
    quit <- true          // 主線程在3秒後，向quit信道寫入數據
    
    time.Sleep(2 * time.Second)
    fmt.Println("main")
}

在for{}的select語句中使用瞭 default後，線程就會無限執行default條件，直到quit信道中讀到數據，否則會一直在一個死循環中運行，從而導致占滿整個CPU資源。

在 for{}的select語句中，不建議使用 default條件。

2.select語句的實際應用

（1）實現 main主線程與 goroutine線程 之間的交互、通信

// 通過控制臺輸入 "bye"， 來控制main函數結束運行
func main()  {
    quit := make(chan bool)
    ch := make(chan string)

    go func() {
        for {
            select {
            case name := <-ch:
                fmt.Printf("from main msg: [%v]\n", name)
                if name == "bye" {
                    quit <- true
                } else {
                    quit <- false
                }
            }
        }
    }()

    for {
        // 控制臺輸入
        fmt.Print("please input string: ")
        scanner := bufio.NewScanner(os.Stdin)
        scanner.Scan()
        ch <- scanner.Text()

        isOver := <- quit
        if isOver {
            break
        }
    }
    fmt.Println("main over")
}

運行：

from main msg: [aaa]
please input string: bbb
from main msg: [bbb]
please input string: bye
from main msg: [bye]
main over

（2）超時實現

func main()  {
    quit := make(chan bool)
    ch := make(chan int)

    go func() {
        for {
            select {
            case num := <- ch:
                fmt.Println("num = ", num)
            case <- time.After(5 * time.Second):
                fmt.Println("超時")
                quit <- true
            }
        }
    }()

    for i := 0; i < 2; i++ {
        ch <- i
        time.Sleep(time.Second)
    }
    <- quit                   // 等待超時後, 結束 main主線程
    fmt.Println("程序結束")
}

到此這篇關於go語言中如何使用select的實現示例的文章就介紹到這瞭,更多相關go語言使用select內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：