48 lines
990 B
Go
48 lines
990 B
Go
// Copyright 2015 The Tango Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package tango
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"net/http"
|
|
"runtime"
|
|
)
|
|
|
|
// Recovery returns a middleware which catch panics and log them
|
|
func Recovery(debug bool) HandlerFunc {
|
|
return func(ctx *Context) {
|
|
defer func() {
|
|
if e := recover(); e != nil {
|
|
var buf bytes.Buffer
|
|
fmt.Fprintf(&buf, "Handler crashed with error: %v", e)
|
|
|
|
for i := 1; ; i++ {
|
|
_, file, line, ok := runtime.Caller(i)
|
|
if !ok {
|
|
break
|
|
} else {
|
|
fmt.Fprintf(&buf, "\n")
|
|
}
|
|
fmt.Fprintf(&buf, "%v:%v", file, line)
|
|
}
|
|
|
|
var content = buf.String()
|
|
ctx.Logger.Error(content)
|
|
|
|
if !ctx.Written() {
|
|
if !debug {
|
|
ctx.Result = InternalServerError(http.StatusText(http.StatusInternalServerError))
|
|
} else {
|
|
ctx.Result = InternalServerError(content)
|
|
}
|
|
}
|
|
}
|
|
}()
|
|
|
|
ctx.Next()
|
|
}
|
|
}
|