溫馨提示×

實時數據處理:利用Go WaitGroup處理數據流

小云
116
2023-10-08 14:00:52
欄目: 編程語言

在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都結束。這樣就可以保證在所有的數據都被處理完之前,主函數不會退出。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女