在Go語言中,可以使用sync.WaitGroup
來處理實時數據流。
sync.WaitGroup
是一個計數信號量,用來等待一組goroutine的結束。它的主要方法有三個:Add()
、Done()
和Wait()
。
首先,需要創建一個sync.WaitGroup
對象,然后在每個goroutine開始之前調用Add()
方法,表示要等待的goroutine數量增加1。在goroutine的結尾處調用Done()
方法,表示該goroutine已經結束。最后,在主goroutine中調用Wait()
方法,等待所有的goroutine都結束。
下面是一個簡單的例子,假設有一個數據流需要實時處理,處理的任務是打印每個數據的平方值:
package main
import (
"fmt"
"sync"
)
func main() {
var wg sync.WaitGroup
dataStream := []int{1, 2, 3, 4, 5}
for _, data := range dataStream {
wg.Add(1)
go process(data, &wg)
}
wg.Wait()
}
func process(data int, wg *sync.WaitGroup) {
defer wg.Done()
result := data * data
fmt.Println(result)
}
在主函數中,首先創建了一個sync.WaitGroup
對象wg
。然后,遍歷數據流,并為每個數據增加了一個等待任務。接著,啟動了一個goroutine來處理每個數據,并傳遞了&wg
作為參數。在process()
函數中,打印了每個數據的平方值,并且在函數結尾處調用了wg.Done()
來表示該goroutine已經結束。
最后,在主函數中調用wg.Wait()
來等待所有的goroutine都結束。這樣就可以保證在所有的數據都被處理完之前,主函數不會退出。