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

package/dist-esm/src/operations/sqlResources.js

@@ -72,14 +72,14 @@ export class SqlResourcesImpl {
         });
     }
     /**
-     * Lists the SQL container under an existing Azure Cosmos DB database account.
+     * Lists the ClientEncryptionKeys under an existing Azure Cosmos DB SQL 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 options The options parameters.
      */
-    listSqlContainers(resourceGroupName, accountName, databaseName, options) {
-        const iter = this.listSqlContainersPagingAll(resourceGroupName, accountName, databaseName, options);
+    listClientEncryptionKeys(resourceGroupName, accountName, databaseName, options) {
+        const iter = this.listClientEncryptionKeysPagingAll(resourceGroupName, accountName, databaseName, options);
         return {
             next() {
                 return iter.next();
@@ -91,22 +91,22 @@ export class SqlResourcesImpl {
                 if (settings === null || settings === void 0 ? void 0 : settings.maxPageSize) {
                     throw new Error("maxPageSize is not supported by this operation.");
                 }
-                return this.listSqlContainersPagingPage(resourceGroupName, accountName, databaseName, options, settings);
+                return this.listClientEncryptionKeysPagingPage(resourceGroupName, accountName, databaseName, options, settings);
             },
         };
     }
-    listSqlContainersPagingPage(resourceGroupName, accountName, databaseName, options, _settings) {
-        return __asyncGenerator(this, arguments, function* listSqlContainersPagingPage_1() {
+    listClientEncryptionKeysPagingPage(resourceGroupName, accountName, databaseName, options, _settings) {
+        return __asyncGenerator(this, arguments, function* listClientEncryptionKeysPagingPage_1() {
             let result;
-            result = yield __await(this._listSqlContainers(resourceGroupName, accountName, databaseName, options));
+            result = yield __await(this._listClientEncryptionKeys(resourceGroupName, accountName, databaseName, options));
             yield yield __await(result.value || []);
         });
     }
-    listSqlContainersPagingAll(resourceGroupName, accountName, databaseName, options) {
-        return __asyncGenerator(this, arguments, function* listSqlContainersPagingAll_1() {
+    listClientEncryptionKeysPagingAll(resourceGroupName, accountName, databaseName, options) {
+        return __asyncGenerator(this, arguments, function* listClientEncryptionKeysPagingAll_1() {
             var _a, e_2, _b, _c;
             try {
-                for (var _d = true, _e = __asyncValues(this.listSqlContainersPagingPage(resourceGroupName, accountName, databaseName, options)), _f; _f = yield __await(_e.next()), _a = _f.done, !_a; _d = true) {
+                for (var _d = true, _e = __asyncValues(this.listClientEncryptionKeysPagingPage(resourceGroupName, accountName, databaseName, options)), _f; _f = yield __await(_e.next()), _a = _f.done, !_a; _d = true) {
                     _c = _f.value;
                     _d = false;
                     const page = _c;
@@ -123,14 +123,14 @@ export class SqlResourcesImpl {
         });
     }
     /**
-     * Lists the ClientEncryptionKeys under an existing Azure Cosmos DB SQL database.
+     * Lists the SQL container 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 options The options parameters.
      */
-    listClientEncryptionKeys(resourceGroupName, accountName, databaseName, options) {
-        const iter = this.listClientEncryptionKeysPagingAll(resourceGroupName, accountName, databaseName, options);
+    listSqlContainers(resourceGroupName, accountName, databaseName, options) {
+        const iter = this.listSqlContainersPagingAll(resourceGroupName, accountName, databaseName, options);
         return {
             next() {
                 return iter.next();
@@ -142,22 +142,22 @@ export class SqlResourcesImpl {
                 if (settings === null || settings === void 0 ? void 0 : settings.maxPageSize) {
                     throw new Error("maxPageSize is not supported by this operation.");
                 }
-                return this.listClientEncryptionKeysPagingPage(resourceGroupName, accountName, databaseName, options, settings);
+                return this.listSqlContainersPagingPage(resourceGroupName, accountName, databaseName, options, settings);
             },
         };
     }
-    listClientEncryptionKeysPagingPage(resourceGroupName, accountName, databaseName, options, _settings) {
-        return __asyncGenerator(this, arguments, function* listClientEncryptionKeysPagingPage_1() {
+    listSqlContainersPagingPage(resourceGroupName, accountName, databaseName, options, _settings) {
+        return __asyncGenerator(this, arguments, function* listSqlContainersPagingPage_1() {
             let result;
-            result = yield __await(this._listClientEncryptionKeys(resourceGroupName, accountName, databaseName, options));
+            result = yield __await(this._listSqlContainers(resourceGroupName, accountName, databaseName, options));
             yield yield __await(result.value || []);
         });
     }
-    listClientEncryptionKeysPagingAll(resourceGroupName, accountName, databaseName, options) {
-        return __asyncGenerator(this, arguments, function* listClientEncryptionKeysPagingAll_1() {
+    listSqlContainersPagingAll(resourceGroupName, accountName, databaseName, options) {
+        return __asyncGenerator(this, arguments, function* listSqlContainersPagingAll_1() {
             var _a, e_3, _b, _c;
             try {
-                for (var _d = true, _e = __asyncValues(this.listClientEncryptionKeysPagingPage(resourceGroupName, accountName, databaseName, options)), _f; _f = yield __await(_e.next()), _a = _f.done, !_a; _d = true) {
+                for (var _d = true, _e = __asyncValues(this.listSqlContainersPagingPage(resourceGroupName, accountName, databaseName, options)), _f; _f = yield __await(_e.next()), _a = _f.done, !_a; _d = true) {
                     _c = _f.value;
                     _d = false;
                     const page = _c;
@@ -762,6 +762,105 @@ export class SqlResourcesImpl {
             return poller.pollUntilDone();
         });
     }
+    /**
+     * Lists the ClientEncryptionKeys under an existing Azure Cosmos DB SQL 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 options The options parameters.
+     */
+    _listClientEncryptionKeys(resourceGroupName, accountName, databaseName, options) {
+        return this.client.sendOperationRequest({ resourceGroupName, accountName, databaseName, options }, listClientEncryptionKeysOperationSpec);
+    }
+    /**
+     * Gets the ClientEncryptionKey under an existing Azure Cosmos DB SQL 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 clientEncryptionKeyName Cosmos DB ClientEncryptionKey name.
+     * @param options The options parameters.
+     */
+    getClientEncryptionKey(resourceGroupName, accountName, databaseName, clientEncryptionKeyName, options) {
+        return this.client.sendOperationRequest({
+            resourceGroupName,
+            accountName,
+            databaseName,
+            clientEncryptionKeyName,
+            options,
+        }, getClientEncryptionKeyOperationSpec);
+    }
+    /**
+     * Create or update a ClientEncryptionKey. This API is meant to be invoked via tools such as the Azure
+     * Powershell (instead of directly).
+     * @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 clientEncryptionKeyName Cosmos DB ClientEncryptionKey name.
+     * @param createUpdateClientEncryptionKeyParameters The parameters to provide for the client encryption
+     *                                                  key.
+     * @param options The options parameters.
+     */
+    beginCreateUpdateClientEncryptionKey(resourceGroupName, accountName, databaseName, clientEncryptionKeyName, createUpdateClientEncryptionKeyParameters, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const directSendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+                return this.client.sendOperationRequest(args, spec);
+            });
+            const sendOperationFn = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+                var _a;
+                let currentRawResponse = undefined;
+                const providedCallback = (_a = args.options) === null || _a === void 0 ? void 0 : _a.onResponse;
+                const callback = (rawResponse, flatResponse) => {
+                    currentRawResponse = rawResponse;
+                    providedCallback === null || providedCallback === void 0 ? void 0 : providedCallback(rawResponse, flatResponse);
+                };
+                const updatedArgs = Object.assign(Object.assign({}, args), { options: Object.assign(Object.assign({}, args.options), { onResponse: callback }) });
+                const flatResponse = yield directSendOperation(updatedArgs, spec);
+                return {
+                    flatResponse,
+                    rawResponse: {
+                        statusCode: currentRawResponse.status,
+                        body: currentRawResponse.parsedBody,
+                        headers: currentRawResponse.headers.toJSON(),
+                    },
+                };
+            });
+            const lro = createLroSpec({
+                sendOperationFn,
+                args: {
+                    resourceGroupName,
+                    accountName,
+                    databaseName,
+                    clientEncryptionKeyName,
+                    createUpdateClientEncryptionKeyParameters,
+                    options,
+                },
+                spec: createUpdateClientEncryptionKeyOperationSpec,
+            });
+            const poller = yield createHttpPoller(lro, {
+                restoreFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
+                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs,
+            });
+            yield poller.poll();
+            return poller;
+        });
+    }
+    /**
+     * Create or update a ClientEncryptionKey. This API is meant to be invoked via tools such as the Azure
+     * Powershell (instead of directly).
+     * @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 clientEncryptionKeyName Cosmos DB ClientEncryptionKey name.
+     * @param createUpdateClientEncryptionKeyParameters The parameters to provide for the client encryption
+     *                                                  key.
+     * @param options The options parameters.
+     */
+    beginCreateUpdateClientEncryptionKeyAndWait(resourceGroupName, accountName, databaseName, clientEncryptionKeyName, createUpdateClientEncryptionKeyParameters, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const poller = yield this.beginCreateUpdateClientEncryptionKey(resourceGroupName, accountName, databaseName, clientEncryptionKeyName, createUpdateClientEncryptionKeyParameters, options);
+            return poller.pollUntilDone();
+        });
+    }
     /**
      * Lists the SQL container under an existing Azure Cosmos DB database account.
      * @param resourceGroupName The name of the resource group. The name is case insensitive.
@@ -916,6 +1015,141 @@ export class SqlResourcesImpl {
             return poller.pollUntilDone();
         });
     }
+    /**
+     * Merges the partitions of a SQL 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 mergeParameters The parameters for the merge operation.
+     * @param options The options parameters.
+     */
+    beginSqlDatabasePartitionMerge(resourceGroupName, accountName, databaseName, mergeParameters, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const directSendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+                return this.client.sendOperationRequest(args, spec);
+            });
+            const sendOperationFn = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+                var _a;
+                let currentRawResponse = undefined;
+                const providedCallback = (_a = args.options) === null || _a === void 0 ? void 0 : _a.onResponse;
+                const callback = (rawResponse, flatResponse) => {
+                    currentRawResponse = rawResponse;
+                    providedCallback === null || providedCallback === void 0 ? void 0 : providedCallback(rawResponse, flatResponse);
+                };
+                const updatedArgs = Object.assign(Object.assign({}, args), { options: Object.assign(Object.assign({}, args.options), { onResponse: callback }) });
+                const flatResponse = yield directSendOperation(updatedArgs, spec);
+                return {
+                    flatResponse,
+                    rawResponse: {
+                        statusCode: currentRawResponse.status,
+                        body: currentRawResponse.parsedBody,
+                        headers: currentRawResponse.headers.toJSON(),
+                    },
+                };
+            });
+            const lro = createLroSpec({
+                sendOperationFn,
+                args: {
+                    resourceGroupName,
+                    accountName,
+                    databaseName,
+                    mergeParameters,
+                    options,
+                },
+                spec: sqlDatabasePartitionMergeOperationSpec,
+            });
+            const poller = yield createHttpPoller(lro, {
+                restoreFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
+                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs,
+                resourceLocationConfig: "location",
+            });
+            yield poller.poll();
+            return poller;
+        });
+    }
+    /**
+     * Merges the partitions of a SQL 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 mergeParameters The parameters for the merge operation.
+     * @param options The options parameters.
+     */
+    beginSqlDatabasePartitionMergeAndWait(resourceGroupName, accountName, databaseName, mergeParameters, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const poller = yield this.beginSqlDatabasePartitionMerge(resourceGroupName, accountName, databaseName, mergeParameters, options);
+            return poller.pollUntilDone();
+        });
+    }
+    /**
+     * Merges the partitions of a SQL 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 containerName Cosmos DB container name.
+     * @param mergeParameters The parameters for the merge operation.
+     * @param options The options parameters.
+     */
+    beginListSqlContainerPartitionMerge(resourceGroupName, accountName, databaseName, containerName, mergeParameters, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const directSendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+                return this.client.sendOperationRequest(args, spec);
+            });
+            const sendOperationFn = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+                var _a;
+                let currentRawResponse = undefined;
+                const providedCallback = (_a = args.options) === null || _a === void 0 ? void 0 : _a.onResponse;
+                const callback = (rawResponse, flatResponse) => {
+                    currentRawResponse = rawResponse;
+                    providedCallback === null || providedCallback === void 0 ? void 0 : providedCallback(rawResponse, flatResponse);
+                };
+                const updatedArgs = Object.assign(Object.assign({}, args), { options: Object.assign(Object.assign({}, args.options), { onResponse: callback }) });
+                const flatResponse = yield directSendOperation(updatedArgs, spec);
+                return {
+                    flatResponse,
+                    rawResponse: {
+                        statusCode: currentRawResponse.status,
+                        body: currentRawResponse.parsedBody,
+                        headers: currentRawResponse.headers.toJSON(),
+                    },
+                };
+            });
+            const lro = createLroSpec({
+                sendOperationFn,
+                args: {
+                    resourceGroupName,
+                    accountName,
+                    databaseName,
+                    containerName,
+                    mergeParameters,
+                    options,
+                },
+                spec: listSqlContainerPartitionMergeOperationSpec,
+            });
+            const poller = yield createHttpPoller(lro, {
+                restoreFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
+                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs,
+                resourceLocationConfig: "location",
+            });
+            yield poller.poll();
+            return poller;
+        });
+    }
+    /**
+     * Merges the partitions of a SQL 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 containerName Cosmos DB container name.
+     * @param mergeParameters The parameters for the merge operation.
+     * @param options The options parameters.
+     */
+    beginListSqlContainerPartitionMergeAndWait(resourceGroupName, accountName, databaseName, containerName, mergeParameters, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const poller = yield this.beginListSqlContainerPartitionMerge(resourceGroupName, accountName, databaseName, containerName, mergeParameters, options);
+            return poller.pollUntilDone();
+        });
+    }
     /**
      * Gets the RUs per second of the SQL container under an existing Azure Cosmos DB database account.
      * @param resourceGroupName The name of the resource group. The name is case insensitive.
@@ -1053,24 +1287,227 @@ export class SqlResourcesImpl {
      * @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 containerName Cosmos DB container name.
+     * @param containerName Cosmos DB container name.
+     * @param options The options parameters.
+     */
+    beginMigrateSqlContainerToAutoscaleAndWait(resourceGroupName, accountName, databaseName, containerName, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const poller = yield this.beginMigrateSqlContainerToAutoscale(resourceGroupName, accountName, databaseName, containerName, options);
+            return poller.pollUntilDone();
+        });
+    }
+    /**
+     * Migrate an Azure Cosmos DB SQL container from autoscale to manual throughput
+     * @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 containerName Cosmos DB container name.
+     * @param options The options parameters.
+     */
+    beginMigrateSqlContainerToManualThroughput(resourceGroupName, accountName, databaseName, containerName, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const directSendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+                return this.client.sendOperationRequest(args, spec);
+            });
+            const sendOperationFn = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+                var _a;
+                let currentRawResponse = undefined;
+                const providedCallback = (_a = args.options) === null || _a === void 0 ? void 0 : _a.onResponse;
+                const callback = (rawResponse, flatResponse) => {
+                    currentRawResponse = rawResponse;
+                    providedCallback === null || providedCallback === void 0 ? void 0 : providedCallback(rawResponse, flatResponse);
+                };
+                const updatedArgs = Object.assign(Object.assign({}, args), { options: Object.assign(Object.assign({}, args.options), { onResponse: callback }) });
+                const flatResponse = yield directSendOperation(updatedArgs, spec);
+                return {
+                    flatResponse,
+                    rawResponse: {
+                        statusCode: currentRawResponse.status,
+                        body: currentRawResponse.parsedBody,
+                        headers: currentRawResponse.headers.toJSON(),
+                    },
+                };
+            });
+            const lro = createLroSpec({
+                sendOperationFn,
+                args: {
+                    resourceGroupName,
+                    accountName,
+                    databaseName,
+                    containerName,
+                    options,
+                },
+                spec: migrateSqlContainerToManualThroughputOperationSpec,
+            });
+            const poller = yield createHttpPoller(lro, {
+                restoreFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
+                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs,
+            });
+            yield poller.poll();
+            return poller;
+        });
+    }
+    /**
+     * Migrate an Azure Cosmos DB SQL container from autoscale to manual throughput
+     * @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 containerName Cosmos DB container name.
+     * @param options The options parameters.
+     */
+    beginMigrateSqlContainerToManualThroughputAndWait(resourceGroupName, accountName, databaseName, containerName, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const poller = yield this.beginMigrateSqlContainerToManualThroughput(resourceGroupName, accountName, databaseName, containerName, options);
+            return poller.pollUntilDone();
+        });
+    }
+    /**
+     * Retrieve throughput distribution for an Azure Cosmos DB SQL 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 SQL database.
+     * @param options The options parameters.
+     */
+    beginSqlDatabaseRetrieveThroughputDistribution(resourceGroupName, accountName, databaseName, retrieveThroughputParameters, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const directSendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+                return this.client.sendOperationRequest(args, spec);
+            });
+            const sendOperationFn = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+                var _a;
+                let currentRawResponse = undefined;
+                const providedCallback = (_a = args.options) === null || _a === void 0 ? void 0 : _a.onResponse;
+                const callback = (rawResponse, flatResponse) => {
+                    currentRawResponse = rawResponse;
+                    providedCallback === null || providedCallback === void 0 ? void 0 : providedCallback(rawResponse, flatResponse);
+                };
+                const updatedArgs = Object.assign(Object.assign({}, args), { options: Object.assign(Object.assign({}, args.options), { onResponse: callback }) });
+                const flatResponse = yield 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: sqlDatabaseRetrieveThroughputDistributionOperationSpec,
+            });
+            const poller = yield createHttpPoller(lro, {
+                restoreFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
+                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs,
+                resourceLocationConfig: "location",
+            });
+            yield poller.poll();
+            return poller;
+        });
+    }
+    /**
+     * Retrieve throughput distribution for an Azure Cosmos DB SQL 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 SQL database.
+     * @param options The options parameters.
+     */
+    beginSqlDatabaseRetrieveThroughputDistributionAndWait(resourceGroupName, accountName, databaseName, retrieveThroughputParameters, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const poller = yield this.beginSqlDatabaseRetrieveThroughputDistribution(resourceGroupName, accountName, databaseName, retrieveThroughputParameters, options);
+            return poller.pollUntilDone();
+        });
+    }
+    /**
+     * Redistribute throughput for an Azure Cosmos DB SQL 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 SQL database.
+     * @param options The options parameters.
+     */
+    beginSqlDatabaseRedistributeThroughput(resourceGroupName, accountName, databaseName, redistributeThroughputParameters, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const directSendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+                return this.client.sendOperationRequest(args, spec);
+            });
+            const sendOperationFn = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+                var _a;
+                let currentRawResponse = undefined;
+                const providedCallback = (_a = args.options) === null || _a === void 0 ? void 0 : _a.onResponse;
+                const callback = (rawResponse, flatResponse) => {
+                    currentRawResponse = rawResponse;
+                    providedCallback === null || providedCallback === void 0 ? void 0 : providedCallback(rawResponse, flatResponse);
+                };
+                const updatedArgs = Object.assign(Object.assign({}, args), { options: Object.assign(Object.assign({}, args.options), { onResponse: callback }) });
+                const flatResponse = yield 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: sqlDatabaseRedistributeThroughputOperationSpec,
+            });
+            const poller = yield createHttpPoller(lro, {
+                restoreFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
+                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs,
+                resourceLocationConfig: "location",
+            });
+            yield poller.poll();
+            return poller;
+        });
+    }
+    /**
+     * Redistribute throughput for an Azure Cosmos DB SQL 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 SQL database.
      * @param options The options parameters.
      */
