54 lines
973 B
Go
54 lines
973 B
Go
|
package oops
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/go-stack/stack"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Error struct {
|
||
|
|
Message string
|
||
|
|
Wrapped error
|
||
|
|
Stack []StackFrame
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e *Error) Error() string {
|
||
|
|
return fmt.Sprintf("%s: %v", e.Message, e.Wrapped)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e *Error) Unwrap() error {
|
||
|
|
return e.Wrapped
|
||
|
|
}
|
||
|
|
|
||
|
|
type StackFrame struct {
|
||
|
|
File string `json:"file"`
|
||
|
|
Line int `json:"line"`
|
||
|
|
Function string `json:"function"`
|
||
|
|
}
|
||
|
|
|
||
|
|
var ZerologStackMarshaler = func(err error) interface{} {
|
||
|
|
if asOops, ok := err.(*Error); ok {
|
||
|
|
return asOops.Stack
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func New(wrapped error, format string, args ...interface{}) error {
|
||
|
|
trace := stack.Trace().TrimRuntime()
|
||
|
|
frames := make([]StackFrame, len(trace))
|
||
|
|
for i, call := range trace {
|
||
|
|
callFrame := call.Frame()
|
||
|
|
frames[i] = StackFrame{
|
||
|
|
File: callFrame.File,
|
||
|
|
Line: callFrame.Line,
|
||
|
|
Function: callFrame.Function,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return &Error{
|
||
|
|
Message: fmt.Sprintf(format, args...),
|
||
|
|
Wrapped: wrapped,
|
||
|
|
Stack: frames,
|
||
|
|
}
|
||
|
|
}
|