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

2022-07-01 12:41:32 +08:00
parent ce7e2a2a9a
commit d200ba4a7b
1 changed files with 2 additions and 4 deletions
--- a/core/mr/mapreduce.go
+++ b/core/mr/mapreduce.go
@@ -376,9 +376,7 @@ type onceChan struct {
 }

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