本站源代码
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

283 lines
7.0KB

  1. // Package errors provides simple error handling primitives.
  2. //
  3. // The traditional error handling idiom in Go is roughly akin to
  4. //
  5. // if err != nil {
  6. // return err
  7. // }
  8. //
  9. // which when applied recursively up the call stack results in error reports
  10. // without context or debugging information. The errors package allows
  11. // programmers to add context to the failure path in their code in a way
  12. // that does not destroy the original value of the error.
  13. //
  14. // Adding context to an error
  15. //
  16. // The errors.Wrap function returns a new error that adds context to the
  17. // original error by recording a stack trace at the point Wrap is called,
  18. // together with the supplied message. For example
  19. //
  20. // _, err := ioutil.ReadAll(r)
  21. // if err != nil {
  22. // return errors.Wrap(err, "read failed")
  23. // }
  24. //
  25. // If additional control is required, the errors.WithStack and
  26. // errors.WithMessage functions destructure errors.Wrap into its component
  27. // operations: annotating an error with a stack trace and with a message,
  28. // respectively.
  29. //
  30. // Retrieving the cause of an error
  31. //
  32. // Using errors.Wrap constructs a stack of errors, adding context to the
  33. // preceding error. Depending on the nature of the error it may be necessary
  34. // to reverse the operation of errors.Wrap to retrieve the original error
  35. // for inspection. Any error value which implements this interface
  36. //
  37. // type causer interface {
  38. // Cause() error
  39. // }
  40. //
  41. // can be inspected by errors.Cause. errors.Cause will recursively retrieve
  42. // the topmost error that does not implement causer, which is assumed to be
  43. // the original cause. For example:
  44. //
  45. // switch err := errors.Cause(err).(type) {
  46. // case *MyError:
  47. // // handle specifically
  48. // default:
  49. // // unknown error
  50. // }
  51. //
  52. // Although the causer interface is not exported by this package, it is
  53. // considered a part of its stable public interface.
  54. //
  55. // Formatted printing of errors
  56. //
  57. // All error values returned from this package implement fmt.Formatter and can
  58. // be formatted by the fmt package. The following verbs are supported:
  59. //
  60. // %s print the error. If the error has a Cause it will be
  61. // printed recursively.
  62. // %v see %s
  63. // %+v extended format. Each Frame of the error's StackTrace will
  64. // be printed in detail.
  65. //
  66. // Retrieving the stack trace of an error or wrapper
  67. //
  68. // New, Errorf, Wrap, and Wrapf record a stack trace at the point they are
  69. // invoked. This information can be retrieved with the following interface:
  70. //
  71. // type stackTracer interface {
  72. // StackTrace() errors.StackTrace
  73. // }
  74. //
  75. // The returned errors.StackTrace type is defined as
  76. //
  77. // type StackTrace []Frame
  78. //
  79. // The Frame type represents a call site in the stack trace. Frame supports
  80. // the fmt.Formatter interface that can be used for printing information about
  81. // the stack trace of this error. For example:
  82. //
  83. // if err, ok := err.(stackTracer); ok {
  84. // for _, f := range err.StackTrace() {
  85. // fmt.Printf("%+s:%d", f)
  86. // }
  87. // }
  88. //
  89. // Although the stackTracer interface is not exported by this package, it is
  90. // considered a part of its stable public interface.
  91. //
  92. // See the documentation for Frame.Format for more details.
  93. package errors
  94. import (
  95. "fmt"
  96. "io"
  97. )
  98. // New returns an error with the supplied message.
  99. // New also records the stack trace at the point it was called.
  100. func New(message string) error {
  101. return &fundamental{
  102. msg: message,
  103. stack: callers(),
  104. }
  105. }
  106. // Errorf formats according to a format specifier and returns the string
  107. // as a value that satisfies error.
  108. // Errorf also records the stack trace at the point it was called.
  109. func Errorf(format string, args ...interface{}) error {
  110. return &fundamental{
  111. msg: fmt.Sprintf(format, args...),
  112. stack: callers(),
  113. }
  114. }
  115. // fundamental is an error that has a message and a stack, but no caller.
  116. type fundamental struct {
  117. msg string
  118. *stack
  119. }
  120. func (f *fundamental) Error() string { return f.msg }
  121. func (f *fundamental) Format(s fmt.State, verb rune) {
  122. switch verb {
  123. case 'v':
  124. if s.Flag('+') {
  125. io.WriteString(s, f.msg)
  126. f.stack.Format(s, verb)
  127. return
  128. }
  129. fallthrough
  130. case 's':
  131. io.WriteString(s, f.msg)
  132. case 'q':
  133. fmt.Fprintf(s, "%q", f.msg)
  134. }
  135. }
  136. // WithStack annotates err with a stack trace at the point WithStack was called.
  137. // If err is nil, WithStack returns nil.
  138. func WithStack(err error) error {
  139. if err == nil {
  140. return nil
  141. }
  142. return &withStack{
  143. err,
  144. callers(),
  145. }
  146. }
  147. type withStack struct {
  148. error
  149. *stack
  150. }
  151. func (w *withStack) Cause() error { return w.error }
  152. func (w *withStack) Format(s fmt.State, verb rune) {
  153. switch verb {
  154. case 'v':
  155. if s.Flag('+') {
  156. fmt.Fprintf(s, "%+v", w.Cause())
  157. w.stack.Format(s, verb)
  158. return
  159. }
  160. fallthrough
  161. case 's':
  162. io.WriteString(s, w.Error())
  163. case 'q':
  164. fmt.Fprintf(s, "%q", w.Error())
  165. }
  166. }
  167. // Wrap returns an error annotating err with a stack trace
  168. // at the point Wrap is called, and the supplied message.
  169. // If err is nil, Wrap returns nil.
  170. func Wrap(err error, message string) error {
  171. if err == nil {
  172. return nil
  173. }
  174. err = &withMessage{
  175. cause: err,
  176. msg: message,
  177. }
  178. return &withStack{
  179. err,
  180. callers(),
  181. }
  182. }
  183. // Wrapf returns an error annotating err with a stack trace
  184. // at the point Wrapf is called, and the format specifier.
  185. // If err is nil, Wrapf returns nil.
  186. func Wrapf(err error, format string, args ...interface{}) error {
  187. if err == nil {
  188. return nil
  189. }
  190. err = &withMessage{
  191. cause: err,
  192. msg: fmt.Sprintf(format, args...),
  193. }
  194. return &withStack{
  195. err,
  196. callers(),
  197. }
  198. }
  199. // WithMessage annotates err with a new message.
  200. // If err is nil, WithMessage returns nil.
  201. func WithMessage(err error, message string) error {
  202. if err == nil {
  203. return nil
  204. }
  205. return &withMessage{
  206. cause: err,
  207. msg: message,
  208. }
  209. }
  210. // WithMessagef annotates err with the format specifier.
  211. // If err is nil, WithMessagef returns nil.
  212. func WithMessagef(err error, format string, args ...interface{}) error {
  213. if err == nil {
  214. return nil
  215. }
  216. return &withMessage{
  217. cause: err,
  218. msg: fmt.Sprintf(format, args...),
  219. }
  220. }
  221. type withMessage struct {
  222. cause error
  223. msg string
  224. }
  225. func (w *withMessage) Error() string { return w.msg + ": " + w.cause.Error() }
  226. func (w *withMessage) Cause() error { return w.cause }
  227. func (w *withMessage) Format(s fmt.State, verb rune) {
  228. switch verb {
  229. case 'v':
  230. if s.Flag('+') {
  231. fmt.Fprintf(s, "%+v\n", w.Cause())
  232. io.WriteString(s, w.msg)
  233. return
  234. }
  235. fallthrough
  236. case 's', 'q':
  237. io.WriteString(s, w.Error())
  238. }
  239. }
  240. // Cause returns the underlying cause of the error, if possible.
  241. // An error value has a cause if it implements the following
  242. // interface:
  243. //
  244. // type causer interface {
  245. // Cause() error
  246. // }
  247. //
  248. // If the error does not implement Cause, the original error will
  249. // be returned. If the error is nil, nil will be returned without further
  250. // investigation.
  251. func Cause(err error) error {
  252. type causer interface {
  253. Cause() error
  254. }
  255. for err != nil {
  256. cause, ok := err.(causer)
  257. if !ok {
  258. break
  259. }
  260. err = cause.Cause()
  261. }
  262. return err
  263. }
上海开阖软件有限公司 沪ICP备12045867号-1