aws-sdk-athena 1.64.0 → 1.65.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ef4f565c264acbdf2e8231ff320670a5a809e877e71265e84529aee3060a3e24
-  data.tar.gz: 8fc2bc53535c4a9ea120231a3ec2deb2eeb82faa6842d98da3e6552c4d1e71e2
+  metadata.gz: 93a5aab54a9e7b9bb75bf9274a0a1db4d32a7fbb6d8803f519065c43f4627b26
+  data.tar.gz: 1b8ff4beb625c0636d13045d3f26f70b296d8500e2313485ec663e3e39d5c4fb
 SHA512:
-  metadata.gz: 7e47325bfb2ddf6cc4a4f05adb381b19c9e25d1601fce6adc0b4748df3f9d856f3725f9aa3a6e6899ca070b15fb460a69f1a19be3702fd09666ee1c3ea6ce06f
-  data.tar.gz: 7b262a69c7faac4b11fa0847ac78928a4b6b8e127a065110834ab9d7bce6850e452cd293c2a9d80402850e940491afafde26bf46ae1b6bb5b7b7186463ee59f8
+  metadata.gz: 160bfa2697be724741708f9cd19a165c989fd2c6f815f33c7a2cb51f6a052303f38cf68308b597e1726791a46d5cf6df1e9a98f07ee9568eb177782228777c17
+  data.tar.gz: 941aced19cae2d832a1687eeea8d368191c214cf06f07b13a3c8dc20938ce8f39b47e4fe536ab74650f6a1a30657f9e1be81b2b707425156191b46cdbca1658c

data/CHANGELOG.md

@@ -1,6 +1,11 @@
 Unreleased Changes
 ------------------
 
+1.65.0 (2023-04-28)
+------------------
+
+* Feature - You can now use capacity reservations on Amazon Athena to run SQL queries on fully-managed compute capacity.
+
 1.64.0 (2023-03-30)
 ------------------
 

data/VERSION

@@ -1 +1 @@
-1.64.0
+1.65.0

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

@@ -546,6 +546,64 @@ module Aws::Athena
       req.send_request(options)
     end
 
+    # Cancels the capacity reservation with the specified name.
+    #
+    # @option params [required, String] :name
+    #   The name of the capacity reservation to cancel.
+    #
+    # @return [Struct] Returns an empty {Seahorse::Client::Response response}.
+    #
+    # @example Request syntax with placeholder values
+    #
+    #   resp = client.cancel_capacity_reservation({
+    #     name: "CapacityReservationName", # required
+    #   })
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/CancelCapacityReservation AWS API Documentation
+    #
+    # @overload cancel_capacity_reservation(params = {})
+    # @param [Hash] params ({})
+    def cancel_capacity_reservation(params = {}, options = {})
+      req = build_request(:cancel_capacity_reservation, params)
+      req.send_request(options)
+    end
+
+    # Creates a capacity reservation with the specified name and number of
+    # requested data processing units.
+    #
+    # @option params [required, Integer] :target_dpus
+    #   The number of requested data processing units.
+    #
+    # @option params [required, String] :name
+    #   The name of the capacity reservation to create.
+    #
+    # @option params [Array<Types::Tag>] :tags
+    #   The tags for the capacity reservation.
+    #
+    # @return [Struct] Returns an empty {Seahorse::Client::Response response}.
+    #
+    # @example Request syntax with placeholder values
+    #
+    #   resp = client.create_capacity_reservation({
+    #     target_dpus: 1, # required
+    #     name: "CapacityReservationName", # required
+    #     tags: [
+    #       {
+    #         key: "TagKey",
+    #         value: "TagValue",
+    #       },
+    #     ],
+    #   })
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/CreateCapacityReservation AWS API Documentation
+    #
+    # @overload create_capacity_reservation(params = {})
+    # @param [Hash] params ({})
+    def create_capacity_reservation(params = {}, options = {})
+      req = build_request(:create_capacity_reservation, params)
+      req.send_request(options)
+    end
+
     # Creates (registers) a data catalog with the specified name and
     # properties. Catalogs created are visible to all users of the same
     # Amazon Web Services account.
