aws-sdk-athena 1.64.0 → 1.66.0

data/lib/aws-sdk-athena/client_api.rb

@@ -15,6 +15,7 @@ module Aws::Athena
 
     AclConfiguration = Shapes::StructureShape.new(name: 'AclConfiguration')
     Age = Shapes::IntegerShape.new(name: 'Age')
+    AllocatedDpusInteger = Shapes::IntegerShape.new(name: 'AllocatedDpusInteger')
     AmazonResourceName = Shapes::StringShape.new(name: 'AmazonResourceName')
     ApplicationDPUSizes = Shapes::StructureShape.new(name: 'ApplicationDPUSizes')
     ApplicationDPUSizesList = Shapes::ListShape.new(name: 'ApplicationDPUSizesList')
@@ -39,6 +40,17 @@ module Aws::Athena
     CalculationStatus = Shapes::StructureShape.new(name: 'CalculationStatus')
     CalculationSummary = Shapes::StructureShape.new(name: 'CalculationSummary')
     CalculationsList = Shapes::ListShape.new(name: 'CalculationsList')
+    CancelCapacityReservationInput = Shapes::StructureShape.new(name: 'CancelCapacityReservationInput')
+    CancelCapacityReservationOutput = Shapes::StructureShape.new(name: 'CancelCapacityReservationOutput')
+    CapacityAllocation = Shapes::StructureShape.new(name: 'CapacityAllocation')
+    CapacityAllocationStatus = Shapes::StringShape.new(name: 'CapacityAllocationStatus')
+    CapacityAssignment = Shapes::StructureShape.new(name: 'CapacityAssignment')
+    CapacityAssignmentConfiguration = Shapes::StructureShape.new(name: 'CapacityAssignmentConfiguration')
+    CapacityAssignmentsList = Shapes::ListShape.new(name: 'CapacityAssignmentsList')
+    CapacityReservation = Shapes::StructureShape.new(name: 'CapacityReservation')
+    CapacityReservationName = Shapes::StringShape.new(name: 'CapacityReservationName')
+    CapacityReservationStatus = Shapes::StringShape.new(name: 'CapacityReservationStatus')
+    CapacityReservationsList = Shapes::ListShape.new(name: 'CapacityReservationsList')
     CatalogNameString = Shapes::StringShape.new(name: 'CatalogNameString')
     ClientRequestToken = Shapes::StringShape.new(name: 'ClientRequestToken')
     CodeBlock = Shapes::StringShape.new(name: 'CodeBlock')
@@ -49,6 +61,8 @@ module Aws::Athena
     ColumnNullable = Shapes::StringShape.new(name: 'ColumnNullable')
     CommentString = Shapes::StringShape.new(name: 'CommentString')
     CoordinatorDpuSize = Shapes::IntegerShape.new(name: 'CoordinatorDpuSize')
+    CreateCapacityReservationInput = Shapes::StructureShape.new(name: 'CreateCapacityReservationInput')
+    CreateCapacityReservationOutput = Shapes::StructureShape.new(name: 'CreateCapacityReservationOutput')
     CreateDataCatalogInput = Shapes::StructureShape.new(name: 'CreateDataCatalogInput')
     CreateDataCatalogOutput = Shapes::StructureShape.new(name: 'CreateDataCatalogOutput')
     CreateNamedQueryInput = Shapes::StructureShape.new(name: 'CreateNamedQueryInput')
@@ -109,6 +123,10 @@ module Aws::Athena
     GetCalculationExecutionResponse = Shapes::StructureShape.new(name: 'GetCalculationExecutionResponse')
     GetCalculationExecutionStatusRequest = Shapes::StructureShape.new(name: 'GetCalculationExecutionStatusRequest')
     GetCalculationExecutionStatusResponse = Shapes::StructureShape.new(name: 'GetCalculationExecutionStatusResponse')
+    GetCapacityAssignmentConfigurationInput = Shapes::StructureShape.new(name: 'GetCapacityAssignmentConfigurationInput')
+    GetCapacityAssignmentConfigurationOutput = Shapes::StructureShape.new(name: 'GetCapacityAssignmentConfigurationOutput')
+    GetCapacityReservationInput = Shapes::StructureShape.new(name: 'GetCapacityReservationInput')
+    GetCapacityReservationOutput = Shapes::StructureShape.new(name: 'GetCapacityReservationOutput')
     GetDataCatalogInput = Shapes::StructureShape.new(name: 'GetDataCatalogInput')
     GetDataCatalogOutput = Shapes::StructureShape.new(name: 'GetDataCatalogOutput')
     GetDatabaseInput = Shapes::StructureShape.new(name: 'GetDatabaseInput')
