@aws-sdk/client-athena 3.321.1 → 3.323.0

package/dist-cjs/protocols/Aws_json1_1.js

@@ -1,8 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.se_StopCalculationExecutionCommand = exports.se_StartSessionCommand = exports.se_StartQueryExecutionCommand = exports.se_StartCalculationExecutionCommand = exports.se_ListWorkGroupsCommand = exports.se_ListTagsForResourceCommand = exports.se_ListTableMetadataCommand = exports.se_ListSessionsCommand = exports.se_ListQueryExecutionsCommand = exports.se_ListPreparedStatementsCommand = exports.se_ListNotebookSessionsCommand = exports.se_ListNotebookMetadataCommand = exports.se_ListNamedQueriesCommand = exports.se_ListExecutorsCommand = exports.se_ListEngineVersionsCommand = exports.se_ListDataCatalogsCommand = exports.se_ListDatabasesCommand = exports.se_ListCalculationExecutionsCommand = exports.se_ListApplicationDPUSizesCommand = exports.se_ImportNotebookCommand = exports.se_GetWorkGroupCommand = exports.se_GetTableMetadataCommand = exports.se_GetSessionStatusCommand = exports.se_GetSessionCommand = exports.se_GetQueryRuntimeStatisticsCommand = exports.se_GetQueryResultsCommand = exports.se_GetQueryExecutionCommand = exports.se_GetPreparedStatementCommand = exports.se_GetNotebookMetadataCommand = exports.se_GetNamedQueryCommand = exports.se_GetDataCatalogCommand = exports.se_GetDatabaseCommand = exports.se_GetCalculationExecutionStatusCommand = exports.se_GetCalculationExecutionCodeCommand = exports.se_GetCalculationExecutionCommand = exports.se_ExportNotebookCommand = exports.se_DeleteWorkGroupCommand = exports.se_DeletePreparedStatementCommand = exports.se_DeleteNotebookCommand = exports.se_DeleteNamedQueryCommand = exports.se_DeleteDataCatalogCommand = exports.se_CreateWorkGroupCommand = exports.se_CreatePresignedNotebookUrlCommand = exports.se_CreatePreparedStatementCommand = exports.se_CreateNotebookCommand = exports.se_CreateNamedQueryCommand = exports.se_CreateDataCatalogCommand = exports.se_BatchGetQueryExecutionCommand = exports.se_BatchGetPreparedStatementCommand = exports.se_BatchGetNamedQueryCommand = void 0;
-exports.de_ListNotebookSessionsCommand = exports.de_ListNotebookMetadataCommand = exports.de_ListNamedQueriesCommand = exports.de_ListExecutorsCommand = exports.de_ListEngineVersionsCommand = exports.de_ListDataCatalogsCommand = exports.de_ListDatabasesCommand = exports.de_ListCalculationExecutionsCommand = exports.de_ListApplicationDPUSizesCommand = exports.de_ImportNotebookCommand = exports.de_GetWorkGroupCommand = exports.de_GetTableMetadataCommand = exports.de_GetSessionStatusCommand = exports.de_GetSessionCommand = exports.de_GetQueryRuntimeStatisticsCommand = exports.de_GetQueryResultsCommand = exports.de_GetQueryExecutionCommand = exports.de_GetPreparedStatementCommand = exports.de_GetNotebookMetadataCommand = exports.de_GetNamedQueryCommand = exports.de_GetDataCatalogCommand = exports.de_GetDatabaseCommand = exports.de_GetCalculationExecutionStatusCommand = exports.de_GetCalculationExecutionCodeCommand = exports.de_GetCalculationExecutionCommand = exports.de_ExportNotebookCommand = exports.de_DeleteWorkGroupCommand = exports.de_DeletePreparedStatementCommand = exports.de_DeleteNotebookCommand = exports.de_DeleteNamedQueryCommand = exports.de_DeleteDataCatalogCommand = exports.de_CreateWorkGroupCommand = exports.de_CreatePresignedNotebookUrlCommand = exports.de_CreatePreparedStatementCommand = exports.de_CreateNotebookCommand = exports.de_CreateNamedQueryCommand = exports.de_CreateDataCatalogCommand = exports.de_BatchGetQueryExecutionCommand = exports.de_BatchGetPreparedStatementCommand = exports.de_BatchGetNamedQueryCommand = exports.se_UpdateWorkGroupCommand = exports.se_UpdatePreparedStatementCommand = exports.se_UpdateNotebookMetadataCommand = exports.se_UpdateNotebookCommand = exports.se_UpdateNamedQueryCommand = exports.se_UpdateDataCatalogCommand = exports.se_UntagResourceCommand = exports.se_TerminateSessionCommand = exports.se_TagResourceCommand = exports.se_StopQueryExecutionCommand = void 0;
-exports.de_UpdateWorkGroupCommand = exports.de_UpdatePreparedStatementCommand = exports.de_UpdateNotebookMetadataCommand = exports.de_UpdateNotebookCommand = exports.de_UpdateNamedQueryCommand = exports.de_UpdateDataCatalogCommand = exports.de_UntagResourceCommand = exports.de_TerminateSessionCommand = exports.de_TagResourceCommand = exports.de_StopQueryExecutionCommand = exports.de_StopCalculationExecutionCommand = exports.de_StartSessionCommand = exports.de_StartQueryExecutionCommand = exports.de_StartCalculationExecutionCommand = exports.de_ListWorkGroupsCommand = exports.de_ListTagsForResourceCommand = exports.de_ListTableMetadataCommand = exports.de_ListSessionsCommand = exports.de_ListQueryExecutionsCommand = exports.de_ListPreparedStatementsCommand = void 0;
+exports.se_ListTagsForResourceCommand = exports.se_ListTableMetadataCommand = exports.se_ListSessionsCommand = exports.se_ListQueryExecutionsCommand = exports.se_ListPreparedStatementsCommand = exports.se_ListNotebookSessionsCommand = exports.se_ListNotebookMetadataCommand = exports.se_ListNamedQueriesCommand = exports.se_ListExecutorsCommand = exports.se_ListEngineVersionsCommand = exports.se_ListDataCatalogsCommand = exports.se_ListDatabasesCommand = exports.se_ListCapacityReservationsCommand = exports.se_ListCalculationExecutionsCommand = exports.se_ListApplicationDPUSizesCommand = exports.se_ImportNotebookCommand = exports.se_GetWorkGroupCommand = exports.se_GetTableMetadataCommand = exports.se_GetSessionStatusCommand = exports.se_GetSessionCommand = exports.se_GetQueryRuntimeStatisticsCommand = exports.se_GetQueryResultsCommand = exports.se_GetQueryExecutionCommand = exports.se_GetPreparedStatementCommand = exports.se_GetNotebookMetadataCommand = exports.se_GetNamedQueryCommand = exports.se_GetDataCatalogCommand = exports.se_GetDatabaseCommand = exports.se_GetCapacityReservationCommand = exports.se_GetCapacityAssignmentConfigurationCommand = exports.se_GetCalculationExecutionStatusCommand = exports.se_GetCalculationExecutionCodeCommand = exports.se_GetCalculationExecutionCommand = exports.se_ExportNotebookCommand = exports.se_DeleteWorkGroupCommand = exports.se_DeletePreparedStatementCommand = exports.se_DeleteNotebookCommand = exports.se_DeleteNamedQueryCommand = exports.se_DeleteDataCatalogCommand = exports.se_CreateWorkGroupCommand = exports.se_CreatePresignedNotebookUrlCommand = exports.se_CreatePreparedStatementCommand = exports.se_CreateNotebookCommand = exports.se_CreateNamedQueryCommand = exports.se_CreateDataCatalogCommand = exports.se_CreateCapacityReservationCommand = exports.se_CancelCapacityReservationCommand = exports.se_BatchGetQueryExecutionCommand = exports.se_BatchGetPreparedStatementCommand = exports.se_BatchGetNamedQueryCommand = void 0;
+exports.de_GetTableMetadataCommand = exports.de_GetSessionStatusCommand = exports.de_GetSessionCommand = exports.de_GetQueryRuntimeStatisticsCommand = exports.de_GetQueryResultsCommand = exports.de_GetQueryExecutionCommand = exports.de_GetPreparedStatementCommand = exports.de_GetNotebookMetadataCommand = exports.de_GetNamedQueryCommand = exports.de_GetDataCatalogCommand = exports.de_GetDatabaseCommand = exports.de_GetCapacityReservationCommand = exports.de_GetCapacityAssignmentConfigurationCommand = exports.de_GetCalculationExecutionStatusCommand = exports.de_GetCalculationExecutionCodeCommand = exports.de_GetCalculationExecutionCommand = exports.de_ExportNotebookCommand = exports.de_DeleteWorkGroupCommand = exports.de_DeletePreparedStatementCommand = exports.de_DeleteNotebookCommand = exports.de_DeleteNamedQueryCommand = exports.de_DeleteDataCatalogCommand = exports.de_CreateWorkGroupCommand = exports.de_CreatePresignedNotebookUrlCommand = exports.de_CreatePreparedStatementCommand = exports.de_CreateNotebookCommand = exports.de_CreateNamedQueryCommand = exports.de_CreateDataCatalogCommand = exports.de_CreateCapacityReservationCommand = exports.de_CancelCapacityReservationCommand = exports.de_BatchGetQueryExecutionCommand = exports.de_BatchGetPreparedStatementCommand = exports.de_BatchGetNamedQueryCommand = exports.se_UpdateWorkGroupCommand = exports.se_UpdatePreparedStatementCommand = exports.se_UpdateNotebookMetadataCommand = exports.se_UpdateNotebookCommand = exports.se_UpdateNamedQueryCommand = exports.se_UpdateDataCatalogCommand = exports.se_UpdateCapacityReservationCommand = exports.se_UntagResourceCommand = exports.se_TerminateSessionCommand = exports.se_TagResourceCommand = exports.se_StopQueryExecutionCommand = exports.se_StopCalculationExecutionCommand = exports.se_StartSessionCommand = exports.se_StartQueryExecutionCommand = exports.se_StartCalculationExecutionCommand = exports.se_PutCapacityAssignmentConfigurationCommand = exports.se_ListWorkGroupsCommand = void 0;
+exports.de_UpdateWorkGroupCommand = exports.de_UpdatePreparedStatementCommand = exports.de_UpdateNotebookMetadataCommand = exports.de_UpdateNotebookCommand = exports.de_UpdateNamedQueryCommand = exports.de_UpdateDataCatalogCommand = exports.de_UpdateCapacityReservationCommand = exports.de_UntagResourceCommand = exports.de_TerminateSessionCommand = exports.de_TagResourceCommand = exports.de_StopQueryExecutionCommand = exports.de_StopCalculationExecutionCommand = exports.de_StartSessionCommand = exports.de_StartQueryExecutionCommand = exports.de_StartCalculationExecutionCommand = exports.de_PutCapacityAssignmentConfigurationCommand = exports.de_ListWorkGroupsCommand = exports.de_ListTagsForResourceCommand = exports.de_ListTableMetadataCommand = exports.de_ListSessionsCommand = exports.de_ListQueryExecutionsCommand = exports.de_ListPreparedStatementsCommand = exports.de_ListNotebookSessionsCommand = exports.de_ListNotebookMetadataCommand = exports.de_ListNamedQueriesCommand = exports.de_ListExecutorsCommand = exports.de_ListEngineVersionsCommand = exports.de_ListDataCatalogsCommand = exports.de_ListDatabasesCommand = exports.de_ListCapacityReservationsCommand = exports.de_ListCalculationExecutionsCommand = exports.de_ListApplicationDPUSizesCommand = exports.de_ImportNotebookCommand = exports.de_GetWorkGroupCommand = void 0;
 const protocol_http_1 = require("@aws-sdk/protocol-http");
 const smithy_client_1 = require("@aws-sdk/smithy-client");
 const uuid_1 = require("uuid");
