unmarshal.go 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. package ec2query
  2. //go:generate go run -tags codegen ../../../models/protocol_tests/generate.go ../../../models/protocol_tests/output/ec2.json unmarshal_test.go
  3. import (
  4. "encoding/xml"
  5. "io"
  6. "github.com/aws/aws-sdk-go/aws/awserr"
  7. "github.com/aws/aws-sdk-go/aws/request"
  8. "github.com/aws/aws-sdk-go/private/protocol/xml/xmlutil"
  9. )
  10. // UnmarshalHandler is a named request handler for unmarshaling ec2query protocol requests
  11. var UnmarshalHandler = request.NamedHandler{Name: "awssdk.ec2query.Unmarshal", Fn: Unmarshal}
  12. // UnmarshalMetaHandler is a named request handler for unmarshaling ec2query protocol request metadata
  13. var UnmarshalMetaHandler = request.NamedHandler{Name: "awssdk.ec2query.UnmarshalMeta", Fn: UnmarshalMeta}
  14. // UnmarshalErrorHandler is a named request handler for unmarshaling ec2query protocol request errors
  15. var UnmarshalErrorHandler = request.NamedHandler{Name: "awssdk.ec2query.UnmarshalError", Fn: UnmarshalError}
  16. // Unmarshal unmarshals a response body for the EC2 protocol.
  17. func Unmarshal(r *request.Request) {
  18. defer r.HTTPResponse.Body.Close()
  19. if r.DataFilled() {
  20. decoder := xml.NewDecoder(r.HTTPResponse.Body)
  21. err := xmlutil.UnmarshalXML(r.Data, decoder, "")
  22. if err != nil {
  23. r.Error = awserr.NewRequestFailure(
  24. awserr.New("SerializationError", "failed decoding EC2 Query response", err),
  25. r.HTTPResponse.StatusCode,
  26. r.RequestID,
  27. )
  28. return
  29. }
  30. }
  31. }
  32. // UnmarshalMeta unmarshals response headers for the EC2 protocol.
  33. func UnmarshalMeta(r *request.Request) {
  34. // TODO implement unmarshaling of request IDs
  35. }
  36. type xmlErrorResponse struct {
  37. XMLName xml.Name `xml:"Response"`
  38. Code string `xml:"Errors>Error>Code"`
  39. Message string `xml:"Errors>Error>Message"`
  40. RequestID string `xml:"RequestID"`
  41. }
  42. // UnmarshalError unmarshals a response error for the EC2 protocol.
  43. func UnmarshalError(r *request.Request) {
  44. defer r.HTTPResponse.Body.Close()
  45. resp := &xmlErrorResponse{}
  46. err := xml.NewDecoder(r.HTTPResponse.Body).Decode(resp)
  47. if err != nil && err != io.EOF {
  48. r.Error = awserr.NewRequestFailure(
  49. awserr.New("SerializationError", "failed decoding EC2 Query error response", err),
  50. r.HTTPResponse.StatusCode,
  51. r.RequestID,
  52. )
  53. } else {
  54. r.Error = awserr.NewRequestFailure(
  55. awserr.New(resp.Code, resp.Message, nil),
  56. r.HTTPResponse.StatusCode,
  57. resp.RequestID,
  58. )
  59. }
  60. }