|
5 | 5 | "fmt"
|
6 | 6 | )
|
7 | 7 |
|
| 8 | +const errorExtID = 3 |
| 9 | + |
8 | 10 | // BoxError is a type representing Tarantool `box.error` object: a single
|
9 | 11 | // MP_ERROR_STACK object with a link to the previous stack error.
|
10 | 12 | // See https://www.tarantool.io/en/doc/latest/reference/reference_lua/box_error/error/
|
@@ -156,25 +158,132 @@ func decodeBoxError(d *decoder) (*BoxError, error) {
|
156 | 158 | return &errorStack[0], nil
|
157 | 159 | }
|
158 | 160 |
|
| 161 | +func encodeBoxError(enc *encoder, boxError *BoxError) error { |
| 162 | + if boxError == nil { |
| 163 | + return fmt.Errorf("msgpack: unexpected nil BoxError on encode") |
| 164 | + } |
| 165 | + |
| 166 | + if err := enc.EncodeMapLen(1); err != nil { |
| 167 | + return err |
| 168 | + } |
| 169 | + if err := encodeUint(enc, keyErrorStack); err != nil { |
| 170 | + return err |
| 171 | + } |
| 172 | + |
| 173 | + var stackDepth = boxError.Depth() |
| 174 | + if err := enc.EncodeArrayLen(stackDepth); err != nil { |
| 175 | + return err |
| 176 | + } |
| 177 | + |
| 178 | + for ; stackDepth > 0; stackDepth-- { |
| 179 | + fieldsLen := len(boxError.Fields) |
| 180 | + |
| 181 | + if fieldsLen > 0 { |
| 182 | + if err := enc.EncodeMapLen(7); err != nil { |
| 183 | + return err |
| 184 | + } |
| 185 | + } else { |
| 186 | + if err := enc.EncodeMapLen(6); err != nil { |
| 187 | + return err |
| 188 | + } |
| 189 | + } |
| 190 | + |
| 191 | + if err := encodeUint(enc, keyErrorType); err != nil { |
| 192 | + return err |
| 193 | + } |
| 194 | + if err := enc.EncodeString(boxError.Type); err != nil { |
| 195 | + return err |
| 196 | + } |
| 197 | + |
| 198 | + if err := encodeUint(enc, keyErrorFile); err != nil { |
| 199 | + return err |
| 200 | + } |
| 201 | + if err := enc.EncodeString(boxError.File); err != nil { |
| 202 | + return err |
| 203 | + } |
| 204 | + |
| 205 | + if err := encodeUint(enc, keyErrorLine); err != nil { |
| 206 | + return err |
| 207 | + } |
| 208 | + if err := enc.EncodeUint64(boxError.Line); err != nil { |
| 209 | + return err |
| 210 | + } |
| 211 | + |
| 212 | + if err := encodeUint(enc, keyErrorMessage); err != nil { |
| 213 | + return err |
| 214 | + } |
| 215 | + if err := enc.EncodeString(boxError.Msg); err != nil { |
| 216 | + return err |
| 217 | + } |
| 218 | + |
| 219 | + if err := encodeUint(enc, keyErrorErrno); err != nil { |
| 220 | + return err |
| 221 | + } |
| 222 | + if err := enc.EncodeUint64(boxError.Errno); err != nil { |
| 223 | + return err |
| 224 | + } |
| 225 | + |
| 226 | + if err := encodeUint(enc, keyErrorErrcode); err != nil { |
| 227 | + return err |
| 228 | + } |
| 229 | + if err := enc.EncodeUint64(boxError.Code); err != nil { |
| 230 | + return err |
| 231 | + } |
| 232 | + |
| 233 | + if fieldsLen > 0 { |
| 234 | + if err := encodeUint(enc, keyErrorFields); err != nil { |
| 235 | + return err |
| 236 | + } |
| 237 | + |
| 238 | + if err := enc.EncodeMapLen(fieldsLen); err != nil { |
| 239 | + return err |
| 240 | + } |
| 241 | + |
| 242 | + for k, v := range boxError.Fields { |
| 243 | + if err := enc.EncodeString(k); err != nil { |
| 244 | + return err |
| 245 | + } |
| 246 | + if err := enc.Encode(v); err != nil { |
| 247 | + return err |
| 248 | + } |
| 249 | + } |
| 250 | + } |
| 251 | + |
| 252 | + if stackDepth > 1 { |
| 253 | + boxError = boxError.Prev |
| 254 | + } |
| 255 | + } |
| 256 | + |
| 257 | + return nil |
| 258 | +} |
| 259 | + |
159 | 260 | // UnmarshalMsgpack deserializes a BoxError value from a MessagePack
|
160 | 261 | // representation.
|
161 | 262 | func (e *BoxError) UnmarshalMsgpack(b []byte) error {
|
162 |
| - var val *BoxError |
163 |
| - var err error |
164 |
| - |
165 | 263 | buf := bytes.NewBuffer(b)
|
166 | 264 | dec := newDecoder(buf)
|
167 | 265 |
|
168 |
| - if val, err = decodeBoxError(dec); err != nil { |
| 266 | + if val, err := decodeBoxError(dec); err != nil { |
169 | 267 | return err
|
170 |
| - } |
171 |
| - |
172 |
| - if val != nil { |
| 268 | + } else { |
173 | 269 | if e == nil {
|
174 | 270 | e = &BoxError{}
|
175 | 271 | }
|
| 272 | + |
176 | 273 | *e = *val
|
| 274 | + |
| 275 | + return nil |
177 | 276 | }
|
| 277 | +} |
178 | 278 |
|
179 |
| - return nil |
| 279 | +// MarshalMsgpack serializes the BoxError into a MessagePack representation. |
| 280 | +func (e *BoxError) MarshalMsgpack() ([]byte, error) { |
| 281 | + var buf bytes.Buffer |
| 282 | + |
| 283 | + enc := newEncoder(&buf) |
| 284 | + if err := encodeBoxError(enc, e); err != nil { |
| 285 | + return nil, err |
| 286 | + } |
| 287 | + |
| 288 | + return buf.Bytes(), nil |
180 | 289 | }
|
0 commit comments