42 lines
825 B
Go
42 lines
825 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 (
|
|
"fmt"
|
|
"net/http"
|
|
"runtime"
|
|
)
|
|
|
|
func Recovery(debug bool) HandlerFunc {
|
|
return func(ctx *Context) {
|
|
defer func() {
|
|
if e := recover(); e != nil {
|
|
content := fmt.Sprintf("Handler crashed with error: %v", e)
|
|
for i := 1; ; i += 1 {
|
|
_, file, line, ok := runtime.Caller(i)
|
|
if !ok {
|
|
break
|
|
} else {
|
|
content += "\n"
|
|
}
|
|
content += fmt.Sprintf("%v %v", file, line)
|
|
}
|
|
|
|
ctx.Logger.Error(content)
|
|
|
|
if !ctx.Written() {
|
|
if !debug {
|
|
content = http.StatusText(http.StatusInternalServerError)
|
|
}
|
|
ctx.Result = InternalServerError(content)
|
|
}
|
|
}
|
|
}()
|
|
|
|
ctx.Next()
|
|
}
|
|
}
|