-    beginMigrateSqlContainerToAutoscaleAndWait(resourceGroupName, accountName, databaseName, containerName, options) {
+    beginSqlDatabaseRedistributeThroughputAndWait(resourceGroupName, accountName, databaseName, redistributeThroughputParameters, options) {
         return __awaiter(this, void 0, void 0, function* () {
-            const poller = yield this.beginMigrateSqlContainerToAutoscale(resourceGroupName, accountName, databaseName, containerName, options);
+            const poller = yield this.beginSqlDatabaseRedistributeThroughput(resourceGroupName, accountName, databaseName, redistributeThroughputParameters, options);
             return poller.pollUntilDone();
         });
     }
     /**
-     * Migrate an Azure Cosmos DB SQL container from autoscale to manual throughput
+     * Retrieve throughput distribution for an Azure Cosmos DB SQL 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 containerName Cosmos DB container name.
+     * @param retrieveThroughputParameters The parameters to provide for retrieving throughput distribution
+     *                                     for the current SQL container.
      * @param options The options parameters.
      */
-    beginMigrateSqlContainerToManualThroughput(resourceGroupName, accountName, databaseName, containerName, options) {
+    beginSqlContainerRetrieveThroughputDistribution(resourceGroupName, accountName, databaseName, containerName, retrieveThroughputParameters, options) {
         return __awaiter(this, void 0, void 0, function* () {
             const directSendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
                 return this.client.sendOperationRequest(args, spec);
@@ -1101,71 +1538,47 @@ export class SqlResourcesImpl {
                     accountName,
                     databaseName,
                     containerName,
+                    retrieveThroughputParameters,
                     options,
                 },
-                spec: migrateSqlContainerToManualThroughputOperationSpec,
+                spec: sqlContainerRetrieveThroughputDistributionOperationSpec,
             });
             const poller = yield createHttpPoller(lro, {
                 restoreFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
                 intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs,
+                resourceLocationConfig: "location",
             });
             yield poller.poll();
             return poller;
         });
     }
     /**
-     * Migrate an Azure Cosmos DB SQL container from autoscale to manual throughput
+     * Retrieve throughput distribution for an Azure Cosmos DB SQL 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 containerName Cosmos DB container name.
+     * @param retrieveThroughputParameters The parameters to provide for retrieving throughput distribution
+     *                                     for the current SQL container.
      * @param options The options parameters.
      */
