41 lines
731 B
Go
41 lines
731 B
Go
package middleware
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"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)
|
|
stackString := string(stack[:length])
|
|
fmt.Fprint(os.Stderr, stackString)
|
|
|
|
message := err.Error() + "<br><br>" + strings.ReplaceAll(stackString, "\n", "<br>")
|
|
_ = ctx.Error(http.StatusInternalServerError, message)
|
|
}()
|
|
|
|
return next(ctx)
|
|
}
|
|
}
|