@@ -1183,6 +1241,77 @@ module Aws::Athena
       req.send_request(options)
     end
 
+    # Gets the capacity assignment configuration for a capacity reservation,
+    # if one exists.
+    #
+    # @option params [required, String] :capacity_reservation_name
+    #   The name of the capacity reservation to retrieve the capacity
+    #   assignment configuration for.
+    #
+    # @return [Types::GetCapacityAssignmentConfigurationOutput] Returns a {Seahorse::Client::Response response} object which responds to the following methods:
+    #
+    #   * {Types::GetCapacityAssignmentConfigurationOutput#capacity_assignment_configuration #capacity_assignment_configuration} => Types::CapacityAssignmentConfiguration
+    #
+    # @example Request syntax with placeholder values
+    #
+    #   resp = client.get_capacity_assignment_configuration({
+    #     capacity_reservation_name: "CapacityReservationName", # required
+    #   })
+    #
+    # @example Response structure
+    #
+    #   resp.capacity_assignment_configuration.capacity_reservation_name #=> String
+    #   resp.capacity_assignment_configuration.capacity_assignments #=> Array
+    #   resp.capacity_assignment_configuration.capacity_assignments[0].work_group_names #=> Array
+    #   resp.capacity_assignment_configuration.capacity_assignments[0].work_group_names[0] #=> String
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/GetCapacityAssignmentConfiguration AWS API Documentation
+    #
+    # @overload get_capacity_assignment_configuration(params = {})
+    # @param [Hash] params ({})
+    def get_capacity_assignment_configuration(params = {}, options = {})
+      req = build_request(:get_capacity_assignment_configuration, params)
+      req.send_request(options)
+    end
+
+    # Returns information about the capacity reservation with the specified
+    # name.
+    #
+    # @option params [required, String] :name
+    #   The name of the capacity reservation.
+    #
+    # @return [Types::GetCapacityReservationOutput] Returns a {Seahorse::Client::Response response} object which responds to the following methods:
+    #
+    #   * {Types::GetCapacityReservationOutput#capacity_reservation #capacity_reservation} => Types::CapacityReservation
+    #
+    # @example Request syntax with placeholder values
+    #
+    #   resp = client.get_capacity_reservation({
+    #     name: "CapacityReservationName", # required
+    #   })
+    #
+    # @example Response structure
+    #
+    #   resp.capacity_reservation.name #=> String
+    #   resp.capacity_reservation.status #=> String, one of "PENDING", "ACTIVE", "CANCELLING", "CANCELLED", "FAILED", "UPDATE_PENDING"
+    #   resp.capacity_reservation.target_dpus #=> Integer
+    #   resp.capacity_reservation.allocated_dpus #=> Integer
+    #   resp.capacity_reservation.last_allocation.status #=> String, one of "PENDING", "SUCCEEDED", "FAILED"
+    #   resp.capacity_reservation.last_allocation.status_message #=> String
+    #   resp.capacity_reservation.last_allocation.request_time #=> Time
+    #   resp.capacity_reservation.last_allocation.request_completion_time #=> Time
+    #   resp.capacity_reservation.last_successful_allocation_time #=> Time
+    #   resp.capacity_reservation.creation_time #=> Time
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/GetCapacityReservation AWS API Documentation
+    #
+    # @overload get_capacity_reservation(params = {})
+    # @param [Hash] params ({})
+    def get_capacity_reservation(params = {}, options = {})
+      req = build_request(:get_capacity_reservation, params)
+      req.send_request(options)
+    end
+
     # Returns the specified data catalog.
     #
     # @option params [required, String] :name
@@ -1910,6 +2039,53 @@ module Aws::Athena
       req.send_request(options)
     end
 
