|
@@ -5,18 +5,22 @@ import (
|
|
"sync/atomic"
|
|
"sync/atomic"
|
|
)
|
|
)
|
|
|
|
|
|
|
|
+// An AtomicFloat64 is an implementation of atomic float64.
|
|
type AtomicFloat64 uint64
|
|
type AtomicFloat64 uint64
|
|
|
|
|
|
|
|
+// NewAtomicFloat64 returns an AtomicFloat64.
|
|
func NewAtomicFloat64() *AtomicFloat64 {
|
|
func NewAtomicFloat64() *AtomicFloat64 {
|
|
return new(AtomicFloat64)
|
|
return new(AtomicFloat64)
|
|
}
|
|
}
|
|
|
|
|
|
|
|
+// ForAtomicFloat64 returns an AtomicFloat64 with given val.
|
|
func ForAtomicFloat64(val float64) *AtomicFloat64 {
|
|
func ForAtomicFloat64(val float64) *AtomicFloat64 {
|
|
f := NewAtomicFloat64()
|
|
f := NewAtomicFloat64()
|
|
f.Set(val)
|
|
f.Set(val)
|
|
return f
|
|
return f
|
|
}
|
|
}
|
|
|
|
|
|
|
|
+// Add adds val to current value.
|
|
func (f *AtomicFloat64) Add(val float64) float64 {
|
|
func (f *AtomicFloat64) Add(val float64) float64 {
|
|
for {
|
|
for {
|
|
old := f.Load()
|
|
old := f.Load()
|
|
@@ -27,14 +31,17 @@ func (f *AtomicFloat64) Add(val float64) float64 {
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
|
|
+// CompareAndSwap compares current value with old, if equals, set the value to val.
|
|
func (f *AtomicFloat64) CompareAndSwap(old, val float64) bool {
|
|
func (f *AtomicFloat64) CompareAndSwap(old, val float64) bool {
|
|
return atomic.CompareAndSwapUint64((*uint64)(f), math.Float64bits(old), math.Float64bits(val))
|
|
return atomic.CompareAndSwapUint64((*uint64)(f), math.Float64bits(old), math.Float64bits(val))
|
|
}
|
|
}
|
|
|
|
|
|
|
|
+// Load loads the current value.
|
|
func (f *AtomicFloat64) Load() float64 {
|
|
func (f *AtomicFloat64) Load() float64 {
|
|
return math.Float64frombits(atomic.LoadUint64((*uint64)(f)))
|
|
return math.Float64frombits(atomic.LoadUint64((*uint64)(f)))
|
|
}
|
|
}
|
|
|
|
|
|
|
|
+// Set sets the current value to val.
|
|
func (f *AtomicFloat64) Set(val float64) {
|
|
func (f *AtomicFloat64) Set(val float64) {
|
|
atomic.StoreUint64((*uint64)(f), math.Float64bits(val))
|
|
atomic.StoreUint64((*uint64)(f), math.Float64bits(val))
|
|
}
|
|
}
|