2017-06-06 15:58:56 +00:00
|
|
|
package query
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/xml"
|
|
|
|
"io/ioutil"
|
|
|
|
|
|
|
|
"github.com/aws/aws-sdk-go/aws/awserr"
|
|
|
|
"github.com/aws/aws-sdk-go/aws/request"
|
|
|
|
)
|
|
|
|
|
|
|
|
type xmlErrorResponse struct {
|
|
|
|
XMLName xml.Name `xml:"ErrorResponse"`
|
|
|
|
Code string `xml:"Error>Code"`
|
|
|
|
Message string `xml:"Error>Message"`
|
|
|
|
RequestID string `xml:"RequestId"`
|
|
|
|
}
|
|
|
|
|
|
|
|
type xmlServiceUnavailableResponse struct {
|
|
|
|
XMLName xml.Name `xml:"ServiceUnavailableException"`
|
|
|
|
}
|
|
|
|
|
|
|
|
// UnmarshalErrorHandler is a name request handler to unmarshal request errors
|
|
|
|
var UnmarshalErrorHandler = request.NamedHandler{Name: "awssdk.query.UnmarshalError", Fn: UnmarshalError}
|
|
|
|
|
|
|
|
// UnmarshalError unmarshals an error response for an AWS Query service.
|
|
|
|
func UnmarshalError(r *request.Request) {
|
|
|
|
defer r.HTTPResponse.Body.Close()
|
|
|
|
|
|
|
|
bodyBytes, err := ioutil.ReadAll(r.HTTPResponse.Body)
|
|
|
|
if err != nil {
|
2018-12-20 21:43:52 +00:00
|
|
|
r.Error = awserr.NewRequestFailure(
|
|
|
|
awserr.New("SerializationError", "failed to read from query HTTP response body", err),
|
|
|
|
r.HTTPResponse.StatusCode,
|
|
|
|
r.RequestID,
|
|
|
|
)
|
2017-06-06 15:58:56 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// First check for specific error
|
|
|
|
resp := xmlErrorResponse{}
|
|
|
|
decodeErr := xml.Unmarshal(bodyBytes, &resp)
|
|
|
|
if decodeErr == nil {
|
|
|
|
reqID := resp.RequestID
|
|
|
|
if reqID == "" {
|
|
|
|
reqID = r.RequestID
|
|
|
|
}
|
|
|
|
r.Error = awserr.NewRequestFailure(
|
|
|
|
awserr.New(resp.Code, resp.Message, nil),
|
|
|
|
r.HTTPResponse.StatusCode,
|
|
|
|
reqID,
|
|
|
|
)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// Check for unhandled error
|
|
|
|
servUnavailResp := xmlServiceUnavailableResponse{}
|
|
|
|
unavailErr := xml.Unmarshal(bodyBytes, &servUnavailResp)
|
|
|
|
if unavailErr == nil {
|
|
|
|
r.Error = awserr.NewRequestFailure(
|
|
|
|
awserr.New("ServiceUnavailableException", "service is unavailable", nil),
|
|
|
|
r.HTTPResponse.StatusCode,
|
|
|
|
r.RequestID,
|
|
|
|
)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// Failed to retrieve any error message from the response body
|
2018-12-20 21:43:52 +00:00
|
|
|
r.Error = awserr.NewRequestFailure(
|
|
|
|
awserr.New("SerializationError",
|
|
|
|
"failed to decode query XML error response", decodeErr),
|
|
|
|
r.HTTPResponse.StatusCode,
|
|
|
|
r.RequestID,
|
|
|
|
)
|
2017-06-06 15:58:56 +00:00
|
|
|
}
|