+    # Lists the capacity reservations for the current account.
+    #
+    # @option params [String] :next_token
+    #   A token generated by the Athena service that specifies where to
+    #   continue pagination if a previous request was truncated.
+    #
+    # @option params [Integer] :max_results
+    #   Specifies the maximum number of results to return.
+    #
+    # @return [Types::ListCapacityReservationsOutput] Returns a {Seahorse::Client::Response response} object which responds to the following methods:
+    #
+    #   * {Types::ListCapacityReservationsOutput#next_token #next_token} => String
+    #   * {Types::ListCapacityReservationsOutput#capacity_reservations #capacity_reservations} => Array&lt;Types::CapacityReservation&gt;
+    #
+    # The returned {Seahorse::Client::Response response} is a pageable response and is Enumerable. For details on usage see {Aws::PageableResponse PageableResponse}.
+    #
+    # @example Request syntax with placeholder values
+    #
+    #   resp = client.list_capacity_reservations({
+    #     next_token: "Token",
+    #     max_results: 1,
+    #   })
+    #
+    # @example Response structure
+    #
+    #   resp.next_token #=> String
+    #   resp.capacity_reservations #=> Array
+    #   resp.capacity_reservations[0].name #=> String
+    #   resp.capacity_reservations[0].status #=> String, one of "PENDING", "ACTIVE", "CANCELLING", "CANCELLED", "FAILED", "UPDATE_PENDING"
+    #   resp.capacity_reservations[0].target_dpus #=> Integer
+    #   resp.capacity_reservations[0].allocated_dpus #=> Integer
+    #   resp.capacity_reservations[0].last_allocation.status #=> String, one of "PENDING", "SUCCEEDED", "FAILED"
+    #   resp.capacity_reservations[0].last_allocation.status_message #=> String
+    #   resp.capacity_reservations[0].last_allocation.request_time #=> Time
+    #   resp.capacity_reservations[0].last_allocation.request_completion_time #=> Time
+    #   resp.capacity_reservations[0].last_successful_allocation_time #=> Time
+    #   resp.capacity_reservations[0].creation_time #=> Time
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/ListCapacityReservations AWS API Documentation
+    #
+    # @overload list_capacity_reservations(params = {})
+    # @param [Hash] params ({})
+    def list_capacity_reservations(params = {}, options = {})
+      req = build_request(:list_capacity_reservations, params)
+      req.send_request(options)
+    end
+
     # Lists the data catalogs in the current Amazon Web Services account.
     #
     # <note markdown="1"> In the Athena console, data catalogs are listed as "data sources" on
@@ -2520,8 +2696,7 @@ module Aws::Athena
       req.send_request(options)
     end
 
-    # Lists the tags associated with an Athena workgroup or data catalog
-    # resource.
+    # Lists the tags associated with an Athena resource.
     #
     # @option params [required, String] :resource_arn
     #   Lists the tags for the resource with the specified ARN.
@@ -2611,6 +2786,40 @@ module Aws::Athena
       req.send_request(options)
     end
 
+    # Puts a new capacity assignment configuration for a specified capacity
+    # reservation. If a capacity assignment configuration already exists for
+    # the capacity reservation, replaces the existing capacity assignment
+    # configuration.
+    #
+    # @option params [required, String] :capacity_reservation_name
+    #   The name of the capacity reservation to put a capacity assignment
+    #   configuration for.
+    #
+    # @option params [required, Array<Types::CapacityAssignment>] :capacity_assignments
+    #   The list of assignments for the capacity assignment configuration.
+    #
+    # @return [Struct] Returns an empty {Seahorse::Client::Response response}.
+    #
+    # @example Request syntax with placeholder values
+    #
+    #   resp = client.put_capacity_assignment_configuration({
+    #     capacity_reservation_name: "CapacityReservationName", # required
+    #     capacity_assignments: [ # required
+    #       {
+    #         work_group_names: ["WorkGroupName"],
+    #       },
+    #     ],
+    #   })
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/PutCapacityAssignmentConfiguration AWS API Documentation
+    #
+    # @overload put_capacity_assignment_configuration(params = {})
+    # @param [Hash] params ({})
+    def put_capacity_assignment_configuration(params = {}, options = {})
+      req = build_request(:put_capacity_assignment_configuration, params)
+      req.send_request(options)
+    end
+
     # Submits calculations for execution within a session. You can supply
     # the code to run as an inline code block within the request.
     #
@@ -2917,30 +3126,29 @@ module Aws::Athena
     end
 
     # Adds one or more tags to an Athena resource. A tag is a label that you
