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

package/src/operations/cassandraClusters.ts

@@ -24,6 +24,12 @@ import {
   CassandraClustersListBySubscriptionResponse,
   CassandraClustersListByResourceGroupOptionalParams,
   CassandraClustersListByResourceGroupResponse,
+  CommandPublicResource,
+  CassandraClustersListCommandOptionalParams,
+  CassandraClustersListCommandResponse,
+  BackupResource,
+  CassandraClustersListBackupsOptionalParams,
+  CassandraClustersListBackupsResponse,
   CassandraClustersGetOptionalParams,
   CassandraClustersGetResponse,
   CassandraClustersDeleteOptionalParams,
@@ -34,6 +40,12 @@ import {
   CommandPostBody,
   CassandraClustersInvokeCommandOptionalParams,
   CassandraClustersInvokeCommandResponse,
+  CassandraClustersInvokeCommandAsyncOptionalParams,
+  CassandraClustersInvokeCommandAsyncResponse,
+  CassandraClustersGetCommandAsyncOptionalParams,
+  CassandraClustersGetCommandAsyncResponse,
+  CassandraClustersGetBackupOptionalParams,
+  CassandraClustersGetBackupResponse,
   CassandraClustersDeallocateOptionalParams,
   CassandraClustersStartOptionalParams,
   CassandraClustersStatusOptionalParams,
@@ -146,6 +158,130 @@ export class CassandraClustersImpl implements CassandraClusters {
     }
   }
 
+  /**
+   * List all commands currently running on ring info
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param clusterName Managed Cassandra cluster name.
+   * @param options The options parameters.
+   */
+  public listCommand(
+    resourceGroupName: string,
+    clusterName: string,
+    options?: CassandraClustersListCommandOptionalParams,
+  ): PagedAsyncIterableIterator<CommandPublicResource> {
+    const iter = this.listCommandPagingAll(
+      resourceGroupName,
+      clusterName,
+      options,
+    );
+    return {
+      next() {
+        return iter.next();
+      },
+      [Symbol.asyncIterator]() {
+        return this;
+      },
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
+        return this.listCommandPagingPage(
+          resourceGroupName,
+          clusterName,
+          options,
+          settings,
+        );
+      },
+    };
+  }
+
+  private async *listCommandPagingPage(
+    resourceGroupName: string,
+    clusterName: string,
+    options?: CassandraClustersListCommandOptionalParams,
+    _settings?: PageSettings,
+  ): AsyncIterableIterator<CommandPublicResource[]> {
+    let result: CassandraClustersListCommandResponse;
+    result = await this._listCommand(resourceGroupName, clusterName, options);
+    yield result.value || [];
+  }
+
+  private async *listCommandPagingAll(
+    resourceGroupName: string,
+    clusterName: string,
+    options?: CassandraClustersListCommandOptionalParams,
+  ): AsyncIterableIterator<CommandPublicResource> {
+    for await (const page of this.listCommandPagingPage(
+      resourceGroupName,
+      clusterName,
+      options,
+    )) {
+      yield* page;
+    }
+  }
+
+  /**
+   * List the backups of this cluster that are available to restore.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param clusterName Managed Cassandra cluster name.
+   * @param options The options parameters.
+   */
+  public listBackups(
+    resourceGroupName: string,
+    clusterName: string,
+    options?: CassandraClustersListBackupsOptionalParams,
+  ): PagedAsyncIterableIterator<BackupResource> {
+    const iter = this.listBackupsPagingAll(
+      resourceGroupName,
+      clusterName,
+      options,
+    );
+    return {
+      next() {
+        return iter.next();
+      },
+      [Symbol.asyncIterator]() {
+        return this;
+      },
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
+        return this.listBackupsPagingPage(
+          resourceGroupName,
+          clusterName,
+          options,
+          settings,
+        );
+      },
+    };
+  }
+
+  private async *listBackupsPagingPage(
+    resourceGroupName: string,
+    clusterName: string,
+    options?: CassandraClustersListBackupsOptionalParams,
+    _settings?: PageSettings,
+  ): AsyncIterableIterator<BackupResource[]> {
+    let result: CassandraClustersListBackupsResponse;
+    result = await this._listBackups(resourceGroupName, clusterName, options);
+    yield result.value || [];
+  }
+
+  private async *listBackupsPagingAll(
+    resourceGroupName: string,
+    clusterName: string,
+    options?: CassandraClustersListBackupsOptionalParams,
+  ): AsyncIterableIterator<BackupResource> {
+    for await (const page of this.listBackupsPagingPage(
+      resourceGroupName,
+      clusterName,
+      options,
+    )) {
+      yield* page;
+    }
+  }
+
   /**
    * List all managed Cassandra clusters in this subscription.
    * @param options The options parameters.
@@ -556,6 +692,172 @@ export class CassandraClustersImpl implements CassandraClusters {
     return poller.pollUntilDone();
   }
 
+  /**
+   * Invoke a command like nodetool for cassandra maintenance asynchronously
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param clusterName Managed Cassandra cluster name.
+   * @param body Specification which command to run where
+   * @param options The options parameters.
+   */
+  async beginInvokeCommandAsync(
+    resourceGroupName: string,
+    clusterName: string,
+    body: CommandPostBody,
+    options?: CassandraClustersInvokeCommandAsyncOptionalParams,
+  ): Promise<
+    SimplePollerLike<
+      OperationState<CassandraClustersInvokeCommandAsyncResponse>,
+      CassandraClustersInvokeCommandAsyncResponse
+    >
+  > {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec,
+    ): Promise<CassandraClustersInvokeCommandAsyncResponse> => {
+      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, clusterName, body, options },
+      spec: invokeCommandAsyncOperationSpec,
+    });
+    const poller = await createHttpPoller<
+      CassandraClustersInvokeCommandAsyncResponse,
+      OperationState<CassandraClustersInvokeCommandAsyncResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
+    });
+    await poller.poll();
+    return poller;
+  }
+
+  /**
+   * Invoke a command like nodetool for cassandra maintenance asynchronously
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param clusterName Managed Cassandra cluster name.
+   * @param body Specification which command to run where
+   * @param options The options parameters.
+   */
+  async beginInvokeCommandAsyncAndWait(
+    resourceGroupName: string,
+    clusterName: string,
+    body: CommandPostBody,
+    options?: CassandraClustersInvokeCommandAsyncOptionalParams,
+  ): Promise<CassandraClustersInvokeCommandAsyncResponse> {
+    const poller = await this.beginInvokeCommandAsync(
+      resourceGroupName,
+      clusterName,
+      body,
+      options,
+    );
+    return poller.pollUntilDone();
+  }
+
+  /**
+   * List all commands currently running on ring info
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param clusterName Managed Cassandra cluster name.
+   * @param options The options parameters.
+   */
+  private _listCommand(
+    resourceGroupName: string,
+    clusterName: string,
+    options?: CassandraClustersListCommandOptionalParams,
+  ): Promise<CassandraClustersListCommandResponse> {
+    return this.client.sendOperationRequest(
+      { resourceGroupName, clusterName, options },
+      listCommandOperationSpec,
+    );
+  }
+
+  /**
+   * Get details about a specified command that was run asynchronously.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param clusterName Managed Cassandra cluster name.
+   * @param commandId Managed Cassandra cluster command id.
+   * @param options The options parameters.
+   */
+  getCommandAsync(
+    resourceGroupName: string,
+    clusterName: string,
+    commandId: string,
+    options?: CassandraClustersGetCommandAsyncOptionalParams,
+  ): Promise<CassandraClustersGetCommandAsyncResponse> {
+    return this.client.sendOperationRequest(
+      { resourceGroupName, clusterName, commandId, options },
+      getCommandAsyncOperationSpec,
+    );
+  }
+
+  /**
+   * List the backups of this cluster that are available to restore.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param clusterName Managed Cassandra cluster name.
+   * @param options The options parameters.
+   */
+  private _listBackups(
+    resourceGroupName: string,
+    clusterName: string,
+    options?: CassandraClustersListBackupsOptionalParams,
+  ): Promise<CassandraClustersListBackupsResponse> {
+    return this.client.sendOperationRequest(
+      { resourceGroupName, clusterName, options },
+      listBackupsOperationSpec,
+    );
+  }
+
+  /**
+   * Get the properties of an individual backup of this cluster that is available to restore.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param clusterName Managed Cassandra cluster name.
+   * @param backupId Id of a restorable backup of a Cassandra cluster.
+   * @param options The options parameters.
+   */
+  getBackup(
+    resourceGroupName: string,
+    clusterName: string,
+    backupId: string,
+    options?: CassandraClustersGetBackupOptionalParams,
+  ): Promise<CassandraClustersGetBackupResponse> {
+    return this.client.sendOperationRequest(
+      { resourceGroupName, clusterName, backupId, options },
+      getBackupOperationSpec,
+    );
+  }
+
   /**
    * Deallocate the Managed Cassandra Cluster and Associated Data Centers. Deallocation will deallocate
    * the host virtual machine of this cluster, and reserved the data disk. This won't do anything on an
@@ -921,6 +1223,124 @@ const invokeCommandOperationSpec: coreClient.OperationSpec = {
   mediaType: "json",
   serializer,
 };
+const invokeCommandAsyncOperationSpec: coreClient.OperationSpec = {
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/cassandraClusters/{clusterName}/invokeCommandAsync",
+  httpMethod: "POST",
+  responses: {
+    200: {
+      bodyMapper: Mappers.CommandPublicResource,
+    },
+    201: {
+      bodyMapper: Mappers.CommandPublicResource,
+    },
+    202: {
+      bodyMapper: Mappers.CommandPublicResource,
+    },
+    204: {
+      bodyMapper: Mappers.CommandPublicResource,
+    },
+    default: {
+      bodyMapper: Mappers.ErrorResponse,
+    },
+  },
+  requestBody: Parameters.body1,
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.resourceGroupName,
+    Parameters.clusterName,
+  ],
+  headerParameters: [Parameters.accept, Parameters.contentType],
+  mediaType: "json",
+  serializer,
+};
+const listCommandOperationSpec: coreClient.OperationSpec = {
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/cassandraClusters/{clusterName}/commands",
+  httpMethod: "GET",
+  responses: {
+    200: {
+      bodyMapper: Mappers.ListCommands,
+    },
+    default: {
+      bodyMapper: Mappers.ErrorResponse,
+    },
+  },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.resourceGroupName,
+    Parameters.clusterName,
+  ],
+  headerParameters: [Parameters.accept],
+  serializer,
+};
+const getCommandAsyncOperationSpec: coreClient.OperationSpec = {
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/cassandraClusters/{clusterName}/commands/{commandId}",
+  httpMethod: "GET",
+  responses: {
+    200: {
+      bodyMapper: Mappers.ListCommands,
+    },
+    default: {
+      bodyMapper: Mappers.ErrorResponse,
+    },
+  },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.resourceGroupName,
+    Parameters.clusterName,
+    Parameters.commandId,
+  ],
+  headerParameters: [Parameters.accept],
+  serializer,
+};
+const listBackupsOperationSpec: coreClient.OperationSpec = {
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/cassandraClusters/{clusterName}/backups",
+  httpMethod: "GET",
+  responses: {
+    200: {
+      bodyMapper: Mappers.ListBackups,
+    },
+    default: {
+      bodyMapper: Mappers.ErrorResponse,
+    },
+  },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.resourceGroupName,
+    Parameters.clusterName,
+  ],
+  headerParameters: [Parameters.accept],
+  serializer,
+};
+const getBackupOperationSpec: coreClient.OperationSpec = {
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/cassandraClusters/{clusterName}/backups/{backupId}",
+  httpMethod: "GET",
+  responses: {
+    200: {
+      bodyMapper: Mappers.BackupResource,
+    },
+    default: {
+      bodyMapper: Mappers.ErrorResponse,
+    },
+  },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.resourceGroupName,
+    Parameters.clusterName,
+    Parameters.backupId,
+  ],
+  headerParameters: [Parameters.accept],
+  serializer,
+};
 const deallocateOperationSpec: coreClient.OperationSpec = {
   path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/cassandraClusters/{clusterName}/deallocate",
   httpMethod: "POST",
@@ -940,7 +1360,7 @@ const deallocateOperationSpec: coreClient.OperationSpec = {
     Parameters.resourceGroupName,
     Parameters.clusterName,
   ],
-  headerParameters: [Parameters.accept],
+  headerParameters: [Parameters.accept, Parameters.xMsForceDeallocate],
   serializer,
 };
 const startOperationSpec: coreClient.OperationSpec = {