aws-sdk-athena 1.58.0 → 1.59.0

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

@@ -81,6 +81,20 @@ module Aws::Athena
       end
     end
 
+    class CreateNotebook
+      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 CreatePreparedStatement
       def self.build(context)
         unless context.config.regional_endpoint
@@ -95,6 +109,20 @@ module Aws::Athena
       end
     end
 
+    class CreatePresignedNotebookUrl
+      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 CreateWorkGroup
       def self.build(context)
         unless context.config.regional_endpoint
@@ -137,6 +165,20 @@ module Aws::Athena
       end
     end
 
+    class DeleteNotebook
+      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 DeletePreparedStatement
       def self.build(context)
         unless context.config.regional_endpoint
@@ -165,6 +207,62 @@ module Aws::Athena
       end
     end
 
+    class ExportNotebook
+      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 GetCalculationExecution
+      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 GetCalculationExecutionCode
+      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 GetCalculationExecutionStatus
+      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
@@ -207,6 +305,20 @@ module Aws::Athena
       end
     end
 
+    class GetNotebookMetadata
+      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 GetPreparedStatement
       def self.build(context)
         unless context.config.regional_endpoint
@@ -263,6 +375,34 @@ module Aws::Athena
       end
     end
 
+    class GetSession
+      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 GetSessionStatus
+      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 GetTableMetadata
       def self.build(context)
         unless context.config.regional_endpoint
@@ -291,6 +431,48 @@ module Aws::Athena
       end
     end
 
+    class ImportNotebook
+      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 ListApplicationDPUSizes
+      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 ListCalculationExecutions
+      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
@@ -333,6 +515,20 @@ module Aws::Athena
       end
     end
 
+    class ListExecutors
+      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 ListNamedQueries
       def self.build(context)
         unless context.config.regional_endpoint
@@ -347,6 +543,34 @@ module Aws::Athena
       end
     end
 
+    class ListNotebookMetadata
+      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 ListNotebookSessions
+      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 ListPreparedStatements
       def self.build(context)
         unless context.config.regional_endpoint
@@ -375,6 +599,20 @@ module Aws::Athena
       end
     end
 
+    class ListSessions
+      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 ListTableMetadata
       def self.build(context)
         unless context.config.regional_endpoint
@@ -417,6 +655,20 @@ module Aws::Athena
       end
     end
 
+    class StartCalculationExecution
+      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 StartQueryExecution
       def self.build(context)
         unless context.config.regional_endpoint
@@ -431,6 +683,34 @@ module Aws::Athena
       end
     end
 
+    class StartSession
+      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 StopCalculationExecution
+      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 StopQueryExecution
       def self.build(context)
         unless context.config.regional_endpoint
@@ -459,6 +739,20 @@ module Aws::Athena
       end
     end
 
+    class TerminateSession
+      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 UntagResource
       def self.build(context)
         unless context.config.regional_endpoint
@@ -501,6 +795,34 @@ module Aws::Athena
       end
     end
 
+    class UpdateNotebook
+      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 UpdateNotebookMetadata
+      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 UpdatePreparedStatement
       def self.build(context)
         unless context.config.regional_endpoint

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

@@ -31,6 +31,7 @@ module Aws::Athena
   # * {InvalidRequestException}
   # * {MetadataException}
   # * {ResourceNotFoundException}
+  # * {SessionAlreadyExistsException}
   # * {TooManyRequestsException}
   #
   # Additionally, error classes are dynamically generated for service errors based on the error code
@@ -109,6 +110,21 @@ module Aws::Athena
       end
     end
 
+    class SessionAlreadyExistsException < ServiceError
+
+      # @param [Seahorse::Client::RequestContext] context
+      # @param [String] message
+      # @param [Aws::Athena::Types::SessionAlreadyExistsException] data
+      def initialize(context, message, data = Aws::EmptyStructure.new)
+        super(context, message, data)
+      end
+
+      # @return [String]
+      def message
+        @message || @data[:message]
+      end
+    end
+
     class TooManyRequestsException < ServiceError
 
       # @param [Seahorse::Client::RequestContext] context

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

@@ -66,24 +66,40 @@ module Aws::Athena
             Aws::Athena::Endpoints::CreateDataCatalog.build(context)
           when :create_named_query
             Aws::Athena::Endpoints::CreateNamedQuery.build(context)
+          when :create_notebook
+            Aws::Athena::Endpoints::CreateNotebook.build(context)
           when :create_prepared_statement
             Aws::Athena::Endpoints::CreatePreparedStatement.build(context)
+          when :create_presigned_notebook_url
+            Aws::Athena::Endpoints::CreatePresignedNotebookUrl.build(context)
           when :create_work_group
             Aws::Athena::Endpoints::CreateWorkGroup.build(context)
           when :delete_data_catalog
             Aws::Athena::Endpoints::DeleteDataCatalog.build(context)
           when :delete_named_query
             Aws::Athena::Endpoints::DeleteNamedQuery.build(context)
