前言

在 Go 語言中，Goroutine 和 Channel 是非常重要的并發(fā)編程概念，它們可以幫助我們解決并發(fā)編程中的各種問題。關(guān)于它們的基本概念和用法，前面的文章 一文初探 Goroutine 與 channel 中已經(jīng)進(jìn)行了介紹。而本文將重點(diǎn)介紹 select，它是協(xié)調(diào)多個(gè) channel 的橋梁。

select 介紹

什么是 select

select 是 Go 語言中的一種控制結(jié)構(gòu)，用于在多個(gè)通信操作中選擇一個(gè)可執(zhí)行的操作。它可以協(xié)調(diào)多個(gè) channel 的讀寫操作，使得我們能夠在多個(gè) channel 中進(jìn)行非阻塞的數(shù)據(jù)傳輸、同步和控制。

為什么需要 select

Go 語言中的 select 語句是一種用于多路復(fù)用通道的機(jī)制，它允許在多個(gè)通道上等待并處理消息。相比于簡(jiǎn)單地使用 for 循環(huán)遍歷通道，使用 select 語句能夠更加高效地管理多個(gè)通道。

以下是一些 select 語句的使用場(chǎng)景：

1.等待多個(gè)通道的消息（多路復(fù)用）

當(dāng)我們需要等待多個(gè)通道的消息時(shí)，使用 select 語句可以非常方便地等待這些通道中的任意一個(gè)通道有消息到達(dá)，從而避免了使用多個(gè)goroutine進(jìn)行同步和等待。

2.超時(shí)等待通道消息

當(dāng)我們需要在一段時(shí)間內(nèi)等待某個(gè)通道有消息到達(dá)時(shí)，使用 select 語句可以與 time 包結(jié)合使用實(shí)現(xiàn)定時(shí)等待。

3.在通道上進(jìn)行非阻塞讀寫

在使用通道進(jìn)行讀寫時(shí)，如果通道沒有數(shù)據(jù)，讀操作或?qū)懖僮鲗?huì)阻塞。但是使用 select 語句結(jié)合 default 分支可以實(shí)現(xiàn)非阻塞讀寫，從而避免了死鎖或死循環(huán)等問題。

因此，select 的主要作用是在處理多個(gè)通道時(shí)提供了一種高效且易于使用的機(jī)制，簡(jiǎn)化了多個(gè) goroutine 的同步和等待，使程序更加可讀、高效和可靠。

select 基礎(chǔ)

語法

select {
    case <- channel1:
        // channel1準(zhǔn)備好了
    case data := <- channel2:
        // channel2準(zhǔn)備好了，并且可以讀取到數(shù)據(jù)data
    case channel3 <- data:
        // channel3準(zhǔn)備好了，并且可以往其中寫入數(shù)據(jù)data
    default:
        // 沒有任何channel準(zhǔn)備好了
}

其中， <- channel1 表示讀取 channel1 的數(shù)據(jù)，data <- channel2 表示用 data 去接收數(shù)據(jù)；channel3 <- data 表示往 channel3 中寫入數(shù)據(jù)。

select 的語法形式類似于 switch，但是它只能用于 channel 操作。在 select 語句中，我們可以定義多個(gè) case，每個(gè) case 都是一個(gè) channel 操作，用于讀取或?qū)懭霐?shù)據(jù)。如果有多個(gè) case 同時(shí)可執(zhí)行，則會(huì)隨機(jī)選擇其中一個(gè)。如果沒有任何可執(zhí)行的 case，則會(huì)執(zhí)行 default 分支（如果存在），或者阻塞等待直到至少有一個(gè) case 可執(zhí)行為止。

基本用法

package main

import (
   "fmt"
   "time"
)

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

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

   go func() {
      time.Sleep(2 * time.Second)
      ch2 <- 2
   }()
   for i := 0; i < 2; i++ {
      select {
      case data, ok := <-ch1:
         if ok {
            fmt.Println("從 ch1 接收到數(shù)據(jù):", data)
         } else {
            fmt.Println("通道已被關(guān)閉")
         }
      case data, ok := <-ch2:
         if ok {
            fmt.Println("從 ch2接收到數(shù)據(jù): ", data)
         } else {
            fmt.Println("通道已被關(guān)閉")
         }
      }
   }

   select {
   case data, ok := <-ch1:
      if ok {
         fmt.Println("從 ch1 接收到數(shù)據(jù):", data)
      } else {
         fmt.Println("通道已被關(guān)閉")
      }
   case data, ok := <-ch2:
      if ok {
         fmt.Println("從 ch2接收到數(shù)據(jù): ", data)
      } else {
         fmt.Println("通道已被關(guān)閉")
      }
   default:
      fmt.Println("沒有接收到數(shù)據(jù)，走 default 分支")
   }
}

