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