使用golang并發(fā)求和，作為對(duì)golang并發(fā)的一個(gè)練習(xí).
為了驗(yàn)證結(jié)果的正確性，要給出最傳統(tǒng)的版本：
func sum1(data []int) int {
s := 0
l := len(data)
for i := 0; i l; i++ {
s += data[i]
}
return s
}
第二種方法
使用N個(gè)goroutine， 然后將N個(gè)分段的和寫入N個(gè)channel中：
func sum2(data []int) int {
s := 0
l := len(data)
const N = 5
seg := l / N
var chs [N]-chan int
for i := 0; i N; i++ {
chs[i] = worker(data[i*seg : (i+1)*seg])
}
for i := 0; i N; i++ {
s += -chs[i]
}
return s
}
func worker(s []int) -chan int {
out := make(chan int)
go func() {
length := len(s)
sum := 0
for i := 0; i length; i++ {
sum += s[i]
}
out - sum
}()
return out
}
對(duì)于一個(gè)求和的任務(wù)來(lái)說，用worker這種“模式”可能 太過麻煩，
看第三種
直接一個(gè)函數(shù)寫出來(lái)：
func sum3(data []int) int {
s := 0
l := len(data)
const N = 5
seg := l / N
var mu sync.Mutex
var wg sync.WaitGroup
wg.Add(N) // 直接加N個(gè)
for i := 0; i N; i++ {
go func(ii int) {
tmpS := data[ii*seg : (ii+1)*seg]
ll := len(tmpS)
mu.Lock()
for i := 0; i ll; i++ {
s += tmpS[i]
}
mu.Unlock()
wg.Done() // 一個(gè)goroutine運(yùn)行完
}(i)
}
wg.Wait() // 等N個(gè)goroutine都運(yùn)行完
return s
}
注意sum3要在讀寫s的地方加鎖，因?yàn)閟可能被多個(gè)goroutine并發(fā)讀寫。
最后一種方法有data race問題
不過運(yùn)行結(jié)果是對(duì)的，看一下思路：
var sum4Tmp int
var sum4mu sync.Mutex
// 這個(gè)有data race問題，可以用WaitGroup改，只是提供一種思路
func sum4(data []int) int {
//s := 0
l := len(data)
const N = 5
seg := l / N
for i := 0; i N; i++ {
go subsum4(data[i*seg : (i+1)*seg])
}
// 這里是>1，因?yàn)橐懦齧ain
// 這種方法不可靠，只是一種思路
for runtime.NumGoroutine() > 1 {
}
// go run -race sum.go會(huì)報(bào)data race問題
// main goroutine對(duì)它讀
// 別的goroutine會(huì)對(duì)它寫（go subsum4)
return sum4Tmp
}
func subsum4(s []int) {
length := len(s)
sum := 0
sum4mu.Lock()
for i := 0; i length; i++ {
sum += s[i]
}
sum4Tmp = sum4Tmp + sum
defer sum4mu.Unlock()
}
最后測(cè)試如下：
首先創(chuàng)建一個(gè)slice, 放1e8(1億)個(gè)整數(shù)(范圍[0,10))進(jìn)去,
然后用4種方法進(jìn)行計(jì)算
func calcTime(f func([]int) int, arr []int, tag string) {
t1 := time.Now().UnixNano()
s := f(arr)
t2 := time.Now().UnixNano() - t1
fmt.Printf("%15s: time: %d, sum: %d\n", tag, t2, s)
}
func main() {
const MAX = 1e8 // 1億
arr := make([]int, MAX)
for i := 0; i MAX; i++ {
arr[i] = rand.Intn(10)
}
calcTime(sum1, arr, "for")
calcTime(sum2, arr, "worker")
calcTime(sum3, arr, "WaitGroup")
calcTime(sum4, arr, "NumGoroutine")
}
我的筆記本輸出結(jié)果：
for: time: 61834200, sum: 450032946
worker: time: 51861100, sum: 450032946
WaitGroup: time: 153628200, sum: 450032946
NumGoroutine: time: 63791300, sum: 450032946
歡迎補(bǔ)充指正！
補(bǔ)充：Golang并發(fā)求和（競(jìng)爭(zhēng)而非分段）
舉例
如果要求2個(gè)goroutine并發(fā)完成1到100的和而不是分段的情況如何解決呢？
解決方案：
var wg sync.WaitGroup
var ch chan int32
var receiveCh chan int32
func add(){
var sum int32
sum = 0
Loop:
for {
select {
case val, ok := -ch:
if ok {
atomic.AddInt32(sum, val)
} else {
break Loop
}
}
}
receiveCh - sum
wg.Done()
}
func main() {
wg.Add(3)
ch = make(chan int32)
receiveCh = make(chan int32, 2)
go func(){
for i := 1; i = 100; i++{
n := i //避免數(shù)據(jù)競(jìng)爭(zhēng)
ch - int32(n)
}
close(ch)
wg.Done()
}()
go add()
go add()
wg.Wait()
close(receiveCh)
var sum int32
sum = 0
for res := range receiveCh{
sum += res
}
fmt.Println("sum:",sum)
}
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教。
您可能感興趣的文章:- golang中的并發(fā)和并行
- 關(guān)于golang高并發(fā)的實(shí)現(xiàn)與注意事項(xiàng)說明
- 基于Golang 高并發(fā)問題的解決方案
- 使用golang編寫一個(gè)并發(fā)工作隊(duì)列
- golang 并發(fā)編程之生產(chǎn)者消費(fèi)者詳解
- Golang 并發(fā)以及通道的使用方式
