66 lines
1.3 KiB
Go
66 lines
1.3 KiB
Go
|
package pkgerrors
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
StackSourceFileName = "source"
|
||
|
|
StackSourceLineName = "line"
|
||
|
|
StackSourceFunctionName = "func"
|
||
|
|
)
|
||
|
|
|
||
|
|
type state struct {
|
||
|
|
b []byte
|
||
|
|
}
|
||
|
|
|
||
|
|
// Write implement fmt.Formatter interface.
|
||
|
|
func (s *state) Write(b []byte) (n int, err error) {
|
||
|
|
s.b = b
|
||
|
|
return len(b), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Width implement fmt.Formatter interface.
|
||
|
|
func (s *state) Width() (wid int, ok bool) {
|
||
|
|
return 0, false
|
||
|
|
}
|
||
|
|
|
||
|
|
// Precision implement fmt.Formatter interface.
|
||
|
|
func (s *state) Precision() (prec int, ok bool) {
|
||
|
|
return 0, false
|
||
|
|
}
|
||
|
|
|
||
|
|
// Flag implement fmt.Formatter interface.
|
||
|
|
func (s *state) Flag(c int) bool {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
func frameField(f errors.Frame, s *state, c rune) string {
|
||
|
|
f.Format(s, c)
|
||
|
|
return string(s.b)
|
||
|
|
}
|
||
|
|
|
||
|
|
// MarshalStack implements pkg/errors stack trace marshaling.
|
||
|
|
//
|
||
|
|
// zerolog.ErrorStackMarshaler = MarshalStack
|
||
|
|
func MarshalStack(err error) interface{} {
|
||
|
|
type stackTracer interface {
|
||
|
|
StackTrace() errors.StackTrace
|
||
|
|
}
|
||
|
|
sterr, ok := err.(stackTracer)
|
||
|
|
if !ok {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
st := sterr.StackTrace()
|
||
|
|
s := &state{}
|
||
|
|
out := make([]map[string]string, 0, len(st))
|
||
|
|
for _, frame := range st {
|
||
|
|
out = append(out, map[string]string{
|
||
|
|
StackSourceFileName: frameField(frame, s, 's'),
|
||
|
|
StackSourceLineName: frameField(frame, s, 'd'),
|
||
|
|
StackSourceFunctionName: frameField(frame, s, 'n'),
|
||
|
|
})
|
||
|
|
}
|
||
|
|
return out
|
||
|
|
}
|