Merge pull request #422 from openziti/add-swap-to-atomic-value

Add swap to atomic value
openziti · Dec 3, 2024 · e7bde0c · e7bde0c
2 parents 1c830a1 + d8e809a
commit e7bde0c
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/concurrenz/atomic_value.go b/concurrenz/atomic_value.go
@@ -36,3 +36,12 @@ func (self *AtomicValue[T]) Load() T {
 func (self *AtomicValue[T]) CompareAndSwap(old, new T) bool {
 	return (*atomic.Value)(self).CompareAndSwap(old, new)
 }
+
+func (self *AtomicValue[T]) Swap(new T) T {
+	result := (*atomic.Value)(self).Swap(new)
+	var old T
+	if v, ok := result.(T); ok {
+		old = v
+	}
+	return old
+}