forked from qiangxue/go-rest-api
-
Notifications
You must be signed in to change notification settings - Fork 0
/
middleware.go
69 lines (62 loc) · 1.74 KB
/
middleware.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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
package errors
import (
"database/sql"
"errors"
"fmt"
"net/http"
"runtime/debug"
"github.com/garaekz/go-rest-api/pkg/log"
routing "github.com/garaekz/ozzo-routing"
validation "github.com/go-ozzo/ozzo-validation/v4"
)
// Handler creates a middleware that handles panics and errors encountered during HTTP request processing.
func Handler(logger log.Logger) routing.Handler {
return func(c *routing.Context) (err error) {
defer func() {
l := logger.With(c.Request.Context())
if e := recover(); e != nil {
var ok bool
if err, ok = e.(error); !ok {
err = fmt.Errorf("%v", e)
}
l.Errorf("recovered from panic (%v): %s", err, debug.Stack())
}
if err != nil {
res := buildErrorResponse(err)
if res.StatusCode() == http.StatusInternalServerError {
l.Errorf("encountered internal server error: %v", err)
}
c.Response.WriteHeader(res.StatusCode())
if err = c.Write(res); err != nil {
l.Errorf("failed writing error response: %v", err)
}
c.Abort() // skip any pending handlers since an error has occurred
err = nil // return nil because the error is already handled
}
}()
return c.Next()
}
}
// buildErrorResponse builds an error response from an error.
func buildErrorResponse(err error) ErrorResponse {
switch err.(type) {
case ErrorResponse:
return err.(ErrorResponse)
case validation.Errors:
return InvalidInput(err.(validation.Errors))
case routing.HTTPError:
switch err.(routing.HTTPError).StatusCode() {
case http.StatusNotFound:
return NotFound("")
default:
return ErrorResponse{
Status: err.(routing.HTTPError).StatusCode(),
Message: err.Error(),
}
}
}
if errors.Is(err, sql.ErrNoRows) {
return NotFound("")
}
return InternalServerError("")
}