aws_sdk_sagemaker/protocol_serde/
shape_container_definition.rs

1// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
2pub fn ser_container_definition(
3    object: &mut ::aws_smithy_json::serialize::JsonObjectWriter,
4    input: &crate::types::ContainerDefinition,
5) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::SerializationError> {
6    if let Some(var_1) = &input.container_hostname {
7        object.key("ContainerHostname").string(var_1.as_str());
8    }
9    if let Some(var_2) = &input.image {
10        object.key("Image").string(var_2.as_str());
11    }
12    if let Some(var_3) = &input.image_config {
13        #[allow(unused_mut)]
14        let mut object_4 = object.key("ImageConfig").start_object();
15        crate::protocol_serde::shape_image_config::ser_image_config(&mut object_4, var_3)?;
16        object_4.finish();
17    }
18    if let Some(var_5) = &input.mode {
19        object.key("Mode").string(var_5.as_str());
20    }
21    if let Some(var_6) = &input.model_data_url {
22        object.key("ModelDataUrl").string(var_6.as_str());
23    }
24    if let Some(var_7) = &input.model_data_source {
25        #[allow(unused_mut)]
26        let mut object_8 = object.key("ModelDataSource").start_object();
27        crate::protocol_serde::shape_model_data_source::ser_model_data_source(&mut object_8, var_7)?;
28        object_8.finish();
29    }
30    if let Some(var_9) = &input.additional_model_data_sources {
31        let mut array_10 = object.key("AdditionalModelDataSources").start_array();
32        for item_11 in var_9 {
33            {
34                #[allow(unused_mut)]
35                let mut object_12 = array_10.value().start_object();
36                crate::protocol_serde::shape_additional_model_data_source::ser_additional_model_data_source(&mut object_12, item_11)?;
37                object_12.finish();
38            }
39        }
40        array_10.finish();
41    }
42    if let Some(var_13) = &input.environment {
43        #[allow(unused_mut)]
44        let mut object_14 = object.key("Environment").start_object();
45        for (key_15, value_16) in var_13 {
46            {
47                object_14.key(key_15.as_str()).string(value_16.as_str());
48            }
49        }
50        object_14.finish();
51    }
52    if let Some(var_17) = &input.model_package_name {
53        object.key("ModelPackageName").string(var_17.as_str());
54    }
55    if let Some(var_18) = &input.inference_specification_name {
56        object.key("InferenceSpecificationName").string(var_18.as_str());
57    }
58    if let Some(var_19) = &input.multi_model_config {
59        #[allow(unused_mut)]
60        let mut object_20 = object.key("MultiModelConfig").start_object();
61        crate::protocol_serde::shape_multi_model_config::ser_multi_model_config(&mut object_20, var_19)?;
62        object_20.finish();
63    }
64    Ok(())
65}
66
67pub(crate) fn de_container_definition<'a, I>(
68    tokens: &mut ::std::iter::Peekable<I>,
69) -> ::std::result::Result<Option<crate::types::ContainerDefinition>, ::aws_smithy_json::deserialize::error::DeserializeError>
70where
71    I: Iterator<Item = Result<::aws_smithy_json::deserialize::Token<'a>, ::aws_smithy_json::deserialize::error::DeserializeError>>,
72{
73    match tokens.next().transpose()? {
74        Some(::aws_smithy_json::deserialize::Token::ValueNull { .. }) => Ok(None),
75        Some(::aws_smithy_json::deserialize::Token::StartObject { .. }) => {
76            #[allow(unused_mut)]
77            let mut builder = crate::types::builders::ContainerDefinitionBuilder::default();
78            loop {
79                match tokens.next().transpose()? {
80                    Some(::aws_smithy_json::deserialize::Token::EndObject { .. }) => break,
81                    Some(::aws_smithy_json::deserialize::Token::ObjectKey { key, .. }) => match key.to_unescaped()?.as_ref() {
82                        "ContainerHostname" => {
83                            builder = builder.set_container_hostname(
84                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
85                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))
86                                    .transpose()?,
87                            );
88                        }
89                        "Image" => {
90                            builder = builder.set_image(
91                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
92                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))
93                                    .transpose()?,
94                            );
95                        }
96                        "ImageConfig" => {
97                            builder = builder.set_image_config(crate::protocol_serde::shape_image_config::de_image_config(tokens)?);
98                        }
99                        "Mode" => {
100                            builder = builder.set_mode(
101                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
102                                    .map(|s| s.to_unescaped().map(|u| crate::types::ContainerMode::from(u.as_ref())))
103                                    .transpose()?,
104                            );
105                        }
106                        "ModelDataUrl" => {
107                            builder = builder.set_model_data_url(
108                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
109                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))
110                                    .transpose()?,
111                            );
112                        }
113                        "ModelDataSource" => {
114                            builder = builder.set_model_data_source(crate::protocol_serde::shape_model_data_source::de_model_data_source(tokens)?);
115                        }
116                        "AdditionalModelDataSources" => {
117                            builder = builder.set_additional_model_data_sources(
118                                crate::protocol_serde::shape_additional_model_data_sources::de_additional_model_data_sources(tokens)?,
119                            );
120                        }
121                        "Environment" => {
122                            builder = builder.set_environment(crate::protocol_serde::shape_environment_map::de_environment_map(tokens)?);
123                        }
124                        "ModelPackageName" => {
125                            builder = builder.set_model_package_name(
126                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
127                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))
128                                    .transpose()?,
129                            );
130                        }
131                        "InferenceSpecificationName" => {
132                            builder = builder.set_inference_specification_name(
133                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
134                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))
135                                    .transpose()?,
136                            );
137                        }
138                        "MultiModelConfig" => {
139                            builder = builder.set_multi_model_config(crate::protocol_serde::shape_multi_model_config::de_multi_model_config(tokens)?);
140                        }
141                        _ => ::aws_smithy_json::deserialize::token::skip_value(tokens)?,
142                    },
143                    other => {
144                        return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(format!(
145                            "expected object key or end object, found: {:?}",
146                            other
147                        )))
148                    }
149                }
150            }
151            Ok(Some(builder.build()))
152        }
153        _ => Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(
154            "expected start object or null",
155        )),
156    }
157}