couchbase 4.6.1 → 4.7.0-dev.1

package/dist/queryindexmanager.js

@@ -1,8 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.QueryIndexManager = exports.CollectionQueryIndexManager = exports.QueryIndex = void 0;
-const bindingutilities_1 = require("./bindingutilities");
 const errors_1 = require("./errors");
+const observability_1 = require("./observability");
+const observabilityhandler_1 = require("./observabilityhandler");
+const observabilitytypes_1 = require("./observabilitytypes");
 const utilities_1 = require("./utilities");
 /**
  * Contains a specific index configuration for the query service.
@@ -44,10 +46,16 @@ class InternalQueryIndexManager {
     /**
      * @internal
      */
-    async createIndex(bucketName, isPrimary, options, callback) {
+    get observabilityInstruments() {
+        return this._cluster.observabilityInstruments;
+    }
+    /**
+     * @internal
+     */
+    async createIndex(bucketName, isPrimary, options, obsReqHandler, callback) {
         const timeout = options.timeout || this._cluster.managementTimeout;
-        return utilities_1.PromiseHelper.wrap((wrapCallback) => {
-            this._cluster.conn.managementQueryIndexCreate({
+        return utilities_1.PromiseHelper.wrapAsync(async () => {
+            const [err, _] = await (0, observability_1.wrapObservableBindingCall)(this._cluster.conn.managementQueryIndexCreate.bind(this._cluster.conn), {
                 bucket_name: bucketName,
                 scope_name: options.scopeName || '',
                 collection_name: options.collectionName || '',
@@ -60,26 +68,25 @@ class InternalQueryIndexManager {
                 num_replicas: options.numReplicas,
                 timeout: timeout,
                 condition: undefined,
-            }, (cppErr) => {
-                const err = (0, bindingutilities_1.errorFromCpp)(cppErr);
-                if (err) {
-                    return wrapCallback(err, null);
-                }
-                wrapCallback(err);
-            });
+            }, obsReqHandler);
+            if (err) {
+                obsReqHandler.endWithError(err);
+                throw err;
+            }
+            obsReqHandler.end();
         }, callback);
     }
     /**
      * @internal
      */
-    async dropIndex(bucketName, isPrimary, options, callback) {
+    async dropIndex(bucketName, isPrimary, options, obsReqHandler, callback) {
         const timeout = options.timeout || this._cluster.managementTimeout;
         // BUG(JSCBC-1066): We need to use a normal drop index for named primary indexes.
         if (options.name) {
             isPrimary = false;
         }
-        return utilities_1.PromiseHelper.wrap((wrapCallback) => {
-            this._cluster.conn.managementQueryIndexDrop({
+        return utilities_1.PromiseHelper.wrapAsync(async () => {
+            const [err, _] = await (0, observability_1.wrapObservableBindingCall)(this._cluster.conn.managementQueryIndexDrop.bind(this._cluster.conn), {
                 bucket_name: bucketName,
                 scope_name: options.scopeName || '',
                 collection_name: options.collectionName || '',
@@ -88,73 +95,71 @@ class InternalQueryIndexManager {
                 is_primary: isPrimary,
                 ignore_if_does_not_exist: options.ignoreIfNotExists || false,
                 timeout: timeout,
-            }, (cppErr) => {
-                const err = (0, bindingutilities_1.errorFromCpp)(cppErr);
-                if (err) {
-                    return wrapCallback(err, null);
-                }
-                wrapCallback(err);
-            });
+            }, obsReqHandler);
+            if (err) {
+                obsReqHandler.endWithError(err);
+                throw err;
+            }
+            obsReqHandler.end();
         }, callback);
     }
     /**
      * @internal
      */
-    async getAllIndexes(bucketName, options, callback) {
+    async getAllIndexes(bucketName, options, obsReqHandler, callback) {
         const timeout = options.timeout || this._cluster.managementTimeout;
-        return utilities_1.PromiseHelper.wrap((wrapCallback) => {
-            this._cluster.conn.managementQueryIndexGetAll({
+        return utilities_1.PromiseHelper.wrapAsync(async () => {
+            const [err, resp] = await (0, observability_1.wrapObservableBindingCall)(this._cluster.conn.managementQueryIndexGetAll.bind(this._cluster.conn), {
                 bucket_name: bucketName,
                 scope_name: options.scopeName || '',
                 collection_name: options.collectionName || '',
                 query_ctx: options.queryContext || this._queryContext,
                 timeout: timeout,
-            }, (cppErr, resp) => {
-                const err = (0, bindingutilities_1.errorFromCpp)(cppErr);
-                if (err) {
-                    return wrapCallback(err, null);
-                }
-                const indexes = resp.indexes.map((row) => new QueryIndex({
-                    isPrimary: row.is_primary,
-                    name: row.name,
-                    state: row.state,
-                    type: row.type,
-                    indexKey: row.index_key,
-                    partition: row.partition,
-                    condition: row.condition,
-                    bucketName: row.bucket_name,
-                    scopeName: row.scope_name,
-                    collectionName: row.collection_name,
-                }));
-                wrapCallback(null, indexes);
-            });
+            }, obsReqHandler);
+            if (err) {
+                obsReqHandler.endWithError(err);
+                throw err;
+            }
+            obsReqHandler.end();
+            return resp.indexes.map((row) => new QueryIndex({
+                isPrimary: row.is_primary,
+                name: row.name,
+                state: row.state,
+                type: row.type,
+                indexKey: row.index_key,
+                partition: row.partition,
+                condition: row.condition,
+                bucketName: row.bucket_name,
+                scopeName: row.scope_name,
+                collectionName: row.collection_name,
+            }));
         }, callback);
     }
     /**
      * @internal
      */
-    async buildDeferredIndexes(bucketName, options, callback) {
+    async buildDeferredIndexes(bucketName, options, obsReqHandler, callback) {
         const timeout = options.timeout || this._cluster.managementTimeout;
-        return utilities_1.PromiseHelper.wrap((wrapCallback) => {
-            this._cluster.conn.managementQueryIndexBuildDeferred({
+        return utilities_1.PromiseHelper.wrapAsync(async () => {
+            const [err, _] = await (0, observability_1.wrapObservableBindingCall)(this._cluster.conn.managementQueryIndexBuildDeferred.bind(this._cluster.conn), {
                 bucket_name: bucketName,
                 scope_name: options.scopeName || '',
                 collection_name: options.collectionName || '',
                 query_ctx: options.queryContext || this._queryContext,
                 timeout: timeout,
-            }, (cppErr) => {
-                const err = (0, bindingutilities_1.errorFromCpp)(cppErr);
-                if (err) {
-                    return wrapCallback(err, null);
-                }
-                wrapCallback(null, null);
-            });
+            }, obsReqHandler);
+            if (err) {
+                obsReqHandler.endWithError(err);
+                throw err;
+            }
+            obsReqHandler.end();
+            return [];
         }, callback);
     }
     /**
      * @internal
      */
-    async watchIndexes(bucketName, indexNames, timeout, options, callback) {
+    async watchIndexes(bucketName, indexNames, timeout, options, obsReqHandler, callback) {
         if (options.watchPrimary) {
             indexNames = [...indexNames, '#primary'];
         }
@@ -162,10 +167,12 @@ class InternalQueryIndexManager {
         return utilities_1.PromiseHelper.wrapAsync(async () => {
             let curInterval = 50;
             for (;;) {
+                const subObsReqHandler = new observabilityhandler_1.ObservableRequestHandler(observabilitytypes_1.QueryIndexMgmtOp.QueryIndexGetAll, this.observabilityInstruments, obsReqHandler.wrappedSpan);
+                subObsReqHandler.setRequestHttpAttributes();
                 // Get all the indexes that are currently registered
                 const foundIdxs = await this.getAllIndexes(bucketName, {
                     timeout: timer.left(),
-                });
+                }, subObsReqHandler);
                 const foundIndexNames = foundIdxs.map((idx) => idx.name);
                 const onlineIdxs = foundIdxs.filter((idx) => idx.state === 'online');
                 const onlineIdxNames = onlineIdxs.map((idx) => idx.name);
@@ -179,6 +186,7 @@ class InternalQueryIndexManager {
                 });
                 // If all the indexes are online, we've succeeded
                 if (allOnline) {
+                    obsReqHandler.end();
                     break;
                 }
                 // Add 500 to our interval to a max of 1000
@@ -189,7 +197,14 @@ class InternalQueryIndexManager {
                     curInterval = Math.min(curInterval, userTimeLeft);
                 }
                 if (curInterval <= 0) {
-                    throw new errors_1.CouchbaseError('Failed to find all indexes online within the alloted time.');
+                    // We don't need to end the top-level span here b/c the caller wraps this in a try/catch that
+                    // will end the span w/ an error accordingly.  But we do need end the sub-span we created for
+                    // the getAllIndexes() call
+                    const err = new errors_1.CouchbaseError('Failed to find all indexes online within the alloted time.');
+                    if (!subObsReqHandler.requestHasEnded) {
+                        subObsReqHandler.endWithError(err);
+                    }
+                    throw err;
                 }
                 // Wait until curInterval expires
                 await new Promise((resolve) => setTimeout(() => resolve(true), curInterval));
@@ -229,16 +244,24 @@ class CollectionQueryIndexManager {
         if (!options) {
             options = {};
         }
-        return this._manager.createIndex(this._bucketName, false, {
-            collectionName: this._collectionName,
-            scopeName: this._scopeName,
-            name: indexName,
-            keys: keys,
-            ignoreIfExists: options.ignoreIfExists,
-            numReplicas: options.numReplicas,
-            deferred: options.deferred,
-            timeout: options.timeout,
-        }, callback);
+        const obsReqHandler = new observabilityhandler_1.ObservableRequestHandler(observabilitytypes_1.QueryIndexMgmtOp.QueryIndexCreate, this._manager.observabilityInstruments, options === null || options === void 0 ? void 0 : options.parentSpan);
+        obsReqHandler.setRequestHttpAttributes();
+        try {
+            return this._manager.createIndex(this._bucketName, false, {
+                collectionName: this._collectionName,
+                scopeName: this._scopeName,
+                name: indexName,
+                keys: keys,
+                ignoreIfExists: options.ignoreIfExists,
+                numReplicas: options.numReplicas,
+                deferred: options.deferred,
+                timeout: options.timeout,
+            }, obsReqHandler, callback);
+        }
+        catch (err) {
+            obsReqHandler.endWithError(err);
+            throw err;
+        }
     }
     /**
      * Creates a new primary query index.
@@ -254,14 +277,22 @@ class CollectionQueryIndexManager {
         if (!options) {
             options = {};
         }
-        return this._manager.createIndex(this._bucketName, true, {
-            collectionName: this._collectionName,
-            scopeName: this._scopeName,
-            name: options.name,
-            ignoreIfExists: options.ignoreIfExists,
-            deferred: options.deferred,
-            timeout: options.timeout,
-        }, callback);
+        const obsReqHandler = new observabilityhandler_1.ObservableRequestHandler(observabilitytypes_1.QueryIndexMgmtOp.QueryIndexCreate, this._manager.observabilityInstruments, options === null || options === void 0 ? void 0 : options.parentSpan);
+        obsReqHandler.setRequestHttpAttributes();
+        try {
+            return this._manager.createIndex(this._bucketName, true, {
+                collectionName: this._collectionName,
+                scopeName: this._scopeName,
+                name: options.name,
+                ignoreIfExists: options.ignoreIfExists,
+                deferred: options.deferred,
+                timeout: options.timeout,
+            }, obsReqHandler, callback);
+        }
+        catch (err) {
+            obsReqHandler.endWithError(err);
+            throw err;
+        }
     }
     /**
      * Drops an existing query index.
@@ -278,13 +309,21 @@ class CollectionQueryIndexManager {
         if (!options) {
             options = {};
         }
-        return this._manager.dropIndex(this._bucketName, false, {
-            collectionName: this._collectionName,
-            scopeName: this._scopeName,
-            name: indexName,
-            ignoreIfNotExists: options.ignoreIfNotExists,
-            timeout: options.timeout,
-        }, callback);
+        const obsReqHandler = new observabilityhandler_1.ObservableRequestHandler(observabilitytypes_1.QueryIndexMgmtOp.QueryIndexDrop, this._manager.observabilityInstruments, options === null || options === void 0 ? void 0 : options.parentSpan);
+        obsReqHandler.setRequestHttpAttributes();
+        try {
+            return this._manager.dropIndex(this._bucketName, false, {
+                collectionName: this._collectionName,
+                scopeName: this._scopeName,
+                name: indexName,
+                ignoreIfNotExists: options.ignoreIfNotExists,
+                timeout: options.timeout,
+            }, obsReqHandler, callback);
+        }
+        catch (err) {
+            obsReqHandler.endWithError(err);
+            throw err;
+        }
     }
     /**
      * Drops an existing primary index.
@@ -300,13 +339,21 @@ class CollectionQueryIndexManager {
         if (!options) {
             options = {};
         }
-        return this._manager.dropIndex(this._bucketName, true, {
-            collectionName: this._collectionName,
-            scopeName: this._scopeName,
-            name: options.name,
-            ignoreIfNotExists: options.ignoreIfNotExists,
-            timeout: options.timeout,
-        }, callback);
+        const obsReqHandler = new observabilityhandler_1.ObservableRequestHandler(observabilitytypes_1.QueryIndexMgmtOp.QueryIndexDrop, this._manager.observabilityInstruments, options === null || options === void 0 ? void 0 : options.parentSpan);
+        obsReqHandler.setRequestHttpAttributes();
+        try {
+            return this._manager.dropIndex(this._bucketName, true, {
+                collectionName: this._collectionName,
+                scopeName: this._scopeName,
+                name: options.name,
+                ignoreIfNotExists: options.ignoreIfNotExists,
+                timeout: options.timeout,
+            }, obsReqHandler, callback);
+        }
+        catch (err) {
+            obsReqHandler.endWithError(err);
+            throw err;
+        }
     }
     /**
      * Returns a list of indexes for a specific bucket.
@@ -322,11 +369,19 @@ class CollectionQueryIndexManager {
         if (!options) {
             options = {};
         }
-        return this._manager.getAllIndexes(this._bucketName, {
-            collectionName: this._collectionName,
-            scopeName: this._scopeName,
-            timeout: options.timeout,
-        }, callback);
+        const obsReqHandler = new observabilityhandler_1.ObservableRequestHandler(observabilitytypes_1.QueryIndexMgmtOp.QueryIndexGetAll, this._manager.observabilityInstruments, options === null || options === void 0 ? void 0 : options.parentSpan);
+        obsReqHandler.setRequestHttpAttributes();
+        try {
+            return this._manager.getAllIndexes(this._bucketName, {
+                collectionName: this._collectionName,
+                scopeName: this._scopeName,
+                timeout: options.timeout,
+            }, obsReqHandler, callback);
+        }
+        catch (err) {
+            obsReqHandler.endWithError(err);
+            throw err;
+        }
     }
     /**
      * Starts building any indexes which were previously created with deferred=true.
@@ -342,11 +397,19 @@ class CollectionQueryIndexManager {
         if (!options) {
             options = {};
         }
-        return this._manager.buildDeferredIndexes(this._bucketName, {
-            collectionName: this._collectionName,
-            scopeName: this._scopeName,
-            timeout: options.timeout,
-        }, callback);
+        const obsReqHandler = new observabilityhandler_1.ObservableRequestHandler(observabilitytypes_1.QueryIndexMgmtOp.QueryIndexBuildDeferred, this._manager.observabilityInstruments, options === null || options === void 0 ? void 0 : options.parentSpan);
+        obsReqHandler.setRequestHttpAttributes();
+        try {
+            return this._manager.buildDeferredIndexes(this._bucketName, {
+                collectionName: this._collectionName,
+                scopeName: this._scopeName,
+                timeout: options.timeout,
+            }, obsReqHandler, callback);
+        }
+        catch (err) {
+            obsReqHandler.endWithError(err);
+            throw err;
+        }
     }
     /**
      * Waits for a number of indexes to finish creation and be ready to use.
@@ -364,11 +427,19 @@ class CollectionQueryIndexManager {
         if (!options) {
             options = {};
         }
-        return this._manager.watchIndexes(this._bucketName, indexNames, timeout, {
-            collectionName: this._collectionName,
-            scopeName: this._scopeName,
-            watchPrimary: options.watchPrimary,
-        }, callback);
+        const obsReqHandler = new observabilityhandler_1.ObservableRequestHandler(observabilitytypes_1.QueryIndexMgmtOp.QueryIndexWatchIndexes, this._manager.observabilityInstruments, options === null || options === void 0 ? void 0 : options.parentSpan);
+        obsReqHandler.setRequestHttpAttributes();
+        try {
+            return this._manager.watchIndexes(this._bucketName, indexNames, timeout, {
+                collectionName: this._collectionName,
+                scopeName: this._scopeName,
+                watchPrimary: options.watchPrimary,
+            }, obsReqHandler, callback);
+        }
+        catch (err) {
+            obsReqHandler.endWithError(err);
+            throw err;
+        }
     }
 }
 exports.CollectionQueryIndexManager = CollectionQueryIndexManager;
@@ -402,16 +473,24 @@ class QueryIndexManager {
         if (!options) {
             options = {};
         }
-        return this._manager.createIndex(bucketName, false, {
-            collectionName: options.collectionName,
-            scopeName: options.scopeName,
-            name: indexName,
-            keys: keys,
-            ignoreIfExists: options.ignoreIfExists,
-            numReplicas: options.numReplicas,
-            deferred: options.deferred,
-            timeout: options.timeout,
-        }, callback);
+        const obsReqHandler = new observabilityhandler_1.ObservableRequestHandler(observabilitytypes_1.QueryIndexMgmtOp.QueryIndexCreate, this._manager.observabilityInstruments, options === null || options === void 0 ? void 0 : options.parentSpan);
+        obsReqHandler.setRequestHttpAttributes();
+        try {
+            return this._manager.createIndex(bucketName, false, {
+                collectionName: options.collectionName,
+                scopeName: options.scopeName,
+                name: indexName,
+                keys: keys,
+                ignoreIfExists: options.ignoreIfExists,
+                numReplicas: options.numReplicas,
+                deferred: options.deferred,
+                timeout: options.timeout,
+            }, obsReqHandler, callback);
+        }
+        catch (err) {
+            obsReqHandler.endWithError(err);
+            throw err;
+        }
     }
     /**
      * Creates a new primary query index.
@@ -428,14 +507,22 @@ class QueryIndexManager {
         if (!options) {
             options = {};
         }
-        return this._manager.createIndex(bucketName, true, {
-            collectionName: options.collectionName,
-            scopeName: options.scopeName,
-            name: options.name,
-            ignoreIfExists: options.ignoreIfExists,
-            deferred: options.deferred,
-            timeout: options.timeout,
-        }, callback);
+        const obsReqHandler = new observabilityhandler_1.ObservableRequestHandler(observabilitytypes_1.QueryIndexMgmtOp.QueryIndexCreate, this._manager.observabilityInstruments, options === null || options === void 0 ? void 0 : options.parentSpan);
+        obsReqHandler.setRequestHttpAttributes();
+        try {
+            return this._manager.createIndex(bucketName, true, {
+                collectionName: options.collectionName,
+                scopeName: options.scopeName,
+                name: options.name,
+                ignoreIfExists: options.ignoreIfExists,
+                deferred: options.deferred,
+                timeout: options.timeout,
+            }, obsReqHandler, callback);
+        }
+        catch (err) {
+            obsReqHandler.endWithError(err);
+            throw err;
+        }
     }
     /**
      * Drops an existing query index.
@@ -453,13 +540,21 @@ class QueryIndexManager {
         if (!options) {
             options = {};
         }
-        return this._manager.dropIndex(bucketName, false, {
-            collectionName: options.collectionName,
-            scopeName: options.scopeName,
-            name: indexName,
-            ignoreIfNotExists: options.ignoreIfNotExists,
-            timeout: options.timeout,
-        }, callback);
+        const obsReqHandler = new observabilityhandler_1.ObservableRequestHandler(observabilitytypes_1.QueryIndexMgmtOp.QueryIndexDrop, this._manager.observabilityInstruments, options === null || options === void 0 ? void 0 : options.parentSpan);
+        obsReqHandler.setRequestHttpAttributes();
+        try {
+            return this._manager.dropIndex(bucketName, false, {
+                collectionName: options.collectionName,
+                scopeName: options.scopeName,
+                name: indexName,
+                ignoreIfNotExists: options.ignoreIfNotExists,
+                timeout: options.timeout,
+            }, obsReqHandler, callback);
+        }
+        catch (err) {
+            obsReqHandler.endWithError(err);
+            throw err;
+        }
     }
     /**
      * Drops an existing primary index.
@@ -476,13 +571,21 @@ class QueryIndexManager {
         if (!options) {
             options = {};
         }
-        return this._manager.dropIndex(bucketName, true, {
-            collectionName: options.collectionName,
-            scopeName: options.scopeName,
-            name: options.name,
-            ignoreIfNotExists: options.ignoreIfNotExists,
-            timeout: options.timeout,
-        }, callback);
+        const obsReqHandler = new observabilityhandler_1.ObservableRequestHandler(observabilitytypes_1.QueryIndexMgmtOp.QueryIndexDrop, this._manager.observabilityInstruments, options === null || options === void 0 ? void 0 : options.parentSpan);
+        obsReqHandler.setRequestHttpAttributes();
+        try {
+            return this._manager.dropIndex(bucketName, true, {
+                collectionName: options.collectionName,
+                scopeName: options.scopeName,
+                name: options.name,
+                ignoreIfNotExists: options.ignoreIfNotExists,
+                timeout: options.timeout,
+            }, obsReqHandler, callback);
+        }
+        catch (err) {
+            obsReqHandler.endWithError(err);
+            throw err;
+        }
     }
     /**
      * Returns a list of indexes for a specific bucket.
@@ -499,11 +602,19 @@ class QueryIndexManager {
         if (!options) {
             options = {};
         }
-        return this._manager.getAllIndexes(bucketName, {
-            collectionName: options.collectionName,
-            scopeName: options.scopeName,
-            timeout: options.timeout,
-        }, callback);
+        const obsReqHandler = new observabilityhandler_1.ObservableRequestHandler(observabilitytypes_1.QueryIndexMgmtOp.QueryIndexGetAll, this._manager.observabilityInstruments, options === null || options === void 0 ? void 0 : options.parentSpan);
+        obsReqHandler.setRequestHttpAttributes();
+        try {
+            return this._manager.getAllIndexes(bucketName, {
+                collectionName: options.collectionName,
+                scopeName: options.scopeName,
+                timeout: options.timeout,
+            }, obsReqHandler, callback);
+        }
+        catch (err) {
+            obsReqHandler.endWithError(err);
+            throw err;
+        }
     }
     /**
      * Starts building any indexes which were previously created with deferred=true.
@@ -520,11 +631,19 @@ class QueryIndexManager {
         if (!options) {
             options = {};
         }
-        return this._manager.buildDeferredIndexes(bucketName, {
-            collectionName: options.collectionName,
-            scopeName: options.scopeName,
-            timeout: options.timeout,
-        }, callback);
+        const obsReqHandler = new observabilityhandler_1.ObservableRequestHandler(observabilitytypes_1.QueryIndexMgmtOp.QueryIndexBuildDeferred, this._manager.observabilityInstruments, options === null || options === void 0 ? void 0 : options.parentSpan);
+        obsReqHandler.setRequestHttpAttributes();
+        try {
+            return this._manager.buildDeferredIndexes(bucketName, {
+                collectionName: options.collectionName,
+                scopeName: options.scopeName,
+                timeout: options.timeout,
+            }, obsReqHandler, callback);
+        }
+        catch (err) {
+            obsReqHandler.endWithError(err);
+            throw err;
+        }
     }
     /**
      * Waits for a number of indexes to finish creation and be ready to use.
@@ -543,11 +662,19 @@ class QueryIndexManager {
         if (!options) {
             options = {};
         }
-        return this._manager.watchIndexes(bucketName, indexNames, timeout, {
-            collectionName: options.collectionName,
-            scopeName: options.scopeName,
-            watchPrimary: options.watchPrimary,
-        }, callback);
+        const obsReqHandler = new observabilityhandler_1.ObservableRequestHandler(observabilitytypes_1.QueryIndexMgmtOp.QueryIndexWatchIndexes, this._manager.observabilityInstruments, options === null || options === void 0 ? void 0 : options.parentSpan);
+        obsReqHandler.setRequestHttpAttributes();
+        try {
+            return this._manager.watchIndexes(bucketName, indexNames, timeout, {
+                collectionName: options.collectionName,
+                scopeName: options.scopeName,
+                watchPrimary: options.watchPrimary,
+            }, obsReqHandler, callback);
+        }
+        catch (err) {
+            obsReqHandler.endWithError(err);
+            throw err;
+        }
     }
 }
 exports.QueryIndexManager = QueryIndexManager;

package/dist/querytypes.d.ts

@@ -1,4 +1,5 @@
 import { MutationState } from './mutationstate';
+import { RequestSpan } from './tracing';
 /**
  * Represents the status of a query.
  *
@@ -310,4 +311,8 @@ export interface QueryOptions {
      * The timeout for this operation, represented in milliseconds.
      */
     timeout?: number;
+    /**
+     * Specifies the parent span for this specific operation.
+     */
+    parentSpan?: RequestSpan;
 }

package/dist/scope.js

@@ -102,7 +102,7 @@ class Scope {
         const options_ = options;
         return utilities_1.PromiseHelper.wrapAsync(() => exec.query(statement, {
             ...options_,
-            queryContext: `${bucket.name}.${this.name}`,
+            queryContext: `\`${bucket.name}\`.\`${this.name}\``,
         }), callback);
     }
     /**
@@ -125,7 +125,7 @@ class Scope {
         const options_ = options;
         return utilities_1.PromiseHelper.wrapAsync(() => exec.query(statement, {
             ...options_,
-            queryContext: `${bucket.name}.${this.name}`,
+            queryContext: `\`${bucket.name}\`.\`${this.name}\``,
         }), callback);
     }
     /**