/
error_query.go
119 lines (109 loc) · 3.98 KB
/
error_query.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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
package gocb
import (
"encoding/json"
gocbcore "github.com/couchbase/gocbcore/v10"
)
// QueryErrorDesc represents a specific error returned from the query service.
type QueryErrorDesc struct {
Code uint32
Message string
Retry bool
Reason map[string]interface{}
}
// MarshalJSON implements the Marshaler interface.
func (e QueryErrorDesc) MarshalJSON() ([]byte, error) {
return json.Marshal(struct {
Code uint32 `json:"code"`
Message string `json:"message"`
Retry bool `json:"retry,omitempty"`
Reason map[string]interface{} `json:"reason,omitempty"`
}{
Code: e.Code,
Message: e.Message,
Retry: e.Retry,
Reason: e.Reason,
})
}
func translateCoreQueryErrorDesc(descs []gocbcore.N1QLErrorDesc) []QueryErrorDesc {
descsOut := make([]QueryErrorDesc, len(descs))
for descIdx, desc := range descs {
descsOut[descIdx] = QueryErrorDesc{
Code: desc.Code,
Message: desc.Message,
Retry: desc.Retry,
Reason: desc.Reason,
}
}
return descsOut
}
// QueryError is the error type of all query errors.
// UNCOMMITTED: This API may change in the future.
type QueryError struct {
InnerError error `json:"-"`
Statement string `json:"statement,omitempty"`
ClientContextID string `json:"client_context_id,omitempty"`
Errors []QueryErrorDesc `json:"errors,omitempty"`
Endpoint string `json:"endpoint,omitempty"`
RetryReasons []RetryReason `json:"retry_reasons,omitempty"`
RetryAttempts uint32 `json:"retry_attempts,omitempty"`
ErrorText string `json:"error_text,omitempty"`
HTTPStatusCode int `json:"http_status_code,omitempty"`
}
// MarshalJSON implements the Marshaler interface.
func (e QueryError) MarshalJSON() ([]byte, error) {
var innerError string
if e.InnerError != nil {
innerError = e.InnerError.Error()
}
return json.Marshal(struct {
InnerError string `json:"msg,omitempty"`
Statement string `json:"statement,omitempty"`
ClientContextID string `json:"client_context_id,omitempty"`
Errors []QueryErrorDesc `json:"errors,omitempty"`
Endpoint string `json:"endpoint,omitempty"`
RetryReasons []RetryReason `json:"retry_reasons,omitempty"`
RetryAttempts uint32 `json:"retry_attempts,omitempty"`
HTTPStatusCode int `json:"http_status_code,omitempty"`
}{
InnerError: innerError,
Statement: e.Statement,
ClientContextID: e.ClientContextID,
Errors: e.Errors,
Endpoint: e.Endpoint,
RetryReasons: e.RetryReasons,
RetryAttempts: e.RetryAttempts,
HTTPStatusCode: e.HTTPStatusCode,
})
}
// Error returns the string representation of this error.
func (e QueryError) Error() string {
errBytes, serErr := json.Marshal(struct {
InnerError error `json:"-"`
Statement string `json:"statement,omitempty"`
ClientContextID string `json:"client_context_id,omitempty"`
Errors []QueryErrorDesc `json:"errors,omitempty"`
Endpoint string `json:"endpoint,omitempty"`
RetryReasons []RetryReason `json:"retry_reasons,omitempty"`
RetryAttempts uint32 `json:"retry_attempts,omitempty"`
ErrorText string `json:"error_text,omitempty"`
HTTPStatusCode int `json:"http_status_code,omitempty"`
}{
InnerError: e.InnerError,
Statement: e.Statement,
ClientContextID: e.ClientContextID,
Errors: e.Errors,
Endpoint: e.Endpoint,
RetryReasons: e.RetryReasons,
RetryAttempts: e.RetryAttempts,
ErrorText: e.ErrorText,
HTTPStatusCode: e.HTTPStatusCode,
})
if serErr != nil {
logErrorf("failed to serialize error to json: %s", serErr.Error())
}
return e.InnerError.Error() + " | " + string(errBytes)
}
// Unwrap returns the underlying cause for this error.
func (e QueryError) Unwrap() error {
return e.InnerError
}