StorageConfiguration.h

 
#pragma once
#include <aws/bedrock-agent/BedrockAgent_EXPORTS.h>
#include <aws/bedrock-agent/model/MongoDbAtlasConfiguration.h>
#include <aws/bedrock-agent/model/NeptuneAnalyticsConfiguration.h>
#include <aws/bedrock-agent/model/OpenSearchManagedClusterConfiguration.h>
#include <aws/bedrock-agent/model/OpenSearchServerlessConfiguration.h>
#include <aws/bedrock-agent/model/PineconeConfiguration.h>
#include <aws/bedrock-agent/model/RdsConfiguration.h>
#include <aws/bedrock-agent/model/RedisEnterpriseCloudConfiguration.h>
#include <aws/bedrock-agent/model/S3VectorsConfiguration.h>
#include <aws/bedrock-agent/model/KnowledgeBaseStorageType.h>
#include <utility>
 
namespace Aws
{
namespace Utils
{
namespace Json
{
  class JsonValue;
  class JsonView;
} // namespace Json
} // namespace Utils
namespace BedrockAgent
{
namespace Model
{
 
  class StorageConfiguration
  {
  public:
    AWS_BEDROCKAGENT_API StorageConfiguration() = default;
    AWS_BEDROCKAGENT_API StorageConfiguration(Aws::Utils::Json::JsonView jsonValue);
    AWS_BEDROCKAGENT_API StorageConfiguration& operator=(Aws::Utils::Json::JsonView jsonValue);
    AWS_BEDROCKAGENT_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
 
