Skip to main content

aws_sdk_sagemakerruntime/protocol_serde/
shape_invoke_endpoint.rs

1// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
2#[allow(clippy::unnecessary_wraps)]
3pub fn de_invoke_endpoint_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::invoke_endpoint::InvokeEndpointOutput, crate::operation::invoke_endpoint::InvokeEndpointError> {
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::invoke_endpoint::InvokeEndpointError::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::invoke_endpoint::InvokeEndpointError::unhandled(generic)),
16    };
17
18    let _error_message = generic.message().map(|msg| msg.to_owned());
19    Err(match error_code {
20        "InternalDependencyException" => crate::operation::invoke_endpoint::InvokeEndpointError::InternalDependencyException({
21            #[allow(unused_mut)]
22            let mut tmp = {
23                #[allow(unused_mut)]
24                let mut output = crate::types::error::builders::InternalDependencyExceptionBuilder::default();
25                output =
26                    crate::protocol_serde::shape_internal_dependency_exception::de_internal_dependency_exception_json_err(_response_body, output)
27                        .map_err(crate::operation::invoke_endpoint::InvokeEndpointError::unhandled)?;
28                let output = output.meta(generic);
29                output.build()
30            };
31            if tmp.message.is_none() {
32                tmp.message = _error_message;
33            }
34            tmp
35        }),
36        "InternalFailure" => crate::operation::invoke_endpoint::InvokeEndpointError::InternalFailure({
37            #[allow(unused_mut)]
38            let mut tmp = {
39                #[allow(unused_mut)]
40                let mut output = crate::types::error::builders::InternalFailureBuilder::default();
41                output = crate::protocol_serde::shape_internal_failure::de_internal_failure_json_err(_response_body, output)
42                    .map_err(crate::operation::invoke_endpoint::InvokeEndpointError::unhandled)?;
43                let output = output.meta(generic);
44                output.build()
45            };
46            if tmp.message.is_none() {
47                tmp.message = _error_message;
48            }
49            tmp
50        }),
51        "ModelError" => crate::operation::invoke_endpoint::InvokeEndpointError::ModelError({
52            #[allow(unused_mut)]
53            let mut tmp = {
54                #[allow(unused_mut)]
55                let mut output = crate::types::error::builders::ModelErrorBuilder::default();
56                output = crate::protocol_serde::shape_model_error::de_model_error_json_err(_response_body, output)
57                    .map_err(crate::operation::invoke_endpoint::InvokeEndpointError::unhandled)?;
58                let output = output.meta(generic);
59                output.build()
60            };
61            if tmp.message.is_none() {
62                tmp.message = _error_message;
63            }
64            tmp
65        }),
66        "ModelNotReadyException" => crate::operation::invoke_endpoint::InvokeEndpointError::ModelNotReadyException({
67            #[allow(unused_mut)]
68            let mut tmp = {
69                #[allow(unused_mut)]
70                let mut output = crate::types::error::builders::ModelNotReadyExceptionBuilder::default();
71                output = crate::protocol_serde::shape_model_not_ready_exception::de_model_not_ready_exception_json_err(_response_body, output)
72                    .map_err(crate::operation::invoke_endpoint::InvokeEndpointError::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        "ServiceUnavailable" => crate::operation::invoke_endpoint::InvokeEndpointError::ServiceUnavailable({
82            #[allow(unused_mut)]
83            let mut tmp = {
84                #[allow(unused_mut)]
85                let mut output = crate::types::error::builders::ServiceUnavailableBuilder::default();
86                output = crate::protocol_serde::shape_service_unavailable::de_service_unavailable_json_err(_response_body, output)
87                    .map_err(crate::operation::invoke_endpoint::InvokeEndpointError::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        "ValidationError" => crate::operation::invoke_endpoint::InvokeEndpointError::ValidationError({
97            #[allow(unused_mut)]
98            let mut tmp = {
99                #[allow(unused_mut)]
100                let mut output = crate::types::error::builders::ValidationErrorBuilder::default();
101                output = crate::protocol_serde::shape_validation_error::de_validation_error_json_err(_response_body, output)
102                    .map_err(crate::operation::invoke_endpoint::InvokeEndpointError::unhandled)?;
103                let output = output.meta(generic);
104                output.build()
105            };
106            if tmp.message.is_none() {
107                tmp.message = _error_message;
108            }
109            tmp
110        }),
111        _ => crate::operation::invoke_endpoint::InvokeEndpointError::generic(generic),
112    })
113}
114
115#[allow(clippy::unnecessary_wraps)]
116pub fn de_invoke_endpoint_http_response(
117    _response_status: u16,
118    _response_headers: &::aws_smithy_runtime_api::http::Headers,
119    _response_body: &[u8],
120) -> std::result::Result<crate::operation::invoke_endpoint::InvokeEndpointOutput, crate::operation::invoke_endpoint::InvokeEndpointError> {
121    Ok({
122        #[allow(unused_mut)]
123        let mut output = crate::operation::invoke_endpoint::builders::InvokeEndpointOutputBuilder::default();
124        output = output.set_body(crate::protocol_serde::shape_invoke_endpoint_output::de_body_payload(_response_body)?);
125        output = output.set_closed_session_id(
126            crate::protocol_serde::shape_invoke_endpoint_output::de_closed_session_id_header(_response_headers).map_err(|_| {
127                crate::operation::invoke_endpoint::InvokeEndpointError::unhandled(
128                    "Failed to parse ClosedSessionId from header `X-Amzn-SageMaker-Closed-Session-Id",
129                )
130            })?,
131        );
132        output = output.set_content_type(
133            crate::protocol_serde::shape_invoke_endpoint_output::de_content_type_header(_response_headers).map_err(|_| {
134                crate::operation::invoke_endpoint::InvokeEndpointError::unhandled("Failed to parse ContentType from header `Content-Type")
135            })?,
136        );
137        output = output.set_custom_attributes(
138            crate::protocol_serde::shape_invoke_endpoint_output::de_custom_attributes_header(_response_headers).map_err(|_| {
139                crate::operation::invoke_endpoint::InvokeEndpointError::unhandled(
140                    "Failed to parse CustomAttributes from header `X-Amzn-SageMaker-Custom-Attributes",
141                )
142            })?,
143        );
144        output = output.set_invoked_production_variant(
145            crate::protocol_serde::shape_invoke_endpoint_output::de_invoked_production_variant_header(_response_headers).map_err(|_| {
146                crate::operation::invoke_endpoint::InvokeEndpointError::unhandled(
147                    "Failed to parse InvokedProductionVariant from header `x-Amzn-Invoked-Production-Variant",
148                )
149            })?,
150        );
151        output = output.set_new_session_id(
152            crate::protocol_serde::shape_invoke_endpoint_output::de_new_session_id_header(_response_headers).map_err(|_| {
153                crate::operation::invoke_endpoint::InvokeEndpointError::unhandled(
154                    "Failed to parse NewSessionId from header `X-Amzn-SageMaker-New-Session-Id",
155                )
156            })?,
157        );
158        output._set_request_id(::aws_types::request_id::RequestId::request_id(_response_headers).map(str::to_string));
159        crate::serde_util::invoke_endpoint_output_output_correct_errors(output).build()
160    })
161}
162
163pub fn ser_invoke_endpoint_headers(
164    input: &crate::operation::invoke_endpoint::InvokeEndpointInput,
165    mut builder: ::http_1x::request::Builder,
166) -> std::result::Result<::http_1x::request::Builder, ::aws_smithy_types::error::operation::BuildError> {
167    if let ::std::option::Option::Some(inner_1) = &input.content_type {
168        let formatted_2 = inner_1.as_str();
169        let header_value = formatted_2;
170        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
171            ::aws_smithy_types::error::operation::BuildError::invalid_field(
172                "content_type",
173                format!("`{}` cannot be used as a header value: {}", &header_value, err),
174            )
175        })?;
176        builder = builder.header("Content-Type", header_value);
177    }
178    if let ::std::option::Option::Some(inner_3) = &input.accept {
179        let formatted_4 = inner_3.as_str();
180        let header_value = formatted_4;
181        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
182            ::aws_smithy_types::error::operation::BuildError::invalid_field(
183                "accept",
184                format!("`{}` cannot be used as a header value: {}", &header_value, err),
185            )
186        })?;
187        builder = builder.header("Accept", header_value);
188    }
189    if let ::std::option::Option::Some(inner_5) = &input.custom_attributes {
190        let formatted_6 = inner_5.as_str();
191        let header_value = formatted_6;
192        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
193            ::aws_smithy_types::error::operation::BuildError::invalid_field(
194                "custom_attributes",
195                format!("`{}` cannot be used as a header value: {}", &"*** Sensitive Data Redacted ***", err),
196            )
197        })?;
198        builder = builder.header("X-Amzn-SageMaker-Custom-Attributes", header_value);
199    }
200    if let ::std::option::Option::Some(inner_7) = &input.target_model {
201        let formatted_8 = inner_7.as_str();
202        let header_value = formatted_8;
203        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
204            ::aws_smithy_types::error::operation::BuildError::invalid_field(
205                "target_model",
206                format!("`{}` cannot be used as a header value: {}", &header_value, err),
207            )
208        })?;
209        builder = builder.header("X-Amzn-SageMaker-Target-Model", header_value);
210    }
211    if let ::std::option::Option::Some(inner_9) = &input.target_variant {
212        let formatted_10 = inner_9.as_str();
213        let header_value = formatted_10;
214        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
215            ::aws_smithy_types::error::operation::BuildError::invalid_field(
216                "target_variant",
217                format!("`{}` cannot be used as a header value: {}", &header_value, err),
218            )
219        })?;
220        builder = builder.header("X-Amzn-SageMaker-Target-Variant", header_value);
221    }
222    if let ::std::option::Option::Some(inner_11) = &input.target_container_hostname {
223        let formatted_12 = inner_11.as_str();
224        let header_value = formatted_12;
225        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
226            ::aws_smithy_types::error::operation::BuildError::invalid_field(
227                "target_container_hostname",
228                format!("`{}` cannot be used as a header value: {}", &header_value, err),
229            )
230        })?;
231        builder = builder.header("X-Amzn-SageMaker-Target-Container-Hostname", header_value);
232    }
233    if let ::std::option::Option::Some(inner_13) = &input.inference_id {
234        let formatted_14 = inner_13.as_str();
235        let header_value = formatted_14;
236        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
237            ::aws_smithy_types::error::operation::BuildError::invalid_field(
238                "inference_id",
239                format!("`{}` cannot be used as a header value: {}", &header_value, err),
240            )
241        })?;
242        builder = builder.header("X-Amzn-SageMaker-Inference-Id", header_value);
243    }
244    if let ::std::option::Option::Some(inner_15) = &input.enable_explanations {
245        let formatted_16 = inner_15.as_str();
246        let header_value = formatted_16;
247        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
248            ::aws_smithy_types::error::operation::BuildError::invalid_field(
249                "enable_explanations",
250                format!("`{}` cannot be used as a header value: {}", &header_value, err),
251            )
252        })?;
253        builder = builder.header("X-Amzn-SageMaker-Enable-Explanations", header_value);
254    }
255    if let ::std::option::Option::Some(inner_17) = &input.inference_component_name {
256        let formatted_18 = inner_17.as_str();
257        let header_value = formatted_18;
258        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
259            ::aws_smithy_types::error::operation::BuildError::invalid_field(
260                "inference_component_name",
261                format!("`{}` cannot be used as a header value: {}", &header_value, err),
262            )
263        })?;
264        builder = builder.header("X-Amzn-SageMaker-Inference-Component", header_value);
265    }
266    if let ::std::option::Option::Some(inner_19) = &input.session_id {
267        let formatted_20 = inner_19.as_str();
268        let header_value = formatted_20;
269        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
270            ::aws_smithy_types::error::operation::BuildError::invalid_field(
271                "session_id",
272                format!("`{}` cannot be used as a header value: {}", &header_value, err),
273            )
274        })?;
275        builder = builder.header("X-Amzn-SageMaker-Session-Id", header_value);
276    }
277    Ok(builder)
278}