aws-sdk-sagemakerruntime 1.100.0

AWS SDK for Amazon SageMaker Runtime
Documentation
// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
#[allow(clippy::unnecessary_wraps)]
pub fn de_invoke_endpoint_async_http_error(
    _response_status: u16,
    _response_headers: &::aws_smithy_runtime_api::http::Headers,
    _response_body: &[u8],
) -> std::result::Result<
    crate::operation::invoke_endpoint_async::InvokeEndpointAsyncOutput,
    crate::operation::invoke_endpoint_async::InvokeEndpointAsyncError,
> {
    #[allow(unused_mut)]
    let mut generic_builder = crate::protocol_serde::parse_http_error_metadata(_response_status, _response_headers, _response_body)
        .map_err(crate::operation::invoke_endpoint_async::InvokeEndpointAsyncError::unhandled)?;
    generic_builder = ::aws_types::request_id::apply_request_id(generic_builder, _response_headers);
    let generic = generic_builder.build();
    let error_code = match generic.code() {
        Some(code) => code,
        None => return Err(crate::operation::invoke_endpoint_async::InvokeEndpointAsyncError::unhandled(generic)),
    };

    let _error_message = generic.message().map(|msg| msg.to_owned());
    Err(match error_code {
        "InternalFailure" => crate::operation::invoke_endpoint_async::InvokeEndpointAsyncError::InternalFailure({
            #[allow(unused_mut)]
            let mut tmp = {
                #[allow(unused_mut)]
                let mut output = crate::types::error::builders::InternalFailureBuilder::default();
                output = crate::protocol_serde::shape_internal_failure::de_internal_failure_json_err(_response_body, output)
                    .map_err(crate::operation::invoke_endpoint_async::InvokeEndpointAsyncError::unhandled)?;
                let output = output.meta(generic);
                output.build()
            };
            if tmp.message.is_none() {
                tmp.message = _error_message;
            }
            tmp
        }),
        "ServiceUnavailable" => crate::operation::invoke_endpoint_async::InvokeEndpointAsyncError::ServiceUnavailable({
            #[allow(unused_mut)]
            let mut tmp = {
                #[allow(unused_mut)]
                let mut output = crate::types::error::builders::ServiceUnavailableBuilder::default();
                output = crate::protocol_serde::shape_service_unavailable::de_service_unavailable_json_err(_response_body, output)
                    .map_err(crate::operation::invoke_endpoint_async::InvokeEndpointAsyncError::unhandled)?;
                let output = output.meta(generic);
                output.build()
            };
            if tmp.message.is_none() {
                tmp.message = _error_message;
            }
            tmp
        }),
        "ValidationError" => crate::operation::invoke_endpoint_async::InvokeEndpointAsyncError::ValidationError({
            #[allow(unused_mut)]
            let mut tmp = {
                #[allow(unused_mut)]
                let mut output = crate::types::error::builders::ValidationErrorBuilder::default();
                output = crate::protocol_serde::shape_validation_error::de_validation_error_json_err(_response_body, output)
                    .map_err(crate::operation::invoke_endpoint_async::InvokeEndpointAsyncError::unhandled)?;
                let output = output.meta(generic);
                output.build()
            };
            if tmp.message.is_none() {
                tmp.message = _error_message;
            }
            tmp
        }),
        _ => crate::operation::invoke_endpoint_async::InvokeEndpointAsyncError::generic(generic),
    })
}

#[allow(clippy::unnecessary_wraps)]
pub fn de_invoke_endpoint_async_http_response(
    _response_status: u16,
    _response_headers: &::aws_smithy_runtime_api::http::Headers,
    _response_body: &[u8],
) -> std::result::Result<
    crate::operation::invoke_endpoint_async::InvokeEndpointAsyncOutput,
    crate::operation::invoke_endpoint_async::InvokeEndpointAsyncError,
> {
    Ok({
        #[allow(unused_mut)]
        let mut output = crate::operation::invoke_endpoint_async::builders::InvokeEndpointAsyncOutputBuilder::default();
        output = crate::protocol_serde::shape_invoke_endpoint_async::de_invoke_endpoint_async(_response_body, output)
            .map_err(crate::operation::invoke_endpoint_async::InvokeEndpointAsyncError::unhandled)?;
        output = output.set_failure_location(
            crate::protocol_serde::shape_invoke_endpoint_async_output::de_failure_location_header(_response_headers).map_err(|_| {
                crate::operation::invoke_endpoint_async::InvokeEndpointAsyncError::unhandled(
                    "Failed to parse FailureLocation from header `X-Amzn-SageMaker-FailureLocation",
                )
            })?,
        );
        output = output.set_output_location(
            crate::protocol_serde::shape_invoke_endpoint_async_output::de_output_location_header(_response_headers).map_err(|_| {
                crate::operation::invoke_endpoint_async::InvokeEndpointAsyncError::unhandled(
                    "Failed to parse OutputLocation from header `X-Amzn-SageMaker-OutputLocation",
                )
            })?,
        );
        output._set_request_id(::aws_types::request_id::RequestId::request_id(_response_headers).map(str::to_string));
        output.build()
    })
}