@@ -29,6 +29,20 @@ const se_BatchGetQueryExecutionCommand = async (input, context) => {
     return buildHttpRpcRequest(context, headers, "/", undefined, body);
 };
 exports.se_BatchGetQueryExecutionCommand = se_BatchGetQueryExecutionCommand;
+const se_CancelCapacityReservationCommand = async (input, context) => {
+    const headers = sharedHeaders("CancelCapacityReservation");
+    let body;
+    body = JSON.stringify((0, smithy_client_1._json)(input));
+    return buildHttpRpcRequest(context, headers, "/", undefined, body);
+};
+exports.se_CancelCapacityReservationCommand = se_CancelCapacityReservationCommand;
+const se_CreateCapacityReservationCommand = async (input, context) => {
+    const headers = sharedHeaders("CreateCapacityReservation");
+    let body;
+    body = JSON.stringify((0, smithy_client_1._json)(input));
+    return buildHttpRpcRequest(context, headers, "/", undefined, body);
+};
+exports.se_CreateCapacityReservationCommand = se_CreateCapacityReservationCommand;
 const se_CreateDataCatalogCommand = async (input, context) => {
     const headers = sharedHeaders("CreateDataCatalog");
     let body;
@@ -134,6 +148,20 @@ const se_GetCalculationExecutionStatusCommand = async (input, context) => {
     return buildHttpRpcRequest(context, headers, "/", undefined, body);
 };
 exports.se_GetCalculationExecutionStatusCommand = se_GetCalculationExecutionStatusCommand;
+const se_GetCapacityAssignmentConfigurationCommand = async (input, context) => {
+    const headers = sharedHeaders("GetCapacityAssignmentConfiguration");
+    let body;
+    body = JSON.stringify((0, smithy_client_1._json)(input));
+    return buildHttpRpcRequest(context, headers, "/", undefined, body);
+};
+exports.se_GetCapacityAssignmentConfigurationCommand = se_GetCapacityAssignmentConfigurationCommand;
+const se_GetCapacityReservationCommand = async (input, context) => {
+    const headers = sharedHeaders("GetCapacityReservation");
+    let body;
+    body = JSON.stringify((0, smithy_client_1._json)(input));
+    return buildHttpRpcRequest(context, headers, "/", undefined, body);
+};
+exports.se_GetCapacityReservationCommand = se_GetCapacityReservationCommand;
 const se_GetDatabaseCommand = async (input, context) => {
     const headers = sharedHeaders("GetDatabase");
     let body;
@@ -239,6 +267,13 @@ const se_ListCalculationExecutionsCommand = async (input, context) => {
     return buildHttpRpcRequest(context, headers, "/", undefined, body);
 };
 exports.se_ListCalculationExecutionsCommand = se_ListCalculationExecutionsCommand;
+const se_ListCapacityReservationsCommand = async (input, context) => {
+    const headers = sharedHeaders("ListCapacityReservations");
+    let body;
+    body = JSON.stringify((0, smithy_client_1._json)(input));
+    return buildHttpRpcRequest(context, headers, "/", undefined, body);
+};
+exports.se_ListCapacityReservationsCommand = se_ListCapacityReservationsCommand;
 const se_ListDatabasesCommand = async (input, context) => {
     const headers = sharedHeaders("ListDatabases");
     let body;
@@ -330,6 +365,13 @@ const se_ListWorkGroupsCommand = async (input, context) => {
     return buildHttpRpcRequest(context, headers, "/", undefined, body);
 };
 exports.se_ListWorkGroupsCommand = se_ListWorkGroupsCommand;
+const se_PutCapacityAssignmentConfigurationCommand = async (input, context) => {
+    const headers = sharedHeaders("PutCapacityAssignmentConfiguration");
+    let body;
+    body = JSON.stringify((0, smithy_client_1._json)(input));
+    return buildHttpRpcRequest(context, headers, "/", undefined, body);
+};
+exports.se_PutCapacityAssignmentConfigurationCommand = se_PutCapacityAssignmentConfigurationCommand;
 const se_StartCalculationExecutionCommand = async (input, context) => {
     const headers = sharedHeaders("StartCalculationExecution");
     let body;
@@ -386,6 +428,13 @@ const se_UntagResourceCommand = async (input, context) => {
     return buildHttpRpcRequest(context, headers, "/", undefined, body);
 };
 exports.se_UntagResourceCommand = se_UntagResourceCommand;
+const se_UpdateCapacityReservationCommand = async (input, context) => {
+    const headers = sharedHeaders("UpdateCapacityReservation");
+    let body;
+    body = JSON.stringify((0, smithy_client_1._json)(input));
+    return buildHttpRpcRequest(context, headers, "/", undefined, body);
+};
+exports.se_UpdateCapacityReservationCommand = se_UpdateCapacityReservationCommand;
 const se_UpdateDataCatalogCommand = async (input, context) => {
     const headers = sharedHeaders("UpdateDataCatalog");
     let body;
@@ -536,6 +585,78 @@ const de_BatchGetQueryExecutionCommandError = async (output, context) => {
             });
     }
 };
+const de_CancelCapacityReservationCommand = async (output, context) => {
+    if (output.statusCode >= 300) {
+        return de_CancelCapacityReservationCommandError(output, context);
+    }
+    const data = await parseBody(output.body, context);
+    let contents = {};
+    contents = (0, smithy_client_1._json)(data);
+    const response = {
+        $metadata: deserializeMetadata(output),
+        ...contents,
+    };
+    return response;
+};
+exports.de_CancelCapacityReservationCommand = de_CancelCapacityReservationCommand;
+const de_CancelCapacityReservationCommandError = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseErrorBody(output.body, context),
+    };
+    const errorCode = loadRestJsonErrorCode(output, parsedOutput.body);
+    switch (errorCode) {
+        case "InternalServerException":
+        case "com.amazonaws.athena#InternalServerException":
+            throw await de_InternalServerExceptionRes(parsedOutput, context);
+        case "InvalidRequestException":
+        case "com.amazonaws.athena#InvalidRequestException":
+            throw await de_InvalidRequestExceptionRes(parsedOutput, context);
+        default:
+            const parsedBody = parsedOutput.body;
+            return throwDefaultError({
+                output,
+                parsedBody,
+                errorCode,
+            });
+    }
+};
+const de_CreateCapacityReservationCommand = async (output, context) => {
+    if (output.statusCode >= 300) {
+        return de_CreateCapacityReservationCommandError(output, context);
+    }
+    const data = await parseBody(output.body, context);
+    let contents = {};
+    contents = (0, smithy_client_1._json)(data);
+    const response = {
+        $metadata: deserializeMetadata(output),
+        ...contents,
+    };
+    return response;
+};
+exports.de_CreateCapacityReservationCommand = de_CreateCapacityReservationCommand;
+const de_CreateCapacityReservationCommandError = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseErrorBody(output.body, context),
+    };
+    const errorCode = loadRestJsonErrorCode(output, parsedOutput.body);
+    switch (errorCode) {
+        case "InternalServerException":
+        case "com.amazonaws.athena#InternalServerException":
+            throw await de_InternalServerExceptionRes(parsedOutput, context);
+        case "InvalidRequestException":
+        case "com.amazonaws.athena#InvalidRequestException":
+            throw await de_InvalidRequestExceptionRes(parsedOutput, context);
+        default:
+            const parsedBody = parsedOutput.body;
+            return throwDefaultError({
+                output,
+                parsedBody,
+                errorCode,
+            });
+    }
+};
 const de_CreateDataCatalogCommand = async (output, context) => {
     if (output.statusCode >= 300) {
         return de_CreateDataCatalogCommandError(output, context);
@@ -1100,6 +1221,78 @@ const de_GetCalculationExecutionStatusCommandError = async (output, context) =>
             });
     }
 };
