aws_sdk_sagemaker/protocol_serde/
shape_inference_component_compute_resource_requirements.rs

1// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
2pub fn ser_inference_component_compute_resource_requirements(
3    object: &mut ::aws_smithy_json::serialize::JsonObjectWriter,
4    input: &crate::types::InferenceComponentComputeResourceRequirements,
5) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::SerializationError> {
6    if let Some(var_1) = &input.number_of_cpu_cores_required {
7        object.key("NumberOfCpuCoresRequired").number(
8            #[allow(clippy::useless_conversion)]
9            ::aws_smithy_types::Number::Float((*var_1).into()),
10        );
11    }
12    if let Some(var_2) = &input.number_of_accelerator_devices_required {
13        object.key("NumberOfAcceleratorDevicesRequired").number(
14            #[allow(clippy::useless_conversion)]
15            ::aws_smithy_types::Number::Float((*var_2).into()),
16        );
17    }
18    if let Some(var_3) = &input.min_memory_required_in_mb {
19        object.key("MinMemoryRequiredInMb").number(
20            #[allow(clippy::useless_conversion)]
21            ::aws_smithy_types::Number::NegInt((*var_3).into()),
22        );
23    }
24    if let Some(var_4) = &input.max_memory_required_in_mb {
25        object.key("MaxMemoryRequiredInMb").number(
26            #[allow(clippy::useless_conversion)]
27            ::aws_smithy_types::Number::NegInt((*var_4).into()),
28        );
29    }
30    Ok(())
31}
32
33pub(crate) fn de_inference_component_compute_resource_requirements<'a, I>(
34    tokens: &mut ::std::iter::Peekable<I>,
35) -> ::std::result::Result<Option<crate::types::InferenceComponentComputeResourceRequirements>, ::aws_smithy_json::deserialize::error::DeserializeError>
36where
37    I: Iterator<Item = Result<::aws_smithy_json::deserialize::Token<'a>, ::aws_smithy_json::deserialize::error::DeserializeError>>,
38{
39    match tokens.next().transpose()? {
40        Some(::aws_smithy_json::deserialize::Token::ValueNull { .. }) => Ok(None),
41        Some(::aws_smithy_json::deserialize::Token::StartObject { .. }) => {
42            #[allow(unused_mut)]
43            let mut builder = crate::types::builders::InferenceComponentComputeResourceRequirementsBuilder::default();
44            loop {
45                match tokens.next().transpose()? {
46                    Some(::aws_smithy_json::deserialize::Token::EndObject { .. }) => break,
47                    Some(::aws_smithy_json::deserialize::Token::ObjectKey { key, .. }) => match key.to_unescaped()?.as_ref() {
48                        "NumberOfCpuCoresRequired" => {
49                            builder = builder.set_number_of_cpu_cores_required(
50                                ::aws_smithy_json::deserialize::token::expect_number_or_null(tokens.next())?.map(|v| v.to_f32_lossy()),
51                            );
52                        }
53                        "NumberOfAcceleratorDevicesRequired" => {
54                            builder = builder.set_number_of_accelerator_devices_required(
55                                ::aws_smithy_json::deserialize::token::expect_number_or_null(tokens.next())?.map(|v| v.to_f32_lossy()),
56                            );
57                        }
58                        "MinMemoryRequiredInMb" => {
59                            builder = builder.set_min_memory_required_in_mb(
60                                ::aws_smithy_json::deserialize::token::expect_number_or_null(tokens.next())?
61                                    .map(i32::try_from)
62                                    .transpose()?,
63                            );
64                        }
65                        "MaxMemoryRequiredInMb" => {
66                            builder = builder.set_max_memory_required_in_mb(
67                                ::aws_smithy_json::deserialize::token::expect_number_or_null(tokens.next())?
68                                    .map(i32::try_from)
69                                    .transpose()?,
70                            );
71                        }
72                        _ => ::aws_smithy_json::deserialize::token::skip_value(tokens)?,
73                    },
74                    other => {
75                        return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(format!(
76                            "expected object key or end object, found: {:?}",
77                            other
78                        )))
79                    }
80                }
81            }
82            Ok(Some(
83                crate::serde_util::inference_component_compute_resource_requirements_correct_errors(builder).build(),
84            ))
85        }
86        _ => Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(
87            "expected start object or null",
88        )),
89    }
90}