@@ -145,6 +163,8 @@ module Aws::Athena
     ListApplicationDPUSizesOutput = Shapes::StructureShape.new(name: 'ListApplicationDPUSizesOutput')
     ListCalculationExecutionsRequest = Shapes::StructureShape.new(name: 'ListCalculationExecutionsRequest')
     ListCalculationExecutionsResponse = Shapes::StructureShape.new(name: 'ListCalculationExecutionsResponse')
+    ListCapacityReservationsInput = Shapes::StructureShape.new(name: 'ListCapacityReservationsInput')
+    ListCapacityReservationsOutput = Shapes::StructureShape.new(name: 'ListCapacityReservationsOutput')
     ListDataCatalogsInput = Shapes::StructureShape.new(name: 'ListDataCatalogsInput')
     ListDataCatalogsOutput = Shapes::StructureShape.new(name: 'ListDataCatalogsOutput')
     ListDatabasesInput = Shapes::StructureShape.new(name: 'ListDatabasesInput')
@@ -174,6 +194,7 @@ module Aws::Athena
     Long = Shapes::IntegerShape.new(name: 'Long')
     MaxApplicationDPUSizesCount = Shapes::IntegerShape.new(name: 'MaxApplicationDPUSizesCount')
     MaxCalculationsCount = Shapes::IntegerShape.new(name: 'MaxCalculationsCount')
+    MaxCapacityReservationsCount = Shapes::IntegerShape.new(name: 'MaxCapacityReservationsCount')
     MaxConcurrentDpus = Shapes::IntegerShape.new(name: 'MaxConcurrentDpus')
     MaxDataCatalogsCount = Shapes::IntegerShape.new(name: 'MaxDataCatalogsCount')
     MaxDatabasesCount = Shapes::IntegerShape.new(name: 'MaxDatabasesCount')
@@ -210,6 +231,8 @@ module Aws::Athena
     PreparedStatementNameList = Shapes::ListShape.new(name: 'PreparedStatementNameList')
     PreparedStatementSummary = Shapes::StructureShape.new(name: 'PreparedStatementSummary')
     PreparedStatementsList = Shapes::ListShape.new(name: 'PreparedStatementsList')
+    PutCapacityAssignmentConfigurationInput = Shapes::StructureShape.new(name: 'PutCapacityAssignmentConfigurationInput')
+    PutCapacityAssignmentConfigurationOutput = Shapes::StructureShape.new(name: 'PutCapacityAssignmentConfigurationOutput')
     QueryExecution = Shapes::StructureShape.new(name: 'QueryExecution')
     QueryExecutionContext = Shapes::StructureShape.new(name: 'QueryExecutionContext')
     QueryExecutionId = Shapes::StringShape.new(name: 'QueryExecutionId')
@@ -275,6 +298,7 @@ module Aws::Athena
     TagResourceInput = Shapes::StructureShape.new(name: 'TagResourceInput')
     TagResourceOutput = Shapes::StructureShape.new(name: 'TagResourceOutput')
     TagValue = Shapes::StringShape.new(name: 'TagValue')
+    TargetDpusInteger = Shapes::IntegerShape.new(name: 'TargetDpusInteger')
     TerminateSessionRequest = Shapes::StructureShape.new(name: 'TerminateSessionRequest')
     TerminateSessionResponse = Shapes::StructureShape.new(name: 'TerminateSessionResponse')
     ThrottleReason = Shapes::StringShape.new(name: 'ThrottleReason')
@@ -290,6 +314,8 @@ module Aws::Athena
     UnprocessedQueryExecutionIdList = Shapes::ListShape.new(name: 'UnprocessedQueryExecutionIdList')
     UntagResourceInput = Shapes::StructureShape.new(name: 'UntagResourceInput')
     UntagResourceOutput = Shapes::StructureShape.new(name: 'UntagResourceOutput')