+          when :delete_notebook
+            Aws::Athena::Endpoints::DeleteNotebook.build(context)
           when :delete_prepared_statement
             Aws::Athena::Endpoints::DeletePreparedStatement.build(context)
           when :delete_work_group
             Aws::Athena::Endpoints::DeleteWorkGroup.build(context)
+          when :export_notebook
+            Aws::Athena::Endpoints::ExportNotebook.build(context)
+          when :get_calculation_execution
+            Aws::Athena::Endpoints::GetCalculationExecution.build(context)
+          when :get_calculation_execution_code
+            Aws::Athena::Endpoints::GetCalculationExecutionCode.build(context)
+          when :get_calculation_execution_status
+            Aws::Athena::Endpoints::GetCalculationExecutionStatus.build(context)
           when :get_data_catalog
             Aws::Athena::Endpoints::GetDataCatalog.build(context)
           when :get_database
             Aws::Athena::Endpoints::GetDatabase.build(context)
           when :get_named_query
             Aws::Athena::Endpoints::GetNamedQuery.build(context)
+          when :get_notebook_metadata
+            Aws::Athena::Endpoints::GetNotebookMetadata.build(context)
           when :get_prepared_statement
             Aws::Athena::Endpoints::GetPreparedStatement.build(context)
           when :get_query_execution
@@ -92,40 +108,70 @@ module Aws::Athena
             Aws::Athena::Endpoints::GetQueryResults.build(context)
           when :get_query_runtime_statistics
             Aws::Athena::Endpoints::GetQueryRuntimeStatistics.build(context)
+          when :get_session
+            Aws::Athena::Endpoints::GetSession.build(context)
+          when :get_session_status
+            Aws::Athena::Endpoints::GetSessionStatus.build(context)
           when :get_table_metadata
             Aws::Athena::Endpoints::GetTableMetadata.build(context)
           when :get_work_group
             Aws::Athena::Endpoints::GetWorkGroup.build(context)
+          when :import_notebook
+            Aws::Athena::Endpoints::ImportNotebook.build(context)
+          when :list_application_dpu_sizes
+            Aws::Athena::Endpoints::ListApplicationDPUSizes.build(context)
+          when :list_calculation_executions
+            Aws::Athena::Endpoints::ListCalculationExecutions.build(context)
           when :list_data_catalogs
             Aws::Athena::Endpoints::ListDataCatalogs.build(context)
           when :list_databases
             Aws::Athena::Endpoints::ListDatabases.build(context)
           when :list_engine_versions
             Aws::Athena::Endpoints::ListEngineVersions.build(context)
+          when :list_executors
+            Aws::Athena::Endpoints::ListExecutors.build(context)
           when :list_named_queries
             Aws::Athena::Endpoints::ListNamedQueries.build(context)
+          when :list_notebook_metadata
+            Aws::Athena::Endpoints::ListNotebookMetadata.build(context)
+          when :list_notebook_sessions
+            Aws::Athena::Endpoints::ListNotebookSessions.build(context)
           when :list_prepared_statements
             Aws::Athena::Endpoints::ListPreparedStatements.build(context)
           when :list_query_executions
             Aws::Athena::Endpoints::ListQueryExecutions.build(context)
+          when :list_sessions
+            Aws::Athena::Endpoints::ListSessions.build(context)
           when :list_table_metadata
             Aws::Athena::Endpoints::ListTableMetadata.build(context)
           when :list_tags_for_resource
             Aws::Athena::Endpoints::ListTagsForResource.build(context)
           when :list_work_groups
             Aws::Athena::Endpoints::ListWorkGroups.build(context)
+          when :start_calculation_execution
+            Aws::Athena::Endpoints::StartCalculationExecution.build(context)
           when :start_query_execution
             Aws::Athena::Endpoints::StartQueryExecution.build(context)
+          when :start_session
+            Aws::Athena::Endpoints::StartSession.build(context)
+          when :stop_calculation_execution
+            Aws::Athena::Endpoints::StopCalculationExecution.build(context)
           when :stop_query_execution
             Aws::Athena::Endpoints::StopQueryExecution.build(context)
           when :tag_resource
             Aws::Athena::Endpoints::TagResource.build(context)
+          when :terminate_session
+            Aws::Athena::Endpoints::TerminateSession.build(context)
           when :untag_resource
             Aws::Athena::Endpoints::UntagResource.build(context)
           when :update_data_catalog
             Aws::Athena::Endpoints::UpdateDataCatalog.build(context)
           when :update_named_query
             Aws::Athena::Endpoints::UpdateNamedQuery.build(context)
+          when :update_notebook
+            Aws::Athena::Endpoints::UpdateNotebook.build(context)
+          when :update_notebook_metadata
+            Aws::Athena::Endpoints::UpdateNotebookMetadata.build(context)
           when :update_prepared_statement
             Aws::Athena::Endpoints::UpdatePreparedStatement.build(context)
           when :update_work_group