refactor: move atomic.Value to value.go, improved handling for zero values

This commit is contained in:
yusing
2025-03-28 07:06:07 +08:00
parent 5d2df3550b
commit 3021672de5
2 changed files with 15 additions and 6 deletions

View File

@@ -0,0 +1,34 @@
package atomic
import (
"encoding/json"
"sync/atomic"
)
type Value[T any] struct {
atomic.Value
}
func (a *Value[T]) Load() T {
if v := a.Value.Load(); v != nil {
return v.(T)
}
var zero T
return zero
}
func (a *Value[T]) Store(v T) {
a.Value.Store(v)
}
func (a *Value[T]) Swap(v T) T {
if v := a.Value.Swap(v); v != nil {
return v.(T)
}
var zero T
return zero
}
func (a *Value[T]) MarshalJSON() ([]byte, error) {
return json.Marshal(a.Load())
}