Upgraded to latest aero version
This commit is contained in:
34
middleware/Recover.go
Normal file
34
middleware/Recover.go
Normal file
@ -0,0 +1,34 @@
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"runtime"
|
||||
|
||||
"github.com/aerogo/aero"
|
||||
)
|
||||
|
||||
// Recover recovers from panics and shows them as the response body.
|
||||
func Recover(next aero.Handler) aero.Handler {
|
||||
return func(ctx aero.Context) error {
|
||||
defer func() {
|
||||
r := recover()
|
||||
|
||||
if r == nil {
|
||||
return
|
||||
}
|
||||
|
||||
err, ok := r.(error)
|
||||
|
||||
if !ok {
|
||||
err = fmt.Errorf("%v", r)
|
||||
}
|
||||
|
||||
stack := make([]byte, 4096)
|
||||
length := runtime.Stack(stack, true)
|
||||
_ = ctx.Error(http.StatusInternalServerError, err, stack[:length])
|
||||
}()
|
||||
|
||||
return next(ctx)
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user