-    beginMigrateSqlContainerToManualThroughputAndWait(resourceGroupName, accountName, databaseName, containerName, options) {
+    beginSqlContainerRetrieveThroughputDistributionAndWait(resourceGroupName, accountName, databaseName, containerName, retrieveThroughputParameters, options) {
         return __awaiter(this, void 0, void 0, function* () {
-            const poller = yield this.beginMigrateSqlContainerToManualThroughput(resourceGroupName, accountName, databaseName, containerName, options);
+            const poller = yield this.beginSqlContainerRetrieveThroughputDistribution(resourceGroupName, accountName, databaseName, containerName, retrieveThroughputParameters, options);
             return poller.pollUntilDone();
         });
     }
     /**
-     * Lists the ClientEncryptionKeys under an existing Azure Cosmos DB SQL 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 options The options parameters.
-     */
-    _listClientEncryptionKeys(resourceGroupName, accountName, databaseName, options) {
-        return this.client.sendOperationRequest({ resourceGroupName, accountName, databaseName, options }, listClientEncryptionKeysOperationSpec);
-    }
-    /**
-     * Gets the ClientEncryptionKey under an existing Azure Cosmos DB SQL 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 clientEncryptionKeyName Cosmos DB ClientEncryptionKey name.
-     * @param options The options parameters.
-     */
-    getClientEncryptionKey(resourceGroupName, accountName, databaseName, clientEncryptionKeyName, options) {
-        return this.client.sendOperationRequest({
-            resourceGroupName,
-            accountName,
-            databaseName,
-            clientEncryptionKeyName,
-            options,
-        }, getClientEncryptionKeyOperationSpec);
-    }
-    /**
-     * Create or update a ClientEncryptionKey. This API is meant to be invoked via tools such as the Azure
-     * Powershell (instead of directly).
+     * Redistribute throughput for an Azure Cosmos DB SQL 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 clientEncryptionKeyName Cosmos DB ClientEncryptionKey name.
-     * @param createUpdateClientEncryptionKeyParameters The parameters to provide for the client encryption
-     *                                                  key.
+     * @param containerName Cosmos DB container name.
+     * @param redistributeThroughputParameters The parameters to provide for redistributing throughput for
+     *                                         the current SQL container.
      * @param options The options parameters.
      */
-    beginCreateUpdateClientEncryptionKey(resourceGroupName, accountName, databaseName, clientEncryptionKeyName, createUpdateClientEncryptionKeyParameters, options) {
+    beginSqlContainerRedistributeThroughput(resourceGroupName, accountName, databaseName, containerName, redistributeThroughputParameters, options) {
         return __awaiter(this, void 0, void 0, function* () {
             const directSendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
                 return this.client.sendOperationRequest(args, spec);
@@ -1195,34 +1608,34 @@ export class SqlResourcesImpl {
                     resourceGroupName,
                     accountName,
                     databaseName,
-                    clientEncryptionKeyName,
-                    createUpdateClientEncryptionKeyParameters,
+                    containerName,
+                    redistributeThroughputParameters,
                     options,
                 },
-                spec: createUpdateClientEncryptionKeyOperationSpec,
+                spec: sqlContainerRedistributeThroughputOperationSpec,
             });
             const poller = yield createHttpPoller(lro, {
                 restoreFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
                 intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs,
+                resourceLocationConfig: "location",
             });
             yield poller.poll();
             return poller;
         });
     }
     /**
-     * Create or update a ClientEncryptionKey. This API is meant to be invoked via tools such as the Azure
-     * Powershell (instead of directly).
+     * Redistribute throughput for an Azure Cosmos DB SQL 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 clientEncryptionKeyName Cosmos DB ClientEncryptionKey name.
-     * @param createUpdateClientEncryptionKeyParameters The parameters to provide for the client encryption
-     *                                                  key.
+     * @param containerName Cosmos DB container name.
+     * @param redistributeThroughputParameters The parameters to provide for redistributing throughput for
+     *                                         the current SQL container.
      * @param options The options parameters.
      */
