@azure/arm-cosmosdb 16.1.1-alpha.20241025.4 → 17.0.0-beta.1

package/src/operations/mongoDBResources.ts

@@ -47,6 +47,16 @@ import {
   MongoDBResourcesMigrateMongoDBDatabaseToAutoscaleResponse,
   MongoDBResourcesMigrateMongoDBDatabaseToManualThroughputOptionalParams,
   MongoDBResourcesMigrateMongoDBDatabaseToManualThroughputResponse,
+  RetrieveThroughputParameters,
+  MongoDBResourcesMongoDBDatabaseRetrieveThroughputDistributionOptionalParams,
+  MongoDBResourcesMongoDBDatabaseRetrieveThroughputDistributionResponse,
+  RedistributeThroughputParameters,
+  MongoDBResourcesMongoDBDatabaseRedistributeThroughputOptionalParams,
+  MongoDBResourcesMongoDBDatabaseRedistributeThroughputResponse,
+  MongoDBResourcesMongoDBContainerRetrieveThroughputDistributionOptionalParams,
+  MongoDBResourcesMongoDBContainerRetrieveThroughputDistributionResponse,
+  MongoDBResourcesMongoDBContainerRedistributeThroughputOptionalParams,
+  MongoDBResourcesMongoDBContainerRedistributeThroughputResponse,
   MongoDBResourcesGetMongoDBCollectionOptionalParams,
   MongoDBResourcesGetMongoDBCollectionResponse,
   MongoDBCollectionCreateUpdateParameters,
@@ -54,6 +64,11 @@ import {
   MongoDBResourcesCreateUpdateMongoDBCollectionResponse,
   MongoDBResourcesDeleteMongoDBCollectionOptionalParams,
   MongoDBResourcesDeleteMongoDBCollectionResponse,
+  MergeParameters,
+  MongoDBResourcesMongoDBDatabasePartitionMergeOptionalParams,
+  MongoDBResourcesMongoDBDatabasePartitionMergeResponse,
+  MongoDBResourcesListMongoDBCollectionPartitionMergeOptionalParams,
+  MongoDBResourcesListMongoDBCollectionPartitionMergeResponse,
   MongoDBResourcesGetMongoDBCollectionThroughputOptionalParams,
   MongoDBResourcesGetMongoDBCollectionThroughputResponse,
   MongoDBResourcesUpdateMongoDBCollectionThroughputOptionalParams,
@@ -917,6 +932,452 @@ export class MongoDBResourcesImpl implements MongoDBResources {
     return poller.pollUntilDone();
   }
 
+  /**
+   * Retrieve throughput distribution for an Azure Cosmos DB MongoDB database
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param accountName Cosmos DB database account name.
+   * @param databaseName Cosmos DB database name.
+   * @param retrieveThroughputParameters The parameters to provide for retrieving throughput distribution
+   *                                     for the current MongoDB database.
+   * @param options The options parameters.
+   */
+  async beginMongoDBDatabaseRetrieveThroughputDistribution(
+    resourceGroupName: string,
+    accountName: string,
+    databaseName: string,
+    retrieveThroughputParameters: RetrieveThroughputParameters,
+    options?: MongoDBResourcesMongoDBDatabaseRetrieveThroughputDistributionOptionalParams,
+  ): Promise<
+    SimplePollerLike<
+      OperationState<MongoDBResourcesMongoDBDatabaseRetrieveThroughputDistributionResponse>,
+      MongoDBResourcesMongoDBDatabaseRetrieveThroughputDistributionResponse
+    >
+  > {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec,
+    ): Promise<MongoDBResourcesMongoDBDatabaseRetrieveThroughputDistributionResponse> => {
+      return this.client.sendOperationRequest(args, spec);
+    };
+    const sendOperationFn = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec,
+    ) => {
+      let currentRawResponse: coreClient.FullOperationResponse | undefined =
+        undefined;
+      const providedCallback = args.options?.onResponse;
+      const callback: coreClient.RawResponseCallback = (
+        rawResponse: coreClient.FullOperationResponse,
+        flatResponse: unknown,
+      ) => {
+        currentRawResponse = rawResponse;
+        providedCallback?.(rawResponse, flatResponse);
+      };
+      const updatedArgs = {
+        ...args,
+        options: {
+          ...args.options,
+          onResponse: callback,
+        },
+      };
+      const flatResponse = await directSendOperation(updatedArgs, spec);
+      return {
+        flatResponse,
+        rawResponse: {
+          statusCode: currentRawResponse!.status,
+          body: currentRawResponse!.parsedBody,
+          headers: currentRawResponse!.headers.toJSON(),
+        },
+      };
+    };
+
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: {
+        resourceGroupName,
+        accountName,
+        databaseName,
+        retrieveThroughputParameters,
+        options,
+      },
+      spec: mongoDBDatabaseRetrieveThroughputDistributionOperationSpec,
+    });
+    const poller = await createHttpPoller<
+      MongoDBResourcesMongoDBDatabaseRetrieveThroughputDistributionResponse,
+      OperationState<MongoDBResourcesMongoDBDatabaseRetrieveThroughputDistributionResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
+      resourceLocationConfig: "location",
+    });
+    await poller.poll();
+    return poller;
+  }
+
+  /**
+   * Retrieve throughput distribution for an Azure Cosmos DB MongoDB database
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param accountName Cosmos DB database account name.
+   * @param databaseName Cosmos DB database name.
+   * @param retrieveThroughputParameters The parameters to provide for retrieving throughput distribution
+   *                                     for the current MongoDB database.
+   * @param options The options parameters.
+   */
+  async beginMongoDBDatabaseRetrieveThroughputDistributionAndWait(
+    resourceGroupName: string,
+    accountName: string,
+    databaseName: string,
+    retrieveThroughputParameters: RetrieveThroughputParameters,
+    options?: MongoDBResourcesMongoDBDatabaseRetrieveThroughputDistributionOptionalParams,
+  ): Promise<MongoDBResourcesMongoDBDatabaseRetrieveThroughputDistributionResponse> {
+    const poller =
+      await this.beginMongoDBDatabaseRetrieveThroughputDistribution(
+        resourceGroupName,
+        accountName,
+        databaseName,
+        retrieveThroughputParameters,
+        options,
+      );
+    return poller.pollUntilDone();
+  }
+
+  /**
+   * Redistribute throughput for an Azure Cosmos DB MongoDB database
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param accountName Cosmos DB database account name.
+   * @param databaseName Cosmos DB database name.
+   * @param redistributeThroughputParameters The parameters to provide for redistributing throughput for
+   *                                         the current MongoDB database.
+   * @param options The options parameters.
+   */
+  async beginMongoDBDatabaseRedistributeThroughput(
+    resourceGroupName: string,
+    accountName: string,
+    databaseName: string,
+    redistributeThroughputParameters: RedistributeThroughputParameters,
+    options?: MongoDBResourcesMongoDBDatabaseRedistributeThroughputOptionalParams,
+  ): Promise<
+    SimplePollerLike<
+      OperationState<MongoDBResourcesMongoDBDatabaseRedistributeThroughputResponse>,
+      MongoDBResourcesMongoDBDatabaseRedistributeThroughputResponse
+    >
+  > {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec,
+    ): Promise<MongoDBResourcesMongoDBDatabaseRedistributeThroughputResponse> => {
+      return this.client.sendOperationRequest(args, spec);
+    };
+    const sendOperationFn = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec,
+    ) => {
+      let currentRawResponse: coreClient.FullOperationResponse | undefined =
+        undefined;
+      const providedCallback = args.options?.onResponse;
+      const callback: coreClient.RawResponseCallback = (
+        rawResponse: coreClient.FullOperationResponse,
+        flatResponse: unknown,
+      ) => {
+        currentRawResponse = rawResponse;
+        providedCallback?.(rawResponse, flatResponse);
+      };
+      const updatedArgs = {
+        ...args,
+        options: {
+          ...args.options,
+          onResponse: callback,
+        },
+      };
+      const flatResponse = await directSendOperation(updatedArgs, spec);
+      return {
+        flatResponse,
+        rawResponse: {
+          statusCode: currentRawResponse!.status,
+          body: currentRawResponse!.parsedBody,
+          headers: currentRawResponse!.headers.toJSON(),
+        },
+      };
+    };
+
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: {
+        resourceGroupName,
+        accountName,
+        databaseName,
+        redistributeThroughputParameters,
+        options,
+      },
+      spec: mongoDBDatabaseRedistributeThroughputOperationSpec,
+    });
+    const poller = await createHttpPoller<
+      MongoDBResourcesMongoDBDatabaseRedistributeThroughputResponse,
+      OperationState<MongoDBResourcesMongoDBDatabaseRedistributeThroughputResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
+      resourceLocationConfig: "location",
+    });
+    await poller.poll();
+    return poller;
+  }
+
+  /**
+   * Redistribute throughput for an Azure Cosmos DB MongoDB database
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param accountName Cosmos DB database account name.
+   * @param databaseName Cosmos DB database name.
+   * @param redistributeThroughputParameters The parameters to provide for redistributing throughput for
+   *                                         the current MongoDB database.
+   * @param options The options parameters.
+   */
+  async beginMongoDBDatabaseRedistributeThroughputAndWait(
+    resourceGroupName: string,
+    accountName: string,
+    databaseName: string,
+    redistributeThroughputParameters: RedistributeThroughputParameters,
+    options?: MongoDBResourcesMongoDBDatabaseRedistributeThroughputOptionalParams,
+  ): Promise<MongoDBResourcesMongoDBDatabaseRedistributeThroughputResponse> {
+    const poller = await this.beginMongoDBDatabaseRedistributeThroughput(
+      resourceGroupName,
+      accountName,
+      databaseName,
+      redistributeThroughputParameters,
+      options,
+    );
+    return poller.pollUntilDone();
+  }
+
+  /**
+   * Retrieve throughput distribution for an Azure Cosmos DB MongoDB container
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param accountName Cosmos DB database account name.
+   * @param databaseName Cosmos DB database name.
+   * @param collectionName Cosmos DB collection name.
+   * @param retrieveThroughputParameters The parameters to provide for retrieving throughput distribution
+   *                                     for the current MongoDB container.
+   * @param options The options parameters.
+   */
+  async beginMongoDBContainerRetrieveThroughputDistribution(
+    resourceGroupName: string,
+    accountName: string,
+    databaseName: string,
+    collectionName: string,
+    retrieveThroughputParameters: RetrieveThroughputParameters,
+    options?: MongoDBResourcesMongoDBContainerRetrieveThroughputDistributionOptionalParams,
+  ): Promise<
+    SimplePollerLike<
+      OperationState<MongoDBResourcesMongoDBContainerRetrieveThroughputDistributionResponse>,
+      MongoDBResourcesMongoDBContainerRetrieveThroughputDistributionResponse
+    >
+  > {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec,
+    ): Promise<MongoDBResourcesMongoDBContainerRetrieveThroughputDistributionResponse> => {
+      return this.client.sendOperationRequest(args, spec);
+    };
+    const sendOperationFn = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec,
+    ) => {
+      let currentRawResponse: coreClient.FullOperationResponse | undefined =
+        undefined;
+      const providedCallback = args.options?.onResponse;
+      const callback: coreClient.RawResponseCallback = (
+        rawResponse: coreClient.FullOperationResponse,
+        flatResponse: unknown,
+      ) => {
+        currentRawResponse = rawResponse;
+        providedCallback?.(rawResponse, flatResponse);
+      };
+      const updatedArgs = {
+        ...args,
+        options: {
+          ...args.options,
+          onResponse: callback,
+        },
+      };
+      const flatResponse = await directSendOperation(updatedArgs, spec);
+      return {
+        flatResponse,
+        rawResponse: {
+          statusCode: currentRawResponse!.status,
+          body: currentRawResponse!.parsedBody,
+          headers: currentRawResponse!.headers.toJSON(),
+        },
+      };
+    };
+
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: {
+        resourceGroupName,
+        accountName,
+        databaseName,
+        collectionName,
+        retrieveThroughputParameters,
+        options,
+      },
+      spec: mongoDBContainerRetrieveThroughputDistributionOperationSpec,
+    });
+    const poller = await createHttpPoller<
+      MongoDBResourcesMongoDBContainerRetrieveThroughputDistributionResponse,
+      OperationState<MongoDBResourcesMongoDBContainerRetrieveThroughputDistributionResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
+      resourceLocationConfig: "location",
+    });
+    await poller.poll();
+    return poller;
+  }
+
+  /**
+   * Retrieve throughput distribution for an Azure Cosmos DB MongoDB container
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param accountName Cosmos DB database account name.
+   * @param databaseName Cosmos DB database name.
+   * @param collectionName Cosmos DB collection name.
+   * @param retrieveThroughputParameters The parameters to provide for retrieving throughput distribution
+   *                                     for the current MongoDB container.
+   * @param options The options parameters.
+   */
+  async beginMongoDBContainerRetrieveThroughputDistributionAndWait(
+    resourceGroupName: string,
+    accountName: string,
+    databaseName: string,
+    collectionName: string,
+    retrieveThroughputParameters: RetrieveThroughputParameters,
+    options?: MongoDBResourcesMongoDBContainerRetrieveThroughputDistributionOptionalParams,
+  ): Promise<MongoDBResourcesMongoDBContainerRetrieveThroughputDistributionResponse> {
+    const poller =
+      await this.beginMongoDBContainerRetrieveThroughputDistribution(
+        resourceGroupName,
+        accountName,
+        databaseName,
+        collectionName,
+        retrieveThroughputParameters,
+        options,
+      );
+    return poller.pollUntilDone();
+  }
+
+  /**
+   * Redistribute throughput for an Azure Cosmos DB MongoDB container
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param accountName Cosmos DB database account name.
+   * @param databaseName Cosmos DB database name.
+   * @param collectionName Cosmos DB collection name.
+   * @param redistributeThroughputParameters The parameters to provide for redistributing throughput for
+   *                                         the current MongoDB container.
+   * @param options The options parameters.
+   */
+  async beginMongoDBContainerRedistributeThroughput(
+    resourceGroupName: string,
+    accountName: string,
+    databaseName: string,
+    collectionName: string,
+    redistributeThroughputParameters: RedistributeThroughputParameters,
+    options?: MongoDBResourcesMongoDBContainerRedistributeThroughputOptionalParams,
+  ): Promise<
+    SimplePollerLike<
+      OperationState<MongoDBResourcesMongoDBContainerRedistributeThroughputResponse>,
+      MongoDBResourcesMongoDBContainerRedistributeThroughputResponse
+    >
+  > {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec,
+    ): Promise<MongoDBResourcesMongoDBContainerRedistributeThroughputResponse> => {
+      return this.client.sendOperationRequest(args, spec);
+    };
+    const sendOperationFn = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec,
+    ) => {
+      let currentRawResponse: coreClient.FullOperationResponse | undefined =
+        undefined;
+      const providedCallback = args.options?.onResponse;
+      const callback: coreClient.RawResponseCallback = (
+        rawResponse: coreClient.FullOperationResponse,
+        flatResponse: unknown,
+      ) => {
+        currentRawResponse = rawResponse;
+        providedCallback?.(rawResponse, flatResponse);
+      };
+      const updatedArgs = {
+        ...args,
+        options: {
+          ...args.options,
+          onResponse: callback,
+        },
+      };
+      const flatResponse = await directSendOperation(updatedArgs, spec);
+      return {
+        flatResponse,
+        rawResponse: {
+          statusCode: currentRawResponse!.status,
+          body: currentRawResponse!.parsedBody,
+          headers: currentRawResponse!.headers.toJSON(),
+        },
+      };
+    };
+
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: {
+        resourceGroupName,
+        accountName,
+        databaseName,
+        collectionName,
+        redistributeThroughputParameters,
+        options,
+      },
+      spec: mongoDBContainerRedistributeThroughputOperationSpec,
+    });
+    const poller = await createHttpPoller<
+      MongoDBResourcesMongoDBContainerRedistributeThroughputResponse,
+      OperationState<MongoDBResourcesMongoDBContainerRedistributeThroughputResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
+      resourceLocationConfig: "location",
+    });
+    await poller.poll();
+    return poller;
+  }
+
+  /**
+   * Redistribute throughput for an Azure Cosmos DB MongoDB container
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param accountName Cosmos DB database account name.
+   * @param databaseName Cosmos DB database name.
+   * @param collectionName Cosmos DB collection name.
+   * @param redistributeThroughputParameters The parameters to provide for redistributing throughput for
+   *                                         the current MongoDB container.
+   * @param options The options parameters.
+   */
+  async beginMongoDBContainerRedistributeThroughputAndWait(
+    resourceGroupName: string,
+    accountName: string,
+    databaseName: string,
+    collectionName: string,
+    redistributeThroughputParameters: RedistributeThroughputParameters,
+    options?: MongoDBResourcesMongoDBContainerRedistributeThroughputOptionalParams,
+  ): Promise<MongoDBResourcesMongoDBContainerRedistributeThroughputResponse> {
+    const poller = await this.beginMongoDBContainerRedistributeThroughput(
+      resourceGroupName,
+      accountName,
+      databaseName,
+      collectionName,
+      redistributeThroughputParameters,
+      options,
+    );
+    return poller.pollUntilDone();
+  }
+
   /**
    * Lists the MongoDB collection under an existing Azure Cosmos DB database account.
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
@@ -937,53 +1398,268 @@ export class MongoDBResourcesImpl implements MongoDBResources {
   }
 
   /**
-   * Gets the MongoDB collection under an existing Azure Cosmos DB database account.
+   * Gets the MongoDB collection under an existing Azure Cosmos DB database account.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param accountName Cosmos DB database account name.
+   * @param databaseName Cosmos DB database name.
+   * @param collectionName Cosmos DB collection name.
+   * @param options The options parameters.
+   */
+  getMongoDBCollection(
+    resourceGroupName: string,
+    accountName: string,
+    databaseName: string,
+    collectionName: string,
+    options?: MongoDBResourcesGetMongoDBCollectionOptionalParams,
+  ): Promise<MongoDBResourcesGetMongoDBCollectionResponse> {
+    return this.client.sendOperationRequest(
+      { resourceGroupName, accountName, databaseName, collectionName, options },
+      getMongoDBCollectionOperationSpec,
+    );
+  }
+
+  /**
+   * Create or update an Azure Cosmos DB MongoDB Collection
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param accountName Cosmos DB database account name.
+   * @param databaseName Cosmos DB database name.
+   * @param collectionName Cosmos DB collection name.
+   * @param createUpdateMongoDBCollectionParameters The parameters to provide for the current MongoDB
+   *                                                Collection.
+   * @param options The options parameters.
+   */
+  async beginCreateUpdateMongoDBCollection(
+    resourceGroupName: string,
+    accountName: string,
+    databaseName: string,
+    collectionName: string,
+    createUpdateMongoDBCollectionParameters: MongoDBCollectionCreateUpdateParameters,
+    options?: MongoDBResourcesCreateUpdateMongoDBCollectionOptionalParams,
+  ): Promise<
+    SimplePollerLike<
+      OperationState<MongoDBResourcesCreateUpdateMongoDBCollectionResponse>,
+      MongoDBResourcesCreateUpdateMongoDBCollectionResponse
+    >
+  > {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec,
+    ): Promise<MongoDBResourcesCreateUpdateMongoDBCollectionResponse> => {
+      return this.client.sendOperationRequest(args, spec);
+    };
+    const sendOperationFn = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec,
+    ) => {
+      let currentRawResponse: coreClient.FullOperationResponse | undefined =
+        undefined;
+      const providedCallback = args.options?.onResponse;
+      const callback: coreClient.RawResponseCallback = (
+        rawResponse: coreClient.FullOperationResponse,
+        flatResponse: unknown,
+      ) => {
+        currentRawResponse = rawResponse;
+        providedCallback?.(rawResponse, flatResponse);
+      };
+      const updatedArgs = {
+        ...args,
+        options: {
+          ...args.options,
+          onResponse: callback,
+        },
+      };
+      const flatResponse = await directSendOperation(updatedArgs, spec);
+      return {
+        flatResponse,
+        rawResponse: {
+          statusCode: currentRawResponse!.status,
+          body: currentRawResponse!.parsedBody,
+          headers: currentRawResponse!.headers.toJSON(),
+        },
+      };
+    };
+
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: {
+        resourceGroupName,
+        accountName,
+        databaseName,
+        collectionName,
+        createUpdateMongoDBCollectionParameters,
+        options,
+      },
+      spec: createUpdateMongoDBCollectionOperationSpec,
+    });
+    const poller = await createHttpPoller<
+      MongoDBResourcesCreateUpdateMongoDBCollectionResponse,
+      OperationState<MongoDBResourcesCreateUpdateMongoDBCollectionResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
+    });
+    await poller.poll();
+    return poller;
+  }
+
+  /**
+   * Create or update an Azure Cosmos DB MongoDB Collection
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param accountName Cosmos DB database account name.
+   * @param databaseName Cosmos DB database name.
+   * @param collectionName Cosmos DB collection name.
+   * @param createUpdateMongoDBCollectionParameters The parameters to provide for the current MongoDB
+   *                                                Collection.
+   * @param options The options parameters.
+   */
+  async beginCreateUpdateMongoDBCollectionAndWait(
+    resourceGroupName: string,
+    accountName: string,
+    databaseName: string,
+    collectionName: string,
+    createUpdateMongoDBCollectionParameters: MongoDBCollectionCreateUpdateParameters,
+    options?: MongoDBResourcesCreateUpdateMongoDBCollectionOptionalParams,
+  ): Promise<MongoDBResourcesCreateUpdateMongoDBCollectionResponse> {
+    const poller = await this.beginCreateUpdateMongoDBCollection(
+      resourceGroupName,
+      accountName,
+      databaseName,
+      collectionName,
+      createUpdateMongoDBCollectionParameters,
+      options,
+    );
+    return poller.pollUntilDone();
+  }
+
+  /**
+   * Deletes an existing Azure Cosmos DB MongoDB Collection.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param accountName Cosmos DB database account name.
+   * @param databaseName Cosmos DB database name.
+   * @param collectionName Cosmos DB collection name.
+   * @param options The options parameters.
+   */
+  async beginDeleteMongoDBCollection(
+    resourceGroupName: string,
+    accountName: string,
+    databaseName: string,
+    collectionName: string,
+    options?: MongoDBResourcesDeleteMongoDBCollectionOptionalParams,
+  ): Promise<
+    SimplePollerLike<
+      OperationState<MongoDBResourcesDeleteMongoDBCollectionResponse>,
+      MongoDBResourcesDeleteMongoDBCollectionResponse
+    >
+  > {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec,
+    ): Promise<MongoDBResourcesDeleteMongoDBCollectionResponse> => {
+      return this.client.sendOperationRequest(args, spec);
+    };
+    const sendOperationFn = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec,
+    ) => {
+      let currentRawResponse: coreClient.FullOperationResponse | undefined =
+        undefined;
+      const providedCallback = args.options?.onResponse;
+      const callback: coreClient.RawResponseCallback = (
+        rawResponse: coreClient.FullOperationResponse,
+        flatResponse: unknown,
+      ) => {
+        currentRawResponse = rawResponse;
+        providedCallback?.(rawResponse, flatResponse);
+      };
+      const updatedArgs = {
+        ...args,
+        options: {
+          ...args.options,
+          onResponse: callback,
+        },
+      };
+      const flatResponse = await directSendOperation(updatedArgs, spec);
+      return {
+        flatResponse,
+        rawResponse: {
+          statusCode: currentRawResponse!.status,
+          body: currentRawResponse!.parsedBody,
+          headers: currentRawResponse!.headers.toJSON(),
+        },
+      };
+    };
+
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: {
+        resourceGroupName,
+        accountName,
+        databaseName,
+        collectionName,
+        options,
+      },
+      spec: deleteMongoDBCollectionOperationSpec,
+    });
+    const poller = await createHttpPoller<
+      MongoDBResourcesDeleteMongoDBCollectionResponse,
+      OperationState<MongoDBResourcesDeleteMongoDBCollectionResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
+    });
+    await poller.poll();
+    return poller;
+  }
+
+  /**
+   * Deletes an existing Azure Cosmos DB MongoDB Collection.
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
    * @param accountName Cosmos DB database account name.
    * @param databaseName Cosmos DB database name.
    * @param collectionName Cosmos DB collection name.
    * @param options The options parameters.
    */