+    UpdateCapacityReservationInput = Shapes::StructureShape.new(name: 'UpdateCapacityReservationInput')
+    UpdateCapacityReservationOutput = Shapes::StructureShape.new(name: 'UpdateCapacityReservationOutput')
     UpdateDataCatalogInput = Shapes::StructureShape.new(name: 'UpdateDataCatalogInput')
     UpdateDataCatalogOutput = Shapes::StructureShape.new(name: 'UpdateDataCatalogOutput')
     UpdateNamedQueryInput = Shapes::StructureShape.new(name: 'UpdateNamedQueryInput')
@@ -307,6 +333,7 @@ module Aws::Athena
     WorkGroupConfigurationUpdates = Shapes::StructureShape.new(name: 'WorkGroupConfigurationUpdates')
     WorkGroupDescriptionString = Shapes::StringShape.new(name: 'WorkGroupDescriptionString')
     WorkGroupName = Shapes::StringShape.new(name: 'WorkGroupName')
+    WorkGroupNamesList = Shapes::ListShape.new(name: 'WorkGroupNamesList')
     WorkGroupState = Shapes::StringShape.new(name: 'WorkGroupState')
     WorkGroupSummary = Shapes::StructureShape.new(name: 'WorkGroupSummary')
     WorkGroupsList = Shapes::ListShape.new(name: 'WorkGroupsList')
@@ -376,6 +403,37 @@ module Aws::Athena
 
     CalculationsList.member = Shapes::ShapeRef.new(shape: CalculationSummary)
 
+    CancelCapacityReservationInput.add_member(:name, Shapes::ShapeRef.new(shape: CapacityReservationName, required: true, location_name: "Name"))
+    CancelCapacityReservationInput.struct_class = Types::CancelCapacityReservationInput
+
+    CancelCapacityReservationOutput.struct_class = Types::CancelCapacityReservationOutput
+
+    CapacityAllocation.add_member(:status, Shapes::ShapeRef.new(shape: CapacityAllocationStatus, required: true, location_name: "Status"))
+    CapacityAllocation.add_member(:status_message, Shapes::ShapeRef.new(shape: String, location_name: "StatusMessage"))
+    CapacityAllocation.add_member(:request_time, Shapes::ShapeRef.new(shape: Timestamp, required: true, location_name: "RequestTime"))
+    CapacityAllocation.add_member(:request_completion_time, Shapes::ShapeRef.new(shape: Timestamp, location_name: "RequestCompletionTime"))
+    CapacityAllocation.struct_class = Types::CapacityAllocation
+
+    CapacityAssignment.add_member(:work_group_names, Shapes::ShapeRef.new(shape: WorkGroupNamesList, location_name: "WorkGroupNames"))
+    CapacityAssignment.struct_class = Types::CapacityAssignment
+
+    CapacityAssignmentConfiguration.add_member(:capacity_reservation_name, Shapes::ShapeRef.new(shape: CapacityReservationName, location_name: "CapacityReservationName"))
+    CapacityAssignmentConfiguration.add_member(:capacity_assignments, Shapes::ShapeRef.new(shape: CapacityAssignmentsList, location_name: "CapacityAssignments"))
+    CapacityAssignmentConfiguration.struct_class = Types::CapacityAssignmentConfiguration
+
+    CapacityAssignmentsList.member = Shapes::ShapeRef.new(shape: CapacityAssignment)
+
+    CapacityReservation.add_member(:name, Shapes::ShapeRef.new(shape: CapacityReservationName, required: true, location_name: "Name"))
+    CapacityReservation.add_member(:status, Shapes::ShapeRef.new(shape: CapacityReservationStatus, required: true, location_name: "Status"))
+    CapacityReservation.add_member(:target_dpus, Shapes::ShapeRef.new(shape: TargetDpusInteger, required: true, location_name: "TargetDpus"))
+    CapacityReservation.add_member(:allocated_dpus, Shapes::ShapeRef.new(shape: AllocatedDpusInteger, required: true, location_name: "AllocatedDpus"))
+    CapacityReservation.add_member(:last_allocation, Shapes::ShapeRef.new(shape: CapacityAllocation, location_name: "LastAllocation"))
+    CapacityReservation.add_member(:last_successful_allocation_time, Shapes::ShapeRef.new(shape: Timestamp, location_name: "LastSuccessfulAllocationTime"))
+    CapacityReservation.add_member(:creation_time, Shapes::ShapeRef.new(shape: Timestamp, required: true, location_name: "CreationTime"))
+    CapacityReservation.struct_class = Types::CapacityReservation
+
+    CapacityReservationsList.member = Shapes::ShapeRef.new(shape: CapacityReservation)
+
     Column.add_member(:name, Shapes::ShapeRef.new(shape: NameString, required: true, location_name: "Name"))
     Column.add_member(:type, Shapes::ShapeRef.new(shape: TypeString, location_name: "Type"))
     Column.add_member(:comment, Shapes::ShapeRef.new(shape: CommentString, location_name: "Comment"))
