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

package/src/operations/graphResources.ts

@@ -0,0 +1,418 @@
+/*
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT License.
+ *
+ * Code generated by Microsoft (R) AutoRest Code Generator.
+ * Changes may cause incorrect behavior and will be lost if the code is regenerated.
+ */
+
+import { PagedAsyncIterableIterator, PageSettings } from "@azure/core-paging";
+import { GraphResources } from "../operationsInterfaces";
+import * as coreClient from "@azure/core-client";
+import * as Mappers from "../models/mappers";
+import * as Parameters from "../models/parameters";
+import { CosmosDBManagementClient } from "../cosmosDBManagementClient";
+import {
+  SimplePollerLike,
+  OperationState,
+  createHttpPoller,
+} from "@azure/core-lro";
+import { createLroSpec } from "../lroImpl";
+import {
+  GraphResourceGetResults,
+  GraphResourcesListGraphsOptionalParams,
+  GraphResourcesListGraphsResponse,
+  GraphResourcesGetGraphOptionalParams,
+  GraphResourcesGetGraphResponse,
+  GraphResourceCreateUpdateParameters,
+  GraphResourcesCreateUpdateGraphOptionalParams,
+  GraphResourcesCreateUpdateGraphResponse,
+  GraphResourcesDeleteGraphResourceOptionalParams,
+} from "../models";
+
+/// <reference lib="esnext.asynciterable" />
+/** Class containing GraphResources operations. */
+export class GraphResourcesImpl implements GraphResources {
+  private readonly client: CosmosDBManagementClient;
+
+  /**
+   * Initialize a new instance of the class GraphResources class.
+   * @param client Reference to the service client
+   */
+  constructor(client: CosmosDBManagementClient) {
+    this.client = client;
+  }
+
+  /**
+   * Lists the graphs 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 options The options parameters.
+   */
+  public listGraphs(
+    resourceGroupName: string,
+    accountName: string,
+    options?: GraphResourcesListGraphsOptionalParams,
+  ): PagedAsyncIterableIterator<GraphResourceGetResults> {
+    const iter = this.listGraphsPagingAll(
+      resourceGroupName,
+      accountName,
+      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.listGraphsPagingPage(
+          resourceGroupName,
+          accountName,
+          options,
+          settings,
+        );
+      },
+    };
+  }
+
+  private async *listGraphsPagingPage(
+    resourceGroupName: string,
+    accountName: string,
+    options?: GraphResourcesListGraphsOptionalParams,
+    _settings?: PageSettings,
+  ): AsyncIterableIterator<GraphResourceGetResults[]> {
+    let result: GraphResourcesListGraphsResponse;
+    result = await this._listGraphs(resourceGroupName, accountName, options);
+    yield result.value || [];
+  }
+
+  private async *listGraphsPagingAll(
+    resourceGroupName: string,
+    accountName: string,
+    options?: GraphResourcesListGraphsOptionalParams,
+  ): AsyncIterableIterator<GraphResourceGetResults> {
+    for await (const page of this.listGraphsPagingPage(
+      resourceGroupName,
+      accountName,
+      options,
+    )) {
+      yield* page;
+    }
+  }
+
+  /**
+   * Lists the graphs 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 options The options parameters.
+   */
+  private _listGraphs(
+    resourceGroupName: string,
+    accountName: string,
+    options?: GraphResourcesListGraphsOptionalParams,
+  ): Promise<GraphResourcesListGraphsResponse> {
+    return this.client.sendOperationRequest(
+      { resourceGroupName, accountName, options },
+      listGraphsOperationSpec,
+    );
+  }
+
+  /**
+   * Gets the Graph resource under an existing Azure Cosmos DB database account with the provided name.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param accountName Cosmos DB database account name.
+   * @param graphName Cosmos DB graph resource name.
+   * @param options The options parameters.
+   */
+  getGraph(
+    resourceGroupName: string,
+    accountName: string,
+    graphName: string,
+    options?: GraphResourcesGetGraphOptionalParams,
+  ): Promise<GraphResourcesGetGraphResponse> {
+    return this.client.sendOperationRequest(
+      { resourceGroupName, accountName, graphName, options },
+      getGraphOperationSpec,
+    );
+  }
+
+  /**
+   * Create or update an Azure Cosmos DB Graph.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param accountName Cosmos DB database account name.
+   * @param graphName Cosmos DB graph resource name.
+   * @param createUpdateGraphParameters The parameters to provide for the current graph.
+   * @param options The options parameters.
+   */
+  async beginCreateUpdateGraph(
+    resourceGroupName: string,
+    accountName: string,
+    graphName: string,
+    createUpdateGraphParameters: GraphResourceCreateUpdateParameters,
+    options?: GraphResourcesCreateUpdateGraphOptionalParams,
+  ): Promise<
+    SimplePollerLike<
+      OperationState<GraphResourcesCreateUpdateGraphResponse>,
+      GraphResourcesCreateUpdateGraphResponse
+    >
+  > {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec,
+    ): Promise<GraphResourcesCreateUpdateGraphResponse> => {
+      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,
+        graphName,
+        createUpdateGraphParameters,
+        options,
+      },
+      spec: createUpdateGraphOperationSpec,
+    });
+    const poller = await createHttpPoller<
+      GraphResourcesCreateUpdateGraphResponse,
+      OperationState<GraphResourcesCreateUpdateGraphResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
+    });
+    await poller.poll();
+    return poller;
+  }
+
+  /**
+   * Create or update an Azure Cosmos DB Graph.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param accountName Cosmos DB database account name.
+   * @param graphName Cosmos DB graph resource name.
+   * @param createUpdateGraphParameters The parameters to provide for the current graph.
+   * @param options The options parameters.
+   */
+  async beginCreateUpdateGraphAndWait(
+    resourceGroupName: string,
+    accountName: string,
+    graphName: string,
+    createUpdateGraphParameters: GraphResourceCreateUpdateParameters,
+    options?: GraphResourcesCreateUpdateGraphOptionalParams,
+  ): Promise<GraphResourcesCreateUpdateGraphResponse> {
+    const poller = await this.beginCreateUpdateGraph(
+      resourceGroupName,
+      accountName,
+      graphName,
+      createUpdateGraphParameters,
+      options,
+    );
+    return poller.pollUntilDone();
+  }
+
+  /**
+   * Deletes an existing Azure Cosmos DB Graph Resource.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param accountName Cosmos DB database account name.
+   * @param graphName Cosmos DB graph resource name.
+   * @param options The options parameters.
+   */
+  async beginDeleteGraphResource(
+    resourceGroupName: string,
+    accountName: string,
+    graphName: string,
+    options?: GraphResourcesDeleteGraphResourceOptionalParams,
+  ): Promise<SimplePollerLike<OperationState<void>, void>> {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec,
+    ): Promise<void> => {
+      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, graphName, options },
+      spec: deleteGraphResourceOperationSpec,
+    });
+    const poller = await createHttpPoller<void, OperationState<void>>(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
+    });
+    await poller.poll();
+    return poller;
+  }
+
+  /**
+   * Deletes an existing Azure Cosmos DB Graph Resource.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param accountName Cosmos DB database account name.
+   * @param graphName Cosmos DB graph resource name.
+   * @param options The options parameters.
+   */
+  async beginDeleteGraphResourceAndWait(
+    resourceGroupName: string,
+    accountName: string,
+    graphName: string,
+    options?: GraphResourcesDeleteGraphResourceOptionalParams,
+  ): Promise<void> {
+    const poller = await this.beginDeleteGraphResource(
+      resourceGroupName,
+      accountName,
+      graphName,
+      options,
+    );
+    return poller.pollUntilDone();
+  }
+}
+// Operation Specifications
+const serializer = coreClient.createSerializer(Mappers, /* isXml */ false);
+
+const listGraphsOperationSpec: coreClient.OperationSpec = {
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/graphs",
+  httpMethod: "GET",
+  responses: {
+    200: {
+      bodyMapper: Mappers.GraphResourcesListResult,
+    },
+  },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.resourceGroupName,
+    Parameters.accountName,
+  ],
+  headerParameters: [Parameters.accept],
+  serializer,
+};
+const getGraphOperationSpec: coreClient.OperationSpec = {
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/graphs/{graphName}",
+  httpMethod: "GET",
+  responses: {
+    200: {
+      bodyMapper: Mappers.GraphResourceGetResults,
+    },
+  },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.resourceGroupName,
+    Parameters.accountName,
+    Parameters.graphName,
+  ],
+  headerParameters: [Parameters.accept],
+  serializer,
+};
+const createUpdateGraphOperationSpec: coreClient.OperationSpec = {
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/graphs/{graphName}",
+  httpMethod: "PUT",
+  responses: {
+    200: {
+      bodyMapper: Mappers.GraphResourceGetResults,
+    },
+    201: {
+      bodyMapper: Mappers.GraphResourceGetResults,
+    },
+    202: {
+      bodyMapper: Mappers.GraphResourceGetResults,
+    },
+    204: {
+      bodyMapper: Mappers.GraphResourceGetResults,
+    },
+  },
+  requestBody: Parameters.createUpdateGraphParameters,
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.resourceGroupName,
+    Parameters.accountName,
+    Parameters.graphName,
+  ],
+  headerParameters: [Parameters.accept, Parameters.contentType],
+  mediaType: "json",
+  serializer,
+};
+const deleteGraphResourceOperationSpec: coreClient.OperationSpec = {
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/graphs/{graphName}",
+  httpMethod: "DELETE",
+  responses: { 200: {}, 201: {}, 202: {}, 204: {} },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.resourceGroupName,
+    Parameters.accountName,
+    Parameters.graphName,
+  ],
+  serializer,
+};

