naza/pkg/nazaatomic/atomic_32bit.go

// Copyright 2021, 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)

// +build 386 arm

package nazaatomic

import (
	"sync"
)

// 注意，因为32位系统下，使用标准库中的atomic操作64位整型有bug，所以32位系统原子操作64位整型时，我们使用mutex

type Int64 struct {
	mu   sync.Mutex
	core int64
}

type Uint64 struct {
	mu   sync.Mutex
	core uint64
}

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

func (obj *Uint64) Load() uint64 {
	obj.mu.Lock()
	ret := obj.core
	obj.mu.Unlock()
	return ret
}

func (obj *Uint64) Store(val uint64) {
	obj.mu.Lock()
	obj.core = val
	obj.mu.Unlock()
}

func (obj *Uint64) Add(delta uint64) (new uint64) {
	obj.mu.Lock()
	obj.core += delta
	new = obj.core
	obj.mu.Unlock()
	return
}

// @param delta 举例，传入3，则减3
func (obj *Uint64) Sub(delta uint64) (new uint64) {
	obj.mu.Lock()
	obj.core += ^uint64(delta - 1)
	new = obj.core
	obj.mu.Unlock()
	return
}

func (obj *Uint64) Increment() (new uint64) {
	obj.mu.Lock()
	obj.core += 1
	new = obj.core
	obj.mu.Unlock()
	return
}

func (obj *Uint64) Decrement() (new uint64) {
	obj.mu.Lock()
	obj.core += ^uint64(0)
	new = obj.core
	obj.mu.Unlock()
	return
}

func (obj *Uint64) CompareAndSwap(old uint64, new uint64) (swapped bool) {
	obj.mu.Lock()
	if obj.core == old {
		obj.core = new
		obj.mu.Unlock()
		return true
	}

	obj.mu.Unlock()
	return false
}

func (obj *Uint64) Swap(new uint64) (old uint64) {
	obj.mu.Lock()
	old = obj.core
	obj.core = new
	obj.mu.Unlock()
	return
}

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

func (obj *Int64) Load() int64 {
	obj.mu.Lock()
	ret := obj.core
	obj.mu.Unlock()
	return ret
}

func (obj *Int64) Store(val int64) {
	obj.mu.Lock()
	obj.core = val
	obj.mu.Unlock()
}

func (obj *Int64) Add(delta int64) (new int64) {
	obj.mu.Lock()
	obj.core += delta
	new = obj.core
	obj.mu.Unlock()
	return
}

// @param delta 举例，传入3，则减3
func (obj *Int64) Sub(delta int64) (new int64) {
	obj.mu.Lock()
	obj.core += ^int64(delta - 1)
	new = obj.core
	obj.mu.Unlock()
	return
}

func (obj *Int64) Increment() (new int64) {
	obj.mu.Lock()
	obj.core += 1
	new = obj.core
	obj.mu.Unlock()
	return
}

func (obj *Int64) Decrement() (new int64) {
	obj.mu.Lock()
	obj.core += ^int64(0)
	new = obj.core
	obj.mu.Unlock()
	return
}

func (obj *Int64) CompareAndSwap(old int64, new int64) (swapped bool) {
	obj.mu.Lock()
	if obj.core == old {
		obj.core = new
		obj.mu.Unlock()
		return true
	}

	obj.mu.Unlock()
	return false
}

func (obj *Int64) Swap(new int64) (old int64) {
	obj.mu.Lock()
	old = obj.core
	obj.core = new
	obj.mu.Unlock()
	return
}