執(zhí)行結(jié)果

從 ch1 接收到數(shù)據(jù): 1
從 ch2接收到數(shù)據(jù):  2
沒有接收到數(shù)據(jù)，走 default 分支

上述示例中，首先創(chuàng)建了兩個(gè) channel，ch1 和 ch2，分別在不同的 goroutine 中向兩個(gè) channel 中寫入數(shù)據(jù)。然后，在主 goroutine 中使用 select 語句監(jiān)聽兩個(gè)channel，一旦某個(gè) channel 上有數(shù)據(jù)流動(dòng)，就打印出相應(yīng)的數(shù)據(jù)。由于 ch1 中的數(shù)據(jù)比 ch2 中的數(shù)據(jù)先到達(dá)，因此首先會(huì)打印出 "從 ch1 接收到數(shù)據(jù): 1"，然后才打印出 "從 ch2接收到數(shù)據(jù): 2"。

為了方便測(cè)試 default 分支，我寫了兩個(gè) select 代碼塊，執(zhí)行到第二個(gè) select 代碼塊的時(shí)候，由于 ch1 和 ch2 都沒有數(shù)據(jù)了，因此執(zhí)行 default 分支，打印 "沒有接收到數(shù)據(jù)，走 default 分支"。

一些使用 select 與 channel 結(jié)合的場(chǎng)景

實(shí)現(xiàn)超時(shí)控制

package main

import (
   "fmt"
   "time"
)

func main() {
   ch := make(chan int)
   go func() {
      time.Sleep(3 * time.Second)
      ch <- 1
   }()

   select {
   case data, ok := <-ch:
      if ok {
         fmt.Println("接收到數(shù)據(jù): ", data)
      } else {
         fmt.Println("通道已被關(guān)閉")
      }
   case <-time.After(2 * time.Second):
      fmt.Println("超時(shí)了！")
   }
}

執(zhí)行結(jié)果為：超時(shí)了！。

在這個(gè)例子中，程序?qū)⒃?3 秒后向 ch 通道里寫入數(shù)據(jù)，而我在 select 代碼塊里設(shè)置的超時(shí)時(shí)間為 2 秒，如果在 2 秒內(nèi)沒有接收到數(shù)據(jù)，則會(huì)觸發(fā)超時(shí)處理。

實(shí)現(xiàn)多任務(wù)并發(fā)控制

package main

import (
   "fmt"
)

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

   for i := 0; i < 10; i++ {
      go func(id int) {
         ch <- id
      }(i)
   }

   for i := 0; i < 10; i++ {
      select {
      case data, ok := <-ch:
         if ok {
            fmt.Println("任務(wù)完成：", data)
         } else {
            fmt.Println("通道已被關(guān)閉")
         }
      }
   }
}

執(zhí)行結(jié)果（每次執(zhí)行的順序都會(huì)不一致）：

任務(wù)完成： 1
任務(wù)完成： 5
任務(wù)完成： 2
任務(wù)完成： 3
任務(wù)完成： 4
任務(wù)完成： 0
任務(wù)完成： 9
任務(wù)完成： 6
任務(wù)完成： 7
任務(wù)完成： 8

在這個(gè)例子中，啟動(dòng)了 10 個(gè) goroutine 并發(fā)執(zhí)行任務(wù)，并使用一個(gè) channel 來接收任務(wù)的完成情況。在主函數(shù)中，使用 select 語句監(jiān)聽這個(gè) channel，每當(dāng)接收到一個(gè)完成的任務(wù)時(shí)，就進(jìn)行處理。

監(jiān)聽多個(gè)通道的消息

package main

import (
   "fmt"
   "time"
)

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

   // 開啟 goroutine 1 用于向通道 ch1 發(fā)送數(shù)據(jù)
   go func() {
      for i := 0; i < 5; i++ {
         ch1 <- i
         time.Sleep(time.Second)
      }
   }()

   // 開啟 goroutine 2 用于向通道 ch2 發(fā)送數(shù)據(jù)
   go func() {
      for i := 5; i < 10; i++ {
         ch2 <- i
         time.Sleep(time.Second)
      }
   }()

   // 主 goroutine 從 ch1 和 ch2 中接收數(shù)據(jù)并打印
   for i := 0; i < 10; i++ {
      select {
      case data := <-ch1:
         fmt.Println("Received from ch1:", data)
      case data := <-ch2:
         fmt.Println("Received from ch2:", data)
      }
   }

   fmt.Println("Done.")
}

