-
Notifications
You must be signed in to change notification settings - Fork 1
/
recoverer.go
38 lines (35 loc) · 960 Bytes
/
recoverer.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
31
32
33
34
35
36
37
38
package middleware
import (
"errors"
"fmt"
"github.com/Nigel2392/router/v3"
"github.com/Nigel2392/router/v3/request"
)
// Recoverer recovers from panics and logs the error,
// if the logger was set.
func Recoverer(onError func(err error, r *request.Request)) router.Middleware {
return func(next router.Handler) router.Handler {
return router.HandleFunc(func(r *request.Request) {
defer func() {
if err := recover(); err != nil {
if DEFAULT_LOGGER != nil {
DEFAULT_LOGGER.Error(FormatMessage(r, "PANIC", "Panic: %v", err))
}
r.Response.Clear()
var newHandler = router.HandleFunc(func(r *request.Request) {
switch err := err.(type) {
case error:
onError(err, r)
case string:
onError(errors.New(err), r)
default:
onError(fmt.Errorf("%v", err), r)
}
})
newHandler.ServeHTTP(r)
}
}()
next.ServeHTTP(r)
})
}
}