16 "github.com/aws/aws-sdk-go/aws"
17 "github.com/aws/aws-sdk-go/aws/awserr"
18 "github.com/aws/aws-sdk-go/aws/request"
21 // UnmarshalHandler is a named request handler for unmarshaling rest protocol requests
22 var UnmarshalHandler = request.NamedHandler{Name: "awssdk.rest.Unmarshal", Fn: Unmarshal}
24 // UnmarshalMetaHandler is a named request handler for unmarshaling rest protocol request metadata
25 var UnmarshalMetaHandler = request.NamedHandler{Name: "awssdk.rest.UnmarshalMeta", Fn: UnmarshalMeta}
27 // Unmarshal unmarshals the REST component of a response in a REST service.
28 func Unmarshal(r *request.Request) {
30 v := reflect.Indirect(reflect.ValueOf(r.Data))
35 // UnmarshalMeta unmarshals the REST metadata of a response in a REST service
36 func UnmarshalMeta(r *request.Request) {
37 r.RequestID = r.HTTPResponse.Header.Get("X-Amzn-Requestid")
38 if r.RequestID == "" {
39 // Alternative version of request id in the header
40 r.RequestID = r.HTTPResponse.Header.Get("X-Amz-Request-Id")
43 v := reflect.Indirect(reflect.ValueOf(r.Data))
44 unmarshalLocationElements(r, v)
48 func unmarshalBody(r *request.Request, v reflect.Value) {
49 if field, ok := v.Type().FieldByName("_"); ok {
50 if payloadName := field.Tag.Get("payload"); payloadName != "" {
51 pfield, _ := v.Type().FieldByName(payloadName)
52 if ptag := pfield.Tag.Get("type"); ptag != "" && ptag != "structure" {
53 payload := v.FieldByName(payloadName)
54 if payload.IsValid() {
55 switch payload.Interface().(type) {
57 defer r.HTTPResponse.Body.Close()
58 b, err := ioutil.ReadAll(r.HTTPResponse.Body)
60 r.Error = awserr.New("SerializationError", "failed to decode REST response", err)
62 payload.Set(reflect.ValueOf(b))
65 defer r.HTTPResponse.Body.Close()
66 b, err := ioutil.ReadAll(r.HTTPResponse.Body)
68 r.Error = awserr.New("SerializationError", "failed to decode REST response", err)
71 payload.Set(reflect.ValueOf(&str))
74 switch payload.Type().String() {
76 payload.Set(reflect.ValueOf(r.HTTPResponse.Body))
78 b, err := ioutil.ReadAll(r.HTTPResponse.Body)
80 r.Error = awserr.New("SerializationError",
81 "failed to read response body", err)
84 payload.Set(reflect.ValueOf(ioutil.NopCloser(bytes.NewReader(b))))
86 io.Copy(ioutil.Discard, r.HTTPResponse.Body)
87 defer r.HTTPResponse.Body.Close()
88 r.Error = awserr.New("SerializationError",
89 "failed to decode REST response",
90 fmt.Errorf("unknown payload type %s", payload.Type()))
99 func unmarshalLocationElements(r *request.Request, v reflect.Value) {
100 for i := 0; i < v.NumField(); i++ {
101 m, field := v.Field(i), v.Type().Field(i)
102 if n := field.Name; n[0:1] == strings.ToLower(n[0:1]) {
107 name := field.Tag.Get("locationName")
112 switch field.Tag.Get("location") {
114 unmarshalStatusCode(m, r.HTTPResponse.StatusCode)
116 err := unmarshalHeader(m, r.HTTPResponse.Header.Get(name), field.Tag)
118 r.Error = awserr.New("SerializationError", "failed to decode REST response", err)
122 prefix := field.Tag.Get("locationName")
123 err := unmarshalHeaderMap(m, r.HTTPResponse.Header, prefix)
125 r.Error = awserr.New("SerializationError", "failed to decode REST response", err)
136 func unmarshalStatusCode(v reflect.Value, statusCode int) {
141 switch v.Interface().(type) {
143 s := int64(statusCode)
144 v.Set(reflect.ValueOf(&s))
148 func unmarshalHeaderMap(r reflect.Value, headers http.Header, prefix string) error {
149 switch r.Interface().(type) {
150 case map[string]*string: // we only support string map value types
151 out := map[string]*string{}
152 for k, v := range headers {
153 k = http.CanonicalHeaderKey(k)
154 if strings.HasPrefix(strings.ToLower(k), strings.ToLower(prefix)) {
155 out[k[len(prefix):]] = &v[0]
158 r.Set(reflect.ValueOf(out))
163 func unmarshalHeader(v reflect.Value, header string, tag reflect.StructTag) error {
164 isJSONValue := tag.Get("type") == "jsonvalue"
166 if len(header) == 0 {
169 } else if !v.IsValid() || (header == "" && v.Elem().Kind() != reflect.String) {
173 switch v.Interface().(type) {
175 v.Set(reflect.ValueOf(&header))
177 b, err := base64.StdEncoding.DecodeString(header)
181 v.Set(reflect.ValueOf(&b))
183 b, err := strconv.ParseBool(header)
187 v.Set(reflect.ValueOf(&b))
189 i, err := strconv.ParseInt(header, 10, 64)
193 v.Set(reflect.ValueOf(&i))
195 f, err := strconv.ParseFloat(header, 64)
199 v.Set(reflect.ValueOf(&f))
201 t, err := time.Parse(RFC822, header)
205 v.Set(reflect.ValueOf(&t))
209 if tag.Get("location") == "header" {
210 b, err = base64.StdEncoding.DecodeString(header)
217 err = json.Unmarshal(b, &m)
221 v.Set(reflect.ValueOf(m))
223 err := fmt.Errorf("Unsupported value for param %v (%s)", v.Interface(), v.Type())