    inline const MongoDbAtlasConfiguration& GetMongoDbAtlasConfiguration() const { return m_mongoDbAtlasConfiguration; }
    inline bool MongoDbAtlasConfigurationHasBeenSet() const { return m_mongoDbAtlasConfigurationHasBeenSet; }
    template<typename MongoDbAtlasConfigurationT = MongoDbAtlasConfiguration>
    void SetMongoDbAtlasConfiguration(MongoDbAtlasConfigurationT&& value) { m_mongoDbAtlasConfigurationHasBeenSet = true; m_mongoDbAtlasConfiguration = std::forward<MongoDbAtlasConfigurationT>(value); }
    template<typename MongoDbAtlasConfigurationT = MongoDbAtlasConfiguration>
    StorageConfiguration& WithMongoDbAtlasConfiguration(MongoDbAtlasConfigurationT&& value) { SetMongoDbAtlasConfiguration(std::forward<MongoDbAtlasConfigurationT>(value)); return *this;}
 
 
    inline const NeptuneAnalyticsConfiguration& GetNeptuneAnalyticsConfiguration() const { return m_neptuneAnalyticsConfiguration; }
    inline bool NeptuneAnalyticsConfigurationHasBeenSet() const { return m_neptuneAnalyticsConfigurationHasBeenSet; }
    template<typename NeptuneAnalyticsConfigurationT = NeptuneAnalyticsConfiguration>
    void SetNeptuneAnalyticsConfiguration(NeptuneAnalyticsConfigurationT&& value) { m_neptuneAnalyticsConfigurationHasBeenSet = true; m_neptuneAnalyticsConfiguration = std::forward<NeptuneAnalyticsConfigurationT>(value); }
    template<typename NeptuneAnalyticsConfigurationT = NeptuneAnalyticsConfiguration>
    StorageConfiguration& WithNeptuneAnalyticsConfiguration(NeptuneAnalyticsConfigurationT&& value) { SetNeptuneAnalyticsConfiguration(std::forward<NeptuneAnalyticsConfigurationT>(value)); return *this;}
 
 
    inline const OpenSearchManagedClusterConfiguration& GetOpensearchManagedClusterConfiguration() const { return m_opensearchManagedClusterConfiguration; }
    inline bool OpensearchManagedClusterConfigurationHasBeenSet() const { return m_opensearchManagedClusterConfigurationHasBeenSet; }
    template<typename OpensearchManagedClusterConfigurationT = OpenSearchManagedClusterConfiguration>
    void SetOpensearchManagedClusterConfiguration(OpensearchManagedClusterConfigurationT&& value) { m_opensearchManagedClusterConfigurationHasBeenSet = true; m_opensearchManagedClusterConfiguration = std::forward<OpensearchManagedClusterConfigurationT>(value); }
    template<typename OpensearchManagedClusterConfigurationT = OpenSearchManagedClusterConfiguration>
    StorageConfiguration& WithOpensearchManagedClusterConfiguration(OpensearchManagedClusterConfigurationT&& value) { SetOpensearchManagedClusterConfiguration(std::forward<OpensearchManagedClusterConfigurationT>(value)); return *this;}
 
 
    inline const OpenSearchServerlessConfiguration& GetOpensearchServerlessConfiguration() const { return m_opensearchServerlessConfiguration; }
    inline bool OpensearchServerlessConfigurationHasBeenSet() const { return m_opensearchServerlessConfigurationHasBeenSet; }
    template<typename OpensearchServerlessConfigurationT = OpenSearchServerlessConfiguration>
    void SetOpensearchServerlessConfiguration(OpensearchServerlessConfigurationT&& value) { m_opensearchServerlessConfigurationHasBeenSet = true; m_opensearchServerlessConfiguration = std::forward<OpensearchServerlessConfigurationT>(value); }
    template<typename OpensearchServerlessConfigurationT = OpenSearchServerlessConfiguration>
    StorageConfiguration& WithOpensearchServerlessConfiguration(OpensearchServerlessConfigurationT&& value) { SetOpensearchServerlessConfiguration(std::forward<OpensearchServerlessConfigurationT>(value)); return *this;}
 
 
    inline const PineconeConfiguration& GetPineconeConfiguration() const { return m_pineconeConfiguration; }
    inline bool PineconeConfigurationHasBeenSet() const { return m_pineconeConfigurationHasBeenSet; }
    template<typename PineconeConfigurationT = PineconeConfiguration>
    void SetPineconeConfiguration(PineconeConfigurationT&& value) { m_pineconeConfigurationHasBeenSet = true; m_pineconeConfiguration = std::forward<PineconeConfigurationT>(value); }
    template<typename PineconeConfigurationT = PineconeConfiguration>
    StorageConfiguration& WithPineconeConfiguration(PineconeConfigurationT&& value) { SetPineconeConfiguration(std::forward<PineconeConfigurationT>(value)); return *this;}
 
 
    inline const RdsConfiguration& GetRdsConfiguration() const { return m_rdsConfiguration; }
    inline bool RdsConfigurationHasBeenSet() const { return m_rdsConfigurationHasBeenSet; }
    template<typename RdsConfigurationT = RdsConfiguration>
    void SetRdsConfiguration(RdsConfigurationT&& value) { m_rdsConfigurationHasBeenSet = true; m_rdsConfiguration = std::forward<RdsConfigurationT>(value); }
    template<typename RdsConfigurationT = RdsConfiguration>
    StorageConfiguration& WithRdsConfiguration(RdsConfigurationT&& value) { SetRdsConfiguration(std::forward<RdsConfigurationT>(value)); return *this;}
 
 
    inline const RedisEnterpriseCloudConfiguration& GetRedisEnterpriseCloudConfiguration() const { return m_redisEnterpriseCloudConfiguration; }
    inline bool RedisEnterpriseCloudConfigurationHasBeenSet() const { return m_redisEnterpriseCloudConfigurationHasBeenSet; }
    template<typename RedisEnterpriseCloudConfigurationT = RedisEnterpriseCloudConfiguration>
    void SetRedisEnterpriseCloudConfiguration(RedisEnterpriseCloudConfigurationT&& value) { m_redisEnterpriseCloudConfigurationHasBeenSet = true; m_redisEnterpriseCloudConfiguration = std::forward<RedisEnterpriseCloudConfigurationT>(value); }
    template<typename RedisEnterpriseCloudConfigurationT = RedisEnterpriseCloudConfiguration>
    StorageConfiguration& WithRedisEnterpriseCloudConfiguration(RedisEnterpriseCloudConfigurationT&& value) { SetRedisEnterpriseCloudConfiguration(std::forward<RedisEnterpriseCloudConfigurationT>(value)); return *this;}
 
 
    inline const S3VectorsConfiguration& GetS3VectorsConfiguration() const { return m_s3VectorsConfiguration; }
    inline bool S3VectorsConfigurationHasBeenSet() const { return m_s3VectorsConfigurationHasBeenSet; }
    template<typename S3VectorsConfigurationT = S3VectorsConfiguration>
    void SetS3VectorsConfiguration(S3VectorsConfigurationT&& value) { m_s3VectorsConfigurationHasBeenSet = true; m_s3VectorsConfiguration = std::forward<S3VectorsConfigurationT>(value); }
    template<typename S3VectorsConfigurationT = S3VectorsConfiguration>
    StorageConfiguration& WithS3VectorsConfiguration(S3VectorsConfigurationT&& value) { SetS3VectorsConfiguration(std::forward<S3VectorsConfigurationT>(value)); return *this;}
 
 
    inline KnowledgeBaseStorageType GetType() const { return m_type; }
    inline bool TypeHasBeenSet() const { return m_typeHasBeenSet; }
    inline void SetType(KnowledgeBaseStorageType value) { m_typeHasBeenSet = true; m_type = value; }
    inline StorageConfiguration& WithType(KnowledgeBaseStorageType value) { SetType(value); return *this;}
  private:
 
    MongoDbAtlasConfiguration m_mongoDbAtlasConfiguration;
    bool m_mongoDbAtlasConfigurationHasBeenSet = false;
 
    NeptuneAnalyticsConfiguration m_neptuneAnalyticsConfiguration;
    bool m_neptuneAnalyticsConfigurationHasBeenSet = false;
 
    OpenSearchManagedClusterConfiguration m_opensearchManagedClusterConfiguration;
    bool m_opensearchManagedClusterConfigurationHasBeenSet = false;
 
    OpenSearchServerlessConfiguration m_opensearchServerlessConfiguration;
    bool m_opensearchServerlessConfigurationHasBeenSet = false;
 
    PineconeConfiguration m_pineconeConfiguration;
    bool m_pineconeConfigurationHasBeenSet = false;
 
    RdsConfiguration m_rdsConfiguration;
    bool m_rdsConfigurationHasBeenSet = false;
 
    RedisEnterpriseCloudConfiguration m_redisEnterpriseCloudConfiguration;
    bool m_redisEnterpriseCloudConfigurationHasBeenSet = false;
 
    S3VectorsConfiguration m_s3VectorsConfiguration;
    bool m_s3VectorsConfigurationHasBeenSet = false;
 
    KnowledgeBaseStorageType m_type{KnowledgeBaseStorageType::NOT_SET};
    bool m_typeHasBeenSet = false;
  };
 
} // namespace Model
} // namespace BedrockAgent
} // namespace Aws