-    # assign to a resource. In Athena, a resource can be a workgroup or data
-    # catalog. Each tag consists of a key and an optional value, both of
-    # which you define. For example, you can use tags to categorize Athena
-    # workgroups or data catalogs by purpose, owner, or environment. Use a
-    # consistent set of tag keys to make it easier to search and filter
-    # workgroups or data catalogs in your account. For best practices, see
-    # [Tagging Best Practices][1]. Tag keys can be from 1 to 128 UTF-8
-    # Unicode characters, and tag values can be from 0 to 256 UTF-8 Unicode
-    # characters. Tags can use letters and numbers representable in UTF-8,
-    # and the following characters: + - = . \_ : / @. Tag keys and values
-    # are case-sensitive. Tag keys must be unique per resource. If you
-    # specify more than one tag, separate them by commas.
+    # assign to a resource. Each tag consists of a key and an optional
+    # value, both of which you define. For example, you can use tags to
+    # categorize Athena workgroups, data catalogs, or capacity reservations
+    # by purpose, owner, or environment. Use a consistent set of tag keys to
+    # make it easier to search and filter the resources in your account. For
+    # best practices, see [Tagging Best Practices][1]. Tag keys can be from
+    # 1 to 128 UTF-8 Unicode characters, and tag values can be from 0 to 256
+    # UTF-8 Unicode characters. Tags can use letters and numbers
+    # representable in UTF-8, and the following characters: + - = . \_ : /
+    # @. Tag keys and values are case-sensitive. Tag keys must be unique per
+    # resource. If you specify more than one tag, separate them by commas.
     #
     #
     #
     # [1]: https://docs.aws.amazon.com/whitepapers/latest/tagging-best-practices/tagging-best-practices.html
     #
     # @option params [required, String] :resource_arn
-    #   Specifies the ARN of the Athena resource (workgroup or data catalog)
-    #   to which tags are to be added.
+    #   Specifies the ARN of the Athena resource to which tags are to be
+    #   added.
     #
     # @option params [required, Array<Types::Tag>] :tags
     #   A collection of one or more tags, separated by commas, to be added to
-    #   an Athena workgroup or data catalog resource.
+    #   an Athena resource.
     #
     # @return [Struct] Returns an empty {Seahorse::Client::Response response}.
     #
@@ -2997,7 +3205,7 @@ module Aws::Athena
       req.send_request(options)
     end
 
-    # Removes one or more tags from a data catalog or workgroup resource.
+    # Removes one or more tags from an Athena resource.
     #
     # @option params [required, String] :resource_arn
     #   Specifies the ARN of the resource from which tags are to be removed.
@@ -3024,6 +3232,33 @@ module Aws::Athena
       req.send_request(options)
     end
 
+    # Updates the number of requested data processing units for the capacity
+    # reservation with the specified name.
+    #
+    # @option params [required, Integer] :target_dpus
+    #   The new number of requested data processing units.
+    #
+    # @option params [required, String] :name
+    #   The name of the capacity reservation.
+    #
+    # @return [Struct] Returns an empty {Seahorse::Client::Response response}.
+    #
+    # @example Request syntax with placeholder values
+    #
+    #   resp = client.update_capacity_reservation({
+    #     target_dpus: 1, # required
+    #     name: "CapacityReservationName", # required
+    #   })
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/UpdateCapacityReservation AWS API Documentation
+    #
+    # @overload update_capacity_reservation(params = {})
+    # @param [Hash] params ({})
+    def update_capacity_reservation(params = {}, options = {})
+      req = build_request(:update_capacity_reservation, params)
+      req.send_request(options)
+    end
+
     # Updates the data catalog that has the specified name.
     #
     # @option params [required, String] :name
@@ -3322,7 +3557,7 @@ module Aws::Athena
         params: params,
         config: config)
       context[:gem_name] = 'aws-sdk-athena'
-      context[:gem_version] = '1.64.0'
+      context[:gem_version] = '1.65.0'
       Seahorse::Client::Request.new(handlers, context)
     end
 

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"))
@@ -564,6 +629,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 +765,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 +944,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 +1211,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 +1298,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 +1359,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 +1537,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 +1735,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 +1954,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 +2053,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

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

@@ -344,6 +344,142 @@ module Aws::Athena
       include Aws::Structure
     end
 
