aws_sdk_dynamodb/protocol_serde/
shape_query.rs

1// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
2#[allow(clippy::unnecessary_wraps)]
3pub fn de_query_http_error(
4    _response_status: u16,
5    _response_headers: &::aws_smithy_runtime_api::http::Headers,
6    _response_body: &[u8],
7) -> std::result::Result<crate::operation::query::QueryOutput, crate::operation::query::QueryError> {
8    #[allow(unused_mut)]
9    let mut generic_builder = crate::protocol_serde::parse_http_error_metadata(_response_status, _response_headers, _response_body)
10        .map_err(crate::operation::query::QueryError::unhandled)?;
11    generic_builder = ::aws_types::request_id::apply_request_id(generic_builder, _response_headers);
12    let generic = generic_builder.build();
13    let error_code = match generic.code() {
14        Some(code) => code,
15        None => return Err(crate::operation::query::QueryError::unhandled(generic)),
16    };
17
18    let _error_message = generic.message().map(|msg| msg.to_owned());
19    Err(match error_code {
20        "InternalServerError" => crate::operation::query::QueryError::InternalServerError({
21            #[allow(unused_mut)]
22            let mut tmp = {
23                #[allow(unused_mut)]
24                let mut output = crate::types::error::builders::InternalServerErrorBuilder::default();
25                output = crate::protocol_serde::shape_internal_server_error::de_internal_server_error_json_err(_response_body, output)
26                    .map_err(crate::operation::query::QueryError::unhandled)?;
27                let output = output.meta(generic);
28                output.build()
29            };
30            if tmp.message.is_none() {
31                tmp.message = _error_message;
32            }
33            tmp
34        }),
35        "InvalidEndpointException" => crate::operation::query::QueryError::InvalidEndpointException({
36            #[allow(unused_mut)]
37            let mut tmp = {
38                #[allow(unused_mut)]
39                let mut output = crate::types::error::builders::InvalidEndpointExceptionBuilder::default();
40                output = crate::protocol_serde::shape_invalid_endpoint_exception::de_invalid_endpoint_exception_json_err(_response_body, output)
41                    .map_err(crate::operation::query::QueryError::unhandled)?;
42                let output = output.meta(generic);
43                output.build()
44            };
45            if tmp.message.is_none() {
46                tmp.message = _error_message;
47            }
48            tmp
49        }),
50        "ProvisionedThroughputExceededException" => {
51            crate::operation::query::QueryError::ProvisionedThroughputExceededException({
52                #[allow(unused_mut)]
53                let mut tmp = {
54                    #[allow(unused_mut)]
55                    let mut output = crate::types::error::builders::ProvisionedThroughputExceededExceptionBuilder::default();
56                    output = crate::protocol_serde::shape_provisioned_throughput_exceeded_exception::de_provisioned_throughput_exceeded_exception_json_err(_response_body, output).map_err(crate::operation::query::QueryError::unhandled)?;
57                    let output = output.meta(generic);
58                    output.build()
59                };
60                if tmp.message.is_none() {
61                    tmp.message = _error_message;
62                }
63                tmp
64            })
65        }
66        "RequestLimitExceeded" => crate::operation::query::QueryError::RequestLimitExceeded({
67            #[allow(unused_mut)]
68            let mut tmp = {
69                #[allow(unused_mut)]
70                let mut output = crate::types::error::builders::RequestLimitExceededBuilder::default();
71                output = crate::protocol_serde::shape_request_limit_exceeded::de_request_limit_exceeded_json_err(_response_body, output)
72                    .map_err(crate::operation::query::QueryError::unhandled)?;
73                let output = output.meta(generic);
74                output.build()
75            };
76            if tmp.message.is_none() {
77                tmp.message = _error_message;
78            }
79            tmp
80        }),
81        "ResourceNotFoundException" => crate::operation::query::QueryError::ResourceNotFoundException({
82            #[allow(unused_mut)]
83            let mut tmp = {
84                #[allow(unused_mut)]
85                let mut output = crate::types::error::builders::ResourceNotFoundExceptionBuilder::default();
86                output = crate::protocol_serde::shape_resource_not_found_exception::de_resource_not_found_exception_json_err(_response_body, output)
87                    .map_err(crate::operation::query::QueryError::unhandled)?;
88                let output = output.meta(generic);
89                output.build()
90            };
91            if tmp.message.is_none() {
92                tmp.message = _error_message;
93            }
94            tmp
95        }),
96        _ => crate::operation::query::QueryError::generic(generic),
97    })
98}
99
100#[allow(clippy::unnecessary_wraps)]
101pub fn de_query_http_response(
102    _response_status: u16,
103    _response_headers: &::aws_smithy_runtime_api::http::Headers,
104    _response_body: &[u8],
105) -> std::result::Result<crate::operation::query::QueryOutput, crate::operation::query::QueryError> {
106    Ok({
107        #[allow(unused_mut)]
108        let mut output = crate::operation::query::builders::QueryOutputBuilder::default();
109        output = crate::protocol_serde::shape_query::de_query(_response_body, output).map_err(crate::operation::query::QueryError::unhandled)?;
110        output._set_request_id(::aws_types::request_id::RequestId::request_id(_response_headers).map(str::to_string));
111        output.build()
112    })
113}
114
115pub fn ser_query_input(
116    input: &crate::operation::query::QueryInput,
117) -> ::std::result::Result<::aws_smithy_types::body::SdkBody, ::aws_smithy_types::error::operation::SerializationError> {
118    let mut out = String::new();
119    let mut object = ::aws_smithy_json::serialize::JsonObjectWriter::new(&mut out);
120    crate::protocol_serde::shape_query_input::ser_query_input_input(&mut object, input)?;
121    object.finish();
122    Ok(::aws_smithy_types::body::SdkBody::from(out))
123}
124
125pub(crate) fn de_query(
126    value: &[u8],
127    mut builder: crate::operation::query::builders::QueryOutputBuilder,
128) -> ::std::result::Result<crate::operation::query::builders::QueryOutputBuilder, ::aws_smithy_json::deserialize::error::DeserializeError> {
129    let mut tokens_owned = ::aws_smithy_json::deserialize::json_token_iter(crate::protocol_serde::or_empty_doc(value)).peekable();
130    let tokens = &mut tokens_owned;
131    ::aws_smithy_json::deserialize::token::expect_start_object(tokens.next())?;
132    loop {
133        match tokens.next().transpose()? {
134            Some(::aws_smithy_json::deserialize::Token::EndObject { .. }) => break,
135            Some(::aws_smithy_json::deserialize::Token::ObjectKey { key, .. }) => match key.to_unescaped()?.as_ref() {
136                "Items" => {
137                    builder = builder.set_items(crate::protocol_serde::shape_item_list::de_item_list(tokens)?);
138                }
139                "Count" => {
140                    builder = builder.set_count(
141                        ::aws_smithy_json::deserialize::token::expect_number_or_null(tokens.next())?
142                            .map(i32::try_from)
143                            .transpose()?,
144                    );
145                }
146                "ScannedCount" => {
147                    builder = builder.set_scanned_count(
148                        ::aws_smithy_json::deserialize::token::expect_number_or_null(tokens.next())?
149                            .map(i32::try_from)
150                            .transpose()?,
151                    );
152                }
153                "LastEvaluatedKey" => {
154                    builder = builder.set_last_evaluated_key(crate::protocol_serde::shape_key::de_key(tokens)?);
155                }
156                "ConsumedCapacity" => {
157                    builder = builder.set_consumed_capacity(crate::protocol_serde::shape_consumed_capacity::de_consumed_capacity(tokens)?);
158                }
159                _ => ::aws_smithy_json::deserialize::token::skip_value(tokens)?,
160            },
161            other => {
162                return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(format!(
163                    "expected object key or end object, found: {:?}",
164                    other
165                )))
166            }
167        }
168    }
169    if tokens.next().is_some() {
170        return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(
171            "found more JSON tokens after completing parsing",
172        ));
173    }
174    Ok(builder)
175}