为什么结果不如flag" -race"那样预期?它期望相同的结果:1000000-带有标志" -race",没有此
https://gist.github.com/romanitalian/f403ceb6e492eaf6ba953cf67d5a22ff
package main
import (
"fmt"
"runtime"
"sync/atomic"
"time"
)
//$ go run -race main_atomic.go
//954203
//
//$ go run main_atomic.go
//1000000
type atomicCounter struct {
val int64
}
func (c *atomicCounter) Add(x int64) {
atomic.AddInt64(&c.val, x)
runtime.Gosched()
}
func (c *atomicCounter) Value() int64 {
return atomic.LoadInt64(&c.val)
}
func main() {
counter := atomicCounter{}
for i := 0; i < 100; i++ {
go func(no int) {
for i := 0; i < 10000; i++ {
counter.Add(1)
}
}(i)
}
time.Sleep(time.Second)
fmt.Println(counter.Value())
}
结果不一样的原因是,time.Sleep(time.Second)
不能保证您的所有goroutines都将在一秒钟的时间段落中执行。即使您执行go run main.go
,也不能保证您每次都会获得相同的结果。如果将time.Milisecond
而不是time.Second
放置,则可以测试它,您会看到更多不一致的结果。
无论您在time.Sleep
方法中放置什么价值,它都不能保证您所有的goroutines都将被执行,这只是意味着您所有的goroutines都不会及时完成。
为了一致的结果,您需要对goroutines进行一些同步。您可以使用WaitGroup
或频道。
使用WaitGroup
:
//rest of the code above is the same
func main() {
counter := atomicCounter{}
var wg sync.WaitGroup
for i := 0; i < 100; i++ {
wg.Add(1)
go func(no int) {
for i := 0; i < 10000; i++ {
counter.Add(1)
}
wg.Done()
}(i)
}
wg.Wait()
fmt.Println(counter.Value())
}
带有频道:
func main() {
valStream := make(chan int)
doneStream := make(chan int)
result := 0
for i := 0; i < 100; i++ {
go func() {
for i := 0; i < 10000; i++ {
valStream <- 1
}
doneStream <- 1
}()
}
go func() {
counter := 0
for count := range doneStream {
counter += count
if counter == 100 {
close(doneStream)
}
}
close(valStream)
}()
for val := range valStream {
result += val
}
fmt.Println(result)
}