pub fn ser_invoke_endpoint_async_headers(
    input: &crate::operation::invoke_endpoint_async::InvokeEndpointAsyncInput,
    mut builder: ::http_1x::request::Builder,
) -> std::result::Result<::http_1x::request::Builder, ::aws_smithy_types::error::operation::BuildError> {
    if let ::std::option::Option::Some(inner_1) = &input.content_type {
        let formatted_2 = inner_1.as_str();
        let header_value = formatted_2;
        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
            ::aws_smithy_types::error::operation::BuildError::invalid_field(
                "content_type",
                format!("`{}` cannot be used as a header value: {}", &header_value, err),
            )
        })?;
        builder = builder.header("X-Amzn-SageMaker-Content-Type", header_value);
    }
    if let ::std::option::Option::Some(inner_3) = &input.accept {
        let formatted_4 = inner_3.as_str();
        let header_value = formatted_4;
        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
            ::aws_smithy_types::error::operation::BuildError::invalid_field(
                "accept",
                format!("`{}` cannot be used as a header value: {}", &header_value, err),
            )
        })?;
        builder = builder.header("X-Amzn-SageMaker-Accept", header_value);
    }
    if let ::std::option::Option::Some(inner_5) = &input.custom_attributes {
        let formatted_6 = inner_5.as_str();
        let header_value = formatted_6;
        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
            ::aws_smithy_types::error::operation::BuildError::invalid_field(
                "custom_attributes",
                format!("`{}` cannot be used as a header value: {}", &"*** Sensitive Data Redacted ***", err),
            )
        })?;
        builder = builder.header("X-Amzn-SageMaker-Custom-Attributes", header_value);
    }
    if let ::std::option::Option::Some(inner_7) = &input.inference_id {
        let formatted_8 = inner_7.as_str();
        let header_value = formatted_8;
        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
            ::aws_smithy_types::error::operation::BuildError::invalid_field(
                "inference_id",
                format!("`{}` cannot be used as a header value: {}", &header_value, err),
            )
        })?;
        builder = builder.header("X-Amzn-SageMaker-Inference-Id", header_value);
    }
    if let ::std::option::Option::Some(inner_9) = &input.input_location {
        let formatted_10 = inner_9.as_str();
        let header_value = formatted_10;
        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
            ::aws_smithy_types::error::operation::BuildError::invalid_field(
                "input_location",
                format!("`{}` cannot be used as a header value: {}", &header_value, err),
            )
        })?;
        builder = builder.header("X-Amzn-SageMaker-InputLocation", header_value);
    }
    if let ::std::option::Option::Some(inner_11) = &input.s3_output_path_extension {
        let formatted_12 = inner_11.as_str();
        let header_value = formatted_12;
        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
            ::aws_smithy_types::error::operation::BuildError::invalid_field(
                "s3_output_path_extension",
                format!("`{}` cannot be used as a header value: {}", &header_value, err),
            )
        })?;
        builder = builder.header("X-Amzn-SageMaker-S3OutputPathExtension", header_value);
    }
    if let ::std::option::Option::Some(inner_13) = &input.filename {
        let formatted_14 = inner_13.as_str();
        let header_value = formatted_14;
        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
            ::aws_smithy_types::error::operation::BuildError::invalid_field(
                "filename",
                format!("`{}` cannot be used as a header value: {}", &header_value, err),
            )
        })?;
        builder = builder.header("X-Amzn-SageMaker-Filename", header_value);
    }
    if let ::std::option::Option::Some(inner_15) = &input.request_ttl_seconds {
        let mut encoder = ::aws_smithy_types::primitive::Encoder::from(*inner_15);
        let formatted_16 = encoder.encode();
        let header_value = formatted_16;
        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
            ::aws_smithy_types::error::operation::BuildError::invalid_field(
                "request_ttl_seconds",
                format!("`{}` cannot be used as a header value: {}", &header_value, err),
            )
        })?;
        builder = builder.header("X-Amzn-SageMaker-RequestTTLSeconds", header_value);
    }
    if let ::std::option::Option::Some(inner_17) = &input.invocation_timeout_seconds {
        let mut encoder = ::aws_smithy_types::primitive::Encoder::from(*inner_17);
        let formatted_18 = encoder.encode();
        let header_value = formatted_18;
        let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
            ::aws_smithy_types::error::operation::BuildError::invalid_field(
                "invocation_timeout_seconds",
                format!("`{}` cannot be used as a header value: {}", &header_value, err),
            )
        })?;
        builder = builder.header("X-Amzn-SageMaker-InvocationTimeoutSeconds", header_value);
    }
    Ok(builder)
}

pub(crate) fn de_invoke_endpoint_async(
    _value: &[u8],
    mut builder: crate::operation::invoke_endpoint_async::builders::InvokeEndpointAsyncOutputBuilder,
) -> ::std::result::Result<
    crate::operation::invoke_endpoint_async::builders::InvokeEndpointAsyncOutputBuilder,
    ::aws_smithy_json::deserialize::error::DeserializeError,
> {
    let mut tokens_owned = ::aws_smithy_json::deserialize::json_token_iter(crate::protocol_serde::or_empty_doc(_value)).peekable();
    let tokens = &mut tokens_owned;
    ::aws_smithy_json::deserialize::token::expect_start_object(tokens.next())?;
    loop {
        match tokens.next().transpose()? {
            Some(::aws_smithy_json::deserialize::Token::EndObject { .. }) => break,
            Some(::aws_smithy_json::deserialize::Token::ObjectKey { key, .. }) => match key.to_unescaped()?.as_ref() {
                "InferenceId" => {
                    builder = builder.set_inference_id(
                        ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
                            .map(|s| s.to_unescaped().map(|u| u.into_owned()))
                            .transpose()?,
                    );
                }
                _ => ::aws_smithy_json::deserialize::token::skip_value(tokens)?,
            },
            other => {
                return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(format!(
                    "expected object key or end object, found: {other:?}"
                )))
            }
        }
    }
    if tokens.next().is_some() {
        return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(
            "found more JSON tokens after completing parsing",
        ));
    }
    Ok(builder)
}