+const de_GetCapacityAssignmentConfigurationCommand = async (output, context) => {
+    if (output.statusCode >= 300) {
+        return de_GetCapacityAssignmentConfigurationCommandError(output, context);
+    }
+    const data = await parseBody(output.body, context);
+    let contents = {};
+    contents = (0, smithy_client_1._json)(data);
+    const response = {
+        $metadata: deserializeMetadata(output),
+        ...contents,
+    };
+    return response;
+};
+exports.de_GetCapacityAssignmentConfigurationCommand = de_GetCapacityAssignmentConfigurationCommand;
+const de_GetCapacityAssignmentConfigurationCommandError = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseErrorBody(output.body, context),
+    };
+    const errorCode = loadRestJsonErrorCode(output, parsedOutput.body);
+    switch (errorCode) {
+        case "InternalServerException":
+        case "com.amazonaws.athena#InternalServerException":
+            throw await de_InternalServerExceptionRes(parsedOutput, context);
+        case "InvalidRequestException":
+        case "com.amazonaws.athena#InvalidRequestException":
+            throw await de_InvalidRequestExceptionRes(parsedOutput, context);
+        default:
+            const parsedBody = parsedOutput.body;
+            return throwDefaultError({
+                output,
+                parsedBody,
+                errorCode,
+            });
+    }
+};
+const de_GetCapacityReservationCommand = async (output, context) => {
+    if (output.statusCode >= 300) {
+        return de_GetCapacityReservationCommandError(output, context);
+    }
+    const data = await parseBody(output.body, context);
+    let contents = {};
+    contents = de_GetCapacityReservationOutput(data, context);
+    const response = {
+        $metadata: deserializeMetadata(output),
+        ...contents,
+    };
+    return response;
+};
+exports.de_GetCapacityReservationCommand = de_GetCapacityReservationCommand;
+const de_GetCapacityReservationCommandError = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseErrorBody(output.body, context),
+    };
+    const errorCode = loadRestJsonErrorCode(output, parsedOutput.body);
+    switch (errorCode) {
+        case "InternalServerException":
+        case "com.amazonaws.athena#InternalServerException":
+            throw await de_InternalServerExceptionRes(parsedOutput, context);
+        case "InvalidRequestException":
+        case "com.amazonaws.athena#InvalidRequestException":
+            throw await de_InvalidRequestExceptionRes(parsedOutput, context);
+        default:
+            const parsedBody = parsedOutput.body;
+            return throwDefaultError({
+                output,
+                parsedBody,
+                errorCode,
+            });
+    }
+};
 const de_GetDatabaseCommand = async (output, context) => {
     if (output.statusCode >= 300) {
         return de_GetDatabaseCommandError(output, context);
@@ -1670,6 +1863,42 @@ const de_ListCalculationExecutionsCommandError = async (output, context) => {
             });
     }
 };