-    beginCreateUpdateClientEncryptionKeyAndWait(resourceGroupName, accountName, databaseName, clientEncryptionKeyName, createUpdateClientEncryptionKeyParameters, options) {
+    beginSqlContainerRedistributeThroughputAndWait(resourceGroupName, accountName, databaseName, containerName, redistributeThroughputParameters, options) {
         return __awaiter(this, void 0, void 0, function* () {
-            const poller = yield this.beginCreateUpdateClientEncryptionKey(resourceGroupName, accountName, databaseName, clientEncryptionKeyName, createUpdateClientEncryptionKeyParameters, options);
+            const poller = yield this.beginSqlContainerRedistributeThroughput(resourceGroupName, accountName, databaseName, containerName, redistributeThroughputParameters, options);
             return poller.pollUntilDone();
         });
     }
@@ -2300,6 +2713,76 @@ const migrateSqlDatabaseToManualThroughputOperationSpec = {
     headerParameters: [Parameters.accept],
     serializer,
 };
+const listClientEncryptionKeysOperationSpec = {
+    path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/sqlDatabases/{databaseName}/clientEncryptionKeys",
+    httpMethod: "GET",
+    responses: {
+        200: {
+            bodyMapper: Mappers.ClientEncryptionKeysListResult,
+        },
+    },
+    queryParameters: [Parameters.apiVersion],
+    urlParameters: [
+        Parameters.$host,
+        Parameters.subscriptionId,
+        Parameters.resourceGroupName,
+        Parameters.accountName,
+        Parameters.databaseName,
+    ],
+    headerParameters: [Parameters.accept],
+    serializer,
+};
+const getClientEncryptionKeyOperationSpec = {
+    path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/sqlDatabases/{databaseName}/clientEncryptionKeys/{clientEncryptionKeyName}",
+    httpMethod: "GET",
+    responses: {
+        200: {
+            bodyMapper: Mappers.ClientEncryptionKeyGetResults,
+        },
+    },
+    queryParameters: [Parameters.apiVersion],
+    urlParameters: [
+        Parameters.$host,
+        Parameters.subscriptionId,
+        Parameters.resourceGroupName,
+        Parameters.accountName,
+        Parameters.databaseName,
+        Parameters.clientEncryptionKeyName,
+    ],
+    headerParameters: [Parameters.accept],
+    serializer,
+};
+const createUpdateClientEncryptionKeyOperationSpec = {
+    path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/sqlDatabases/{databaseName}/clientEncryptionKeys/{clientEncryptionKeyName}",
+    httpMethod: "PUT",
+    responses: {
+        200: {
+            bodyMapper: Mappers.ClientEncryptionKeyGetResults,
+        },
+        201: {
+            bodyMapper: Mappers.ClientEncryptionKeyGetResults,
+        },
+        202: {
+            bodyMapper: Mappers.ClientEncryptionKeyGetResults,
+        },
+        204: {
+            bodyMapper: Mappers.ClientEncryptionKeyGetResults,
+        },
+    },
+    requestBody: Parameters.createUpdateClientEncryptionKeyParameters,
+    queryParameters: [Parameters.apiVersion],
+    urlParameters: [
+        Parameters.$host,
+        Parameters.subscriptionId,
+        Parameters.resourceGroupName,
+        Parameters.accountName,
+        Parameters.databaseName,
+        Parameters.clientEncryptionKeyName,
+    ],
+    headerParameters: [Parameters.accept, Parameters.contentType],
+    mediaType: "json",
+    serializer,
+};
 const listSqlContainersOperationSpec = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/sqlDatabases/{databaseName}/containers",
     httpMethod: "GET",
@@ -2398,6 +2881,73 @@ const deleteSqlContainerOperationSpec = {
     ],
     serializer,
 };
