naza/pkg/nazaatomic/atomic.go

// Copyright 2019, Chef.  All rights reserved.
// https://github.com/q191201771/naza
//
// Use of this source code is governed by a MIT-style license
// that can be found in the License file.
//
// Author: Chef (191201771@qq.com)

package nazaatomic

import "sync/atomic"

type Bool struct {
	core Int32
}

type Int32 struct {
	core int32
}

type Uint32 struct {
	core uint32
}

type Int64 struct {
	core int64
}

type Uint64 struct {
	core uint64
}

// ----------------------------------------------------------------------------

func (obj *Int32) Load() int32 {
	return atomic.LoadInt32(&obj.core)
}

func (obj *Int32) Store(val int32) {
	atomic.StoreInt32(&obj.core, val)
}

func (obj *Int32) Add(delta int32) (new int32) {
	return atomic.AddInt32(&obj.core, delta)
}

// @param delta 举例，传入3，则减3
func (obj *Int32) Sub(delta int32) (new int32) {
	return atomic.AddInt32(&obj.core, -delta)
}

func (obj *Int32) Increment() (new int32) {
	return atomic.AddInt32(&obj.core, 1)
}

func (obj *Int32) Decrement() (new int32) {
	return atomic.AddInt32(&obj.core, -1)
}

func (obj *Int32) CompareAndSwap(old int32, new int32) (swapped bool) {
	return atomic.CompareAndSwapInt32(&obj.core, old, new)
}

func (obj *Int32) Swap(new int32) (old int32) {
	return atomic.SwapInt32(&obj.core, new)
}

// ----------------------------------------------------------------------------

func (obj *Uint32) Load() uint32 {
	return atomic.LoadUint32(&obj.core)
}

func (obj *Uint32) Store(val uint32) {
	atomic.StoreUint32(&obj.core, val)
}

func (obj *Uint32) Add(delta uint32) (new uint32) {
	return atomic.AddUint32(&obj.core, delta)
}

// @param delta 举例，传入3，则减3
func (obj *Uint32) Sub(delta uint32) (new uint32) {
	return atomic.AddUint32(&obj.core, ^uint32(delta-1))
}

func (obj *Uint32) Increment() (new uint32) {
	return atomic.AddUint32(&obj.core, 1)
}

func (obj *Uint32) Decrement() (new uint32) {
	return atomic.AddUint32(&obj.core, ^uint32(0))
}

func (obj *Uint32) CompareAndSwap(old uint32, new uint32) (swapped bool) {
	return atomic.CompareAndSwapUint32(&obj.core, old, new)
}

func (obj *Uint32) Swap(new uint32) (old uint32) {
	return atomic.SwapUint32(&obj.core, new)
}

// ----------------------------------------------------------------------------

func (obj *Uint64) Load() uint64 {
	return atomic.LoadUint64(&obj.core)
}

func (obj *Uint64) Store(val uint64) {
	atomic.StoreUint64(&obj.core, val)
}

func (obj *Uint64) Add(delta uint64) (new uint64) {
	return atomic.AddUint64(&obj.core, delta)
}

// @param delta 举例，传入3，则减3
func (obj *Uint64) Sub(delta uint64) (new uint64) {
	return atomic.AddUint64(&obj.core, ^uint64(delta-1))
}

func (obj *Uint64) Increment() (new uint64) {
	return atomic.AddUint64(&obj.core, 1)
}

func (obj *Uint64) Decrement() (new uint64) {
	return atomic.AddUint64(&obj.core, ^uint64(0))
}

func (obj *Uint64) CompareAndSwap(old uint64, new uint64) (swapped bool) {
	return atomic.CompareAndSwapUint64(&obj.core, old, new)
}

func (obj *Uint64) Swap(new uint64) (old uint64) {
	return atomic.SwapUint64(&obj.core, new)
}

// ----------------------------------------------------------------------------

func (obj *Int64) Load() int64 {
	return atomic.LoadInt64(&obj.core)
}

func (obj *Int64) Store(val int64) {
	atomic.StoreInt64(&obj.core, val)
}

func (obj *Int64) Add(delta int64) (new int64) {
	return atomic.AddInt64(&obj.core, delta)
}

// @param delta 举例，传入3，则减3
func (obj *Int64) Sub(delta int64) (new int64) {
	return atomic.AddInt64(&obj.core, -delta)
}

func (obj *Int64) Increment() (new int64) {
	return atomic.AddInt64(&obj.core, 1)
}

func (obj *Int64) Decrement() (new int64) {
	return atomic.AddInt64(&obj.core, -1)
}

func (obj *Int64) CompareAndSwap(old int64, new int64) (swapped bool) {
	return atomic.CompareAndSwapInt64(&obj.core, old, new)
}

func (obj *Int64) Swap(new int64) (old int64) {
	return atomic.SwapInt64(&obj.core, new)
}

// ----------------------------------------------------------------------------

func (obj *Bool) Load() bool {
	return int32tobool(obj.core.Load())
}

func (obj *Bool) Store(val bool) {
	obj.core.Store(booltoint32(val))
}

func (obj *Bool) CompareAndSwap(old bool, new bool) (swapped bool) {
	return obj.core.CompareAndSwap(booltoint32(old), booltoint32(new))
}

func (obj *Bool) Swap(new bool) (old bool) {
	return int32tobool(obj.core.Swap(booltoint32(new)))
}

func booltoint32(val bool) int32 {
	if val {
		return 1
	}
	return 0
}

func int32tobool(val int32) bool {
	if val == 1 {
		return true
	}
	return false
}