+const de_ListCapacityReservationsCommand = async (output, context) => {
+    if (output.statusCode >= 300) {
+        return de_ListCapacityReservationsCommandError(output, context);
+    }
+    const data = await parseBody(output.body, context);
+    let contents = {};
+    contents = de_ListCapacityReservationsOutput(data, context);
+    const response = {
+        $metadata: deserializeMetadata(output),
+        ...contents,
+    };
+    return response;
+};
+exports.de_ListCapacityReservationsCommand = de_ListCapacityReservationsCommand;
+const de_ListCapacityReservationsCommandError = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseErrorBody(output.body, context),
+    };
+    const errorCode = loadRestJsonErrorCode(output, parsedOutput.body);
+    switch (errorCode) {
+        case "InternalServerException":
+        case "com.amazonaws.athena#InternalServerException":
+            throw await de_InternalServerExceptionRes(parsedOutput, context);
+        case "InvalidRequestException":
+        case "com.amazonaws.athena#InvalidRequestException":
+            throw await de_InvalidRequestExceptionRes(parsedOutput, context);
+        default:
+            const parsedBody = parsedOutput.body;
+            return throwDefaultError({
+                output,
+                parsedBody,
+                errorCode,
+            });
+    }
+};
 const de_ListDatabasesCommand = async (output, context) => {
     if (output.statusCode >= 300) {
         return de_ListDatabasesCommandError(output, context);
@@ -2159,6 +2388,42 @@ const de_ListWorkGroupsCommandError = async (output, context) => {
             });
     }
 };
