pointer_go118.go 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. // Copyright (c) 2022 Uber Technologies, Inc.
  2. //
  3. // Permission is hereby granted, free of charge, to any person obtaining a copy
  4. // of this software and associated documentation files (the "Software"), to deal
  5. // in the Software without restriction, including without limitation the rights
  6. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  7. // copies of the Software, and to permit persons to whom the Software is
  8. // furnished to do so, subject to the following conditions:
  9. //
  10. // The above copyright notice and this permission notice shall be included in
  11. // all copies or substantial portions of the Software.
  12. //
  13. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  14. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  15. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  16. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  17. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  18. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  19. // THE SOFTWARE.
  20. //go:build go1.18 && !go1.19
  21. // +build go1.18,!go1.19
  22. package atomic
  23. import "unsafe"
  24. type Pointer[T any] struct {
  25. _ nocmp // disallow non-atomic comparison
  26. p UnsafePointer
  27. }
  28. // NewPointer creates a new Pointer.
  29. func NewPointer[T any](v *T) *Pointer[T] {
  30. var p Pointer[T]
  31. if v != nil {
  32. p.p.Store(unsafe.Pointer(v))
  33. }
  34. return &p
  35. }
  36. // Load atomically loads the wrapped value.
  37. func (p *Pointer[T]) Load() *T {
  38. return (*T)(p.p.Load())
  39. }
  40. // Store atomically stores the passed value.
  41. func (p *Pointer[T]) Store(val *T) {
  42. p.p.Store(unsafe.Pointer(val))
  43. }
  44. // Swap atomically swaps the wrapped pointer and returns the old value.
  45. func (p *Pointer[T]) Swap(val *T) (old *T) {
  46. return (*T)(p.p.Swap(unsafe.Pointer(val)))
  47. }
  48. // CompareAndSwap is an atomic compare-and-swap.
  49. func (p *Pointer[T]) CompareAndSwap(old, new *T) (swapped bool) {
  50. return p.p.CompareAndSwap(unsafe.Pointer(old), unsafe.Pointer(new))
  51. }