forked from gojek/fiber
-
Notifications
You must be signed in to change notification settings - Fork 0
/
response.go
51 lines (43 loc) · 993 Bytes
/
response.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
package fiber
import (
"github.com/gojek/fiber/errors"
"github.com/gojek/fiber/protocol"
)
type Response interface {
IsSuccess() bool
Payload() []byte
StatusCode() int
BackendName() string
WithBackendName(string) Response
}
type ErrorResponse struct {
*CachedPayload
code int
backend string
}
func (resp *ErrorResponse) IsSuccess() bool {
return false
}
func (resp *ErrorResponse) BackendName() string {
return resp.backend
}
func (resp *ErrorResponse) WithBackendName(backendName string) Response {
resp.backend = backendName
return resp
}
func (resp *ErrorResponse) StatusCode() int {
return resp.code
}
func NewErrorResponse(err error) Response {
var fiberErr *errors.FiberError
if castedError, ok := err.(*errors.FiberError); ok {
fiberErr = castedError
} else {
fiberErr = errors.NewFiberError(protocol.HTTP, err)
}
payload, _ := fiberErr.ToJSON()
return &ErrorResponse{
CachedPayload: NewCachedPayload(payload),
code: fiberErr.Code,
}
}