forked from labstack/echo
/
recover.go
30 lines (26 loc) · 809 Bytes
/
recover.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
package middleware
import (
"fmt"
"runtime"
"github.com/labstack/echo"
)
// Recover returns a middleware which recovers from panics anywhere in the chain
// and handles the control to centralized HTTPErrorHandler.
func Recover() echo.MiddlewareFunc {
// TODO: Provide better stack trace `https://github.com/go-errors/errors` `https://github.com/docker/libcontainer/tree/master/stacktrace`
return func(h echo.HandlerFunc) echo.HandlerFunc {
return func(c *echo.Context) *echo.HTTPError {
defer func() {
if err := recover(); err != nil {
trace := make([]byte, 1<<16)
n := runtime.Stack(trace, true)
c.Error(&echo.HTTPError{
Error: fmt.Errorf("echo => panic recover\n %v\n stack trace %d bytes\n %s",
err, n, trace[:n]),
})
}
}()
return h(c)
}
}
}