zeromicro/go-zero
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: CompareAndSwapInt32 may be better than AddInt32 (#2077)

Browse Source
This commit is contained in:
wxc
2022-07-01 12:41:32 +08:00
committed by GitHub
parent ce7e2a2a9a
commit d200ba4a7b
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
core/mr/mapreduce.go
View File

@@ -376,9 +376,7 @@ type onceChan struct {
} }
func (oc *onceChan) write(val interface{}) { func (oc *onceChan) write(val interface{}) {
if atomic.AddInt32(&oc.wrote, 1) > 1 { if atomic.CompareAndSwapInt32(&oc.wrote, 0, 1) {
return oc.channel <- val
} }
oc.channel <- val
} }