You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
717 B
32 lines
717 B
package blademaster
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http/httputil"
|
|
"os"
|
|
"runtime"
|
|
|
|
"github.com/bilibili/kratos/pkg/log"
|
|
)
|
|
|
|
// Recovery returns a middleware that recovers from any panics and writes a 500 if there was one.
|
|
func Recovery() HandlerFunc {
|
|
return func(c *Context) {
|
|
defer func() {
|
|
var rawReq []byte
|
|
if err := recover(); err != nil {
|
|
const size = 64 << 10
|
|
buf := make([]byte, size)
|
|
buf = buf[:runtime.Stack(buf, false)]
|
|
if c.Request != nil {
|
|
rawReq, _ = httputil.DumpRequest(c.Request, false)
|
|
}
|
|
pl := fmt.Sprintf("http call panic: %s\n%v\n%s\n", string(rawReq), err, buf)
|
|
fmt.Fprintf(os.Stderr, pl)
|
|
log.Error(pl)
|
|
c.AbortWithStatus(500)
|
|
}
|
|
}()
|
|
c.Next()
|
|
}
|
|
}
|
|
|