執(zhí)行結(jié)果（每次執(zhí)行程序打印的順序都不一致）：

Received from ch2: 5
Received from ch1: 0
Received from ch1: 1
Received from ch2: 6
Received from ch1: 2
Received from ch2: 7
Received from ch1: 3
Received from ch2: 8
Received from ch1: 4
Received from ch2: 9
Done.

該示例代碼中，通過使用 select 多路復(fù)用，可以同時(shí)監(jiān)聽多個(gè)通道的數(shù)據(jù)，并避免了使用多個(gè) goroutine 進(jìn)行同步和等待的問題。

使用 default 實(shí)現(xiàn)非阻塞讀寫

import (
   "fmt"
   "time"
)

func main() {
   ch := make(chan int, 1)

   go func() {
      for i := 1; i <= 5; i++ {
         ch <- i
         time.Sleep(1 * time.Second)
      }
      close(ch)
   }()

   for {
      select {
      case val, ok := <-ch:
         if ok {
            fmt.Println(val)
         } else {
            ch = nil
         }
      default:
         fmt.Println("No value ready")
         time.Sleep(500 * time.Millisecond)
      }
      if ch == nil {
         break
      }
   }
}

執(zhí)行結(jié)果（每次執(zhí)行程序打印的順序都不一致）：

No value ready
1
No value ready
2
No value ready
No value ready
3
No value ready
No value ready
4
No value ready
No value ready
5
No value ready
No value ready

這個(gè)代碼中，使用了 default 分支來實(shí)現(xiàn)非阻塞的通道讀取和寫入操作。在 select 語句中，如果有通道已經(jīng)準(zhǔn)備好進(jìn)行讀寫操作，那么就會(huì)執(zhí)行相應(yīng)的分支。但是如果沒有任何通道準(zhǔn)備好讀寫，那么就會(huì)執(zhí)行 default 分支中的代碼。

select 的注意事項(xiàng)

以下是關(guān)于 select 語句的一些注意事項(xiàng)：

• select 語句只能用于通信操作，如 channel 的讀寫，不能用于普通的計(jì)算或函數(shù)調(diào)用。
• select 語句會(huì)阻塞，直到至少有一個(gè) case 語句滿足條件。 如果有多個(gè) case 語句滿足條件，則會(huì)隨機(jī)選擇一個(gè)執(zhí)行。
• 如果沒有 case 語句滿足條件，并且有 default 語句，則會(huì)執(zhí)行 default 語句。
• 在 select 語句中使用 channel 時(shí)，必須保證 channel 是已經(jīng)初始化的。
• 如果一個(gè)通道被關(guān)閉，那么仍然可以從它中讀取數(shù)據(jù)，直到它被清空，此時(shí)會(huì)返回通道元素類型的零值和一個(gè)布爾值，指示通道是否已關(guān)閉。

總之，在使用 select 語句時(shí)，要仔細(xì)考慮每個(gè) case 語句的條件和執(zhí)行順序，避免死鎖和其他問題。

總結(jié)

本文主要介紹了 Go 語言中的 select 語句。在文章中，首先介紹了 select 的基本概念，包括它是一種用于在多個(gè)通道之間進(jìn)行選擇的語句，以及為什么需要使用 select。

接下來，文章詳細(xì)介紹了 select 的基礎(chǔ)知識(shí)，包括語法和基礎(chǔ)用法。在語法方面，講解了 select 語句的基本結(jié)構(gòu)以及如何使用 case 子句進(jìn)行通道選擇。在基礎(chǔ)用法方面，介紹了如何使用 select 語句進(jìn)行通道的讀取和寫入操作，并講解了一些注意事項(xiàng)。

在接下來的內(nèi)容中，文章列舉了一些使用 select 與 channel 結(jié)合的場(chǎng)景。這些場(chǎng)景包括實(shí)現(xiàn)超時(shí)控制、實(shí)現(xiàn)多任務(wù)并發(fā)控制、監(jiān)聽多個(gè)通道的消息以及使用 default 實(shí)現(xiàn)非阻塞讀寫。對(duì)于每個(gè)場(chǎng)景，文章都詳細(xì)介紹了如何使用 select 語句實(shí)現(xiàn)。

最后，文章總結(jié)了 select 的注意事項(xiàng)，包括選擇的通道必須是可讀或可寫的通道、select 語句中的 case 子句必須是通道操作或者空的 default 子句，不能是其他類型的語句等等。

到此這篇關(guān)于一文帶你學(xué)會(huì)Go select語句輕松實(shí)現(xiàn)高效并發(fā)的文章就介紹到這了,更多相關(guān)Go select語句內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論