在Go語言中,可以使用sync.WaitGroup
來處理實(shí)時(shí)數(shù)據(jù)流。
sync.WaitGroup
是一個(gè)計(jì)數(shù)信號量,用來等待一組goroutine的結(jié)束。它的主要方法有三個(gè):Add()
、Done()
和Wait()
。
首先,需要?jiǎng)?chuàng)建一個(gè)sync.WaitGroup
對象,然后在每個(gè)goroutine開始之前調(diào)用Add()
方法,表示要等待的goroutine數(shù)量增加1。在goroutine的結(jié)尾處調(diào)用Done()
方法,表示該goroutine已經(jīng)結(jié)束。最后,在主goroutine中調(diào)用Wait()
方法,等待所有的goroutine都結(jié)束。
下面是一個(gè)簡單的例子,假設(shè)有一個(gè)數(shù)據(jù)流需要實(shí)時(shí)處理,處理的任務(wù)是打印每個(gè)數(shù)據(jù)的平方值:
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)
}
在主函數(shù)中,首先創(chuàng)建了一個(gè)sync.WaitGroup
對象wg
。然后,遍歷數(shù)據(jù)流,并為每個(gè)數(shù)據(jù)增加了一個(gè)等待任務(wù)。接著,啟動(dòng)了一個(gè)goroutine來處理每個(gè)數(shù)據(jù),并傳遞了&wg
作為參數(shù)。在process()
函數(shù)中,打印了每個(gè)數(shù)據(jù)的平方值,并且在函數(shù)結(jié)尾處調(diào)用了wg.Done()
來表示該goroutine已經(jīng)結(jié)束。
最后,在主函數(shù)中調(diào)用wg.Wait()
來等待所有的goroutine都結(jié)束。這樣就可以保證在所有的數(shù)據(jù)都被處理完之前,主函數(shù)不會退出。