+const sqlDatabasePartitionMergeOperationSpec = {
+    path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/sqlDatabases/{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 listSqlContainerPartitionMergeOperationSpec = {
+    path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/sqlDatabases/{databaseName}/containers/{containerName}/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.containerName,
+    ],
+    headerParameters: [Parameters.accept, Parameters.contentType],
+    mediaType: "json",
+    serializer,
+};
 const getSqlContainerThroughputOperationSpec = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/sqlDatabases/{databaseName}/containers/{containerName}/throughputSettings/default",
     httpMethod: "GET",
@@ -2513,14 +3063,27 @@ const migrateSqlContainerToManualThroughputOperationSpec = {
     headerParameters: [Parameters.accept],
     serializer,
 };
-const listClientEncryptionKeysOperationSpec = {
-    path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/sqlDatabases/{databaseName}/clientEncryptionKeys",
-    httpMethod: "GET",
+const sqlDatabaseRetrieveThroughputDistributionOperationSpec = {
+    path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/sqlDatabases/{databaseName}/throughputSettings/default/retrieveThroughputDistribution",
+    httpMethod: "POST",
     responses: {
         200: {
-            bodyMapper: Mappers.ClientEncryptionKeysListResult,
+            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,
@@ -2529,17 +3092,31 @@ const listClientEncryptionKeysOperationSpec = {
         Parameters.accountName,
         Parameters.databaseName,
     ],
-    headerParameters: [Parameters.accept],
+    headerParameters: [Parameters.accept, Parameters.contentType],
+    mediaType: "json",
     serializer,
 };
-const getClientEncryptionKeyOperationSpec = {
-    path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/sqlDatabases/{databaseName}/clientEncryptionKeys/{clientEncryptionKeyName}",
-    httpMethod: "GET",
+const sqlDatabaseRedistributeThroughputOperationSpec = {
+    path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/sqlDatabases/{databaseName}/throughputSettings/default/redistributeThroughput",
+    httpMethod: "POST",
     responses: {
         200: {
-            bodyMapper: Mappers.ClientEncryptionKeyGetResults,
+            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,
@@ -2547,29 +3124,32 @@ const getClientEncryptionKeyOperationSpec = {
         Parameters.resourceGroupName,
         Parameters.accountName,
         Parameters.databaseName,
-        Parameters.clientEncryptionKeyName,
     ],
-    headerParameters: [Parameters.accept],
+    headerParameters: [Parameters.accept, Parameters.contentType],
+    mediaType: "json",
     serializer,
 };
-const createUpdateClientEncryptionKeyOperationSpec = {
-    path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/sqlDatabases/{databaseName}/clientEncryptionKeys/{clientEncryptionKeyName}",
-    httpMethod: "PUT",
+const sqlContainerRetrieveThroughputDistributionOperationSpec = {
+    path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/sqlDatabases/{databaseName}/containers/{containerName}/throughputSettings/default/retrieveThroughputDistribution",
+    httpMethod: "POST",
     responses: {
         200: {
-            bodyMapper: Mappers.ClientEncryptionKeyGetResults,
+            bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
         },
         201: {
-            bodyMapper: Mappers.ClientEncryptionKeyGetResults,
+            bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
         },
         202: {
-            bodyMapper: Mappers.ClientEncryptionKeyGetResults,
+            bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
         },
         204: {
-            bodyMapper: Mappers.ClientEncryptionKeyGetResults,
+            bodyMapper: Mappers.PhysicalPartitionThroughputInfoResult,
+        },
+        default: {
+            bodyMapper: Mappers.ErrorResponse,
         },
     },
-    requestBody: Parameters.createUpdateClientEncryptionKeyParameters,
+    requestBody: Parameters.retrieveThroughputParameters,
     queryParameters: [Parameters.apiVersion],
     urlParameters: [
         Parameters.$host,
@@ -2577,7 +3157,41 @@ const createUpdateClientEncryptionKeyOperationSpec = {
         Parameters.resourceGroupName,
         Parameters.accountName,
         Parameters.databaseName,
-        Parameters.clientEncryptionKeyName,
+        Parameters.containerName,
+    ],
+    headerParameters: [Parameters.accept, Parameters.contentType],
+    mediaType: "json",
+    serializer,
+};
+const sqlContainerRedistributeThroughputOperationSpec = {
+    path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/sqlDatabases/{databaseName}/containers/{containerName}/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.containerName,
     ],
     headerParameters: [Parameters.accept, Parameters.contentType],
     mediaType: "json",