+    # @!attribute [rw] name
+    #   The name of the capacity reservation to cancel.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/CancelCapacityReservationInput AWS API Documentation
+    #
+    class CancelCapacityReservationInput < Struct.new(
+      :name)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/CancelCapacityReservationOutput AWS API Documentation
+    #
+    class CancelCapacityReservationOutput < Aws::EmptyStructure; end
+
+    # Contains the submission time of a single allocation request for a
+    # capacity reservation and the most recent status of the attempted
+    # allocation.
+    #
+    # @!attribute [rw] status
+    #   The status of the capacity allocation.
+    #   @return [String]
+    #
+    # @!attribute [rw] status_message
+    #   The status message of the capacity allocation.
+    #   @return [String]
+    #
+    # @!attribute [rw] request_time
+    #   The time when the capacity allocation was requested.
+    #   @return [Time]
+    #
+    # @!attribute [rw] request_completion_time
+    #   The time when the capacity allocation request was completed.
+    #   @return [Time]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/CapacityAllocation AWS API Documentation
+    #
+    class CapacityAllocation < Struct.new(
+      :status,
+      :status_message,
+      :request_time,
+      :request_completion_time)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # A mapping between one or more workgroups and a capacity reservation.
+    #
+    # @!attribute [rw] work_group_names
+    #   The list of workgroup names for the capacity assignment.
+    #   @return [Array<String>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/CapacityAssignment AWS API Documentation
+    #
+    class CapacityAssignment < Struct.new(
+      :work_group_names)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Assigns Athena workgroups (and hence their queries) to capacity
+    # reservations. A capacity reservation can have only one capacity
+    # assignment configuration, but the capacity assignment configuration
+    # can be made up of multiple individual assignments. Each assignment
+    # specifies how Athena queries can consume capacity from the capacity
+    # reservation that their workgroup is mapped to.
+    #
+    # @!attribute [rw] capacity_reservation_name
+    #   The name of the reservation that the capacity assignment
+    #   configuration is for.
+    #   @return [String]
+    #
+    # @!attribute [rw] capacity_assignments
+    #   The list of assignments that make up the capacity assignment
+    #   configuration.
+    #   @return [Array<Types::CapacityAssignment>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/CapacityAssignmentConfiguration AWS API Documentation
+    #
+    class CapacityAssignmentConfiguration < Struct.new(
+      :capacity_reservation_name,
+      :capacity_assignments)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # A reservation for a specified number of data processing units (DPUs).
+    # When a reservation is initially created, it has no DPUs. Athena
+    # allocates DPUs until the allocated amount equals the requested amount.
+    #
+    # @!attribute [rw] name
+    #   The name of the capacity reservation.
+    #   @return [String]
+    #
+    # @!attribute [rw] status
+    #   The status of the capacity reservation.
+    #   @return [String]
+    #
+    # @!attribute [rw] target_dpus
+    #   The number of data processing units requested.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] allocated_dpus
+    #   The number of data processing units currently allocated.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] last_allocation
+    #   Contains the submission time of a single allocation request for a
+    #   capacity reservation and the most recent status of the attempted
+    #   allocation.
+    #   @return [Types::CapacityAllocation]
+    #
+    # @!attribute [rw] last_successful_allocation_time
+    #   The time of the most recent capacity allocation that succeeded.
+    #   @return [Time]
+    #
+    # @!attribute [rw] creation_time
+    #   The time in UTC epoch millis when the capacity reservation was
+    #   created.
+    #   @return [Time]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/CapacityReservation AWS API Documentation
+    #
+    class CapacityReservation < Struct.new(
+      :name,
+      :status,
+      :target_dpus,
+      :allocated_dpus,
+      :last_allocation,
+      :last_successful_allocation_time,
+      :creation_time)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Contains metadata for a column in a table.
     #
     # @!attribute [rw] name
@@ -429,6 +565,32 @@ module Aws::Athena
       include Aws::Structure
     end
 
