mirror of
https://github.com/letic/terraform-provider-google.git
synced 2024-09-07 01:26:43 +00:00
67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
![]() |
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 {
|
||
|
r.Error = awserr.New("SerializationError", "failed to read from query HTTP response body", err)
|
||
|
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
|
||
|
r.Error = awserr.New("SerializationError",
|
||
|
"failed to decode query XML error response", decodeErr)
|
||
|
}
|