diff options
Diffstat (limited to 'vendor/codeberg.org/gruf/go-atomics/flags.go')
-rw-r--r-- | vendor/codeberg.org/gruf/go-atomics/flags.go | 97 |
1 files changed, 97 insertions, 0 deletions
diff --git a/vendor/codeberg.org/gruf/go-atomics/flags.go b/vendor/codeberg.org/gruf/go-atomics/flags.go new file mode 100644 index 000000000..42176bece --- /dev/null +++ b/vendor/codeberg.org/gruf/go-atomics/flags.go @@ -0,0 +1,97 @@ +package atomics + +import ( + "sync/atomic" + + "codeberg.org/gruf/go-bitutil" +) + +// Flags32 provides user-friendly means of performing atomic operations on bitutil.Flags32 types. +type Flags32 bitutil.Flags32 + +// NewFlags32 will return a new Flags32 instance initialized with zero value. +func NewFlags32() *Flags32 { + return new(Flags32) +} + +// Get will atomically load a(n) bitutil.Flags32 value contained within f, and check if bit value is set. +func (f *Flags32) Get(bit uint8) bool { + return f.Load().Get(bit) +} + +// Set performs a compare-and-swap for a(n) bitutil.Flags32 with bit value set, at address contained within f. +func (f *Flags32) Set(bit uint8) bool { + cur := f.Load() + return f.CAS(cur, cur.Set(bit)) +} + +// Unset performs a compare-and-swap for a(n) bitutil.Flags32 with bit value unset, at address contained within f. +func (f *Flags32) Unset(bit uint8) bool { + cur := f.Load() + return f.CAS(cur, cur.Unset(bit)) +} + +// Store will atomically store bitutil.Flags32 value in address contained within f. +func (f *Flags32) Store(val bitutil.Flags32) { + atomic.StoreUint32((*uint32)(f), uint32(val)) +} + +// Load will atomically load bitutil.Flags32 value at address contained within f. +func (f *Flags32) Load() bitutil.Flags32 { + return bitutil.Flags32(atomic.LoadUint32((*uint32)(f))) +} + +// CAS performs a compare-and-swap for a(n) bitutil.Flags32 value at address contained within f. +func (f *Flags32) CAS(cmp, swp bitutil.Flags32) bool { + return atomic.CompareAndSwapUint32((*uint32)(f), uint32(cmp), uint32(swp)) +} + +// Swap atomically stores new bitutil.Flags32 value into address contained within f, and returns previous value. +func (f *Flags32) Swap(swp bitutil.Flags32) bitutil.Flags32 { + return bitutil.Flags32(atomic.SwapUint32((*uint32)(f), uint32(swp))) +} + +// Flags64 provides user-friendly means of performing atomic operations on bitutil.Flags64 types. +type Flags64 bitutil.Flags64 + +// NewFlags64 will return a new Flags64 instance initialized with zero value. +func NewFlags64() *Flags64 { + return new(Flags64) +} + +// Get will atomically load a(n) bitutil.Flags64 value contained within f, and check if bit value is set. +func (f *Flags64) Get(bit uint8) bool { + return f.Load().Get(bit) +} + +// Set performs a compare-and-swap for a(n) bitutil.Flags64 with bit value set, at address contained within f. +func (f *Flags64) Set(bit uint8) bool { + cur := f.Load() + return f.CAS(cur, cur.Set(bit)) +} + +// Unset performs a compare-and-swap for a(n) bitutil.Flags64 with bit value unset, at address contained within f. +func (f *Flags64) Unset(bit uint8) bool { + cur := f.Load() + return f.CAS(cur, cur.Unset(bit)) +} + +// Store will atomically store bitutil.Flags64 value in address contained within f. +func (f *Flags64) Store(val bitutil.Flags64) { + atomic.StoreUint64((*uint64)(f), uint64(val)) +} + +// Load will atomically load bitutil.Flags64 value at address contained within f. +func (f *Flags64) Load() bitutil.Flags64 { + return bitutil.Flags64(atomic.LoadUint64((*uint64)(f))) +} + +// CAS performs a compare-and-swap for a(n) bitutil.Flags64 value at address contained within f. +func (f *Flags64) CAS(cmp, swp bitutil.Flags64) bool { + return atomic.CompareAndSwapUint64((*uint64)(f), uint64(cmp), uint64(swp)) +} + +// Swap atomically stores new bitutil.Flags64 value into address contained within f, and returns previous value. +func (f *Flags64) Swap(swp bitutil.Flags64) bitutil.Flags64 { + return bitutil.Flags64(atomic.SwapUint64((*uint64)(f), uint64(swp))) +} |