package/src/operations/gremlinResources.ts

@@ -1723,8 +1723,8 @@ const getGremlinGraphOperationSpec: coreClient.OperationSpec = {
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.accountName,
-    Parameters.databaseName,
     Parameters.graphName,
+    Parameters.databaseName,
   ],
   headerParameters: [Parameters.accept],
   serializer,
@@ -1753,8 +1753,8 @@ const createUpdateGremlinGraphOperationSpec: coreClient.OperationSpec = {
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.accountName,
-    Parameters.databaseName,
     Parameters.graphName,
+    Parameters.databaseName,
   ],
   headerParameters: [Parameters.accept, Parameters.contentType],
   mediaType: "json",
@@ -1783,8 +1783,8 @@ const deleteGremlinGraphOperationSpec: coreClient.OperationSpec = {
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.accountName,
-    Parameters.databaseName,
     Parameters.graphName,
+    Parameters.databaseName,
   ],
   serializer,
 };
@@ -1802,8 +1802,8 @@ const getGremlinGraphThroughputOperationSpec: coreClient.OperationSpec = {
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.accountName,
-    Parameters.databaseName,
     Parameters.graphName,
+    Parameters.databaseName,
   ],
   headerParameters: [Parameters.accept],
   serializer,
@@ -1832,8 +1832,8 @@ const updateGremlinGraphThroughputOperationSpec: coreClient.OperationSpec = {
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.accountName,
-    Parameters.databaseName,
     Parameters.graphName,
+    Parameters.databaseName,
   ],
   headerParameters: [Parameters.accept, Parameters.contentType],
   mediaType: "json",