@@ -397,6 +455,13 @@ module Aws::Athena
 
     ColumnList.member = Shapes::ShapeRef.new(shape: Column)
 
+    CreateCapacityReservationInput.add_member(:target_dpus, Shapes::ShapeRef.new(shape: TargetDpusInteger, required: true, location_name: "TargetDpus"))
+    CreateCapacityReservationInput.add_member(:name, Shapes::ShapeRef.new(shape: CapacityReservationName, required: true, location_name: "Name"))
+    CreateCapacityReservationInput.add_member(:tags, Shapes::ShapeRef.new(shape: TagList, location_name: "Tags"))
+    CreateCapacityReservationInput.struct_class = Types::CreateCapacityReservationInput
+
+    CreateCapacityReservationOutput.struct_class = Types::CreateCapacityReservationOutput
+
     CreateDataCatalogInput.add_member(:name, Shapes::ShapeRef.new(shape: CatalogNameString, required: true, location_name: "Name"))
     CreateDataCatalogInput.add_member(:type, Shapes::ShapeRef.new(shape: DataCatalogType, required: true, location_name: "Type"))
     CreateDataCatalogInput.add_member(:description, Shapes::ShapeRef.new(shape: DescriptionString, location_name: "Description"))
@@ -509,6 +574,7 @@ module Aws::Athena
     EngineConfiguration.add_member(:max_concurrent_dpus, Shapes::ShapeRef.new(shape: MaxConcurrentDpus, required: true, location_name: "MaxConcurrentDpus"))
     EngineConfiguration.add_member(:default_executor_dpu_size, Shapes::ShapeRef.new(shape: DefaultExecutorDpuSize, location_name: "DefaultExecutorDpuSize"))
     EngineConfiguration.add_member(:additional_configs, Shapes::ShapeRef.new(shape: ParametersMap, location_name: "AdditionalConfigs"))
+    EngineConfiguration.add_member(:spark_properties, Shapes::ShapeRef.new(shape: ParametersMap, location_name: "SparkProperties"))
     EngineConfiguration.struct_class = Types::EngineConfiguration
 
     EngineVersion.add_member(:selected_engine_version, Shapes::ShapeRef.new(shape: NameString, location_name: "SelectedEngineVersion"))
@@ -564,6 +630,18 @@ module Aws::Athena
     GetCalculationExecutionStatusResponse.add_member(:statistics, Shapes::ShapeRef.new(shape: CalculationStatistics, location_name: "Statistics"))
     GetCalculationExecutionStatusResponse.struct_class = Types::GetCalculationExecutionStatusResponse
 
+    GetCapacityAssignmentConfigurationInput.add_member(:capacity_reservation_name, Shapes::ShapeRef.new(shape: CapacityReservationName, required: true, location_name: "CapacityReservationName"))
+    GetCapacityAssignmentConfigurationInput.struct_class = Types::GetCapacityAssignmentConfigurationInput
+
+    GetCapacityAssignmentConfigurationOutput.add_member(:capacity_assignment_configuration, Shapes::ShapeRef.new(shape: CapacityAssignmentConfiguration, required: true, location_name: "CapacityAssignmentConfiguration"))
+    GetCapacityAssignmentConfigurationOutput.struct_class = Types::GetCapacityAssignmentConfigurationOutput
+
+    GetCapacityReservationInput.add_member(:name, Shapes::ShapeRef.new(shape: CapacityReservationName, required: true, location_name: "Name"))
+    GetCapacityReservationInput.struct_class = Types::GetCapacityReservationInput
+
+    GetCapacityReservationOutput.add_member(:capacity_reservation, Shapes::ShapeRef.new(shape: CapacityReservation, required: true, location_name: "CapacityReservation"))
+    GetCapacityReservationOutput.struct_class = Types::GetCapacityReservationOutput
+
     GetDataCatalogInput.add_member(:name, Shapes::ShapeRef.new(shape: CatalogNameString, required: true, location_name: "Name"))
     GetDataCatalogInput.struct_class = Types::GetDataCatalogInput
 
