feat(client-batch): AWS Batch: Resource Aware Scheduling feature support

aws · Feb 26, 2025 · a48eb96 · a48eb96
1 parent c338212
commit a48eb96
Show file tree

Hide file tree

Showing 21 changed files with 2,949 additions and 116 deletions.
diff --git a/clients/client-batch/README.md b/clients/client-batch/README.md
@@ -229,6 +229,14 @@ CreateComputeEnvironment
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/batch/command/CreateComputeEnvironmentCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/CreateComputeEnvironmentCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/CreateComputeEnvironmentCommandOutput/)
 
+</details>
+<details>
+<summary>
+CreateConsumableResource
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/batch/command/CreateConsumableResourceCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/CreateConsumableResourceCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/CreateConsumableResourceCommandOutput/)
+
 </details>
 <details>
 <summary>
@@ -253,6 +261,14 @@ DeleteComputeEnvironment
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/batch/command/DeleteComputeEnvironmentCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/DeleteComputeEnvironmentCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/DeleteComputeEnvironmentCommandOutput/)
 
+</details>
+<details>
+<summary>
+DeleteConsumableResource
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/batch/command/DeleteConsumableResourceCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/DeleteConsumableResourceCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/DeleteConsumableResourceCommandOutput/)
+
 </details>
 <details>
 <summary>
@@ -285,6 +301,14 @@ DescribeComputeEnvironments
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/batch/command/DescribeComputeEnvironmentsCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/DescribeComputeEnvironmentsCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/DescribeComputeEnvironmentsCommandOutput/)
 
+</details>
+<details>
+<summary>
+DescribeConsumableResource
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/batch/command/DescribeConsumableResourceCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/DescribeConsumableResourceCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/DescribeConsumableResourceCommandOutput/)
+
 </details>
 <details>
 <summary>
@@ -325,6 +349,14 @@ GetJobQueueSnapshot
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/batch/command/GetJobQueueSnapshotCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/GetJobQueueSnapshotCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/GetJobQueueSnapshotCommandOutput/)
 
+</details>
+<details>
+<summary>
+ListConsumableResources
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/batch/command/ListConsumableResourcesCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/ListConsumableResourcesCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/ListConsumableResourcesCommandOutput/)
+
 </details>
 <details>
 <summary>
@@ -333,6 +365,14 @@ ListJobs
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/batch/command/ListJobsCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/ListJobsCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/ListJobsCommandOutput/)
 
+</details>
+<details>
+<summary>
+ListJobsByConsumableResource
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/batch/command/ListJobsByConsumableResourceCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/ListJobsByConsumableResourceCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/ListJobsByConsumableResourceCommandOutput/)
+
 </details>
 <details>
 <summary>
@@ -397,6 +437,14 @@ UpdateComputeEnvironment
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/batch/command/UpdateComputeEnvironmentCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/UpdateComputeEnvironmentCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/UpdateComputeEnvironmentCommandOutput/)
 
+</details>
+<details>
+<summary>
+UpdateConsumableResource
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/batch/command/UpdateConsumableResourceCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/UpdateConsumableResourceCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-batch/Interface/UpdateConsumableResourceCommandOutput/)
+
 </details>
 <details>
 <summary>

diff --git a/clients/client-batch/package.json b/clients/client-batch/package.json
@@ -56,7 +56,9 @@
     "@smithy/util-middleware": "^4.0.1",
     "@smithy/util-retry": "^4.0.1",
     "@smithy/util-utf8": "^4.0.0",
