前言

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

select 介紹

什么是 select

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

為什么需要 select

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

以下是一些 select 語句的使用場景：

1.等待多個通道的消息（多路復用）

當我們需要等待多個通道的消息時，使用 select 語句可以非常方便地等待這些通道中的任意一個通道有消息到達，從而避免了使用多個goroutine進行同步和等待。

2.超時等待通道消息

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

3.在通道上進行非阻塞讀寫

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

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

select 基礎

語法

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

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

select 的語法形式類似于 switch，但是它只能用于 channel 操作。在 select 語句中，我們可以定義多個 case，每個 case 都是一個 channel 操作，用于讀取或?qū)懭霐?shù)據(jù)。如果有多個 case 同時可執(zhí)行，則會隨機選擇其中一個。如果沒有任何可執(zhí)行的 case，則會執(zhí)行 default 分支（如果存在），或者阻塞等待直到至少有一個 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("通道已被關閉")
         }
      case data, ok := <-ch2:
         if ok {
            fmt.Println("從 ch2接收到數(shù)據(jù): ", data)
         } else {
            fmt.Println("通道已被關閉")
         }
      }
   }

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

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

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

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

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

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

實現(xiàn)超時控制

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("通道已被關閉")
      }
   case <-time.After(2 * time.Second):
      fmt.Println("超時了！")
   }
}

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

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

實現(xiàn)多任務并發(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("任務完成：", data)
         } else {
            fmt.Println("通道已被關閉")
         }
      }
   }
}

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

任務完成： 1
任務完成： 5
任務完成： 2
任務完成： 3
任務完成： 4
任務完成： 0
任務完成： 9
任務完成： 6
任務完成： 7
任務完成： 8

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

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

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 多路復用，可以同時監(jiān)聽多個通道的數(shù)據(jù)，并避免了使用多個 goroutine 進行同步和等待的問題。

使用 default 實現(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

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

select 的注意事項

以下是關于 select 語句的一些注意事項：

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

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

總結(jié)

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

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

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

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

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

最新評論