@@ -688,6 +766,14 @@ module Aws::Athena
     ListCalculationExecutionsResponse.add_member(:calculations, Shapes::ShapeRef.new(shape: CalculationsList, location_name: "Calculations"))
     ListCalculationExecutionsResponse.struct_class = Types::ListCalculationExecutionsResponse
 
+    ListCapacityReservationsInput.add_member(:next_token, Shapes::ShapeRef.new(shape: Token, location_name: "NextToken"))
+    ListCapacityReservationsInput.add_member(:max_results, Shapes::ShapeRef.new(shape: MaxCapacityReservationsCount, location_name: "MaxResults"))
+    ListCapacityReservationsInput.struct_class = Types::ListCapacityReservationsInput
+
+    ListCapacityReservationsOutput.add_member(:next_token, Shapes::ShapeRef.new(shape: Token, location_name: "NextToken"))
+    ListCapacityReservationsOutput.add_member(:capacity_reservations, Shapes::ShapeRef.new(shape: CapacityReservationsList, required: true, location_name: "CapacityReservations"))
+    ListCapacityReservationsOutput.struct_class = Types::ListCapacityReservationsOutput
+
     ListDataCatalogsInput.add_member(:next_token, Shapes::ShapeRef.new(shape: Token, location_name: "NextToken"))
     ListDataCatalogsInput.add_member(:max_results, Shapes::ShapeRef.new(shape: MaxDataCatalogsCount, location_name: "MaxResults"))
     ListDataCatalogsInput.struct_class = Types::ListDataCatalogsInput
@@ -859,6 +945,12 @@ module Aws::Athena
 
     PreparedStatementsList.member = Shapes::ShapeRef.new(shape: PreparedStatementSummary)
 
+    PutCapacityAssignmentConfigurationInput.add_member(:capacity_reservation_name, Shapes::ShapeRef.new(shape: CapacityReservationName, required: true, location_name: "CapacityReservationName"))
+    PutCapacityAssignmentConfigurationInput.add_member(:capacity_assignments, Shapes::ShapeRef.new(shape: CapacityAssignmentsList, required: true, location_name: "CapacityAssignments"))
+    PutCapacityAssignmentConfigurationInput.struct_class = Types::PutCapacityAssignmentConfigurationInput
+
+    PutCapacityAssignmentConfigurationOutput.struct_class = Types::PutCapacityAssignmentConfigurationOutput
+
     QueryExecution.add_member(:query_execution_id, Shapes::ShapeRef.new(shape: QueryExecutionId, location_name: "QueryExecutionId"))
     QueryExecution.add_member(:query, Shapes::ShapeRef.new(shape: QueryString, location_name: "Query"))
     QueryExecution.add_member(:statement_type, Shapes::ShapeRef.new(shape: StatementType, location_name: "StatementType"))
@@ -1120,6 +1212,12 @@ module Aws::Athena
 
     UntagResourceOutput.struct_class = Types::UntagResourceOutput
 
+    UpdateCapacityReservationInput.add_member(:target_dpus, Shapes::ShapeRef.new(shape: TargetDpusInteger, required: true, location_name: "TargetDpus"))
+    UpdateCapacityReservationInput.add_member(:name, Shapes::ShapeRef.new(shape: CapacityReservationName, required: true, location_name: "Name"))
+    UpdateCapacityReservationInput.struct_class = Types::UpdateCapacityReservationInput
+
+    UpdateCapacityReservationOutput.struct_class = Types::UpdateCapacityReservationOutput
+
     UpdateDataCatalogInput.add_member(:name, Shapes::ShapeRef.new(shape: CatalogNameString, required: true, location_name: "Name"))
     UpdateDataCatalogInput.add_member(:type, Shapes::ShapeRef.new(shape: DataCatalogType, required: true, location_name: "Type"))
     UpdateDataCatalogInput.add_member(:description, Shapes::ShapeRef.new(shape: DescriptionString, location_name: "Description"))
@@ -1201,6 +1299,8 @@ module Aws::Athena
     WorkGroupConfigurationUpdates.add_member(:enable_minimum_encryption_configuration, Shapes::ShapeRef.new(shape: BoxedBoolean, location_name: "EnableMinimumEncryptionConfiguration"))
     WorkGroupConfigurationUpdates.struct_class = Types::WorkGroupConfigurationUpdates
 
