~ubuntu-branches/debian/stretch/golang-github-aws-aws-sdk-go/stretch

package ec2query

 

//go:generate go run ../../fixtures/protocol/generate.go ../../fixtures/protocol/output/ec2.json unmarshal_test.go

 

import (

        "encoding/xml"

        "io"

 

        "github.com/aws/aws-sdk-go/aws/awserr"

        "github.com/aws/aws-sdk-go/aws/request"

        "github.com/aws/aws-sdk-go/internal/protocol/xml/xmlutil"

)

 

// Unmarshal unmarshals a response body for the EC2 protocol.

func Unmarshal(r *request.Request) {

        defer r.HTTPResponse.Body.Close()

        if r.DataFilled() {

                decoder := xml.NewDecoder(r.HTTPResponse.Body)

                err := xmlutil.UnmarshalXML(r.Data, decoder, "")

                if err != nil {

                        r.Error = awserr.New("SerializationError", "failed decoding EC2 Query response", err)

                        return

                }

        }

}

 

// UnmarshalMeta unmarshals response headers for the EC2 protocol.

func UnmarshalMeta(r *request.Request) {

        // TODO implement unmarshaling of request IDs

}

 

type xmlErrorResponse struct {

        XMLName   xml.Name `xml:"Response"`

        Code      string   `xml:"Errors>Error>Code"`

        Message   string   `xml:"Errors>Error>Message"`

        RequestID string   `xml:"RequestId"`

}

 

// UnmarshalError unmarshals a response error for the EC2 protocol.

func UnmarshalError(r *request.Request) {

        defer r.HTTPResponse.Body.Close()

 

        resp := &xmlErrorResponse{}

        err := xml.NewDecoder(r.HTTPResponse.Body).Decode(resp)

        if err != nil && err != io.EOF {

                r.Error = awserr.New("SerializationError", "failed decoding EC2 Query error response", err)

        } else {

                r.Error = awserr.NewRequestFailure(

                        awserr.New(resp.Code, resp.Message, nil),

                        r.HTTPResponse.StatusCode,

                        resp.RequestID,

                )

        }

}