+    # @!attribute [rw] target_dpus
+    #   The number of requested data processing units.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] name
+    #   The name of the capacity reservation to create.
+    #   @return [String]
+    #
+    # @!attribute [rw] tags
+    #   The tags for the capacity reservation.
+    #   @return [Array<Types::Tag>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/CreateCapacityReservationInput AWS API Documentation
+    #
+    class CreateCapacityReservationInput < Struct.new(
+      :target_dpus,
+      :name,
+      :tags)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/CreateCapacityReservationOutput AWS API Documentation
+    #
+    class CreateCapacityReservationOutput < Aws::EmptyStructure; end
+
     # @!attribute [rw] name
     #   The name of the data catalog to create. The catalog name must be
     #   unique for the Amazon Web Services account and can use a maximum of
@@ -1011,7 +1173,7 @@ module Aws::Athena
     # @!attribute [rw] coordinator_dpu_size
     #   The number of DPUs to use for the coordinator. A coordinator is a
     #   special executor that orchestrates processing work and manages other
-    #   executors in a notebook session.
+    #   executors in a notebook session. The default is 1.
     #   @return [Integer]
     #
     # @!attribute [rw] max_concurrent_dpus
@@ -1021,7 +1183,7 @@ module Aws::Athena
     # @!attribute [rw] default_executor_dpu_size
     #   The default number of DPUs to use for executors. An executor is the
     #   smallest unit of compute that a notebook session can request from
-    #   Athena.
+    #   Athena. The default is 1.
     #   @return [Integer]
     #
     # @!attribute [rw] additional_configs
@@ -1282,6 +1444,56 @@ module Aws::Athena
       include Aws::Structure
     end
 
+    # @!attribute [rw] capacity_reservation_name
+    #   The name of the capacity reservation to retrieve the capacity
+    #   assignment configuration for.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/GetCapacityAssignmentConfigurationInput AWS API Documentation
+    #
+    class GetCapacityAssignmentConfigurationInput < Struct.new(
+      :capacity_reservation_name)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] capacity_assignment_configuration
+    #   The requested capacity assignment configuration for the specified
+    #   capacity reservation.
+    #   @return [Types::CapacityAssignmentConfiguration]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/GetCapacityAssignmentConfigurationOutput AWS API Documentation
+    #
+    class GetCapacityAssignmentConfigurationOutput < Struct.new(
+      :capacity_assignment_configuration)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] name
+    #   The name of the capacity reservation.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/GetCapacityReservationInput AWS API Documentation
+    #
+    class GetCapacityReservationInput < Struct.new(
+      :name)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] capacity_reservation
+    #   The requested capacity reservation structure.
+    #   @return [Types::CapacityReservation]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/GetCapacityReservationOutput AWS API Documentation
+    #
+    class GetCapacityReservationOutput < Struct.new(
+      :capacity_reservation)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] name
     #   The name of the data catalog to return.
     #   @return [String]
@@ -1860,6 +2072,44 @@ module Aws::Athena
       include Aws::Structure
     end
 
+    # @!attribute [rw] next_token
+    #   A token generated by the Athena service that specifies where to
+    #   continue pagination if a previous request was truncated.
+    #   @return [String]
+    #
+    # @!attribute [rw] max_results
+    #   Specifies the maximum number of results to return.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/ListCapacityReservationsInput AWS API Documentation
+    #
+    class ListCapacityReservationsInput < Struct.new(
+      :next_token,
+      :max_results)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] next_token
+    #   A token generated by the Athena service that specifies where to
+    #   continue pagination if a previous request was truncated. To obtain
+    #   the next set of pages, pass in the NextToken from the response
+    #   object of the previous page call.
+    #   @return [String]
+    #
+    # @!attribute [rw] capacity_reservations
+    #   The capacity reservations for the current account.
+    #   @return [Array<Types::CapacityReservation>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/ListCapacityReservationsOutput AWS API Documentation
+    #
+    class ListCapacityReservationsOutput < Struct.new(
+      :next_token,
+      :capacity_reservations)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] next_token
     #   A token generated by the Athena service that specifies where to
     #   continue pagination if a previous request was truncated. To obtain
@@ -2666,6 +2916,28 @@ module Aws::Athena
       include Aws::Structure
     end
 