-    "tslib": "^2.6.2"
+    "@types/uuid": "^9.0.1",
+    "tslib": "^2.6.2",
+    "uuid": "^9.0.1"
   },
   "devDependencies": {
     "@tsconfig/node18": "18.2.4",

diff --git a/clients/client-batch/src/Batch.ts b/clients/client-batch/src/Batch.ts
@@ -9,6 +9,11 @@ import {
   CreateComputeEnvironmentCommandInput,
   CreateComputeEnvironmentCommandOutput,
 } from "./commands/CreateComputeEnvironmentCommand";
+import {
+  CreateConsumableResourceCommand,
+  CreateConsumableResourceCommandInput,
+  CreateConsumableResourceCommandOutput,
+} from "./commands/CreateConsumableResourceCommand";
 import {
   CreateJobQueueCommand,
   CreateJobQueueCommandInput,
@@ -24,6 +29,11 @@ import {
   DeleteComputeEnvironmentCommandInput,
   DeleteComputeEnvironmentCommandOutput,
 } from "./commands/DeleteComputeEnvironmentCommand";
+import {
+  DeleteConsumableResourceCommand,
+  DeleteConsumableResourceCommandInput,
+  DeleteConsumableResourceCommandOutput,
+} from "./commands/DeleteConsumableResourceCommand";
 import {
   DeleteJobQueueCommand,
   DeleteJobQueueCommandInput,
@@ -44,6 +54,11 @@ import {
   DescribeComputeEnvironmentsCommandInput,
   DescribeComputeEnvironmentsCommandOutput,
 } from "./commands/DescribeComputeEnvironmentsCommand";
+import {
+  DescribeConsumableResourceCommand,
+  DescribeConsumableResourceCommandInput,
+  DescribeConsumableResourceCommandOutput,
+} from "./commands/DescribeConsumableResourceCommand";
 import {
   DescribeJobDefinitionsCommand,
   DescribeJobDefinitionsCommandInput,
@@ -69,6 +84,16 @@ import {
   GetJobQueueSnapshotCommandInput,
   GetJobQueueSnapshotCommandOutput,
 } from "./commands/GetJobQueueSnapshotCommand";
+import {
+  ListConsumableResourcesCommand,
+  ListConsumableResourcesCommandInput,
+  ListConsumableResourcesCommandOutput,
+} from "./commands/ListConsumableResourcesCommand";
+import {
+  ListJobsByConsumableResourceCommand,
+  ListJobsByConsumableResourceCommandInput,
+  ListJobsByConsumableResourceCommandOutput,
+} from "./commands/ListJobsByConsumableResourceCommand";
 import { ListJobsCommand, ListJobsCommandInput, ListJobsCommandOutput } from "./commands/ListJobsCommand";
 import {
   ListSchedulingPoliciesCommand,
@@ -102,6 +127,11 @@ import {
   UpdateComputeEnvironmentCommandInput,
   UpdateComputeEnvironmentCommandOutput,
 } from "./commands/UpdateComputeEnvironmentCommand";
+import {
+  UpdateConsumableResourceCommand,
+  UpdateConsumableResourceCommandInput,
+  UpdateConsumableResourceCommandOutput,
+} from "./commands/UpdateConsumableResourceCommand";
 import {
   UpdateJobQueueCommand,
   UpdateJobQueueCommandInput,
@@ -116,19 +146,24 @@ import {
 const commands = {
   CancelJobCommand,
   CreateComputeEnvironmentCommand,
+  CreateConsumableResourceCommand,
   CreateJobQueueCommand,
   CreateSchedulingPolicyCommand,
   DeleteComputeEnvironmentCommand,
+  DeleteConsumableResourceCommand,
   DeleteJobQueueCommand,
   DeleteSchedulingPolicyCommand,
   DeregisterJobDefinitionCommand,
   DescribeComputeEnvironmentsCommand,
+  DescribeConsumableResourceCommand,
   DescribeJobDefinitionsCommand,
   DescribeJobQueuesCommand,
   DescribeJobsCommand,
   DescribeSchedulingPoliciesCommand,
   GetJobQueueSnapshotCommand,
+  ListConsumableResourcesCommand,
   ListJobsCommand,
+  ListJobsByConsumableResourceCommand,
   ListSchedulingPoliciesCommand,
   ListTagsForResourceCommand,
   RegisterJobDefinitionCommand,
@@ -137,6 +172,7 @@ const commands = {
   TerminateJobCommand,
   UntagResourceCommand,
   UpdateComputeEnvironmentCommand,
+  UpdateConsumableResourceCommand,
   UpdateJobQueueCommand,
   UpdateSchedulingPolicyCommand,
 };
@@ -170,6 +206,23 @@ export interface Batch {
     cb: (err: any, data?: CreateComputeEnvironmentCommandOutput) => void
   ): void;
 
+  /**
+   * @see {@link CreateConsumableResourceCommand}
+   */
+  createConsumableResource(
+    args: CreateConsumableResourceCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<CreateConsumableResourceCommandOutput>;
+  createConsumableResource(
+    args: CreateConsumableResourceCommandInput,
+    cb: (err: any, data?: CreateConsumableResourceCommandOutput) => void
+  ): void;
+  createConsumableResource(
+    args: CreateConsumableResourceCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: CreateConsumableResourceCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link CreateJobQueueCommand}
    */
@@ -218,6 +271,23 @@ export interface Batch {
     cb: (err: any, data?: DeleteComputeEnvironmentCommandOutput) => void
   ): void;
 
+  /**
+   * @see {@link DeleteConsumableResourceCommand}
+   */
+  deleteConsumableResource(
+    args: DeleteConsumableResourceCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<DeleteConsumableResourceCommandOutput>;
+  deleteConsumableResource(
+    args: DeleteConsumableResourceCommandInput,
+    cb: (err: any, data?: DeleteConsumableResourceCommandOutput) => void
+  ): void;
+  deleteConsumableResource(
+    args: DeleteConsumableResourceCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: DeleteConsumableResourceCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link DeleteJobQueueCommand}
    */
@@ -284,6 +354,23 @@ export interface Batch {
     cb: (err: any, data?: DescribeComputeEnvironmentsCommandOutput) => void
   ): void;
 
+  /**
+   * @see {@link DescribeConsumableResourceCommand}
+   */
+  describeConsumableResource(
+    args: DescribeConsumableResourceCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<DescribeConsumableResourceCommandOutput>;
+  describeConsumableResource(
+    args: DescribeConsumableResourceCommandInput,
+    cb: (err: any, data?: DescribeConsumableResourceCommandOutput) => void
+  ): void;
+  describeConsumableResource(
+    args: DescribeConsumableResourceCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: DescribeConsumableResourceCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link DescribeJobDefinitionsCommand}
    */
@@ -365,6 +452,24 @@ export interface Batch {
     cb: (err: any, data?: GetJobQueueSnapshotCommandOutput) => void
   ): void;
 
+  /**
+   * @see {@link ListConsumableResourcesCommand}
+   */
+  listConsumableResources(): Promise<ListConsumableResourcesCommandOutput>;
+  listConsumableResources(
+    args: ListConsumableResourcesCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<ListConsumableResourcesCommandOutput>;
+  listConsumableResources(
+    args: ListConsumableResourcesCommandInput,
+    cb: (err: any, data?: ListConsumableResourcesCommandOutput) => void
+  ): void;
+  listConsumableResources(
+    args: ListConsumableResourcesCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: ListConsumableResourcesCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link ListJobsCommand}
    */
@@ -377,6 +482,23 @@ export interface Batch {
     cb: (err: any, data?: ListJobsCommandOutput) => void
   ): void;
 
+  /**
+   * @see {@link ListJobsByConsumableResourceCommand}
+   */
+  listJobsByConsumableResource(
+    args: ListJobsByConsumableResourceCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<ListJobsByConsumableResourceCommandOutput>;
+  listJobsByConsumableResource(
+    args: ListJobsByConsumableResourceCommandInput,
+    cb: (err: any, data?: ListJobsByConsumableResourceCommandOutput) => void
+  ): void;
+  listJobsByConsumableResource(
+    args: ListJobsByConsumableResourceCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: ListJobsByConsumableResourceCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link ListSchedulingPoliciesCommand}
    */
@@ -490,6 +612,23 @@ export interface Batch {
     cb: (err: any, data?: UpdateComputeEnvironmentCommandOutput) => void
   ): void;
 
+  /**
+   * @see {@link UpdateConsumableResourceCommand}
+   */
+  updateConsumableResource(
+    args: UpdateConsumableResourceCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<UpdateConsumableResourceCommandOutput>;
+  updateConsumableResource(
+    args: UpdateConsumableResourceCommandInput,
+    cb: (err: any, data?: UpdateConsumableResourceCommandOutput) => void
+  ): void;
+  updateConsumableResource(
+    args: UpdateConsumableResourceCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: UpdateConsumableResourceCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link UpdateJobQueueCommand}
    */