+    WorkGroupNamesList.member = Shapes::ShapeRef.new(shape: WorkGroupName)
+
     WorkGroupSummary.add_member(:name, Shapes::ShapeRef.new(shape: WorkGroupName, location_name: "Name"))
     WorkGroupSummary.add_member(:state, Shapes::ShapeRef.new(shape: WorkGroupState, location_name: "State"))
     WorkGroupSummary.add_member(:description, Shapes::ShapeRef.new(shape: WorkGroupDescriptionString, location_name: "Description"))
@@ -1260,6 +1360,26 @@ module Aws::Athena
         o.errors << Shapes::ShapeRef.new(shape: InvalidRequestException)
       end)
 
+      api.add_operation(:cancel_capacity_reservation, Seahorse::Model::Operation.new.tap do |o|
+        o.name = "CancelCapacityReservation"
+        o.http_method = "POST"
+        o.http_request_uri = "/"
+        o.input = Shapes::ShapeRef.new(shape: CancelCapacityReservationInput)
+        o.output = Shapes::ShapeRef.new(shape: CancelCapacityReservationOutput)
+        o.errors << Shapes::ShapeRef.new(shape: InvalidRequestException)
+        o.errors << Shapes::ShapeRef.new(shape: InternalServerException)
+      end)
+
+      api.add_operation(:create_capacity_reservation, Seahorse::Model::Operation.new.tap do |o|
+        o.name = "CreateCapacityReservation"
+        o.http_method = "POST"
+        o.http_request_uri = "/"
+        o.input = Shapes::ShapeRef.new(shape: CreateCapacityReservationInput)
+        o.output = Shapes::ShapeRef.new(shape: CreateCapacityReservationOutput)
+        o.errors << Shapes::ShapeRef.new(shape: InternalServerException)
+        o.errors << Shapes::ShapeRef.new(shape: InvalidRequestException)
+      end)
+
       api.add_operation(:create_data_catalog, Seahorse::Model::Operation.new.tap do |o|
         o.name = "CreateDataCatalog"
         o.http_method = "POST"
@@ -1418,6 +1538,26 @@ module Aws::Athena
         o.errors << Shapes::ShapeRef.new(shape: ResourceNotFoundException)
       end)
 
+      api.add_operation(:get_capacity_assignment_configuration, Seahorse::Model::Operation.new.tap do |o|
+        o.name = "GetCapacityAssignmentConfiguration"
+        o.http_method = "POST"
+        o.http_request_uri = "/"
+        o.input = Shapes::ShapeRef.new(shape: GetCapacityAssignmentConfigurationInput)
+        o.output = Shapes::ShapeRef.new(shape: GetCapacityAssignmentConfigurationOutput)
+        o.errors << Shapes::ShapeRef.new(shape: InvalidRequestException)
+        o.errors << Shapes::ShapeRef.new(shape: InternalServerException)
+      end)
+
+      api.add_operation(:get_capacity_reservation, Seahorse::Model::Operation.new.tap do |o|
+        o.name = "GetCapacityReservation"
+        o.http_method = "POST"
+        o.http_request_uri = "/"
+        o.input = Shapes::ShapeRef.new(shape: GetCapacityReservationInput)
+        o.output = Shapes::ShapeRef.new(shape: GetCapacityReservationOutput)
+        o.errors << Shapes::ShapeRef.new(shape: InvalidRequestException)
+        o.errors << Shapes::ShapeRef.new(shape: InternalServerException)
+      end)
+
       api.add_operation(:get_data_catalog, Seahorse::Model::Operation.new.tap do |o|
         o.name = "GetDataCatalog"
         o.http_method = "POST"
@@ -1596,6 +1736,22 @@ module Aws::Athena
         )
       end)
 
+      api.add_operation(:list_capacity_reservations, Seahorse::Model::Operation.new.tap do |o|
+        o.name = "ListCapacityReservations"
+        o.http_method = "POST"
+        o.http_request_uri = "/"
+        o.input = Shapes::ShapeRef.new(shape: ListCapacityReservationsInput)
+        o.output = Shapes::ShapeRef.new(shape: ListCapacityReservationsOutput)
+        o.errors << Shapes::ShapeRef.new(shape: InternalServerException)
+        o.errors << Shapes::ShapeRef.new(shape: InvalidRequestException)
+        o[:pager] = Aws::Pager.new(
+          limit_key: "max_results",
+          tokens: {
+            "next_token" => "next_token"
+          }
+        )
+      end)
+
       api.add_operation(:list_data_catalogs, Seahorse::Model::Operation.new.tap do |o|
         o.name = "ListDataCatalogs"
         o.http_method = "POST"
@@ -1799,6 +1955,16 @@ module Aws::Athena
         )
       end)
 