@@ -1865,8 +1865,8 @@ const migrateGremlinGraphToAutoscaleOperationSpec: coreClient.OperationSpec = {
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.accountName,
-    Parameters.databaseName,
     Parameters.graphName,
+    Parameters.databaseName,
   ],
   headerParameters: [Parameters.accept],
   serializer,
@@ -1898,8 +1898,8 @@ const migrateGremlinGraphToManualThroughputOperationSpec: coreClient.OperationSp
       Parameters.subscriptionId,
       Parameters.resourceGroupName,
       Parameters.accountName,
-      Parameters.databaseName,
       Parameters.graphName,
+      Parameters.databaseName,
     ],
     headerParameters: [Parameters.accept],
     serializer,
@@ -1932,8 +1932,8 @@ const retrieveContinuousBackupInformationOperationSpec: coreClient.OperationSpec
       Parameters.subscriptionId,
       Parameters.resourceGroupName,
       Parameters.accountName,
-      Parameters.databaseName,
       Parameters.graphName,
+      Parameters.databaseName,
     ],
     headerParameters: [Parameters.accept, Parameters.contentType],
     mediaType: "json",

package/src/operations/index.ts

@@ -19,14 +19,17 @@ export * from "./collectionPartitionRegion";
 export * from "./collectionPartition";
 export * from "./partitionKeyRangeId";
 export * from "./partitionKeyRangeIdRegion";
+export * from "./graphResources";
 export * from "./sqlResources";
 export * from "./mongoDBResources";
 export * from "./tableResources";
 export * from "./cassandraResources";
 export * from "./gremlinResources";
 export * from "./locations";
+export * from "./dataTransferJobs";
 export * from "./cassandraClusters";
 export * from "./cassandraDataCenters";
+export * from "./networkSecurityPerimeterConfigurations";
 export * from "./notebookWorkspaces";
 export * from "./privateEndpointConnections";
 export * from "./privateLinkResources";
@@ -43,3 +46,8 @@ export * from "./restorableGremlinResources";
 export * from "./restorableTables";
 export * from "./restorableTableResources";
 export * from "./service";
+export * from "./throughputPools";
+export * from "./throughputPool";
+export * from "./throughputPoolAccounts";
+export * from "./throughputPoolAccount";
+export * from "./chaosFault";