aws-sdk-bedrockagent 1.131.0

AWS SDK for Agents for Amazon Bedrock
Documentation
// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
pub fn ser_storage_configuration(
    object: &mut ::aws_smithy_json::serialize::JsonObjectWriter,
    input: &crate::types::StorageConfiguration,
) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::SerializationError> {
    {
        object.key("type").string(input.r#type.as_str());
    }
    if let Some(var_1) = &input.opensearch_serverless_configuration {
        #[allow(unused_mut)]
        let mut object_2 = object.key("opensearchServerlessConfiguration").start_object();
        crate::protocol_serde::shape_open_search_serverless_configuration::ser_open_search_serverless_configuration(&mut object_2, var_1)?;
        object_2.finish();
    }
    if let Some(var_3) = &input.opensearch_managed_cluster_configuration {
        #[allow(unused_mut)]
        let mut object_4 = object.key("opensearchManagedClusterConfiguration").start_object();
        crate::protocol_serde::shape_open_search_managed_cluster_configuration::ser_open_search_managed_cluster_configuration(&mut object_4, var_3)?;
        object_4.finish();
    }
    if let Some(var_5) = &input.pinecone_configuration {
        #[allow(unused_mut)]
        let mut object_6 = object.key("pineconeConfiguration").start_object();
        crate::protocol_serde::shape_pinecone_configuration::ser_pinecone_configuration(&mut object_6, var_5)?;
        object_6.finish();
    }
    if let Some(var_7) = &input.redis_enterprise_cloud_configuration {
        #[allow(unused_mut)]
        let mut object_8 = object.key("redisEnterpriseCloudConfiguration").start_object();
        crate::protocol_serde::shape_redis_enterprise_cloud_configuration::ser_redis_enterprise_cloud_configuration(&mut object_8, var_7)?;
        object_8.finish();
    }
    if let Some(var_9) = &input.rds_configuration {
        #[allow(unused_mut)]
        let mut object_10 = object.key("rdsConfiguration").start_object();
        crate::protocol_serde::shape_rds_configuration::ser_rds_configuration(&mut object_10, var_9)?;
        object_10.finish();
    }
    if let Some(var_11) = &input.mongo_db_atlas_configuration {
        #[allow(unused_mut)]
        let mut object_12 = object.key("mongoDbAtlasConfiguration").start_object();
        crate::protocol_serde::shape_mongo_db_atlas_configuration::ser_mongo_db_atlas_configuration(&mut object_12, var_11)?;
        object_12.finish();
    }
    if let Some(var_13) = &input.neptune_analytics_configuration {
        #[allow(unused_mut)]
        let mut object_14 = object.key("neptuneAnalyticsConfiguration").start_object();
        crate::protocol_serde::shape_neptune_analytics_configuration::ser_neptune_analytics_configuration(&mut object_14, var_13)?;
        object_14.finish();
    }
    if let Some(var_15) = &input.s3_vectors_configuration {
        #[allow(unused_mut)]
        let mut object_16 = object.key("s3VectorsConfiguration").start_object();
        crate::protocol_serde::shape_s3_vectors_configuration::ser_s3_vectors_configuration(&mut object_16, var_15)?;
        object_16.finish();
    }
    Ok(())
}

pub(crate) fn de_storage_configuration<'a, I>(
    tokens: &mut ::std::iter::Peekable<I>,
    _value: &'a [u8],
) -> ::std::result::Result<Option<crate::types::StorageConfiguration>, ::aws_smithy_json::deserialize::error::DeserializeError>
where
    I: Iterator<Item = Result<::aws_smithy_json::deserialize::Token<'a>, ::aws_smithy_json::deserialize::error::DeserializeError>>,
{
    match tokens.next().transpose()? {
        Some(::aws_smithy_json::deserialize::Token::ValueNull { .. }) => Ok(None),
        Some(::aws_smithy_json::deserialize::Token::StartObject { .. }) => {
            #[allow(unused_mut)]
            let mut builder = crate::types::builders::StorageConfigurationBuilder::default();
            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() {
                            "type" => {
                                builder = builder.set_type(
                                    ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
                                        .map(|s| s.to_unescaped().map(|u| crate::types::KnowledgeBaseStorageType::from(u.as_ref())))
                                        .transpose()?,
                                );
                            }
                            "opensearchServerlessConfiguration" => {
                                builder = builder.set_opensearch_serverless_configuration(
                                    crate::protocol_serde::shape_open_search_serverless_configuration::de_open_search_serverless_configuration(
                                        tokens, _value,
                                    )?,
                                );
                            }
                            "opensearchManagedClusterConfiguration" => {
                                builder = builder.set_opensearch_managed_cluster_configuration(
                                    crate::protocol_serde::shape_open_search_managed_cluster_configuration::de_open_search_managed_cluster_configuration(tokens, _value)?
                                );
                            }
                            "pineconeConfiguration" => {
                                builder = builder.set_pinecone_configuration(
                                    crate::protocol_serde::shape_pinecone_configuration::de_pinecone_configuration(tokens, _value)?,
                                );
                            }
                            "redisEnterpriseCloudConfiguration" => {
                                builder = builder.set_redis_enterprise_cloud_configuration(
                                    crate::protocol_serde::shape_redis_enterprise_cloud_configuration::de_redis_enterprise_cloud_configuration(
                                        tokens, _value,
                                    )?,
                                );
                            }
                            "rdsConfiguration" => {
                                builder = builder
                                    .set_rds_configuration(crate::protocol_serde::shape_rds_configuration::de_rds_configuration(tokens, _value)?);
                            }
                            "mongoDbAtlasConfiguration" => {
                                builder = builder.set_mongo_db_atlas_configuration(
                                    crate::protocol_serde::shape_mongo_db_atlas_configuration::de_mongo_db_atlas_configuration(tokens, _value)?,
                                );
                            }
                            "neptuneAnalyticsConfiguration" => {
                                builder = builder.set_neptune_analytics_configuration(
                                    crate::protocol_serde::shape_neptune_analytics_configuration::de_neptune_analytics_configuration(tokens, _value)?,
                                );
                            }
                            "s3VectorsConfiguration" => {
                                builder = builder.set_s3_vectors_configuration(
                                    crate::protocol_serde::shape_s3_vectors_configuration::de_s3_vectors_configuration(tokens, _value)?,
                                );
                            }
                            _ => ::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:?}"
                        )))
                    }
                }
            }
            Ok(Some(crate::serde_util::storage_configuration_correct_errors(builder).build().map_err(
                |err| ::aws_smithy_json::deserialize::error::DeserializeError::custom_source("Response was invalid", err),
            )?))
        }
        _ => Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(
            "expected start object or null",
        )),
    }
}