@azure/arm-storagecache 4.3.0 → 5.0.0

package/src/operations/storageTargetOperations.ts

@@ -3,212 +3,372 @@
  * 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.
+ * Changes may cause incorrect behavior and will be lost if the code is regenerated.
  */
 
-import * as msRest from "@azure/ms-rest-js";
-import * as msRestAzure from "@azure/ms-rest-azure-js";
-import * as Mappers from "../models/storageTargetOperationsMappers";
+import { StorageTargetOperations } from "../operationsInterfaces";
+import * as coreClient from "@azure/core-client";
+import * as Mappers from "../models/mappers";
 import * as Parameters from "../models/parameters";
-import { StorageCacheManagementClientContext } from "../storageCacheManagementClientContext";
+import { StorageCacheManagementClient } from "../storageCacheManagementClient";
+import { PollerLike, PollOperationState, LroEngine } from "@azure/core-lro";
+import { LroImpl } from "../lroImpl";
+import {
+  StorageTargetFlushOptionalParams,
+  StorageTargetSuspendOptionalParams,
+  StorageTargetResumeOptionalParams
+} from "../models";
 
-/** Class representing a StorageTargetOperations. */
-export class StorageTargetOperations {
-  private readonly client: StorageCacheManagementClientContext;
+/** Class containing StorageTargetOperations operations. */
+export class StorageTargetOperationsImpl implements StorageTargetOperations {
+  private readonly client: StorageCacheManagementClient;
 
   /**
-   * Create a StorageTargetOperations.
-   * @param {StorageCacheManagementClientContext} client Reference to the service client.
+   * Initialize a new instance of the class StorageTargetOperations class.
+   * @param client Reference to the service client
    */
-  constructor(client: StorageCacheManagementClientContext) {
+  constructor(client: StorageCacheManagementClient) {
     this.client = client;
   }
 
   /**
-   * Tells the cache to write all dirty data to the Storage Target's backend storage. Client requests
-   * to this storage target's namespace will return errors until the flush operation completes.
+   * Tells the cache to write all dirty data to the Storage Target's backend storage. Client requests to
+   * this storage target's namespace will return errors until the flush operation completes.
    * @param resourceGroupName Target resource group.
-   * @param cacheName Name of Cache. Length of name must not be greater than 80 and chars must be
-   * from the [-0-9a-zA-Z_] char class.
+   * @param cacheName Name of Cache. Length of name must not be greater than 80 and chars must be from
+   *                  the [-0-9a-zA-Z_] char class.
    * @param storageTargetName Name of Storage Target.
-   * @param [options] The optional parameters
-   * @returns Promise<msRest.RestResponse>
+   * @param options The options parameters.
    */
-  flush(resourceGroupName: string, cacheName: string, storageTargetName: string, options?: msRest.RequestOptionsBase): Promise<msRest.RestResponse> {
-    return this.beginFlush(resourceGroupName,cacheName,storageTargetName,options)
-      .then(lroPoller => lroPoller.pollUntilFinished());
+  async beginFlush(
+    resourceGroupName: string,
+    cacheName: string,
+    storageTargetName: string,
+    options?: StorageTargetFlushOptionalParams
+  ): Promise<PollerLike<PollOperationState<void>, void>> {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ): Promise<void> => {
+      return this.client.sendOperationRequest(args, spec);
+    };
+    const sendOperation = 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 = new LroImpl(
+      sendOperation,
+      { resourceGroupName, cacheName, storageTargetName, options },
+      flushOperationSpec
+    );
+    return new LroEngine(lro, {
+      resumeFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
+      lroResourceLocationConfig: "azure-async-operation"
+    });
   }
 
   /**
-   * Suspends client access to a storage target.
+   * Tells the cache to write all dirty data to the Storage Target's backend storage. Client requests to
+   * this storage target's namespace will return errors until the flush operation completes.
    * @param resourceGroupName Target resource group.
-   * @param cacheName Name of Cache. Length of name must not be greater than 80 and chars must be
-   * from the [-0-9a-zA-Z_] char class.
+   * @param cacheName Name of Cache. Length of name must not be greater than 80 and chars must be from
+   *                  the [-0-9a-zA-Z_] char class.
    * @param storageTargetName Name of Storage Target.
-   * @param [options] The optional parameters
-   * @returns Promise<msRest.RestResponse>
+   * @param options The options parameters.
    */
-  suspend(resourceGroupName: string, cacheName: string, storageTargetName: string, options?: msRest.RequestOptionsBase): Promise<msRest.RestResponse> {
-    return this.beginSuspend(resourceGroupName,cacheName,storageTargetName,options)
-      .then(lroPoller => lroPoller.pollUntilFinished());
+  async beginFlushAndWait(
+    resourceGroupName: string,
+    cacheName: string,
+    storageTargetName: string,
+    options?: StorageTargetFlushOptionalParams
+  ): Promise<void> {
+    const poller = await this.beginFlush(
+      resourceGroupName,
+      cacheName,
+      storageTargetName,
+      options
+    );
+    return poller.pollUntilDone();
   }
 
   /**
-   * Resumes client access to a previously suspended storage target.
+   * Suspends client access to a storage target.
    * @param resourceGroupName Target resource group.
-   * @param cacheName Name of Cache. Length of name must not be greater than 80 and chars must be
-   * from the [-0-9a-zA-Z_] char class.
+   * @param cacheName Name of Cache. Length of name must not be greater than 80 and chars must be from
+   *                  the [-0-9a-zA-Z_] char class.
    * @param storageTargetName Name of Storage Target.
-   * @param [options] The optional parameters
-   * @returns Promise<msRest.RestResponse>
+   * @param options The options parameters.
    */
-  resume(resourceGroupName: string, cacheName: string, storageTargetName: string, options?: msRest.RequestOptionsBase): Promise<msRest.RestResponse> {
-    return this.beginResume(resourceGroupName,cacheName,storageTargetName,options)
-      .then(lroPoller => lroPoller.pollUntilFinished());
+  async beginSuspend(
+    resourceGroupName: string,
+    cacheName: string,
+    storageTargetName: string,
+    options?: StorageTargetSuspendOptionalParams
+  ): Promise<PollerLike<PollOperationState<void>, void>> {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ): Promise<void> => {
+      return this.client.sendOperationRequest(args, spec);
+    };
+    const sendOperation = 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 = new LroImpl(
+      sendOperation,
+      { resourceGroupName, cacheName, storageTargetName, options },
+      suspendOperationSpec
+    );
+    return new LroEngine(lro, {
+      resumeFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
+      lroResourceLocationConfig: "azure-async-operation"
+    });
   }
 
   /**
-   * Tells the cache to write all dirty data to the Storage Target's backend storage. Client requests
-   * to this storage target's namespace will return errors until the flush operation completes.
+   * Suspends client access to a storage target.
    * @param resourceGroupName Target resource group.
-   * @param cacheName Name of Cache. Length of name must not be greater than 80 and chars must be
-   * from the [-0-9a-zA-Z_] char class.
+   * @param cacheName Name of Cache. Length of name must not be greater than 80 and chars must be from
+   *                  the [-0-9a-zA-Z_] char class.
    * @param storageTargetName Name of Storage Target.
-   * @param [options] The optional parameters
-   * @returns Promise<msRestAzure.LROPoller>
+   * @param options The options parameters.
    */
-  beginFlush(resourceGroupName: string, cacheName: string, storageTargetName: string, options?: msRest.RequestOptionsBase): Promise<msRestAzure.LROPoller> {
-    return this.client.sendLRORequest(
-      {
-        resourceGroupName,
-        cacheName,
-        storageTargetName,
-        options
-      },
-      beginFlushOperationSpec,
-      options);
+  async beginSuspendAndWait(
+    resourceGroupName: string,
+    cacheName: string,
+    storageTargetName: string,
+    options?: StorageTargetSuspendOptionalParams
+  ): Promise<void> {
+    const poller = await this.beginSuspend(
+      resourceGroupName,
+      cacheName,
+      storageTargetName,
+      options
+    );
+    return poller.pollUntilDone();
   }
 
   /**
-   * Suspends client access to a storage target.
+   * Resumes client access to a previously suspended storage target.
    * @param resourceGroupName Target resource group.
-   * @param cacheName Name of Cache. Length of name must not be greater than 80 and chars must be
-   * from the [-0-9a-zA-Z_] char class.
+   * @param cacheName Name of Cache. Length of name must not be greater than 80 and chars must be from
+   *                  the [-0-9a-zA-Z_] char class.
    * @param storageTargetName Name of Storage Target.
-   * @param [options] The optional parameters
-   * @returns Promise<msRestAzure.LROPoller>
+   * @param options The options parameters.
    */
-  beginSuspend(resourceGroupName: string, cacheName: string, storageTargetName: string, options?: msRest.RequestOptionsBase): Promise<msRestAzure.LROPoller> {
-    return this.client.sendLRORequest(
-      {
-        resourceGroupName,
-        cacheName,
-        storageTargetName,
-        options
-      },
-      beginSuspendOperationSpec,
-      options);
+  async beginResume(
+    resourceGroupName: string,
+    cacheName: string,
+    storageTargetName: string,
+    options?: StorageTargetResumeOptionalParams
+  ): Promise<PollerLike<PollOperationState<void>, void>> {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ): Promise<void> => {
+      return this.client.sendOperationRequest(args, spec);
+    };
+    const sendOperation = 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 = new LroImpl(
+      sendOperation,
+      { resourceGroupName, cacheName, storageTargetName, options },
+      resumeOperationSpec
+    );
+    return new LroEngine(lro, {
+      resumeFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
+      lroResourceLocationConfig: "azure-async-operation"
+    });
   }
 
   /**
    * Resumes client access to a previously suspended storage target.
    * @param resourceGroupName Target resource group.
-   * @param cacheName Name of Cache. Length of name must not be greater than 80 and chars must be
-   * from the [-0-9a-zA-Z_] char class.
+   * @param cacheName Name of Cache. Length of name must not be greater than 80 and chars must be from
+   *                  the [-0-9a-zA-Z_] char class.
    * @param storageTargetName Name of Storage Target.
-   * @param [options] The optional parameters
-   * @returns Promise<msRestAzure.LROPoller>
+   * @param options The options parameters.
    */
-  beginResume(resourceGroupName: string, cacheName: string, storageTargetName: string, options?: msRest.RequestOptionsBase): Promise<msRestAzure.LROPoller> {
-    return this.client.sendLRORequest(
-      {
-        resourceGroupName,
-        cacheName,
-        storageTargetName,
-        options
-      },
-      beginResumeOperationSpec,
-      options);
+  async beginResumeAndWait(
+    resourceGroupName: string,
+    cacheName: string,
+    storageTargetName: string,
+    options?: StorageTargetResumeOptionalParams
+  ): Promise<void> {
+    const poller = await this.beginResume(
+      resourceGroupName,
+      cacheName,
+      storageTargetName,
+      options
+    );
+    return poller.pollUntilDone();
   }
 }
-
 // Operation Specifications
-const serializer = new msRest.Serializer(Mappers);
-const beginFlushOperationSpec: msRest.OperationSpec = {
+const serializer = coreClient.createSerializer(Mappers, /* isXml */ false);
+
+const flushOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.StorageCache/caches/{cacheName}/storageTargets/{storageTargetName}/flush",
   httpMethod: "POST",
-  path: "subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.StorageCache/caches/{cacheName}/storageTargets/{storageTargetName}/flush",
-  urlParameters: [
-    Parameters.resourceGroupName,
-    Parameters.subscriptionId,
-    Parameters.cacheName,
-    Parameters.storageTargetName
-  ],
-  queryParameters: [
-    Parameters.apiVersion
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
   responses: {
     200: {},
+    201: {},
     202: {},
     204: {},
     default: {
       bodyMapper: Mappers.CloudError
     }
   },
-  serializer
-};
-
-const beginSuspendOperationSpec: msRest.OperationSpec = {
-  httpMethod: "POST",
-  path: "subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.StorageCache/caches/{cacheName}/storageTargets/{storageTargetName}/suspend",
+  queryParameters: [Parameters.apiVersion],
   urlParameters: [
-    Parameters.resourceGroupName,
+    Parameters.$host,
     Parameters.subscriptionId,
+    Parameters.resourceGroupName,
     Parameters.cacheName,
     Parameters.storageTargetName
   ],
-  queryParameters: [
-    Parameters.apiVersion
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
+  headerParameters: [Parameters.accept],
+  serializer
+};
+const suspendOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.StorageCache/caches/{cacheName}/storageTargets/{storageTargetName}/suspend",
+  httpMethod: "POST",
   responses: {
     200: {},
+    201: {},
     202: {},
     204: {},
     default: {
       bodyMapper: Mappers.CloudError
     }
   },
-  serializer
-};
-
-const beginResumeOperationSpec: msRest.OperationSpec = {
-  httpMethod: "POST",
-  path: "subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.StorageCache/caches/{cacheName}/storageTargets/{storageTargetName}/resume",
+  queryParameters: [Parameters.apiVersion],
   urlParameters: [
-    Parameters.resourceGroupName,
+    Parameters.$host,
     Parameters.subscriptionId,
+    Parameters.resourceGroupName,
     Parameters.cacheName,
     Parameters.storageTargetName
   ],
-  queryParameters: [
-    Parameters.apiVersion
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
+  headerParameters: [Parameters.accept],
+  serializer
+};
+const resumeOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.StorageCache/caches/{cacheName}/storageTargets/{storageTargetName}/resume",
+  httpMethod: "POST",
   responses: {
     200: {},
+    201: {},
     202: {},
     204: {},
     default: {
       bodyMapper: Mappers.CloudError
     }
   },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.resourceGroupName,
+    Parameters.cacheName,
+    Parameters.storageTargetName
+  ],
+  headerParameters: [Parameters.accept],
   serializer
 };