mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-12-05 01:52:46 +00:00
59 lines
1.4 KiB
Go
59 lines
1.4 KiB
Go
|
package atomics
|
||
|
|
||
|
import "sync"
|
||
|
|
||
|
// State provides user-friendly means of performing atomic-like
|
||
|
// operations on a uint32 state, and allowing callbacks on successful
|
||
|
// state change. This is a bit of a misnomer being where it is, as it
|
||
|
// actually uses a mutex under-the-hood.
|
||
|
type State struct {
|
||
|
mutex sync.Mutex
|
||
|
state uint32
|
||
|
}
|
||
|
|
||
|
// Store will update State value safely within mutex lock.
|
||
|
func (st *State) Store(val uint32) {
|
||
|
st.mutex.Lock()
|
||
|
st.state = val
|
||
|
st.mutex.Unlock()
|
||
|
}
|
||
|
|
||
|
// Load will get value of State safely within mutex lock.
|
||
|
func (st *State) Load() uint32 {
|
||
|
st.mutex.Lock()
|
||
|
state := st.state
|
||
|
st.mutex.Unlock()
|
||
|
return state
|
||
|
}
|
||
|
|
||
|
// WithLock performs fn within State mutex lock, useful if you want
|
||
|
// to just use State's mutex for locking instead of creating another.
|
||
|
func (st *State) WithLock(fn func()) {
|
||
|
st.mutex.Lock()
|
||
|
defer st.mutex.Unlock()
|
||
|
fn()
|
||
|
}
|
||
|
|
||
|
// Update performs fn within State mutex lock, with the current state
|
||
|
// value provided as an argument, and return value used to update state.
|
||
|
func (st *State) Update(fn func(state uint32) uint32) {
|
||
|
st.mutex.Lock()
|
||
|
defer st.mutex.Unlock()
|
||
|
st.state = fn(st.state)
|
||
|
}
|
||
|
|
||
|
// CAS performs a compare-and-swap on State, calling fn on success. Success value is also returned.
|
||
|
func (st *State) CAS(cmp, swp uint32, fn func()) (ok bool) {
|
||
|
// Acquire lock
|
||
|
st.mutex.Lock()
|
||
|
defer st.mutex.Unlock()
|
||
|
|
||
|
// Perform CAS operation, fn() on success
|
||
|
if ok = (st.state == cmp); ok {
|
||
|
st.state = swp
|
||
|
fn()
|
||
|
}
|
||
|
|
||
|
return
|
||
|
}
|