+const de_PutCapacityAssignmentConfigurationCommand = async (output, context) => {
+    if (output.statusCode >= 300) {
+        return de_PutCapacityAssignmentConfigurationCommandError(output, context);
+    }
+    const data = await parseBody(output.body, context);
+    let contents = {};
+    contents = (0, smithy_client_1._json)(data);
+    const response = {
+        $metadata: deserializeMetadata(output),
+        ...contents,
+    };
+    return response;
+};
+exports.de_PutCapacityAssignmentConfigurationCommand = de_PutCapacityAssignmentConfigurationCommand;
+const de_PutCapacityAssignmentConfigurationCommandError = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseErrorBody(output.body, context),
+    };
+    const errorCode = loadRestJsonErrorCode(output, parsedOutput.body);
+    switch (errorCode) {
+        case "InternalServerException":
+        case "com.amazonaws.athena#InternalServerException":
+            throw await de_InternalServerExceptionRes(parsedOutput, context);
+        case "InvalidRequestException":
+        case "com.amazonaws.athena#InvalidRequestException":
+            throw await de_InvalidRequestExceptionRes(parsedOutput, context);
+        default:
+            const parsedBody = parsedOutput.body;
+            return throwDefaultError({
+                output,
+                parsedBody,
+                errorCode,
+            });
+    }
+};
 const de_StartCalculationExecutionCommand = async (output, context) => {
     if (output.statusCode >= 300) {
         return de_StartCalculationExecutionCommandError(output, context);
@@ -2474,6 +2739,42 @@ const de_UntagResourceCommandError = async (output, context) => {
             });
     }
 };
