|
- package errors
-
- import (
- "fmt"
- "io"
- "path"
- "runtime"
- "strings"
- )
-
- // Frame represents a program counter inside a stack frame.
- type Frame uintptr
-
- // pc returns the program counter for this frame;
- // multiple frames may have the same PC value.
- func (f Frame) pc() uintptr { return uintptr(f) - 1 }
-
- // file returns the full path to the file that contains the
- // function for this Frame's pc.
- func (f Frame) file() string {
- fn := runtime.FuncForPC(f.pc())
- if fn == nil {
- return "unknown"
- }
- file, _ := fn.FileLine(f.pc())
- return file
- }
-
- // line returns the line number of source code of the
- // function for this Frame's pc.
- func (f Frame) line() int {
- fn := runtime.FuncForPC(f.pc())
- if fn == nil {
- return 0
- }
- _, line := fn.FileLine(f.pc())
- return line
- }
-
- // Format formats the frame according to the fmt.Formatter interface.
- //
- // %s source file
- // %d source line
- // %n function name
- // %v equivalent to %s:%d
- //
- // Format accepts flags that alter the printing of some verbs, as follows:
- //
- // %+s function name and path of source file relative to the compile time
- // GOPATH separated by \n\t (<funcname>\n\t<path>)
- // %+v equivalent to %+s:%d
- func (f Frame) Format(s fmt.State, verb rune) {
- switch verb {
- case 's':
- switch {
- case s.Flag('+'):
- pc := f.pc()
- fn := runtime.FuncForPC(pc)
- if fn == nil {
- io.WriteString(s, "unknown")
- } else {
- file, _ := fn.FileLine(pc)
- fmt.Fprintf(s, "%s\n\t%s", fn.Name(), file)
- }
- default:
- io.WriteString(s, path.Base(f.file()))
- }
- case 'd':
- fmt.Fprintf(s, "%d", f.line())
- case 'n':
- name := runtime.FuncForPC(f.pc()).Name()
- io.WriteString(s, funcname(name))
- case 'v':
- f.Format(s, 's')
- io.WriteString(s, ":")
- f.Format(s, 'd')
- }
- }
-
- // StackTrace is stack of Frames from innermost (newest) to outermost (oldest).
- type StackTrace []Frame
-
- // Format formats the stack of Frames according to the fmt.Formatter interface.
- //
- // %s lists source files for each Frame in the stack
- // %v lists the source file and line number for each Frame in the stack
- //
- // Format accepts flags that alter the printing of some verbs, as follows:
- //
- // %+v Prints filename, function, and line number for each Frame in the stack.
- func (st StackTrace) Format(s fmt.State, verb rune) {
- switch verb {
- case 'v':
- switch {
- case s.Flag('+'):
- for _, f := range st {
- fmt.Fprintf(s, "\n%+v", f)
- }
- case s.Flag('#'):
- fmt.Fprintf(s, "%#v", []Frame(st))
- default:
- fmt.Fprintf(s, "%v", []Frame(st))
- }
- case 's':
- fmt.Fprintf(s, "%s", []Frame(st))
- }
- }
-
- // stack represents a stack of program counters.
- type stack []uintptr
-
- func (s *stack) Format(st fmt.State, verb rune) {
- switch verb {
- case 'v':
- switch {
- case st.Flag('+'):
- for _, pc := range *s {
- f := Frame(pc)
- fmt.Fprintf(st, "\n%+v", f)
- }
- }
- }
- }
-
- func (s *stack) StackTrace() StackTrace {
- f := make([]Frame, len(*s))
- for i := 0; i < len(f); i++ {
- f[i] = Frame((*s)[i])
- }
- return f
- }
-
- func callers() *stack {
- const depth = 32
- var pcs [depth]uintptr
- n := runtime.Callers(3, pcs[:])
- var st stack = pcs[0:n]
- return &st
- }
-
- // funcname removes the path prefix component of a function's name reported by func.Name().
- func funcname(name string) string {
- i := strings.LastIndex(name, "/")
- name = name[i+1:]
- i = strings.Index(name, ".")
- return name[i+1:]
- }
|