Update AtomicBool to use atomic memory operation (#1939)

src/util/atomicbool.go

@@ -1,32 +1,34 @@
 package util
 
-import "sync"
+import (
+	"sync/atomic"
+)
+
+func convertBoolToInt32(b bool) int32 {
+	if b {
+		return 1
+	}
+	return 0
+}
 
 // AtomicBool is a boxed-class that provides synchronized access to the
 // underlying boolean value
 type AtomicBool struct {
-	mutex sync.Mutex
+	state int32 // "1" is true, "0" is false
-	state bool
 }
 
 // NewAtomicBool returns a new AtomicBool
 func NewAtomicBool(initialState bool) *AtomicBool {
-	return &AtomicBool{
+	return &AtomicBool{state: convertBoolToInt32(initialState)}
-		mutex: sync.Mutex{},
-		state: initialState}
 }
 
 // Get returns the current boolean value synchronously
 func (a *AtomicBool) Get() bool {
-	a.mutex.Lock()
+	return atomic.LoadInt32(&a.state) == 1
-	defer a.mutex.Unlock()
-	return a.state
 }
 
 // Set updates the boolean value synchronously
 func (a *AtomicBool) Set(newState bool) bool {
-	a.mutex.Lock()
+	atomic.StoreInt32(&a.state, convertBoolToInt32(newState))
-	defer a.mutex.Unlock()
+	return newState
-	a.state = newState
-	return a.state
 }