+const de_UpdateCapacityReservationCommand = async (output, context) => {
+    if (output.statusCode >= 300) {
+        return de_UpdateCapacityReservationCommandError(output, context);
+    }
+    const data = await parseBody(output.body, context);
+    let contents = {};
+    contents = (0, smithy_client_1._json)(data);
+    const response = {
+        $metadata: deserializeMetadata(output),
+        ...contents,
+    };
+    return response;
+};
+exports.de_UpdateCapacityReservationCommand = de_UpdateCapacityReservationCommand;
+const de_UpdateCapacityReservationCommandError = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseErrorBody(output.body, context),
+    };
+    const errorCode = loadRestJsonErrorCode(output, parsedOutput.body);
+    switch (errorCode) {
+        case "InternalServerException":
+        case "com.amazonaws.athena#InternalServerException":
+            throw await de_InternalServerExceptionRes(parsedOutput, context);
+        case "InvalidRequestException":
+        case "com.amazonaws.athena#InvalidRequestException":
+            throw await de_InvalidRequestExceptionRes(parsedOutput, context);
+        default:
+            const parsedBody = parsedOutput.body;
+            return throwDefaultError({
+                output,
+                parsedBody,
+                errorCode,
+            });
+    }
+};
 const de_UpdateDataCatalogCommand = async (output, context) => {
     if (output.statusCode >= 300) {
         return de_UpdateDataCatalogCommandError(output, context);
@@ -2819,6 +3120,33 @@ const de_CalculationSummary = (output, context) => {
         Status: (_) => de_CalculationStatus(_, context),
     });
 };
