error.go 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. // Copyright (c) 2016 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. package atomic
  21. // Error is an atomic type-safe wrapper around Value for errors
  22. type Error struct{ v Value }
  23. // errorHolder is non-nil holder for error object.
  24. // atomic.Value panics on saving nil object, so err object needs to be
  25. // wrapped with valid object first.
  26. type errorHolder struct{ err error }
  27. // NewError creates new atomic error object
  28. func NewError(err error) *Error {
  29. e := &Error{}
  30. if err != nil {
  31. e.Store(err)
  32. }
  33. return e
  34. }
  35. // Load atomically loads the wrapped error
  36. func (e *Error) Load() error {
  37. v := e.v.Load()
  38. if v == nil {
  39. return nil
  40. }
  41. eh := v.(errorHolder)
  42. return eh.err
  43. }
  44. // Store atomically stores error.
  45. // NOTE: a holder object is allocated on each Store call.
  46. func (e *Error) Store(err error) {
  47. e.v.Store(errorHolder{err: err})
  48. }