@azure/search-documents 12.3.0-beta.1 → 13.0.0

package/dist/esm/searchIndexerClient.js

@@ -1,14 +1,14 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
 import { isTokenCredential } from "@azure/core-auth";
-import { bearerTokenAuthenticationPolicy } from "@azure/core-rest-pipeline";
-import { SearchServiceClient as GeneratedClient } from "./generated/service/searchServiceClient.js";
+import { bearerTokenAuthenticationPolicy, bearerTokenAuthenticationPolicyName, } from "@azure/core-rest-pipeline";
+import { SearchIndexerClient as GeneratedClient } from "./searchIndexer/searchIndexerClient.js";
 import { logger } from "./logger.js";
 import { createOdataMetadataPolicy } from "./odataMetadataPolicy.js";
 import { createSearchApiKeyCredentialPolicy } from "./searchApiKeyCredentialPolicy.js";
 import { KnownSearchAudience } from "./searchAudience.js";
 import * as utils from "./serviceUtils.js";
-import { createSpan } from "./tracing.js";
+import { tracingClient } from "./tracing.js";
 /**
  * Class to perform operations to manage
  * (create, update, list/delete)
@@ -54,6 +54,7 @@ export class SearchIndexerClient {
         this.endpoint = endpoint;
         const internalClientPipelineOptions = {
             ...options,
+            apiVersion: options.serviceVersion ?? options.apiVersion ?? utils.defaultServiceVersion,
             ...{
                 loggingOptions: {
                     logger: logger.info,
@@ -71,8 +72,10 @@ export class SearchIndexerClient {
         this.serviceVersion =
             options.serviceVersion ?? options.apiVersion ?? utils.defaultServiceVersion;
         this.apiVersion = this.serviceVersion;
-        this.client = new GeneratedClient(this.endpoint, this.serviceVersion, internalClientPipelineOptions);
+        this.client = new GeneratedClient(this.endpoint, credential, internalClientPipelineOptions);
         this.pipeline = this.client.pipeline;
+        // Replaced with a custom policy below
+        this.pipeline.removePolicy({ name: bearerTokenAuthenticationPolicyName });
         if (isTokenCredential(credential)) {
             const scope = options.audience
                 ? `${options.audience}/.default`
@@ -89,21 +92,10 @@ export class SearchIndexerClient {
      * @param options - Options to the list indexers operation.
      */
     async listIndexers(options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-listIndexers", options);
-        try {
-            const result = await this.client.indexers.list(updatedOptions);
+        return tracingClient.withSpan("SearchIndexerClient-listIndexers", options, async (updatedOptions) => {
+            const result = await this.client.getIndexers(updatedOptions);
             return result.indexers.map(utils.generatedSearchIndexerToPublicSearchIndexer);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Retrieves a list of names of existing indexers in the service.
@@ -111,45 +103,23 @@ export class SearchIndexerClient {
      */
     // eslint-disable-next-line @azure/azure-sdk/ts-naming-options
     async listIndexersNames(options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-listIndexersNames", options);
-        try {
-            const result = await this.client.indexers.list({
+        return tracingClient.withSpan("SearchIndexerClient-listIndexersNames", options, async (updatedOptions) => {
+            const result = await this.client.getIndexers({
                 ...updatedOptions,
                 select: "name",
             });
             return result.indexers.map((idx) => idx.name);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Retrieves a list of existing data sources in the service.
      * @param options - Options to the list indexers operation.
      */
     async listDataSourceConnections(options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-listDataSourceConnections", options);
-        try {
-            const result = await this.client.dataSources.list(updatedOptions);
+        return tracingClient.withSpan("SearchIndexerClient-listDataSourceConnections", options, async (updatedOptions) => {
+            const result = await this.client.getDataSourceConnections(updatedOptions);
             return result.dataSources.map(utils.generatedDataSourceToPublicDataSource);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Retrieves a list of names of existing data sources in the service.
@@ -158,45 +128,23 @@ export class SearchIndexerClient {
     async listDataSourceConnectionsNames(
     // eslint-disable-next-line @azure/azure-sdk/ts-naming-options
     options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-listDataSourceConnectionsNames", options);
-        try {
-            const result = await this.client.dataSources.list({
+        return tracingClient.withSpan("SearchIndexerClient-listDataSourceConnectionsNames", options, async (updatedOptions) => {
+            const result = await this.client.getDataSourceConnections({
                 ...updatedOptions,
                 select: "name",
             });
             return result.dataSources.map((ds) => ds.name);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Retrieves a list of existing Skillsets in the service.
      * @param options - Options to the list Skillsets operation.
      */
     async listSkillsets(options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-listSkillsets", options);
-        try {
-            const result = await this.client.skillsets.list(updatedOptions);
+        return tracingClient.withSpan("SearchIndexerClient-listSkillsets", options, async (updatedOptions) => {
+            const result = await this.client.getSkillsets(updatedOptions);
             return result.skillsets.map(utils.generatedSkillsetToPublicSkillset);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Retrieves a list of names of existing Skillsets in the service.
@@ -204,24 +152,13 @@ export class SearchIndexerClient {
      */
     // eslint-disable-next-line @azure/azure-sdk/ts-naming-options
     async listSkillsetsNames(options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-listSkillsetsNames", options);
-        try {
-            const result = await this.client.skillsets.list({
+        return tracingClient.withSpan("SearchIndexerClient-listSkillsetsNames", options, async (updatedOptions) => {
+            const result = await this.client.getSkillsets({
                 ...updatedOptions,
                 select: "name",
             });
             return result.skillsets.map((sks) => sks.name);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Retrieves information about an Indexer.
@@ -229,21 +166,10 @@ export class SearchIndexerClient {
      * @param options - Additional optional arguments.
      */
     async getIndexer(indexerName, options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-getIndexer", options);
-        try {
-            const result = await this.client.indexers.get(indexerName, updatedOptions);
+        return tracingClient.withSpan("SearchIndexerClient-getIndexer", options, async (updatedOptions) => {
+            const result = await this.client.getIndexer(indexerName, updatedOptions);
             return utils.generatedSearchIndexerToPublicSearchIndexer(result);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Retrieves information about a DataSource
@@ -251,21 +177,10 @@ export class SearchIndexerClient {
      * @param options - Additional optional arguments
      */
     async getDataSourceConnection(dataSourceConnectionName, options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-getDataSourceConnection", options);
-        try {
-            const result = await this.client.dataSources.get(dataSourceConnectionName, updatedOptions);
+        return tracingClient.withSpan("SearchIndexerClient-getDataSourceConnection", options, async (updatedOptions) => {
+            const result = await this.client.getDataSourceConnection(dataSourceConnectionName, updatedOptions);
             return utils.generatedDataSourceToPublicDataSource(result);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Retrieves information about an Skillset.
@@ -273,21 +188,10 @@ export class SearchIndexerClient {
      * @param options - Additional optional arguments.
      */
     async getSkillset(skillsetName, options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-getSkillset", options);
-        try {
-            const result = await this.client.skillsets.get(skillsetName, updatedOptions);
+        return tracingClient.withSpan("SearchIndexerClient-getSkillset", options, async (updatedOptions) => {
+            const result = await this.client.getSkillset(skillsetName, updatedOptions);
             return utils.generatedSkillsetToPublicSkillset(result);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Creates a new indexer in a search service.
@@ -295,21 +199,10 @@ export class SearchIndexerClient {
      * @param options - Additional optional arguments.
      */
     async createIndexer(indexer, options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-createIndexer", options);
-        try {
-            const result = await this.client.indexers.create(utils.publicSearchIndexerToGeneratedSearchIndexer(indexer), updatedOptions);
+        return tracingClient.withSpan("SearchIndexerClient-createIndexer", options, async (updatedOptions) => {
+            const result = await this.client.createIndexer(utils.publicSearchIndexerToGeneratedSearchIndexer(indexer), updatedOptions);
             return utils.generatedSearchIndexerToPublicSearchIndexer(result);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Creates a new dataSource in a search service.
@@ -317,21 +210,10 @@ export class SearchIndexerClient {
      * @param options - Additional optional arguments.
      */
     async createDataSourceConnection(dataSourceConnection, options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-createDataSourceConnection", options);
-        try {
-            const result = await this.client.dataSources.create(utils.publicDataSourceToGeneratedDataSource(dataSourceConnection), updatedOptions);
+        return tracingClient.withSpan("SearchIndexerClient-createDataSourceConnection", options, async (updatedOptions) => {
+            const result = await this.client.createDataSourceConnection(utils.publicDataSourceToGeneratedDataSource(dataSourceConnection), updatedOptions);
             return utils.generatedDataSourceToPublicDataSource(result);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Creates a new skillset in a search service.
@@ -339,21 +221,10 @@ export class SearchIndexerClient {
      * @param options - Additional optional arguments.
      */
     async createSkillset(skillset, options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-createSkillset", options);
-        try {
-            const result = await this.client.skillsets.create(utils.publicSkillsetToGeneratedSkillset(skillset), updatedOptions);
+        return tracingClient.withSpan("SearchIndexerClient-createSkillset", options, async (updatedOptions) => {
+            const result = await this.client.createSkillset(utils.publicSkillsetToGeneratedSkillset(skillset), updatedOptions);
             return utils.generatedSkillsetToPublicSkillset(result);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Creates a new indexer or modifies an existing one.
@@ -361,26 +232,15 @@ export class SearchIndexerClient {
      * @param options - Additional optional arguments.
      */
     async createOrUpdateIndexer(indexer, options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-createOrUpdateIndexer", options);
-        const { onlyIfUnchanged, ...restOptions } = updatedOptions;
-        try {
+        return tracingClient.withSpan("SearchIndexerClient-createOrUpdateIndexer", options, async (updatedOptions) => {
+            const { onlyIfUnchanged, ...restOptions } = updatedOptions;
             const etag = onlyIfUnchanged ? indexer.etag : undefined;
-            const result = await this.client.indexers.createOrUpdate(indexer.name, utils.publicSearchIndexerToGeneratedSearchIndexer(indexer), {
+            const result = await this.client.createOrUpdateIndexer(utils.publicSearchIndexerToGeneratedSearchIndexer(indexer), indexer.name, {
                 ...restOptions,
                 ifMatch: etag,
             });
             return utils.generatedSearchIndexerToPublicSearchIndexer(result);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Creates a new datasource or modifies an existing one.
@@ -388,25 +248,14 @@ export class SearchIndexerClient {
      * @param options - Additional optional arguments.
      */
     async createOrUpdateDataSourceConnection(dataSourceConnection, options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-createOrUpdateDataSourceConnection", options);
-        try {
+        return tracingClient.withSpan("SearchIndexerClient-createOrUpdateDataSourceConnection", options, async (updatedOptions) => {
             const etag = options.onlyIfUnchanged ? dataSourceConnection.etag : undefined;
-            const result = await this.client.dataSources.createOrUpdate(dataSourceConnection.name, utils.publicDataSourceToGeneratedDataSource(dataSourceConnection), {
+            const result = await this.client.createOrUpdateDataSourceConnection(utils.publicDataSourceToGeneratedDataSource(dataSourceConnection), dataSourceConnection.name, {
                 ...updatedOptions,
                 ifMatch: etag,
             });
             return utils.generatedDataSourceToPublicDataSource(result);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Creates a new Skillset or modifies an existing one.
@@ -414,25 +263,14 @@ export class SearchIndexerClient {
      * @param options - Additional optional arguments.
      */
     async createOrUpdateSkillset(skillset, options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-createOrUpdateSkillset", options);
-        try {
+        return tracingClient.withSpan("SearchIndexerClient-createOrUpdateSkillset", options, async (updatedOptions) => {
             const etag = options.onlyIfUnchanged ? skillset.etag : undefined;
-            const result = await this.client.skillsets.createOrUpdate(skillset.name, utils.publicSkillsetToGeneratedSkillset(skillset), {
+            const result = await this.client.createOrUpdateSkillset(utils.publicSkillsetToGeneratedSkillset(skillset), skillset.name, {
                 ...updatedOptions,
                 ifMatch: etag,
             });
             return utils.generatedSkillsetToPublicSkillset(result);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Deletes an existing indexer.
@@ -440,29 +278,18 @@ export class SearchIndexerClient {
      * @param options - Additional optional arguments.
      */
     async deleteIndexer(indexer, options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-deleteIndexer", options);
-        try {
+        return tracingClient.withSpan("SearchIndexerClient-deleteIndexer", options, async (updatedOptions) => {
             const indexerName = typeof indexer === "string" ? indexer : indexer.name;
             const etag = typeof indexer === "string"
                 ? undefined
                 : options.onlyIfUnchanged
                     ? indexer.etag
                     : undefined;
-            await this.client.indexers.delete(indexerName, {
+            await this.client.deleteIndexer(indexerName, {
                 ...updatedOptions,
                 ifMatch: etag,
             });
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Deletes an existing datasource.
@@ -470,29 +297,20 @@ export class SearchIndexerClient {
      * @param options - Additional optional arguments.
      */
     async deleteDataSourceConnection(dataSourceConnection, options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-deleteDataSourceConnection", options);
-        try {
-            const dataSourceConnectionName = typeof dataSourceConnection === "string" ? dataSourceConnection : dataSourceConnection.name;
+        return tracingClient.withSpan("SearchIndexerClient-deleteDataSourceConnection", options, async (updatedOptions) => {
+            const dataSourceConnectionName = typeof dataSourceConnection === "string"
+                ? dataSourceConnection
+                : dataSourceConnection.name;
             const etag = typeof dataSourceConnection === "string"
                 ? undefined
                 : options.onlyIfUnchanged
                     ? dataSourceConnection.etag
                     : undefined;
-            await this.client.dataSources.delete(dataSourceConnectionName, {
+            await this.client.deleteDataSourceConnection(dataSourceConnectionName, {
                 ...updatedOptions,
                 ifMatch: etag,
             });
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Deletes an existing Skillset.
@@ -500,29 +318,18 @@ export class SearchIndexerClient {
      * @param options - Additional optional arguments.
      */
     async deleteSkillset(skillset, options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-deleteSkillset", options);
-        try {
+        return tracingClient.withSpan("SearchIndexerClient-deleteSkillset", options, async (updatedOptions) => {
             const skillsetName = typeof skillset === "string" ? skillset : skillset.name;
             const etag = typeof skillset === "string"
                 ? undefined
                 : options.onlyIfUnchanged
                     ? skillset.etag
                     : undefined;
-            await this.client.skillsets.delete(skillsetName, {
+            await this.client.deleteSkillset(skillsetName, {
                 ...updatedOptions,
                 ifMatch: etag,
             });
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Returns the current status and execution history of an indexer.
@@ -530,21 +337,9 @@ export class SearchIndexerClient {
      * @param options - Additional optional arguments.
      */
     async getIndexerStatus(indexerName, options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-getIndexerStatus", options);
-        try {
-            const result = await this.client.indexers.getStatus(indexerName, updatedOptions);
-            return result;
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        return tracingClient.withSpan("SearchIndexerClient-getIndexerStatus", options, async (updatedOptions) => {
+            return this.client.getIndexerStatus(indexerName, updatedOptions);
+        });
     }
     /**
      * Resets the change tracking state associated with an indexer.
@@ -552,20 +347,9 @@ export class SearchIndexerClient {
      * @param options - Additional optional arguments.
      */
     async resetIndexer(indexerName, options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-resetIndexer", options);
-        try {
-            await this.client.indexers.reset(indexerName, updatedOptions);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        return tracingClient.withSpan("SearchIndexerClient-resetIndexer", options, async (updatedOptions) => {
+            await this.client.resetIndexer(indexerName, updatedOptions);
+        });
     }
     /**
      * Runs an indexer on-demand.
@@ -573,69 +357,9 @@ export class SearchIndexerClient {
      * @param options - Additional optional arguments.
      */
     async runIndexer(indexerName, options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-runIndexer", options);
-        try {
-            await this.client.indexers.run(indexerName, updatedOptions);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
-    }
-    /**
-     * Resets specific documents in the datasource to be selectively re-ingested by the indexer.
-     * @param indexerName - The name of the indexer to reset documents for.
-     * @param options - Additional optional arguments.
-     */
-    async resetDocuments(indexerName, options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-resetDocs", options);
-        try {
-            await this.client.indexers.resetDocs(indexerName, {
-                ...updatedOptions,
-                keysOrIds: {
-                    documentKeys: updatedOptions.documentKeys,
-                    datasourceDocumentIds: updatedOptions.datasourceDocumentIds,
-                },
-            });
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
-    }
-    /**
-     * Reset an existing skillset in a search service.
-     * @param skillsetName - The name of the skillset to reset.
-     * @param skillNames - The names of skills to reset.
-     * @param options - The options parameters.
-     */
-    async resetSkills(skillsetName, options = {}) {
-        const { span, updatedOptions } = createSpan("SearchIndexerClient-resetSkills", options);
-        try {
-            await this.client.skillsets.resetSkills(skillsetName, { skillNames: options.skillNames }, updatedOptions);
-        }
-        catch (e) {
-            span.setStatus({
-                status: "error",
-                error: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        return tracingClient.withSpan("SearchIndexerClient-runIndexer", options, async (updatedOptions) => {
+            await this.client.runIndexer(indexerName, updatedOptions);
+        });
     }
 }
 //# sourceMappingURL=searchIndexerClient.js.map