+      api.add_operation(:put_capacity_assignment_configuration, Seahorse::Model::Operation.new.tap do |o|
+        o.name = "PutCapacityAssignmentConfiguration"
+        o.http_method = "POST"
+        o.http_request_uri = "/"
+        o.input = Shapes::ShapeRef.new(shape: PutCapacityAssignmentConfigurationInput)
+        o.output = Shapes::ShapeRef.new(shape: PutCapacityAssignmentConfigurationOutput)
+        o.errors << Shapes::ShapeRef.new(shape: InvalidRequestException)
+        o.errors << Shapes::ShapeRef.new(shape: InternalServerException)
+      end)
+
       api.add_operation(:start_calculation_execution, Seahorse::Model::Operation.new.tap do |o|
         o.name = "StartCalculationExecution"
         o.http_method = "POST"
@@ -1888,6 +2054,16 @@ module Aws::Athena
         o.errors << Shapes::ShapeRef.new(shape: ResourceNotFoundException)
       end)
 
+      api.add_operation(:update_capacity_reservation, Seahorse::Model::Operation.new.tap do |o|
+        o.name = "UpdateCapacityReservation"
+        o.http_method = "POST"
+        o.http_request_uri = "/"
+        o.input = Shapes::ShapeRef.new(shape: UpdateCapacityReservationInput)
+        o.output = Shapes::ShapeRef.new(shape: UpdateCapacityReservationOutput)
+        o.errors << Shapes::ShapeRef.new(shape: InvalidRequestException)
+        o.errors << Shapes::ShapeRef.new(shape: InternalServerException)
+      end)
+
       api.add_operation(:update_data_catalog, Seahorse::Model::Operation.new.tap do |o|
         o.name = "UpdateDataCatalog"
         o.http_method = "POST"

data/lib/aws-sdk-athena/endpoints.rb

@@ -53,6 +53,34 @@ module Aws::Athena
       end
     end
 
+    class CancelCapacityReservation
+      def self.build(context)
+        unless context.config.regional_endpoint
+          endpoint = context.config.endpoint.to_s
+        end
+        Aws::Athena::EndpointParameters.new(
+          region: context.config.region,
+          use_dual_stack: context.config.use_dualstack_endpoint,
+          use_fips: context.config.use_fips_endpoint,
+          endpoint: endpoint,
+        )
+      end
+    end
+
+    class CreateCapacityReservation
+      def self.build(context)
+        unless context.config.regional_endpoint
+          endpoint = context.config.endpoint.to_s
+        end
+        Aws::Athena::EndpointParameters.new(
+          region: context.config.region,
+          use_dual_stack: context.config.use_dualstack_endpoint,
+          use_fips: context.config.use_fips_endpoint,
+          endpoint: endpoint,
+        )
+      end
+    end
+
     class CreateDataCatalog
       def self.build(context)
         unless context.config.regional_endpoint
@@ -263,6 +291,34 @@ module Aws::Athena
       end
     end
 
+    class GetCapacityAssignmentConfiguration
+      def self.build(context)
+        unless context.config.regional_endpoint
+          endpoint = context.config.endpoint.to_s
+        end
+        Aws::Athena::EndpointParameters.new(
+          region: context.config.region,
+          use_dual_stack: context.config.use_dualstack_endpoint,
+          use_fips: context.config.use_fips_endpoint,
+          endpoint: endpoint,
+        )
+      end
+    end
+
+    class GetCapacityReservation
+      def self.build(context)
+        unless context.config.regional_endpoint
+          endpoint = context.config.endpoint.to_s
+        end
+        Aws::Athena::EndpointParameters.new(
+          region: context.config.region,
+          use_dual_stack: context.config.use_dualstack_endpoint,
+          use_fips: context.config.use_fips_endpoint,
+          endpoint: endpoint,
+        )
+      end
+    end
+
     class GetDataCatalog
       def self.build(context)
         unless context.config.regional_endpoint
@@ -473,6 +529,20 @@ module Aws::Athena
       end
     end
 
+    class ListCapacityReservations
+      def self.build(context)
+        unless context.config.regional_endpoint
+          endpoint = context.config.endpoint.to_s
+        end
+        Aws::Athena::EndpointParameters.new(
+          region: context.config.region,
+          use_dual_stack: context.config.use_dualstack_endpoint,
+          use_fips: context.config.use_fips_endpoint,
+          endpoint: endpoint,
+        )
+      end
+    end
+
     class ListDataCatalogs
       def self.build(context)
         unless context.config.regional_endpoint
@@ -655,6 +725,20 @@ module Aws::Athena
       end
     end
 
+    class PutCapacityAssignmentConfiguration
+      def self.build(context)
+        unless context.config.regional_endpoint
+          endpoint = context.config.endpoint.to_s
+        end
+        Aws::Athena::EndpointParameters.new(
+          region: context.config.region,
+          use_dual_stack: context.config.use_dualstack_endpoint,
+          use_fips: context.config.use_fips_endpoint,
+          endpoint: endpoint,
+        )
+      end
+    end
+
     class StartCalculationExecution
       def self.build(context)
         unless context.config.regional_endpoint
@@ -767,6 +851,20 @@ module Aws::Athena
       end
     end
 
+    class UpdateCapacityReservation
+      def self.build(context)
+        unless context.config.regional_endpoint
+          endpoint = context.config.endpoint.to_s
+        end
+        Aws::Athena::EndpointParameters.new(
+          region: context.config.region,
+          use_dual_stack: context.config.use_dualstack_endpoint,
+          use_fips: context.config.use_fips_endpoint,
+          endpoint: endpoint,
+        )
+      end
+    end
+
     class UpdateDataCatalog
       def self.build(context)
         unless context.config.regional_endpoint

data/lib/aws-sdk-athena/plugins/endpoints.rb

@@ -62,6 +62,10 @@ module Aws::Athena
             Aws::Athena::Endpoints::BatchGetPreparedStatement.build(context)
           when :batch_get_query_execution
             Aws::Athena::Endpoints::BatchGetQueryExecution.build(context)
+          when :cancel_capacity_reservation
+            Aws::Athena::Endpoints::CancelCapacityReservation.build(context)
+          when :create_capacity_reservation
+            Aws::Athena::Endpoints::CreateCapacityReservation.build(context)
           when :create_data_catalog
             Aws::Athena::Endpoints::CreateDataCatalog.build(context)
           when :create_named_query
@@ -92,6 +96,10 @@ module Aws::Athena
             Aws::Athena::Endpoints::GetCalculationExecutionCode.build(context)
           when :get_calculation_execution_status
             Aws::Athena::Endpoints::GetCalculationExecutionStatus.build(context)
+          when :get_capacity_assignment_configuration
+            Aws::Athena::Endpoints::GetCapacityAssignmentConfiguration.build(context)
+          when :get_capacity_reservation
+            Aws::Athena::Endpoints::GetCapacityReservation.build(context)
           when :get_data_catalog
             Aws::Athena::Endpoints::GetDataCatalog.build(context)
           when :get_database
@@ -122,6 +130,8 @@ module Aws::Athena
             Aws::Athena::Endpoints::ListApplicationDPUSizes.build(context)
           when :list_calculation_executions
             Aws::Athena::Endpoints::ListCalculationExecutions.build(context)
+          when :list_capacity_reservations
+            Aws::Athena::Endpoints::ListCapacityReservations.build(context)
           when :list_data_catalogs
             Aws::Athena::Endpoints::ListDataCatalogs.build(context)
           when :list_databases
@@ -148,6 +158,8 @@ module Aws::Athena
             Aws::Athena::Endpoints::ListTagsForResource.build(context)
           when :list_work_groups
             Aws::Athena::Endpoints::ListWorkGroups.build(context)
+          when :put_capacity_assignment_configuration
+            Aws::Athena::Endpoints::PutCapacityAssignmentConfiguration.build(context)
           when :start_calculation_execution
             Aws::Athena::Endpoints::StartCalculationExecution.build(context)
           when :start_query_execution
@@ -164,6 +176,8 @@ module Aws::Athena
             Aws::Athena::Endpoints::TerminateSession.build(context)
           when :untag_resource
             Aws::Athena::Endpoints::UntagResource.build(context)
+          when :update_capacity_reservation
+            Aws::Athena::Endpoints::UpdateCapacityReservation.build(context)
           when :update_data_catalog
             Aws::Athena::Endpoints::UpdateDataCatalog.build(context)
           when :update_named_query