淺談Golang中select的用法

達聞西發表於2019-02-16
Golang

在go中，select主要是和channel有關，大概的格式如下：

select{
case <- ch1:
// do something
case i := <- ch2
// use i do something
default:
// ...
}

這個語法看起來和switch語句很接近，但是具體select有什麼用呢？我們來看一段程式碼：

  • code-1
func main() {
    ch := make(chan int)
    for i := 0; i < 10; i++{
        select {
        case ch <- i:
        case x := <- ch:
            fmt.Println(x)
        }
    }
}

這段程式碼裡我們先建立了一個channel(管道)，這個管道是無緩衝的，那麼這段程式碼是否能正常的執行呢？

答案是不能的，這段程式碼會產生一個deadlock。

那麼我們稍微修改一下這個程式碼，

ch := make(chan int, 1)

在這裡我們將剛才無緩衝的管道改變為有緩衝且緩衝大小為1的管道，再次執行之後，會顯示正確的結果，但是這個結果比較微妙：

0
2
4
6
8

因為這個管道的緩衝值只有1，那麼同一時間只會有一個case執行，這個channel不是空的就是滿的。
所以這個結果是比較固定的，但是為什麼只會輸出這麼幾個數字呢？那我們來一起看看具體是怎麼執行的。

在第一次進入迴圈的時候，i為0，進入到select中，開始由上向下來發現哪一個case可以執行，當計算表示式
ch <- i,也就是向管道寫入資料的時候，因為這個管道現在有緩衝，那麼在向管道寫完資料之後，此時的case便執行完成，然後就跳出select，開始進行下一次的迴圈，當i=1的時候(現在這個管道里面的資料是0)，再次進入select中，此時還是開始計算ch <- i 表示式，但是現在管道里面是有資料的，再次向管道中寫入資料，那麼會使該傳送操作阻塞，此時該case便無法再執行，那麼select將會繼續向下執行下一個case，在下一個case中，有一個管道的接收操作x := <- ch，在這裡管道里有之前第一次迴圈的時候放入的0這個資料，那麼在這裡就會將管道的資料賦值給x，從而列印出第一個資料0，那麼後面的資料就和之前的過程是一樣的了。

看到這裡，大概就能明白select的作用了，順便說一下，select的case語句中，都是對應一個I/O操作，準確的說是對應一個channel的I/O操作，那麼到這裡也應該可以理解為什麼在code-1中，一個無緩衝的channel能在那段程式碼中產生一個deadlock。

這裡還引出幾個概念：channel, buffer channel, goroutine等。

OK！今天就先到這裡，下次再繼續分享Golang的特色之一，goroutine。

相關文章