+    # @!attribute [rw] capacity_reservation_name
+    #   The name of the capacity reservation to put a capacity assignment
+    #   configuration for.
+    #   @return [String]
+    #
+    # @!attribute [rw] capacity_assignments
+    #   The list of assignments for the capacity assignment configuration.
+    #   @return [Array<Types::CapacityAssignment>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/PutCapacityAssignmentConfigurationInput AWS API Documentation
+    #
+    class PutCapacityAssignmentConfigurationInput < Struct.new(
+      :capacity_reservation_name,
+      :capacity_assignments)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/PutCapacityAssignmentConfigurationOutput AWS API Documentation
+    #
+    class PutCapacityAssignmentConfigurationOutput < Aws::EmptyStructure; end
+
     # Information about a single instance of a query execution.
     #
     # @!attribute [rw] query_execution_id
@@ -3913,14 +4185,14 @@ module Aws::Athena
       include Aws::Structure
     end
 
-    # A label that you assign to a resource. In Athena, a resource can be a
-    # workgroup or data catalog. Each tag consists of a key and an optional
-    # value, both of which you define. For example, you can use tags to
-    # categorize Athena workgroups or data catalogs by purpose, owner, or
-    # environment. Use a consistent set of tag keys to make it easier to
-    # search and filter workgroups or data catalogs in your account. For
-    # best practices, see [Tagging Best Practices][1]. Tag keys can be from
-    # 1 to 128 UTF-8 Unicode characters, and tag values can be from 0 to 256
+    # A label that you assign to a resource. Athena resources include
+    # workgroups, data catalogs, and capacity reservations. Each tag
+    # consists of a key and an optional value, both of which you define. For
+    # example, you can use tags to categorize Athena resources by purpose,
+    # owner, or environment. Use a consistent set of tag keys to make it
+    # easier to search and filter the resources in your account. For best
+    # practices, see [Tagging Best Practices][1]. Tag keys can be from 1 to
+    # 128 UTF-8 Unicode characters, and tag values can be from 0 to 256
     # UTF-8 Unicode characters. Tags can use letters and numbers
     # representable in UTF-8, and the following characters: + - = . \_ : /
     # @. Tag keys and values are case-sensitive. Tag keys must be unique per
@@ -3954,13 +4226,13 @@ module Aws::Athena
     end
 
     # @!attribute [rw] resource_arn
-    #   Specifies the ARN of the Athena resource (workgroup or data catalog)
-    #   to which tags are to be added.
+    #   Specifies the ARN of the Athena resource to which tags are to be
+    #   added.
     #   @return [String]
     #
     # @!attribute [rw] tags
     #   A collection of one or more tags, separated by commas, to be added
-    #   to an Athena workgroup or data catalog resource.
+    #   to an Athena resource.
     #   @return [Array<Types::Tag>]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/TagResourceInput AWS API Documentation
@@ -4148,6 +4420,27 @@ module Aws::Athena
     #
     class UntagResourceOutput < Aws::EmptyStructure; end
 
+    # @!attribute [rw] target_dpus
+    #   The new number of requested data processing units.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] name
+    #   The name of the capacity reservation.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/UpdateCapacityReservationInput AWS API Documentation
+    #
+    class UpdateCapacityReservationInput < Struct.new(
+      :target_dpus,
+      :name)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @see http://docs.aws.amazon.com/goto/WebAPI/athena-2017-05-18/UpdateCapacityReservationOutput AWS API Documentation
+    #
+    class UpdateCapacityReservationOutput < Aws::EmptyStructure; end
+
     # @!attribute [rw] name
     #   The name of the data catalog to update. The catalog name must be
     #   unique for the Amazon Web Services account and can use a maximum of

data/lib/aws-sdk-athena.rb

@@ -52,6 +52,6 @@ require_relative 'aws-sdk-athena/customizations'
 # @!group service
 module Aws::Athena
 
-  GEM_VERSION = '1.64.0'
+  GEM_VERSION = '1.65.0'
 
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aws-sdk-athena
 version: !ruby/object:Gem::Version
-  version: 1.64.0
+  version: 1.65.0
 platform: ruby
 authors:
 - Amazon Web Services
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-03-30 00:00:00.000000000 Z
+date: 2023-04-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core