-  getMongoDBCollection(
+  async beginDeleteMongoDBCollectionAndWait(
     resourceGroupName: string,
     accountName: string,
     databaseName: string,
     collectionName: string,
-    options?: MongoDBResourcesGetMongoDBCollectionOptionalParams,
-  ): Promise<MongoDBResourcesGetMongoDBCollectionResponse> {
-    return this.client.sendOperationRequest(
-      { resourceGroupName, accountName, databaseName, collectionName, options },
-      getMongoDBCollectionOperationSpec,
+    options?: MongoDBResourcesDeleteMongoDBCollectionOptionalParams,
+  ): Promise<MongoDBResourcesDeleteMongoDBCollectionResponse> {
+    const poller = await this.beginDeleteMongoDBCollection(
+      resourceGroupName,
+      accountName,
+      databaseName,
+      collectionName,
+      options,
     );
+    return poller.pollUntilDone();
   }
 
   /**
-   * Create or update an Azure Cosmos DB MongoDB Collection
+   * Merges the partitions of a MongoDB database
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
    * @param accountName Cosmos DB database account name.
    * @param databaseName Cosmos DB database name.
-   * @param collectionName Cosmos DB collection name.
-   * @param createUpdateMongoDBCollectionParameters The parameters to provide for the current MongoDB
-   *                                                Collection.
+   * @param mergeParameters The parameters for the merge operation.
    * @param options The options parameters.
    */
-  async beginCreateUpdateMongoDBCollection(
+  async beginMongoDBDatabasePartitionMerge(
     resourceGroupName: string,
     accountName: string,
     databaseName: string,
-    collectionName: string,
-    createUpdateMongoDBCollectionParameters: MongoDBCollectionCreateUpdateParameters,
-    options?: MongoDBResourcesCreateUpdateMongoDBCollectionOptionalParams,
+    mergeParameters: MergeParameters,
+    options?: MongoDBResourcesMongoDBDatabasePartitionMergeOptionalParams,
   ): Promise<
     SimplePollerLike<
-      OperationState<MongoDBResourcesCreateUpdateMongoDBCollectionResponse>,
-      MongoDBResourcesCreateUpdateMongoDBCollectionResponse
+      OperationState<MongoDBResourcesMongoDBDatabasePartitionMergeResponse>,
+      MongoDBResourcesMongoDBDatabasePartitionMergeResponse
     >
   > {
     const directSendOperation = async (
       args: coreClient.OperationArguments,
       spec: coreClient.OperationSpec,
-    ): Promise<MongoDBResourcesCreateUpdateMongoDBCollectionResponse> => {
+    ): Promise<MongoDBResourcesMongoDBDatabasePartitionMergeResponse> => {
       return this.client.sendOperationRequest(args, spec);
     };
     const sendOperationFn = async (
@@ -1024,76 +1700,74 @@ export class MongoDBResourcesImpl implements MongoDBResources {
         resourceGroupName,
         accountName,
         databaseName,
-        collectionName,
-        createUpdateMongoDBCollectionParameters,
+        mergeParameters,
         options,
       },
-      spec: createUpdateMongoDBCollectionOperationSpec,
+      spec: mongoDBDatabasePartitionMergeOperationSpec,
     });
     const poller = await createHttpPoller<
-      MongoDBResourcesCreateUpdateMongoDBCollectionResponse,
-      OperationState<MongoDBResourcesCreateUpdateMongoDBCollectionResponse>
+      MongoDBResourcesMongoDBDatabasePartitionMergeResponse,
+      OperationState<MongoDBResourcesMongoDBDatabasePartitionMergeResponse>
     >(lro, {
       restoreFrom: options?.resumeFrom,
       intervalInMs: options?.updateIntervalInMs,
+      resourceLocationConfig: "location",
     });
     await poller.poll();
     return poller;
   }
 
   /**
-   * Create or update an Azure Cosmos DB MongoDB Collection
+   * Merges the partitions of a MongoDB database
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
    * @param accountName Cosmos DB database account name.
    * @param databaseName Cosmos DB database name.
-   * @param collectionName Cosmos DB collection name.
-   * @param createUpdateMongoDBCollectionParameters The parameters to provide for the current MongoDB
-   *                                                Collection.
+   * @param mergeParameters The parameters for the merge operation.
    * @param options The options parameters.
    */
-  async beginCreateUpdateMongoDBCollectionAndWait(
+  async beginMongoDBDatabasePartitionMergeAndWait(
     resourceGroupName: string,
     accountName: string,
     databaseName: string,
-    collectionName: string,
-    createUpdateMongoDBCollectionParameters: MongoDBCollectionCreateUpdateParameters,
-    options?: MongoDBResourcesCreateUpdateMongoDBCollectionOptionalParams,
-  ): Promise<MongoDBResourcesCreateUpdateMongoDBCollectionResponse> {
-    const poller = await this.beginCreateUpdateMongoDBCollection(
+    mergeParameters: MergeParameters,
+    options?: MongoDBResourcesMongoDBDatabasePartitionMergeOptionalParams,
+  ): Promise<MongoDBResourcesMongoDBDatabasePartitionMergeResponse> {
+    const poller = await this.beginMongoDBDatabasePartitionMerge(
       resourceGroupName,
       accountName,
       databaseName,
-      collectionName,
-      createUpdateMongoDBCollectionParameters,
+      mergeParameters,
       options,
     );
     return poller.pollUntilDone();
   }
 
   /**
-   * Deletes an existing Azure Cosmos DB MongoDB Collection.
+   * Merges the partitions of a MongoDB Collection
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
    * @param accountName Cosmos DB database account name.
    * @param databaseName Cosmos DB database name.
    * @param collectionName Cosmos DB collection name.
+   * @param mergeParameters The parameters for the merge operation.
    * @param options The options parameters.
    */
-  async beginDeleteMongoDBCollection(
+  async beginListMongoDBCollectionPartitionMerge(
     resourceGroupName: string,
     accountName: string,
     databaseName: string,
     collectionName: string,
-    options?: MongoDBResourcesDeleteMongoDBCollectionOptionalParams,
+    mergeParameters: MergeParameters,
+    options?: MongoDBResourcesListMongoDBCollectionPartitionMergeOptionalParams,
   ): Promise<
     SimplePollerLike<
-      OperationState<MongoDBResourcesDeleteMongoDBCollectionResponse>,
-      MongoDBResourcesDeleteMongoDBCollectionResponse
+      OperationState<MongoDBResourcesListMongoDBCollectionPartitionMergeResponse>,
+      MongoDBResourcesListMongoDBCollectionPartitionMergeResponse
     >
   > {
     const directSendOperation = async (
       args: coreClient.OperationArguments,
       spec: coreClient.OperationSpec,
-    ): Promise<MongoDBResourcesDeleteMongoDBCollectionResponse> => {
+    ): Promise<MongoDBResourcesListMongoDBCollectionPartitionMergeResponse> => {
       return this.client.sendOperationRequest(args, spec);
     };
     const sendOperationFn = async (
@@ -1135,41 +1809,46 @@ export class MongoDBResourcesImpl implements MongoDBResources {
         accountName,
         databaseName,
         collectionName,
+        mergeParameters,
         options,
       },
-      spec: deleteMongoDBCollectionOperationSpec,
+      spec: listMongoDBCollectionPartitionMergeOperationSpec,
     });
     const poller = await createHttpPoller<
-      MongoDBResourcesDeleteMongoDBCollectionResponse,
-      OperationState<MongoDBResourcesDeleteMongoDBCollectionResponse>
+      MongoDBResourcesListMongoDBCollectionPartitionMergeResponse,
+      OperationState<MongoDBResourcesListMongoDBCollectionPartitionMergeResponse>
     >(lro, {
       restoreFrom: options?.resumeFrom,
       intervalInMs: options?.updateIntervalInMs,
+      resourceLocationConfig: "location",
     });
     await poller.poll();
     return poller;
   }
 
   /**
-   * Deletes an existing Azure Cosmos DB MongoDB Collection.
+   * Merges the partitions of a MongoDB Collection
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
    * @param accountName Cosmos DB database account name.
    * @param databaseName Cosmos DB database name.
    * @param collectionName Cosmos DB collection name.
+   * @param mergeParameters The parameters for the merge operation.
    * @param options The options parameters.
    */
-  async beginDeleteMongoDBCollectionAndWait(
+  async beginListMongoDBCollectionPartitionMergeAndWait(
     resourceGroupName: string,
     accountName: string,
     databaseName: string,
     collectionName: string,
-    options?: MongoDBResourcesDeleteMongoDBCollectionOptionalParams,
-  ): Promise<MongoDBResourcesDeleteMongoDBCollectionResponse> {
-    const poller = await this.beginDeleteMongoDBCollection(
+    mergeParameters: MergeParameters,
+    options?: MongoDBResourcesListMongoDBCollectionPartitionMergeOptionalParams,
+  ): Promise<MongoDBResourcesListMongoDBCollectionPartitionMergeResponse> {
+    const poller = await this.beginListMongoDBCollectionPartitionMerge(
       resourceGroupName,
       accountName,
       databaseName,
       collectionName,
+      mergeParameters,
       options,
     );
     return poller.pollUntilDone();
@@ -2303,6 +2982,144 @@ const migrateMongoDBDatabaseToManualThroughputOperationSpec: coreClient.Operatio
     headerParameters: [Parameters.accept],
     serializer,
   };
+const mongoDBDatabaseRetrieveThroughputDistributionOperationSpec: coreClient.OperationSpec =
+  {
+    path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/mongodbDatabases/{databaseName}/throughputSettings/default/retrieveThroughputDistribution",
+    httpMethod: "POST",
+    responses: {
+      200: {
+        bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
+      },
+      201: {
+        bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
+      },
+      202: {
+        bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
+      },
+      204: {
+        bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
+      },
+      default: {
+        bodyMapper: Mappers.ErrorResponse,
+      },
+    },
+    requestBody: Parameters.retrieveThroughputParameters,
+    queryParameters: [Parameters.apiVersion],
+    urlParameters: [
+      Parameters.$host,
+      Parameters.subscriptionId,
+      Parameters.resourceGroupName,
+      Parameters.accountName,
+      Parameters.databaseName,
+    ],
+    headerParameters: [Parameters.accept, Parameters.contentType],
+    mediaType: "json",
+    serializer,
+  };
+const mongoDBDatabaseRedistributeThroughputOperationSpec: coreClient.OperationSpec =
+  {
+    path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/mongodbDatabases/{databaseName}/throughputSettings/default/redistributeThroughput",
+    httpMethod: "POST",
+    responses: {
+      200: {
+        bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
+      },
+      201: {
+        bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
+      },
+      202: {
+        bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
+      },
+      204: {
+        bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
+      },
+      default: {
+        bodyMapper: Mappers.ErrorResponse,
+      },
+    },
+    requestBody: Parameters.redistributeThroughputParameters,
+    queryParameters: [Parameters.apiVersion],
+    urlParameters: [
+      Parameters.$host,
+      Parameters.subscriptionId,
+      Parameters.resourceGroupName,
+      Parameters.accountName,
+      Parameters.databaseName,
+    ],
+    headerParameters: [Parameters.accept, Parameters.contentType],
+    mediaType: "json",
+    serializer,
+  };
+const mongoDBContainerRetrieveThroughputDistributionOperationSpec: coreClient.OperationSpec =
+  {
+    path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/mongodbDatabases/{databaseName}/collections/{collectionName}/throughputSettings/default/retrieveThroughputDistribution",
+    httpMethod: "POST",
+    responses: {
+      200: {
+        bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
+      },
+      201: {
+        bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
+      },
+      202: {
+        bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
+      },
+      204: {
+        bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
+      },
+      default: {
+        bodyMapper: Mappers.ErrorResponse,
+      },
+    },
+    requestBody: Parameters.retrieveThroughputParameters,
+    queryParameters: [Parameters.apiVersion],
+    urlParameters: [
+      Parameters.$host,
+      Parameters.subscriptionId,
+      Parameters.resourceGroupName,
+      Parameters.accountName,
+      Parameters.databaseName,
+      Parameters.collectionName,
+    ],
+    headerParameters: [Parameters.accept, Parameters.contentType],
+    mediaType: "json",
+    serializer,
+  };
+const mongoDBContainerRedistributeThroughputOperationSpec: coreClient.OperationSpec =
+  {
+    path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/mongodbDatabases/{databaseName}/collections/{collectionName}/throughputSettings/default/redistributeThroughput",
+    httpMethod: "POST",
+    responses: {
+      200: {
+        bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
+      },
+      201: {
+        bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
+      },
+      202: {
+        bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
+      },
+      204: {
+        bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
+      },
+      default: {
+        bodyMapper: Mappers.ErrorResponse,
+      },
+    },
+    requestBody: Parameters.redistributeThroughputParameters,
+    queryParameters: [Parameters.apiVersion],
+    urlParameters: [
+      Parameters.$host,
+      Parameters.subscriptionId,
+      Parameters.resourceGroupName,
+      Parameters.accountName,
+      Parameters.databaseName,
+      Parameters.collectionName,
+    ],
+    headerParameters: [Parameters.accept, Parameters.contentType],
+    mediaType: "json",
+    serializer,
+  };
 const listMongoDBCollectionsOperationSpec: coreClient.OperationSpec = {
   path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/mongodbDatabases/{databaseName}/collections",
   httpMethod: "GET",
@@ -2401,6 +3218,74 @@ const deleteMongoDBCollectionOperationSpec: coreClient.OperationSpec = {
   ],
   serializer,
 };
+const mongoDBDatabasePartitionMergeOperationSpec: coreClient.OperationSpec = {
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/mongodbDatabases/{databaseName}/partitionMerge",
+  httpMethod: "POST",
+  responses: {
+    200: {
+      bodyMapper: Mappers.PhysicalPartitionStorageInfoCollection,
+    },
+    201: {
+      bodyMapper: Mappers.PhysicalPartitionStorageInfoCollection,
+    },
+    202: {
+      bodyMapper: Mappers.PhysicalPartitionStorageInfoCollection,
+    },
+    204: {
+      bodyMapper: Mappers.PhysicalPartitionStorageInfoCollection,
+    },
+    default: {
+      bodyMapper: Mappers.ErrorResponse,
+    },
+  },
+  requestBody: Parameters.mergeParameters,
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.resourceGroupName,
+    Parameters.accountName,
+    Parameters.databaseName,
+  ],
+  headerParameters: [Parameters.accept, Parameters.contentType],
+  mediaType: "json",
+  serializer,
+};
+const listMongoDBCollectionPartitionMergeOperationSpec: coreClient.OperationSpec =
+  {
+    path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/mongodbDatabases/{databaseName}/collections/{collectionName}/partitionMerge",
+    httpMethod: "POST",
+    responses: {
+      200: {
+        bodyMapper: Mappers.PhysicalPartitionStorageInfoCollection,
+      },
+      201: {
+        bodyMapper: Mappers.PhysicalPartitionStorageInfoCollection,
+      },
+      202: {
+        bodyMapper: Mappers.PhysicalPartitionStorageInfoCollection,
+      },
+      204: {
+        bodyMapper: Mappers.PhysicalPartitionStorageInfoCollection,
+      },
+      default: {
+        bodyMapper: Mappers.ErrorResponse,
+      },
+    },
+    requestBody: Parameters.mergeParameters,
+    queryParameters: [Parameters.apiVersion],
+    urlParameters: [
+      Parameters.$host,
+      Parameters.subscriptionId,
+      Parameters.resourceGroupName,
+      Parameters.accountName,
+      Parameters.databaseName,
+      Parameters.collectionName,
+    ],
+    headerParameters: [Parameters.accept, Parameters.contentType],
+    mediaType: "json",
+    serializer,
+  };
 const getMongoDBCollectionThroughputOperationSpec: coreClient.OperationSpec = {
   path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/mongodbDatabases/{databaseName}/collections/{collectionName}/throughputSettings/default",
   httpMethod: "GET",