aws_sdk_sagemaker/protocol_serde/
shape_dataset_definition.rs

1// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
2pub fn ser_dataset_definition(
3    object: &mut ::aws_smithy_json::serialize::JsonObjectWriter,
4    input: &crate::types::DatasetDefinition,
5) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::SerializationError> {
6    if let Some(var_1) = &input.athena_dataset_definition {
7        #[allow(unused_mut)]
8        let mut object_2 = object.key("AthenaDatasetDefinition").start_object();
9        crate::protocol_serde::shape_athena_dataset_definition::ser_athena_dataset_definition(&mut object_2, var_1)?;
10        object_2.finish();
11    }
12    if let Some(var_3) = &input.redshift_dataset_definition {
13        #[allow(unused_mut)]
14        let mut object_4 = object.key("RedshiftDatasetDefinition").start_object();
15        crate::protocol_serde::shape_redshift_dataset_definition::ser_redshift_dataset_definition(&mut object_4, var_3)?;
16        object_4.finish();
17    }
18    if let Some(var_5) = &input.local_path {
19        object.key("LocalPath").string(var_5.as_str());
20    }
21    if let Some(var_6) = &input.data_distribution_type {
22        object.key("DataDistributionType").string(var_6.as_str());
23    }
24    if let Some(var_7) = &input.input_mode {
25        object.key("InputMode").string(var_7.as_str());
26    }
27    Ok(())
28}
29
30pub(crate) fn de_dataset_definition<'a, I>(
31    tokens: &mut ::std::iter::Peekable<I>,
32) -> ::std::result::Result<Option<crate::types::DatasetDefinition>, ::aws_smithy_json::deserialize::error::DeserializeError>
33where
34    I: Iterator<Item = Result<::aws_smithy_json::deserialize::Token<'a>, ::aws_smithy_json::deserialize::error::DeserializeError>>,
35{
36    match tokens.next().transpose()? {
37        Some(::aws_smithy_json::deserialize::Token::ValueNull { .. }) => Ok(None),
38        Some(::aws_smithy_json::deserialize::Token::StartObject { .. }) => {
39            #[allow(unused_mut)]
40            let mut builder = crate::types::builders::DatasetDefinitionBuilder::default();
41            loop {
42                match tokens.next().transpose()? {
43                    Some(::aws_smithy_json::deserialize::Token::EndObject { .. }) => break,
44                    Some(::aws_smithy_json::deserialize::Token::ObjectKey { key, .. }) => match key.to_unescaped()?.as_ref() {
45                        "AthenaDatasetDefinition" => {
46                            builder = builder.set_athena_dataset_definition(
47                                crate::protocol_serde::shape_athena_dataset_definition::de_athena_dataset_definition(tokens)?,
48                            );
49                        }
50                        "RedshiftDatasetDefinition" => {
51                            builder = builder.set_redshift_dataset_definition(
52                                crate::protocol_serde::shape_redshift_dataset_definition::de_redshift_dataset_definition(tokens)?,
53                            );
54                        }
55                        "LocalPath" => {
56                            builder = builder.set_local_path(
57                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
58                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))
59                                    .transpose()?,
60                            );
61                        }
62                        "DataDistributionType" => {
63                            builder = builder.set_data_distribution_type(
64                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
65                                    .map(|s| s.to_unescaped().map(|u| crate::types::DataDistributionType::from(u.as_ref())))
66                                    .transpose()?,
67                            );
68                        }
69                        "InputMode" => {
70                            builder = builder.set_input_mode(
71                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
72                                    .map(|s| s.to_unescaped().map(|u| crate::types::InputMode::from(u.as_ref())))
73                                    .transpose()?,
74                            );
75                        }
76                        _ => ::aws_smithy_json::deserialize::token::skip_value(tokens)?,
77                    },
78                    other => {
79                        return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(format!(
80                            "expected object key or end object, found: {:?}",
81                            other
82                        )))
83                    }
84                }
85            }
86            Ok(Some(builder.build()))
87        }
88        _ => Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(
89            "expected start object or null",
90        )),
91    }
92}