+const de_CapacityAllocation = (output, context) => {
+    return (0, smithy_client_1.take)(output, {
+        RequestCompletionTime: (_) => (0, smithy_client_1.expectNonNull)((0, smithy_client_1.parseEpochTimestamp)((0, smithy_client_1.expectNumber)(_))),
+        RequestTime: (_) => (0, smithy_client_1.expectNonNull)((0, smithy_client_1.parseEpochTimestamp)((0, smithy_client_1.expectNumber)(_))),
+        Status: smithy_client_1.expectString,
+        StatusMessage: smithy_client_1.expectString,
+    });
+};
+const de_CapacityReservation = (output, context) => {
+    return (0, smithy_client_1.take)(output, {
+        AllocatedDpus: smithy_client_1.expectInt32,
+        CreationTime: (_) => (0, smithy_client_1.expectNonNull)((0, smithy_client_1.parseEpochTimestamp)((0, smithy_client_1.expectNumber)(_))),
+        LastAllocation: (_) => de_CapacityAllocation(_, context),
+        LastSuccessfulAllocationTime: (_) => (0, smithy_client_1.expectNonNull)((0, smithy_client_1.parseEpochTimestamp)((0, smithy_client_1.expectNumber)(_))),
+        Name: smithy_client_1.expectString,
+        Status: smithy_client_1.expectString,
+        TargetDpus: smithy_client_1.expectInt32,
+    });
+};
+const de_CapacityReservationsList = (output, context) => {
+    const retVal = (output || [])
+        .filter((e) => e != null)
+        .map((entry) => {
+        return de_CapacityReservation(entry, context);
+    });
+    return retVal;
+};
 const de_ExportNotebookOutput = (output, context) => {
     return (0, smithy_client_1.take)(output, {
         NotebookMetadata: (_) => de_NotebookMetadata(_, context),
@@ -2842,6 +3170,11 @@ const de_GetCalculationExecutionStatusResponse = (output, context) => {
         Status: (_) => de_CalculationStatus(_, context),
     });
 };
+const de_GetCapacityReservationOutput = (output, context) => {
+    return (0, smithy_client_1.take)(output, {
+        CapacityReservation: (_) => de_CapacityReservation(_, context),
+    });
+};
 const de_GetNotebookMetadataOutput = (output, context) => {
     return (0, smithy_client_1.take)(output, {
         NotebookMetadata: (_) => de_NotebookMetadata(_, context),
@@ -2897,6 +3230,12 @@ const de_ListCalculationExecutionsResponse = (output, context) => {
         NextToken: smithy_client_1.expectString,
     });
 };
+const de_ListCapacityReservationsOutput = (output, context) => {
+    return (0, smithy_client_1.take)(output, {
+        CapacityReservations: (_) => de_CapacityReservationsList(_, context),
+        NextToken: smithy_client_1.expectString,
+    });
+};
 const de_ListNotebookMetadataOutput = (output, context) => {
     return (0, smithy_client_1.take)(output, {
         NextToken: smithy_client_1.expectString,

package/dist-es/Athena.js

@@ -3,6 +3,8 @@ import { AthenaClient } from "./AthenaClient";
 import { BatchGetNamedQueryCommand, } from "./commands/BatchGetNamedQueryCommand";
 import { BatchGetPreparedStatementCommand, } from "./commands/BatchGetPreparedStatementCommand";
 import { BatchGetQueryExecutionCommand, } from "./commands/BatchGetQueryExecutionCommand";
+import { CancelCapacityReservationCommand, } from "./commands/CancelCapacityReservationCommand";
+import { CreateCapacityReservationCommand, } from "./commands/CreateCapacityReservationCommand";
 import { CreateDataCatalogCommand, } from "./commands/CreateDataCatalogCommand";
 import { CreateNamedQueryCommand, } from "./commands/CreateNamedQueryCommand";
 import { CreateNotebookCommand, } from "./commands/CreateNotebookCommand";
@@ -18,6 +20,8 @@ import { ExportNotebookCommand, } from "./commands/ExportNotebookCommand";
 import { GetCalculationExecutionCodeCommand, } from "./commands/GetCalculationExecutionCodeCommand";
 import { GetCalculationExecutionCommand, } from "./commands/GetCalculationExecutionCommand";
 import { GetCalculationExecutionStatusCommand, } from "./commands/GetCalculationExecutionStatusCommand";
+import { GetCapacityAssignmentConfigurationCommand, } from "./commands/GetCapacityAssignmentConfigurationCommand";
+import { GetCapacityReservationCommand, } from "./commands/GetCapacityReservationCommand";
 import { GetDatabaseCommand } from "./commands/GetDatabaseCommand";
 import { GetDataCatalogCommand, } from "./commands/GetDataCatalogCommand";
 import { GetNamedQueryCommand, } from "./commands/GetNamedQueryCommand";
@@ -33,6 +37,7 @@ import { GetWorkGroupCommand, } from "./commands/GetWorkGroupCommand";
 import { ImportNotebookCommand, } from "./commands/ImportNotebookCommand";
 import { ListApplicationDPUSizesCommand, } from "./commands/ListApplicationDPUSizesCommand";
 import { ListCalculationExecutionsCommand, } from "./commands/ListCalculationExecutionsCommand";
+import { ListCapacityReservationsCommand, } from "./commands/ListCapacityReservationsCommand";
 import { ListDatabasesCommand, } from "./commands/ListDatabasesCommand";
 import { ListDataCatalogsCommand, } from "./commands/ListDataCatalogsCommand";
 import { ListEngineVersionsCommand, } from "./commands/ListEngineVersionsCommand";
@@ -46,6 +51,7 @@ import { ListSessionsCommand, } from "./commands/ListSessionsCommand";
 import { ListTableMetadataCommand, } from "./commands/ListTableMetadataCommand";
 import { ListTagsForResourceCommand, } from "./commands/ListTagsForResourceCommand";
 import { ListWorkGroupsCommand, } from "./commands/ListWorkGroupsCommand";
+import { PutCapacityAssignmentConfigurationCommand, } from "./commands/PutCapacityAssignmentConfigurationCommand";
 import { StartCalculationExecutionCommand, } from "./commands/StartCalculationExecutionCommand";
 import { StartQueryExecutionCommand, } from "./commands/StartQueryExecutionCommand";
 import { StartSessionCommand, } from "./commands/StartSessionCommand";
@@ -54,6 +60,7 @@ import { StopQueryExecutionCommand, } from "./commands/StopQueryExecutionCommand
 import { TagResourceCommand } from "./commands/TagResourceCommand";
 import { TerminateSessionCommand, } from "./commands/TerminateSessionCommand";
 import { UntagResourceCommand, } from "./commands/UntagResourceCommand";
+import { UpdateCapacityReservationCommand, } from "./commands/UpdateCapacityReservationCommand";
 import { UpdateDataCatalogCommand, } from "./commands/UpdateDataCatalogCommand";
 import { UpdateNamedQueryCommand, } from "./commands/UpdateNamedQueryCommand";
 import { UpdateNotebookCommand, } from "./commands/UpdateNotebookCommand";
@@ -64,6 +71,8 @@ const commands = {
     BatchGetNamedQueryCommand,
     BatchGetPreparedStatementCommand,
     BatchGetQueryExecutionCommand,
+    CancelCapacityReservationCommand,
+    CreateCapacityReservationCommand,
     CreateDataCatalogCommand,
     CreateNamedQueryCommand,
     CreateNotebookCommand,
@@ -79,6 +88,8 @@ const commands = {
     GetCalculationExecutionCommand,
     GetCalculationExecutionCodeCommand,
     GetCalculationExecutionStatusCommand,
+    GetCapacityAssignmentConfigurationCommand,
+    GetCapacityReservationCommand,
     GetDatabaseCommand,
     GetDataCatalogCommand,
     GetNamedQueryCommand,
@@ -94,6 +105,7 @@ const commands = {
     ImportNotebookCommand,
     ListApplicationDPUSizesCommand,
     ListCalculationExecutionsCommand,
+    ListCapacityReservationsCommand,
     ListDatabasesCommand,
     ListDataCatalogsCommand,
     ListEngineVersionsCommand,
@@ -107,6 +119,7 @@ const commands = {
     ListTableMetadataCommand,
     ListTagsForResourceCommand,
     ListWorkGroupsCommand,
+    PutCapacityAssignmentConfigurationCommand,
     StartCalculationExecutionCommand,
     StartQueryExecutionCommand,
     StartSessionCommand,
@@ -115,6 +128,7 @@ const commands = {
     TagResourceCommand,
     TerminateSessionCommand,
     UntagResourceCommand,
+    UpdateCapacityReservationCommand,
     UpdateDataCatalogCommand,
     UpdateNamedQueryCommand,
     UpdateNotebookCommand,

package/dist-es/commands/CancelCapacityReservationCommand.js

@@ -0,0 +1,41 @@
+import { getEndpointPlugin } from "@aws-sdk/middleware-endpoint";
+import { getSerdePlugin } from "@aws-sdk/middleware-serde";
+import { Command as $Command } from "@aws-sdk/smithy-client";
+import { de_CancelCapacityReservationCommand, se_CancelCapacityReservationCommand } from "../protocols/Aws_json1_1";
+export class CancelCapacityReservationCommand extends $Command {
+    static getEndpointParameterInstructions() {
+        return {
+            UseFIPS: { type: "builtInParams", name: "useFipsEndpoint" },
+            Endpoint: { type: "builtInParams", name: "endpoint" },
+            Region: { type: "builtInParams", name: "region" },
+            UseDualStack: { type: "builtInParams", name: "useDualstackEndpoint" },
+        };
+    }
+    constructor(input) {
+        super();
+        this.input = input;
+    }
+    resolveMiddleware(clientStack, configuration, options) {
+        this.middlewareStack.use(getSerdePlugin(configuration, this.serialize, this.deserialize));
+        this.middlewareStack.use(getEndpointPlugin(configuration, CancelCapacityReservationCommand.getEndpointParameterInstructions()));
+        const stack = clientStack.concat(this.middlewareStack);
+        const { logger } = configuration;
+        const clientName = "AthenaClient";
+        const commandName = "CancelCapacityReservationCommand";
+        const handlerExecutionContext = {
+            logger,
+            clientName,
+            commandName,
+            inputFilterSensitiveLog: (_) => _,
+            outputFilterSensitiveLog: (_) => _,
+        };
+        const { requestHandler } = configuration;
+        return stack.resolve((request) => requestHandler.handle(request.request, options || {}), handlerExecutionContext);
+    }
+    serialize(input, context) {
+        return se_CancelCapacityReservationCommand(input, context);
+    }
+    deserialize(output, context) {
+        return de_CancelCapacityReservationCommand(output, context);
+    }
+}

package/dist-es/commands/CreateCapacityReservationCommand.js

@@ -0,0 +1,41 @@
+import { getEndpointPlugin } from "@aws-sdk/middleware-endpoint";
+import { getSerdePlugin } from "@aws-sdk/middleware-serde";
+import { Command as $Command } from "@aws-sdk/smithy-client";
+import { de_CreateCapacityReservationCommand, se_CreateCapacityReservationCommand } from "../protocols/Aws_json1_1";
+export class CreateCapacityReservationCommand extends $Command {
+    static getEndpointParameterInstructions() {
+        return {
+            UseFIPS: { type: "builtInParams", name: "useFipsEndpoint" },
+            Endpoint: { type: "builtInParams", name: "endpoint" },
+            Region: { type: "builtInParams", name: "region" },
+            UseDualStack: { type: "builtInParams", name: "useDualstackEndpoint" },
+        };
+    }
+    constructor(input) {
+        super();
+        this.input = input;
+    }
+    resolveMiddleware(clientStack, configuration, options) {
+        this.middlewareStack.use(getSerdePlugin(configuration, this.serialize, this.deserialize));
+        this.middlewareStack.use(getEndpointPlugin(configuration, CreateCapacityReservationCommand.getEndpointParameterInstructions()));
+        const stack = clientStack.concat(this.middlewareStack);
+        const { logger } = configuration;
+        const clientName = "AthenaClient";
+        const commandName = "CreateCapacityReservationCommand";
+        const handlerExecutionContext = {
+            logger,
+            clientName,
+            commandName,
+            inputFilterSensitiveLog: (_) => _,
+            outputFilterSensitiveLog: (_) => _,
+        };
+        const { requestHandler } = configuration;
+        return stack.resolve((request) => requestHandler.handle(request.request, options || {}), handlerExecutionContext);
+    }
+    serialize(input, context) {
+        return se_CreateCapacityReservationCommand(input, context);
+    }
+    deserialize(output, context) {
+        return de_CreateCapacityReservationCommand(output, context);
+    }
+}

package/dist-es/commands/GetCapacityAssignmentConfigurationCommand.js

@@ -0,0 +1,41 @@
+import { getEndpointPlugin } from "@aws-sdk/middleware-endpoint";
+import { getSerdePlugin } from "@aws-sdk/middleware-serde";
+import { Command as $Command } from "@aws-sdk/smithy-client";
+import { de_GetCapacityAssignmentConfigurationCommand, se_GetCapacityAssignmentConfigurationCommand, } from "../protocols/Aws_json1_1";
+export class GetCapacityAssignmentConfigurationCommand extends $Command {
+    static getEndpointParameterInstructions() {
+        return {
+            UseFIPS: { type: "builtInParams", name: "useFipsEndpoint" },
+            Endpoint: { type: "builtInParams", name: "endpoint" },
+            Region: { type: "builtInParams", name: "region" },
+            UseDualStack: { type: "builtInParams", name: "useDualstackEndpoint" },
+        };
+    }
+    constructor(input) {
+        super();
+        this.input = input;
+    }
+    resolveMiddleware(clientStack, configuration, options) {
+        this.middlewareStack.use(getSerdePlugin(configuration, this.serialize, this.deserialize));
+        this.middlewareStack.use(getEndpointPlugin(configuration, GetCapacityAssignmentConfigurationCommand.getEndpointParameterInstructions()));
+        const stack = clientStack.concat(this.middlewareStack);
+        const { logger } = configuration;
+        const clientName = "AthenaClient";
+        const commandName = "GetCapacityAssignmentConfigurationCommand";
+        const handlerExecutionContext = {
+            logger,
+            clientName,
+            commandName,
+            inputFilterSensitiveLog: (_) => _,
+            outputFilterSensitiveLog: (_) => _,
+        };
+        const { requestHandler } = configuration;
+        return stack.resolve((request) => requestHandler.handle(request.request, options || {}), handlerExecutionContext);
+    }
+    serialize(input, context) {
+        return se_GetCapacityAssignmentConfigurationCommand(input, context);
+    }
+    deserialize(output, context) {
+        return de_GetCapacityAssignmentConfigurationCommand(output, context);
+    }
+}