2021-01-17 21:46:25 +08:00
|
|
|
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
2018-03-27 15:09:17 +08:00
|
|
|
//
|
|
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
|
|
// If a copy of the MIT was not distributed with this file,
|
2019-02-02 16:18:25 +08:00
|
|
|
// You can obtain one at https://github.com/gogf/gf.
|
2018-03-27 15:09:17 +08:00
|
|
|
|
|
|
|
package gtype
|
|
|
|
|
|
|
|
import (
|
2019-09-29 15:59:09 +08:00
|
|
|
"github.com/gogf/gf/util/gconv"
|
2019-04-15 22:55:12 +08:00
|
|
|
"math"
|
2019-09-29 15:59:09 +08:00
|
|
|
"strconv"
|
2019-04-15 22:55:12 +08:00
|
|
|
"sync/atomic"
|
|
|
|
"unsafe"
|
2018-03-27 15:09:17 +08:00
|
|
|
)
|
|
|
|
|
2019-12-20 23:23:50 +08:00
|
|
|
// Float64 is a struct for concurrent-safe operation for type float64.
|
2018-03-27 15:09:17 +08:00
|
|
|
type Float64 struct {
|
2019-04-15 22:55:12 +08:00
|
|
|
value uint64
|
2018-03-27 15:09:17 +08:00
|
|
|
}
|
|
|
|
|
2019-12-20 23:23:50 +08:00
|
|
|
// NewFloat64 creates and returns a concurrent-safe object for float64 type,
|
2019-04-15 22:55:12 +08:00
|
|
|
// with given initial value <value>.
|
2019-06-19 09:06:52 +08:00
|
|
|
func NewFloat64(value ...float64) *Float64 {
|
|
|
|
if len(value) > 0 {
|
|
|
|
return &Float64{
|
|
|
|
value: math.Float64bits(value[0]),
|
2019-04-15 22:55:12 +08:00
|
|
|
}
|
2019-06-19 09:06:52 +08:00
|
|
|
}
|
|
|
|
return &Float64{}
|
2018-03-27 15:09:17 +08:00
|
|
|
}
|
|
|
|
|
2019-04-15 22:55:12 +08:00
|
|
|
// Clone clones and returns a new concurrent-safe object for float64 type.
|
2019-06-18 08:37:21 +08:00
|
|
|
func (v *Float64) Clone() *Float64 {
|
2019-06-19 09:06:52 +08:00
|
|
|
return NewFloat64(v.Val())
|
2018-08-28 17:06:49 +08:00
|
|
|
}
|
|
|
|
|
2019-05-08 17:21:18 +08:00
|
|
|
// Set atomically stores <value> into t.value and returns the previous value of t.value.
|
2019-06-18 08:37:21 +08:00
|
|
|
func (v *Float64) Set(value float64) (old float64) {
|
2019-06-19 09:06:52 +08:00
|
|
|
return math.Float64frombits(atomic.SwapUint64(&v.value, math.Float64bits(value)))
|
2018-03-27 15:09:17 +08:00
|
|
|
}
|
|
|
|
|
2019-12-20 23:23:50 +08:00
|
|
|
// Val atomically loads and returns t.value.
|
2019-06-18 08:37:21 +08:00
|
|
|
func (v *Float64) Val() float64 {
|
2019-06-19 09:06:52 +08:00
|
|
|
return math.Float64frombits(atomic.LoadUint64(&v.value))
|
2018-03-27 15:09:17 +08:00
|
|
|
}
|
|
|
|
|
2019-05-08 17:21:18 +08:00
|
|
|
// Add atomically adds <delta> to t.value and returns the new value.
|
2019-06-18 08:37:21 +08:00
|
|
|
func (v *Float64) Add(delta float64) (new float64) {
|
2019-04-15 22:55:12 +08:00
|
|
|
for {
|
2019-06-18 08:37:21 +08:00
|
|
|
old := math.Float64frombits(v.value)
|
2019-06-19 09:06:52 +08:00
|
|
|
new = old + delta
|
2019-04-15 22:55:12 +08:00
|
|
|
if atomic.CompareAndSwapUint64(
|
2019-06-18 08:37:21 +08:00
|
|
|
(*uint64)(unsafe.Pointer(&v.value)),
|
2019-04-15 22:55:12 +08:00
|
|
|
math.Float64bits(old),
|
|
|
|
math.Float64bits(new),
|
|
|
|
) {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return
|
2018-04-27 20:35:56 +08:00
|
|
|
}
|
2019-06-18 08:37:21 +08:00
|
|
|
|
|
|
|
// Cas executes the compare-and-swap operation for value.
|
2020-01-16 21:04:28 +08:00
|
|
|
func (v *Float64) Cas(old, new float64) (swapped bool) {
|
2019-09-29 20:12:59 +08:00
|
|
|
return atomic.CompareAndSwapUint64(&v.value, math.Float64bits(old), math.Float64bits(new))
|
2019-06-18 08:37:21 +08:00
|
|
|
}
|
2019-09-29 15:59:09 +08:00
|
|
|
|
2019-10-01 16:03:18 +08:00
|
|
|
// String implements String interface for string printing.
|
|
|
|
func (v *Float64) String() string {
|
|
|
|
return strconv.FormatFloat(v.Val(), 'g', -1, 64)
|
|
|
|
}
|
|
|
|
|
2019-09-29 15:59:09 +08:00
|
|
|
// MarshalJSON implements the interface MarshalJSON for json.Marshal.
|
|
|
|
func (v *Float64) MarshalJSON() ([]byte, error) {
|
2019-09-29 20:12:59 +08:00
|
|
|
return gconv.UnsafeStrToBytes(strconv.FormatFloat(v.Val(), 'g', -1, 64)), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// UnmarshalJSON implements the interface UnmarshalJSON for json.Unmarshal.
|
|
|
|
func (v *Float64) UnmarshalJSON(b []byte) error {
|
|
|
|
v.Set(gconv.Float64(gconv.UnsafeBytesToStr(b)))
|
|
|
|
return nil
|
2019-09-29 15:59:09 +08:00
|
|
|
}
|
2020-01-20 19:56:42 +08:00
|
|
|
|
|
|
|
// UnmarshalValue is an interface implement which sets any type of value for <v>.
|
|
|
|
func (v *Float64) UnmarshalValue(value interface{}) error {
|
|
|
|
v.Set(gconv.Float64(value))
|
|
|
|
return nil
|
|
|
|
}
|