-
Notifications
You must be signed in to change notification settings - Fork 1
/
atomic-reference.go
43 lines (36 loc) · 979 Bytes
/
atomic-reference.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
/*
© 2022–present Harald Rudell <harald.rudell@gmail.com> (https://haraldrudell.github.io/haraldrudell/)
ISC License
*/
package parl
import (
"sync/atomic"
"unsafe"
)
// AtomicReference holds a typed reference that is accessed atomically.
type AtomicReference[T any] struct {
reference *T
}
func MakeAtomicReference[T any]() (reference AtomicReference[T]) {
return AtomicReference[T]{}
}
func (ref *AtomicReference[T]) Get() (reference *T) {
return (*T)(atomic.LoadPointer(
(*unsafe.Pointer)(unsafe.Pointer(&ref.reference)),
))
}
func (ref *AtomicReference[T]) Put(reference *T) (r0 *T) {
r0 = (*T)(atomic.SwapPointer(
(*unsafe.Pointer)(unsafe.Pointer(&ref.reference)),
unsafe.Pointer(reference),
))
return
}
func (ref *AtomicReference[T]) PutIf(reference *T, expected *T) (swapped bool) {
swapped = atomic.CompareAndSwapPointer(
(*unsafe.Pointer)(unsafe.Pointer(&ref.reference)),
unsafe.Pointer(expected),
unsafe.Pointer(reference),
)
return
}