feat(client-kafka): Amazon MSK now supports intelligent rebalancing for MSK Express brokers.

awstools · awstools · commit 265e5dae15a5 · 2025-11-10T19:18:38.000Z
diff --git a/clients/client-kafka/README.md b/clients/client-kafka/README.md
@@ -594,6 +594,14 @@ UpdateMonitoring
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/kafka/command/UpdateMonitoringCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-kafka/Interface/UpdateMonitoringCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-kafka/Interface/UpdateMonitoringCommandOutput/)
 
+</details>
+<details>
+<summary>
+UpdateRebalancing
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/kafka/command/UpdateRebalancingCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-kafka/Interface/UpdateRebalancingCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-kafka/Interface/UpdateRebalancingCommandOutput/)
+
 </details>
 <details>
 <summary>
diff --git a/clients/client-kafka/src/Kafka.ts b/clients/client-kafka/src/Kafka.ts
@@ -239,6 +239,11 @@ import {
   UpdateMonitoringCommandInput,
   UpdateMonitoringCommandOutput,
 } from "./commands/UpdateMonitoringCommand";
+import {
+  UpdateRebalancingCommand,
+  UpdateRebalancingCommandInput,
+  UpdateRebalancingCommandOutput,
+} from "./commands/UpdateRebalancingCommand";
 import {
   UpdateReplicationInfoCommand,
   UpdateReplicationInfoCommandInput,
@@ -306,6 +311,7 @@ const commands = {
   UpdateConfigurationCommand,
   UpdateConnectivityCommand,
   UpdateMonitoringCommand,
+  UpdateRebalancingCommand,
   UpdateReplicationInfoCommand,
   UpdateSecurityCommand,
   UpdateStorageCommand,
@@ -1098,6 +1104,23 @@ export interface Kafka {
     cb: (err: any, data?: UpdateMonitoringCommandOutput) => void
   ): void;
 
+  /**
+   * @see {@link UpdateRebalancingCommand}
+   */
+  updateRebalancing(
+    args: UpdateRebalancingCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<UpdateRebalancingCommandOutput>;
+  updateRebalancing(
+    args: UpdateRebalancingCommandInput,
+    cb: (err: any, data?: UpdateRebalancingCommandOutput) => void
+  ): void;
+  updateRebalancing(
+    args: UpdateRebalancingCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: UpdateRebalancingCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link UpdateReplicationInfoCommand}
    */
diff --git a/clients/client-kafka/src/KafkaClient.ts b/clients/client-kafka/src/KafkaClient.ts
@@ -174,6 +174,7 @@ import {
 } from "./commands/UpdateConfigurationCommand";
 import { UpdateConnectivityCommandInput, UpdateConnectivityCommandOutput } from "./commands/UpdateConnectivityCommand";
 import { UpdateMonitoringCommandInput, UpdateMonitoringCommandOutput } from "./commands/UpdateMonitoringCommand";
+import { UpdateRebalancingCommandInput, UpdateRebalancingCommandOutput } from "./commands/UpdateRebalancingCommand";
 import {
   UpdateReplicationInfoCommandInput,
   UpdateReplicationInfoCommandOutput,
@@ -244,6 +245,7 @@ export type ServiceInputTypes =
   | UpdateConfigurationCommandInput
   | UpdateConnectivityCommandInput
   | UpdateMonitoringCommandInput
+  | UpdateRebalancingCommandInput
   | UpdateReplicationInfoCommandInput
   | UpdateSecurityCommandInput
   | UpdateStorageCommandInput;
@@ -301,6 +303,7 @@ export type ServiceOutputTypes =
   | UpdateConfigurationCommandOutput
   | UpdateConnectivityCommandOutput
   | UpdateMonitoringCommandOutput
+  | UpdateRebalancingCommandOutput
   | UpdateReplicationInfoCommandOutput
   | UpdateSecurityCommandOutput
   | UpdateStorageCommandOutput;
diff --git a/clients/client-kafka/src/commands/CreateClusterCommand.ts b/clients/client-kafka/src/commands/CreateClusterCommand.ts
@@ -80,6 +80,9 @@ export interface CreateClusterCommandOutput extends CreateClusterResponse, __Met
  *       "STRING_VALUE",
  *     ],
  *   },
+ *   Rebalancing: { // Rebalancing
+ *     Status: "PAUSED" || "ACTIVE",
+ *   },
  *   ClientAuthentication: { // ClientAuthentication
  *     Sasl: { // Sasl
  *       Scram: { // Scram
diff --git a/clients/client-kafka/src/commands/CreateClusterV2Command.ts b/clients/client-kafka/src/commands/CreateClusterV2Command.ts
@@ -85,6 +85,9 @@ export interface CreateClusterV2CommandOutput extends CreateClusterV2Response, _
  *         "STRING_VALUE",
  *       ],
  *     },
+ *     Rebalancing: { // Rebalancing
+ *       Status: "PAUSED" || "ACTIVE",
+ *     },
  *     ClientAuthentication: { // ClientAuthentication
  *       Sasl: { // Sasl
  *         Scram: { // Scram
diff --git a/clients/client-kafka/src/commands/DescribeClusterCommand.ts b/clients/client-kafka/src/commands/DescribeClusterCommand.ts
@@ -87,6 +87,9 @@ export interface DescribeClusterCommandOutput extends DescribeClusterResponse, _
  * //         "STRING_VALUE",
  * //       ],
  * //     },
+ * //     Rebalancing: { // Rebalancing
+ * //       Status: "PAUSED" || "ACTIVE",
+ * //     },
  * //     ClientAuthentication: { // ClientAuthentication
  * //       Sasl: { // Sasl
  * //         Scram: { // Scram
diff --git a/clients/client-kafka/src/commands/DescribeClusterOperationCommand.ts b/clients/client-kafka/src/commands/DescribeClusterOperationCommand.ts
@@ -166,6 +166,9 @@ export interface DescribeClusterOperationCommandOutput extends DescribeClusterOp
  * //           Number("double"),
  * //         ],
  * //       },
+ * //       Rebalancing: { // Rebalancing
+ * //         Status: "PAUSED" || "ACTIVE",
+ * //       },
  * //     },
  * //     TargetClusterInfo: {
  * //       BrokerEBSVolumeInfo: [
@@ -270,6 +273,9 @@ export interface DescribeClusterOperationCommandOutput extends DescribeClusterOp
  * //           Number("double"),
  * //         ],
  * //       },
+ * //       Rebalancing: {
+ * //         Status: "PAUSED" || "ACTIVE",
+ * //       },
  * //     },
  * //     VpcConnectionInfo: { // VpcConnectionInfo
  * //       VpcConnectionArn: "STRING_VALUE",
diff --git a/clients/client-kafka/src/commands/DescribeClusterOperationV2Command.ts b/clients/client-kafka/src/commands/DescribeClusterOperationV2Command.ts
@@ -167,6 +167,9 @@ export interface DescribeClusterOperationV2CommandOutput extends DescribeCluster
  * //             Number("double"),
  * //           ],
  * //         },
+ * //         Rebalancing: { // Rebalancing
+ * //           Status: "PAUSED" || "ACTIVE",
+ * //         },
  * //       },
  * //       TargetClusterInfo: {
  * //         BrokerEBSVolumeInfo: [
@@ -271,6 +274,9 @@ export interface DescribeClusterOperationV2CommandOutput extends DescribeCluster
  * //             Number("double"),
  * //           ],
  * //         },
+ * //         Rebalancing: {
+ * //           Status: "PAUSED" || "ACTIVE",
+ * //         },
  * //       },
  * //       VpcConnectionInfo: { // VpcConnectionInfo
  * //         VpcConnectionArn: "STRING_VALUE",
diff --git a/clients/client-kafka/src/commands/DescribeClusterV2Command.ts b/clients/client-kafka/src/commands/DescribeClusterV2Command.ts
@@ -101,6 +101,9 @@ export interface DescribeClusterV2CommandOutput extends DescribeClusterV2Respons
  * //           "STRING_VALUE",
  * //         ],
  * //       },
+ * //       Rebalancing: { // Rebalancing
+ * //         Status: "PAUSED" || "ACTIVE",
+ * //       },
  * //       CurrentBrokerSoftwareInfo: { // BrokerSoftwareInfo
  * //         ConfigurationArn: "STRING_VALUE",
  * //         ConfigurationRevision: Number("long"),
diff --git a/clients/client-kafka/src/commands/ListClusterOperationsCommand.ts b/clients/client-kafka/src/commands/ListClusterOperationsCommand.ts
@@ -169,6 +169,9 @@ export interface ListClusterOperationsCommandOutput extends ListClusterOperation
  * //             Number("double"),
  * //           ],
  * //         },
+ * //         Rebalancing: { // Rebalancing
+ * //           Status: "PAUSED" || "ACTIVE",
+ * //         },
  * //       },
  * //       TargetClusterInfo: {
  * //         BrokerEBSVolumeInfo: [
@@ -273,6 +276,9 @@ export interface ListClusterOperationsCommandOutput extends ListClusterOperation
  * //             Number("double"),
  * //           ],
  * //         },
+ * //         Rebalancing: {
+ * //           Status: "PAUSED" || "ACTIVE",
+ * //         },
  * //       },
  * //       VpcConnectionInfo: { // VpcConnectionInfo
  * //         VpcConnectionArn: "STRING_VALUE",
diff --git a/clients/client-kafka/src/commands/ListClustersCommand.ts b/clients/client-kafka/src/commands/ListClustersCommand.ts
@@ -90,6 +90,9 @@ export interface ListClustersCommandOutput extends ListClustersResponse, __Metad
  * //           "STRING_VALUE",
  * //         ],
  * //       },
+ * //       Rebalancing: { // Rebalancing
+ * //         Status: "PAUSED" || "ACTIVE",
+ * //       },
  * //       ClientAuthentication: { // ClientAuthentication
  * //         Sasl: { // Sasl
  * //           Scram: { // Scram
diff --git a/clients/client-kafka/src/commands/ListClustersV2Command.ts b/clients/client-kafka/src/commands/ListClustersV2Command.ts
@@ -105,6 +105,9 @@ export interface ListClustersV2CommandOutput extends ListClustersV2Response, __M
  * //             "STRING_VALUE",
  * //           ],
  * //         },
+ * //         Rebalancing: { // Rebalancing
+ * //           Status: "PAUSED" || "ACTIVE",
+ * //         },
  * //         CurrentBrokerSoftwareInfo: { // BrokerSoftwareInfo
  * //           ConfigurationArn: "STRING_VALUE",
  * //           ConfigurationRevision: Number("long"),
diff --git a/clients/client-kafka/src/commands/UpdateRebalancingCommand.ts b/clients/client-kafka/src/commands/UpdateRebalancingCommand.ts
@@ -0,0 +1,121 @@
+// smithy-typescript generated code
+import { getEndpointPlugin } from "@smithy/middleware-endpoint";
+import { getSerdePlugin } from "@smithy/middleware-serde";
+import { Command as $Command } from "@smithy/smithy-client";
+import { MetadataBearer as __MetadataBearer } from "@smithy/types";
+
+import { commonParams } from "../endpoint/EndpointParameters";
+import { KafkaClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../KafkaClient";
+import { UpdateRebalancingRequest, UpdateRebalancingResponse } from "../models/models_0";
+import { de_UpdateRebalancingCommand, se_UpdateRebalancingCommand } from "../protocols/Aws_restJson1";
+
+/**
+ * @public
+ */
+export type { __MetadataBearer };
+export { $Command };
+/**
+ * @public
+ *
+ * The input for {@link UpdateRebalancingCommand}.
+ */
+export interface UpdateRebalancingCommandInput extends UpdateRebalancingRequest {}
+/**
+ * @public
+ *
+ * The output of {@link UpdateRebalancingCommand}.
+ */
+export interface UpdateRebalancingCommandOutput extends UpdateRebalancingResponse, __MetadataBearer {}
+
+/**
+ * <p>Use this resource to update the intelligent rebalancing status of an Amazon MSK Provisioned cluster with Express brokers.</p>
+ * @example
+ * Use a bare-bones client and the command you need to make an API call.
+ * ```javascript
+ * import { KafkaClient, UpdateRebalancingCommand } from "@aws-sdk/client-kafka"; // ES Modules import
+ * // const { KafkaClient, UpdateRebalancingCommand } = require("@aws-sdk/client-kafka"); // CommonJS import
+ * // import type { KafkaClientConfig } from "@aws-sdk/client-kafka";
+ * const config = {}; // type is KafkaClientConfig
+ * const client = new KafkaClient(config);
+ * const input = { // UpdateRebalancingRequest
+ *   ClusterArn: "STRING_VALUE", // required
+ *   CurrentVersion: "STRING_VALUE", // required
+ *   Rebalancing: { // Rebalancing
+ *     Status: "PAUSED" || "ACTIVE",
+ *   },
+ * };
+ * const command = new UpdateRebalancingCommand(input);
+ * const response = await client.send(command);
+ * // { // UpdateRebalancingResponse
+ * //   ClusterArn: "STRING_VALUE",
+ * //   ClusterOperationArn: "STRING_VALUE",
+ * // };
+ *
+ * ```
+ *
+ * @param UpdateRebalancingCommandInput - {@link UpdateRebalancingCommandInput}
+ * @returns {@link UpdateRebalancingCommandOutput}
+ * @see {@link UpdateRebalancingCommandInput} for command's `input` shape.
+ * @see {@link UpdateRebalancingCommandOutput} for command's `response` shape.
+ * @see {@link KafkaClientResolvedConfig | config} for KafkaClient's `config` shape.
+ *
+ * @throws {@link BadRequestException} (client fault)
+ *  <p>Returns information about an error.</p>
+ *
+ * @throws {@link ForbiddenException} (client fault)
+ *  <p>Returns information about an error.</p>
+ *
+ * @throws {@link InternalServerErrorException} (server fault)
+ *  <p>Returns information about an error.</p>
+ *
+ * @throws {@link NotFoundException} (client fault)
+ *  <p>Returns information about an error.</p>
+ *
+ * @throws {@link ServiceUnavailableException} (server fault)
+ *  <p>Returns information about an error.</p>
+ *
+ * @throws {@link TooManyRequestsException} (client fault)
+ *  <p>Returns information about an error.</p>
+ *
+ * @throws {@link UnauthorizedException} (client fault)
+ *  <p>Returns information about an error.</p>
+ *
+ * @throws {@link KafkaServiceException}
+ * <p>Base exception class for all service exceptions from Kafka service.</p>
+ *
+ *
+ * @public
+ */
+export class UpdateRebalancingCommand extends $Command
+  .classBuilder<
+    UpdateRebalancingCommandInput,
+    UpdateRebalancingCommandOutput,
+    KafkaClientResolvedConfig,
+    ServiceInputTypes,
+    ServiceOutputTypes
+  >()
+  .ep(commonParams)
+  .m(function (this: any, Command: any, cs: any, config: KafkaClientResolvedConfig, o: any) {
+    return [
+      getSerdePlugin(config, this.serialize, this.deserialize),
+      getEndpointPlugin(config, Command.getEndpointParameterInstructions()),
+    ];
+  })
+  .s("Kafka", "UpdateRebalancing", {})
+  .n("KafkaClient", "UpdateRebalancingCommand")
+  .f(void 0, void 0)
+  .ser(se_UpdateRebalancingCommand)
+  .de(de_UpdateRebalancingCommand)
+  .build() {
+  /** @internal type navigation helper, not in runtime. */
+  protected declare static __types: {
+    api: {
+      input: UpdateRebalancingRequest;
+      output: UpdateRebalancingResponse;
+    };
+    sdk: {
+      input: UpdateRebalancingCommandInput;
+      output: UpdateRebalancingCommandOutput;
+    };
+  };
+}
diff --git a/clients/client-kafka/src/commands/index.ts b/clients/client-kafka/src/commands/index.ts
@@ -48,6 +48,7 @@ export * from "./UpdateClusterKafkaVersionCommand";
 export * from "./UpdateConfigurationCommand";
 export * from "./UpdateConnectivityCommand";
 export * from "./UpdateMonitoringCommand";
+export * from "./UpdateRebalancingCommand";
 export * from "./UpdateReplicationInfoCommand";
 export * from "./UpdateSecurityCommand";
 export * from "./UpdateStorageCommand";
diff --git a/clients/client-kafka/src/models/models_0.ts b/clients/client-kafka/src/models/models_0.ts
diff --git a/clients/client-kafka/src/protocols/Aws_restJson1.ts b/clients/client-kafka/src/protocols/Aws_restJson1.ts
diff --git a/codegen/sdk-codegen/aws-models/kafka.json b/codegen/sdk-codegen/aws-models/kafka.json
