autotel-plugins 0.19.25 → 0.19.27

package/dist/bigquery.js

@@ -4,25 +4,25 @@ import { runWithSpan, finalizeSpan } from 'autotel/trace-helpers';
 // src/bigquery/index.ts
 
 // src/common/constants.ts
-var SEMATTRS_DB_SYSTEM_NAME = "db.system.name";
-var SEMATTRS_DB_NAMESPACE = "db.namespace";
-var SEMATTRS_DB_COLLECTION_NAME = "db.collection.name";
-var SEMATTRS_DB_OPERATION_NAME = "db.operation.name";
-var SEMATTRS_DB_QUERY_TEXT = "db.query.text";
-var SEMATTRS_DB_QUERY_SUMMARY = "db.query.summary";
-var SEMATTRS_GCP_BIGQUERY_JOB_ID = "gcp.bigquery.job.id";
-var SEMATTRS_GCP_BIGQUERY_JOB_LOCATION = "gcp.bigquery.job.location";
-var SEMATTRS_GCP_BIGQUERY_PROJECT_ID = "gcp.bigquery.project.id";
-var SEMATTRS_GCP_BIGQUERY_DESTINATION_TABLE = "gcp.bigquery.destination.table";
-var SEMATTRS_GCP_BIGQUERY_QUERY_HASH = "gcp.bigquery.query.hash";
-var SEMATTRS_GCP_BIGQUERY_ROWS_AFFECTED = "gcp.bigquery.rows.affected";
-var SEMATTRS_GCP_BIGQUERY_ROWS_RETURNED = "gcp.bigquery.rows.returned";
-var DEFAULT_TRACER_NAME = "autotel-plugins/bigquery";
-var DEFAULT_DB_SYSTEM_NAME = "gcp.bigquery";
-var INSTRUMENTED_FLAG = "__autotelBigQueryInstrumented";
-var PROTOTYPE_INSTRUMENTED_FLAG = "__autotelBigQueryPrototypeInstrumented";
-var CONFIG_STORAGE_KEY = "__autotelBigQueryConfig";
-var TRACER_STORAGE_KEY = "__autotelBigQueryTracer";
+var SEMATTRS_DB_SYSTEM_NAME = 'db.system.name';
+var SEMATTRS_DB_NAMESPACE = 'db.namespace';
+var SEMATTRS_DB_COLLECTION_NAME = 'db.collection.name';
+var SEMATTRS_DB_OPERATION_NAME = 'db.operation.name';
+var SEMATTRS_DB_QUERY_TEXT = 'db.query.text';
+var SEMATTRS_DB_QUERY_SUMMARY = 'db.query.summary';
+var SEMATTRS_GCP_BIGQUERY_JOB_ID = 'gcp.bigquery.job.id';
+var SEMATTRS_GCP_BIGQUERY_JOB_LOCATION = 'gcp.bigquery.job.location';
+var SEMATTRS_GCP_BIGQUERY_PROJECT_ID = 'gcp.bigquery.project.id';
+var SEMATTRS_GCP_BIGQUERY_DESTINATION_TABLE = 'gcp.bigquery.destination.table';
+var SEMATTRS_GCP_BIGQUERY_QUERY_HASH = 'gcp.bigquery.query.hash';
+var SEMATTRS_GCP_BIGQUERY_ROWS_AFFECTED = 'gcp.bigquery.rows.affected';
+var SEMATTRS_GCP_BIGQUERY_ROWS_RETURNED = 'gcp.bigquery.rows.returned';
+var DEFAULT_TRACER_NAME = 'autotel-plugins/bigquery';
+var DEFAULT_DB_SYSTEM_NAME = 'gcp.bigquery';
+var INSTRUMENTED_FLAG = '__autotelBigQueryInstrumented';
+var PROTOTYPE_INSTRUMENTED_FLAG = '__autotelBigQueryPrototypeInstrumented';
+var CONFIG_STORAGE_KEY = '__autotelBigQueryConfig';
+var TRACER_STORAGE_KEY = '__autotelBigQueryTracer';
 function getInstanceConfig(instance) {
   if (instance?.[CONFIG_STORAGE_KEY]) {
     return instance[CONFIG_STORAGE_KEY];
@@ -70,13 +70,13 @@ function extractOperationType(query) {
 function createQuerySummary(query) {
   const operation = extractOperationType(query);
   if (!operation) {
-    return "UNKNOWN";
+    return 'UNKNOWN';
   }
   const patterns = {
     SELECT: /FROM\s+(?<table>[\w.]+)/iu,
     INSERT: /INTO\s+(?<table>[\w.]+)/iu,
     UPDATE: /UPDATE\s+(?<table>[\w.]+)/iu,
-    DELETE: /FROM\s+(?<table>[\w.]+)/iu
+    DELETE: /FROM\s+(?<table>[\w.]+)/iu,
   };
   const pattern = patterns[operation];
   if (pattern) {
@@ -91,9 +91,9 @@ function createQuerySummary(query) {
 function sanitizeQuery(query) {
   let sanitized = query.replaceAll(/'(?:[^'\\]|\\.)*'/gu, "'?'");
   sanitized = sanitized.replaceAll(/"(?:[^"\\]|\\.)*"/gu, '"?"');
-  sanitized = sanitized.replaceAll(/\b\d+\.?\d*\b/gu, "?");
-  sanitized = sanitized.replaceAll(/\b(?:true|false)\b/giu, "?");
-  sanitized = sanitized.replaceAll(/\bNULL\b/giu, "?");
+  sanitized = sanitized.replaceAll(/\b\d+\.?\d*\b/gu, '?');
+  sanitized = sanitized.replaceAll(/\b(?:true|false)\b/giu, '?');
+  sanitized = sanitized.replaceAll(/\bNULL\b/giu, '?');
   return sanitized;
 }
 function truncateText(text, maxLength) {
@@ -122,7 +122,7 @@ function extractTableReference(obj) {
       return {
         projectId: obj.dataset?.parent?.projectId || obj.parent?.projectId,
         datasetId: obj.dataset?.id || obj.parent?.datasetId,
-        tableId: obj.id
+        tableId: obj.id,
       };
     }
     if (obj.metadata?.tableReference) {
@@ -137,10 +137,11 @@ function createSpan(tracer, operationName, target, projectId, config = {}) {
   const spanName = target ? `${operationName} ${target}` : operationName;
   const attributes = {
     [SEMATTRS_DB_SYSTEM_NAME]: DEFAULT_DB_SYSTEM_NAME,
-    [SEMATTRS_DB_OPERATION_NAME]: operationName
+    [SEMATTRS_DB_OPERATION_NAME]: operationName,
   };
   if (projectId || config.projectId) {
-    attributes[SEMATTRS_GCP_BIGQUERY_PROJECT_ID] = projectId || config.projectId;
+    attributes[SEMATTRS_GCP_BIGQUERY_PROJECT_ID] =
+      projectId || config.projectId;
   }
   if (config.location) {
     attributes[SEMATTRS_GCP_BIGQUERY_JOB_LOCATION] = config.location;
@@ -149,43 +150,47 @@ function createSpan(tracer, operationName, target, projectId, config = {}) {
 }
 function instrumentQueryMethod(BigQuery) {
   const originalQuery = BigQuery.prototype.query;
-  if (typeof originalQuery !== "function") {
+  if (typeof originalQuery !== 'function') {
     return;
   }
-  BigQuery.prototype.query = function instrumentedQuery(query, options, callback) {
+  BigQuery.prototype.query = function instrumentedQuery(
+    query,
+    options,
+    callback,
+  ) {
     const config = getInstanceConfig(this);
     const tracer = getInstanceTracer(this);
     if (!config || !tracer) {
       return originalQuery.call(this, query, options, callback);
     }
-    const queryText = typeof query === "string" ? query : query.query;
+    const queryText = typeof query === 'string' ? query : query.query;
     const projectId = extractProjectId(this);
     const location = extractLocation(this, options);
-    const operation = queryText ? extractOperationType(queryText) : "QUERY";
-    const summary = queryText ? createQuerySummary(queryText) : "QUERY";
+    const operation = queryText ? extractOperationType(queryText) : 'QUERY';
+    const summary = queryText ? createQuerySummary(queryText) : 'QUERY';
     const span = createSpan(
       tracer,
-      operation || "QUERY",
+      operation || 'QUERY',
       summary,
       projectId,
-      config
+      config,
     );
     if (location) {
       span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_LOCATION, location);
     }
     span.setAttribute(SEMATTRS_DB_QUERY_SUMMARY, summary);
-    if (config.captureQueryText !== "never" && queryText) {
-      if (config.captureQueryText === "raw") {
+    if (config.captureQueryText !== 'never' && queryText) {
+      if (config.captureQueryText === 'raw') {
         const truncated = truncateText(
           queryText,
-          config.maxQueryTextLength || 1e3
+          config.maxQueryTextLength || 1e3,
         );
         span.setAttribute(SEMATTRS_DB_QUERY_TEXT, truncated);
-      } else if (config.captureQueryText === "sanitized") {
+      } else if (config.captureQueryText === 'sanitized') {
         const sanitized = sanitizeQuery(queryText);
         const truncated = truncateText(
           sanitized,
-          config.maxQueryTextLength || 1e3
+          config.maxQueryTextLength || 1e3,
         );
         span.setAttribute(SEMATTRS_DB_QUERY_TEXT, truncated);
       }
@@ -196,16 +201,18 @@ function instrumentQueryMethod(BigQuery) {
     if (options?.destination) {
       const dest = extractTableReference(options.destination);
       if (dest.tableId) {
-        const destTable = dest.datasetId ? `${dest.datasetId}.${dest.tableId}` : dest.tableId;
+        const destTable = dest.datasetId
+          ? `${dest.datasetId}.${dest.tableId}`
+          : dest.tableId;
         span.setAttribute(SEMATTRS_GCP_BIGQUERY_DESTINATION_TABLE, destTable);
       }
     }
-    if (typeof callback === "function") {
+    if (typeof callback === 'function') {
       const wrappedCallback = (err, ...args) => {
         if (err) {
           finalizeSpan(
             span,
-            err instanceof Error ? err : new Error(String(err))
+            err instanceof Error ? err : new Error(String(err)),
           );
         } else {
           const [rows, response] = args;
@@ -215,7 +222,7 @@ function instrumentQueryMethod(BigQuery) {
           if (response?.jobReference?.jobId) {
             span.setAttribute(
               SEMATTRS_GCP_BIGQUERY_JOB_ID,
-              response.jobReference.jobId
+              response.jobReference.jobId,
             );
           }
           finalizeSpan(span);
@@ -227,32 +234,34 @@ function instrumentQueryMethod(BigQuery) {
     return runWithSpan(span, () => {
       try {
         const result = originalQuery.call(this, query, options);
-        return Promise.resolve(result).then(([rows, response]) => {
-          if (Array.isArray(rows)) {
-            span.setAttribute(
-              SEMATTRS_GCP_BIGQUERY_ROWS_RETURNED,
-              rows.length
+        return Promise.resolve(result)
+          .then(([rows, response]) => {
+            if (Array.isArray(rows)) {
+              span.setAttribute(
+                SEMATTRS_GCP_BIGQUERY_ROWS_RETURNED,
+                rows.length,
+              );
+            }
+            if (response?.jobReference?.jobId) {
+              span.setAttribute(
+                SEMATTRS_GCP_BIGQUERY_JOB_ID,
+                response.jobReference.jobId,
+              );
+            }
+            finalizeSpan(span);
+            return [rows, response];
+          })
+          .catch((error) => {
+            finalizeSpan(
+              span,
+              error instanceof Error ? error : new Error(String(error)),
             );
-          }
-          if (response?.jobReference?.jobId) {
-            span.setAttribute(
-              SEMATTRS_GCP_BIGQUERY_JOB_ID,
-              response.jobReference.jobId
-            );
-          }
-          finalizeSpan(span);
-          return [rows, response];
-        }).catch((error) => {
-          finalizeSpan(
-            span,
-            error instanceof Error ? error : new Error(String(error))
-          );
-          throw error;
-        });
+            throw error;
+          });
       } catch (error) {
         finalizeSpan(
           span,
-          error instanceof Error ? error : new Error(String(error))
+          error instanceof Error ? error : new Error(String(error)),
         );
         throw error;
       }
@@ -261,43 +270,45 @@ function instrumentQueryMethod(BigQuery) {
 }
 function instrumentCreateQueryJob(BigQuery) {
   const originalCreateQueryJob = BigQuery.prototype.createQueryJob;
-  if (typeof originalCreateQueryJob !== "function") {
+  if (typeof originalCreateQueryJob !== 'function') {
     return;
   }
-  BigQuery.prototype.createQueryJob = function instrumentedCreateQueryJob(options) {
+  BigQuery.prototype.createQueryJob = function instrumentedCreateQueryJob(
+    options,
+  ) {
     const config = getInstanceConfig(this);
     const tracer = getInstanceTracer(this);
     if (!config || !tracer) {
       return originalCreateQueryJob.call(this, options);
     }
-    const queryText = typeof options === "string" ? options : options.query;
+    const queryText = typeof options === 'string' ? options : options.query;
     const projectId = extractProjectId(this);
     const location = extractLocation(this, options);
-    const operation = queryText ? extractOperationType(queryText) : "QUERY";
-    const summary = queryText ? createQuerySummary(queryText) : "QUERY";
+    const operation = queryText ? extractOperationType(queryText) : 'QUERY';
+    const summary = queryText ? createQuerySummary(queryText) : 'QUERY';
     const span = createSpan(
       tracer,
-      `${operation || "QUERY"}_JOB`,
+      `${operation || 'QUERY'}_JOB`,
       summary,
       projectId,
-      config
+      config,
     );
     if (location) {
       span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_LOCATION, location);
     }
     span.setAttribute(SEMATTRS_DB_QUERY_SUMMARY, summary);
-    if (config.captureQueryText !== "never" && queryText) {
-      if (config.captureQueryText === "raw") {
+    if (config.captureQueryText !== 'never' && queryText) {
+      if (config.captureQueryText === 'raw') {
         const truncated = truncateText(
           queryText,
-          config.maxQueryTextLength || 1e3
+          config.maxQueryTextLength || 1e3,
         );
         span.setAttribute(SEMATTRS_DB_QUERY_TEXT, truncated);
-      } else if (config.captureQueryText === "sanitized") {
+      } else if (config.captureQueryText === 'sanitized') {
         const sanitized = sanitizeQuery(queryText);
         const truncated = truncateText(
           sanitized,
-          config.maxQueryTextLength || 1e3
+          config.maxQueryTextLength || 1e3,
         );
         span.setAttribute(SEMATTRS_DB_QUERY_TEXT, truncated);
       }
@@ -308,35 +319,39 @@ function instrumentCreateQueryJob(BigQuery) {
     if (options?.destination) {
       const dest = extractTableReference(options.destination);
       if (dest.tableId) {
-        const destTable = dest.datasetId ? `${dest.datasetId}.${dest.tableId}` : dest.tableId;
+        const destTable = dest.datasetId
+          ? `${dest.datasetId}.${dest.tableId}`
+          : dest.tableId;
         span.setAttribute(SEMATTRS_GCP_BIGQUERY_DESTINATION_TABLE, destTable);
       }
     }
     return runWithSpan(span, () => {
       try {
         const result = originalCreateQueryJob.call(this, options);
-        return Promise.resolve(result).then(([job, response]) => {
-          if (job?.id) {
-            span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_ID, job.id);
-          } else if (response?.jobReference?.jobId) {
-            span.setAttribute(
-              SEMATTRS_GCP_BIGQUERY_JOB_ID,
-              response.jobReference.jobId
+        return Promise.resolve(result)
+          .then(([job, response]) => {
+            if (job?.id) {
+              span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_ID, job.id);
+            } else if (response?.jobReference?.jobId) {
+              span.setAttribute(
+                SEMATTRS_GCP_BIGQUERY_JOB_ID,
+                response.jobReference.jobId,
+              );
+            }
+            finalizeSpan(span);
+            return [job, response];
+          })
+          .catch((error) => {
+            finalizeSpan(
+              span,
+              error instanceof Error ? error : new Error(String(error)),
             );
-          }
-          finalizeSpan(span);
-          return [job, response];
-        }).catch((error) => {
-          finalizeSpan(
-            span,
-            error instanceof Error ? error : new Error(String(error))
-          );
-          throw error;
-        });
+            throw error;
+          });
       } catch (error) {
         finalizeSpan(
           span,
-          error instanceof Error ? error : new Error(String(error))
+          error instanceof Error ? error : new Error(String(error)),
         );
         throw error;
       }
@@ -345,7 +360,7 @@ function instrumentCreateQueryJob(BigQuery) {
 }
 function instrumentTableInsert(Table) {
   const originalInsert = Table.prototype.insert;
-  if (typeof originalInsert !== "function") {
+  if (typeof originalInsert !== 'function') {
     return;
   }
   Table.prototype.insert = function instrumentedInsert(rows, options) {
@@ -358,10 +373,14 @@ function instrumentTableInsert(Table) {
     const projectId = extractProjectId(this.dataset?.parent || this.parent);
     const location = extractLocation(
       this.dataset?.parent || this.parent,
-      options
+      options,
     );
-    const target = tableRef.tableId ? tableRef.datasetId ? `${tableRef.datasetId}.${tableRef.tableId}` : tableRef.tableId : void 0;
-    const span = createSpan(tracer, "INSERT", target, projectId, config);
+    const target = tableRef.tableId
+      ? tableRef.datasetId
+        ? `${tableRef.datasetId}.${tableRef.tableId}`
+        : tableRef.tableId
+      : void 0;
+    const span = createSpan(tracer, 'INSERT', target, projectId, config);
     if (location) {
       span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_LOCATION, location);
     }
@@ -376,26 +395,28 @@ function instrumentTableInsert(Table) {
     return runWithSpan(span, () => {
       try {
         const result = originalInsert.call(this, rows, options);
-        return Promise.resolve(result).then((response) => {
-          if (response?.insertErrors?.length > 0) {
-            span.setAttribute(
-              "gcp.bigquery.insert.errors",
-              response.insertErrors.length
+        return Promise.resolve(result)
+          .then((response) => {
+            if (response?.insertErrors?.length > 0) {
+              span.setAttribute(
+                'gcp.bigquery.insert.errors',
+                response.insertErrors.length,
+              );
+            }
+            finalizeSpan(span);
+            return response;
+          })
+          .catch((error) => {
+            finalizeSpan(
+              span,
+              error instanceof Error ? error : new Error(String(error)),
             );
-          }
-          finalizeSpan(span);
-          return response;
-        }).catch((error) => {
-          finalizeSpan(
-            span,
-            error instanceof Error ? error : new Error(String(error))
-          );
-          throw error;
-        });
+            throw error;
+          });
       } catch (error) {
         finalizeSpan(
           span,
-          error instanceof Error ? error : new Error(String(error))
+          error instanceof Error ? error : new Error(String(error)),
         );
         throw error;
       }
@@ -404,7 +425,7 @@ function instrumentTableInsert(Table) {
 }
 function instrumentTableGetRows(Table) {
   const originalGetRows = Table.prototype.getRows;
-  if (typeof originalGetRows !== "function") {
+  if (typeof originalGetRows !== 'function') {
     return;
   }
   Table.prototype.getRows = function instrumentedGetRows(options) {
@@ -417,10 +438,14 @@ function instrumentTableGetRows(Table) {
     const projectId = extractProjectId(this.dataset?.parent || this.parent);
     const location = extractLocation(
       this.dataset?.parent || this.parent,
-      options
+      options,
     );
-    const target = tableRef.tableId ? tableRef.datasetId ? `${tableRef.datasetId}.${tableRef.tableId}` : tableRef.tableId : void 0;
-    const span = createSpan(tracer, "SELECT", target, projectId, config);
+    const target = tableRef.tableId
+      ? tableRef.datasetId
+        ? `${tableRef.datasetId}.${tableRef.tableId}`
+        : tableRef.tableId
+      : void 0;
+    const span = createSpan(tracer, 'SELECT', target, projectId, config);
     if (location) {
       span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_LOCATION, location);
     }
@@ -433,26 +458,28 @@ function instrumentTableGetRows(Table) {
     return runWithSpan(span, () => {
       try {
         const result = originalGetRows.call(this, options);
-        return Promise.resolve(result).then(([rows, nextQuery, apiResponse]) => {
-          if (Array.isArray(rows)) {
-            span.setAttribute(
-              SEMATTRS_GCP_BIGQUERY_ROWS_RETURNED,
-              rows.length
+        return Promise.resolve(result)
+          .then(([rows, nextQuery, apiResponse]) => {
+            if (Array.isArray(rows)) {
+              span.setAttribute(
+                SEMATTRS_GCP_BIGQUERY_ROWS_RETURNED,
+                rows.length,
+              );
+            }
+            finalizeSpan(span);
+            return [rows, nextQuery, apiResponse];
+          })
+          .catch((error) => {
+            finalizeSpan(
+              span,
+              error instanceof Error ? error : new Error(String(error)),
             );
-          }
-          finalizeSpan(span);
-          return [rows, nextQuery, apiResponse];
-        }).catch((error) => {
-          finalizeSpan(
-            span,
-            error instanceof Error ? error : new Error(String(error))
-          );
-          throw error;
-        });
+            throw error;
+          });
       } catch (error) {
         finalizeSpan(
           span,
-          error instanceof Error ? error : new Error(String(error))
+          error instanceof Error ? error : new Error(String(error)),
         );
         throw error;
       }
@@ -461,10 +488,13 @@ function instrumentTableGetRows(Table) {
 }
 function instrumentTableCreateLoadJob(Table) {
   const originalCreateLoadJob = Table.prototype.createLoadJob;
-  if (typeof originalCreateLoadJob !== "function") {
+  if (typeof originalCreateLoadJob !== 'function') {
     return;
   }
-  Table.prototype.createLoadJob = function instrumentedCreateLoadJob(source, metadata) {
+  Table.prototype.createLoadJob = function instrumentedCreateLoadJob(
+    source,
+    metadata,
+  ) {
     const config = getInstanceConfig(this);
     const tracer = getInstanceTracer(this);
     if (!config || !tracer) {
@@ -474,10 +504,14 @@ function instrumentTableCreateLoadJob(Table) {
     const projectId = extractProjectId(this.dataset?.parent || this.parent);
     const location = extractLocation(
       this.dataset?.parent || this.parent,
-      metadata
+      metadata,
     );
-    const target = tableRef.tableId ? tableRef.datasetId ? `${tableRef.datasetId}.${tableRef.tableId}` : tableRef.tableId : void 0;
-    const span = createSpan(tracer, "LOAD", target, projectId, config);
+    const target = tableRef.tableId
+      ? tableRef.datasetId
+        ? `${tableRef.datasetId}.${tableRef.tableId}`
+        : tableRef.tableId
+      : void 0;
+    const span = createSpan(tracer, 'LOAD', target, projectId, config);
     if (location) {
       span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_LOCATION, location);
     }
@@ -488,28 +522,30 @@ function instrumentTableCreateLoadJob(Table) {
       span.setAttribute(SEMATTRS_DB_COLLECTION_NAME, tableRef.tableId);
     }
     if (metadata?.sourceFormat) {
-      span.setAttribute("gcp.bigquery.source.format", metadata.sourceFormat);
+      span.setAttribute('gcp.bigquery.source.format', metadata.sourceFormat);
     }
     return runWithSpan(span, () => {
       try {
         const result = originalCreateLoadJob.call(this, source, metadata);
-        return Promise.resolve(result).then(([job, response]) => {
-          if (job?.id) {
-            span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_ID, job.id);
-          }
-          finalizeSpan(span);
-          return [job, response];
-        }).catch((error) => {
-          finalizeSpan(
-            span,
-            error instanceof Error ? error : new Error(String(error))
-          );
-          throw error;
-        });
+        return Promise.resolve(result)
+          .then(([job, response]) => {
+            if (job?.id) {
+              span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_ID, job.id);
+            }
+            finalizeSpan(span);
+            return [job, response];
+          })
+          .catch((error) => {
+            finalizeSpan(
+              span,
+              error instanceof Error ? error : new Error(String(error)),
+            );
+            throw error;
+          });
       } catch (error) {
         finalizeSpan(
           span,
-          error instanceof Error ? error : new Error(String(error))
+          error instanceof Error ? error : new Error(String(error)),
         );
         throw error;
       }
@@ -518,10 +554,13 @@ function instrumentTableCreateLoadJob(Table) {
 }
 function instrumentTableCreateCopyJob(Table) {
   const originalCreateCopyJob = Table.prototype.createCopyJob;
-  if (typeof originalCreateCopyJob !== "function") {
+  if (typeof originalCreateCopyJob !== 'function') {
     return;
   }
-  Table.prototype.createCopyJob = function instrumentedCreateCopyJob(destination, metadata) {
+  Table.prototype.createCopyJob = function instrumentedCreateCopyJob(
+    destination,
+    metadata,
+  ) {
     const config = getInstanceConfig(this);
     const tracer = getInstanceTracer(this);
     if (!config || !tracer) {
@@ -531,16 +570,24 @@ function instrumentTableCreateCopyJob(Table) {
     const projectId = extractProjectId(this.dataset?.parent || this.parent);
     const location = extractLocation(
       this.dataset?.parent || this.parent,
-      metadata
+      metadata,
     );
-    const source = sourceRef.tableId ? sourceRef.datasetId ? `${sourceRef.datasetId}.${sourceRef.tableId}` : sourceRef.tableId : "unknown";
-    const span = createSpan(tracer, "COPY", source, projectId, config);
+    const source = sourceRef.tableId
+      ? sourceRef.datasetId
+        ? `${sourceRef.datasetId}.${sourceRef.tableId}`
+        : sourceRef.tableId
+      : 'unknown';
+    const span = createSpan(tracer, 'COPY', source, projectId, config);
     if (location) {
       span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_LOCATION, location);
     }
     if (destination) {
       const destRef = extractTableReference(destination);
-      const destTable = destRef.tableId ? destRef.datasetId ? `${destRef.datasetId}.${destRef.tableId}` : destRef.tableId : void 0;
+      const destTable = destRef.tableId
+        ? destRef.datasetId
+          ? `${destRef.datasetId}.${destRef.tableId}`
+          : destRef.tableId
+        : void 0;
       if (destTable) {
         span.setAttribute(SEMATTRS_GCP_BIGQUERY_DESTINATION_TABLE, destTable);
       }
@@ -548,23 +595,25 @@ function instrumentTableCreateCopyJob(Table) {
     return runWithSpan(span, () => {
       try {
         const result = originalCreateCopyJob.call(this, destination, metadata);
-        return Promise.resolve(result).then(([job, response]) => {
-          if (job?.id) {
-            span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_ID, job.id);
-          }
-          finalizeSpan(span);
-          return [job, response];
-        }).catch((error) => {
-          finalizeSpan(
-            span,
-            error instanceof Error ? error : new Error(String(error))
-          );
-          throw error;
-        });
+        return Promise.resolve(result)
+          .then(([job, response]) => {
+            if (job?.id) {
+              span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_ID, job.id);
+            }
+            finalizeSpan(span);
+            return [job, response];
+          })
+          .catch((error) => {
+            finalizeSpan(
+              span,
+              error instanceof Error ? error : new Error(String(error)),
+            );
+            throw error;
+          });
       } catch (error) {
         finalizeSpan(
           span,
-          error instanceof Error ? error : new Error(String(error))
+          error instanceof Error ? error : new Error(String(error)),
         );
         throw error;
       }
@@ -573,10 +622,13 @@ function instrumentTableCreateCopyJob(Table) {
 }
 function instrumentTableCreateExtractJob(Table) {
   const originalCreateExtractJob = Table.prototype.createExtractJob;
-  if (typeof originalCreateExtractJob !== "function") {
+  if (typeof originalCreateExtractJob !== 'function') {
     return;
   }
-  Table.prototype.createExtractJob = function instrumentedCreateExtractJob(destination, metadata) {
+  Table.prototype.createExtractJob = function instrumentedCreateExtractJob(
+    destination,
+    metadata,
+  ) {
     const config = getInstanceConfig(this);
     const tracer = getInstanceTracer(this);
     if (!config || !tracer) {
@@ -586,10 +638,14 @@ function instrumentTableCreateExtractJob(Table) {
     const projectId = extractProjectId(this.dataset?.parent || this.parent);
     const location = extractLocation(
       this.dataset?.parent || this.parent,
-      metadata
+      metadata,
     );
-    const source = sourceRef.tableId ? sourceRef.datasetId ? `${sourceRef.datasetId}.${sourceRef.tableId}` : sourceRef.tableId : "unknown";
-    const span = createSpan(tracer, "EXTRACT", source, projectId, config);
+    const source = sourceRef.tableId
+      ? sourceRef.datasetId
+        ? `${sourceRef.datasetId}.${sourceRef.tableId}`
+        : sourceRef.tableId
+      : 'unknown';
+    const span = createSpan(tracer, 'EXTRACT', source, projectId, config);
     if (location) {
       span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_LOCATION, location);
     }
@@ -599,39 +655,41 @@ function instrumentTableCreateExtractJob(Table) {
     if (sourceRef.tableId) {
       span.setAttribute(SEMATTRS_DB_COLLECTION_NAME, sourceRef.tableId);
     }
-    if (typeof destination === "string") {
-      span.setAttribute("gcp.bigquery.destination.uri", destination);
+    if (typeof destination === 'string') {
+      span.setAttribute('gcp.bigquery.destination.uri', destination);
     } else if (Array.isArray(destination) && destination.length > 0) {
-      span.setAttribute("gcp.bigquery.destination.uri", destination[0]);
+      span.setAttribute('gcp.bigquery.destination.uri', destination[0]);
     }
     const format = metadata?.destinationFormat || metadata?.format;
     if (format) {
-      span.setAttribute("gcp.bigquery.destination.format", format);
+      span.setAttribute('gcp.bigquery.destination.format', format);
     }
     return runWithSpan(span, () => {
       try {
         const result = originalCreateExtractJob.call(
           this,
           destination,
-          metadata
+          metadata,
         );
-        return Promise.resolve(result).then(([job, response]) => {
-          if (job?.id) {
-            span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_ID, job.id);
-          }
-          finalizeSpan(span);
-          return [job, response];
-        }).catch((error) => {
-          finalizeSpan(
-            span,
-            error instanceof Error ? error : new Error(String(error))
-          );
-          throw error;
-        });
+        return Promise.resolve(result)
+          .then(([job, response]) => {
+            if (job?.id) {
+              span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_ID, job.id);
+            }
+            finalizeSpan(span);
+            return [job, response];
+          })
+          .catch((error) => {
+            finalizeSpan(
+              span,
+              error instanceof Error ? error : new Error(String(error)),
+            );
+            throw error;
+          });
       } catch (error) {
         finalizeSpan(
           span,
-          error instanceof Error ? error : new Error(String(error))
+          error instanceof Error ? error : new Error(String(error)),
         );
         throw error;
       }
@@ -640,10 +698,12 @@ function instrumentTableCreateExtractJob(Table) {
 }
 function instrumentJobGetQueryResults(Job) {
   const originalGetQueryResults = Job.prototype.getQueryResults;
-  if (typeof originalGetQueryResults !== "function") {
+  if (typeof originalGetQueryResults !== 'function') {
     return;
   }
-  Job.prototype.getQueryResults = function instrumentedGetQueryResults(options) {
+  Job.prototype.getQueryResults = function instrumentedGetQueryResults(
+    options,
+  ) {
     const config = getInstanceConfig(this);
     const tracer = getInstanceTracer(this);
     if (!config || !tracer) {
@@ -654,10 +714,10 @@ function instrumentJobGetQueryResults(Job) {
     const location = this.location || options?.location;
     const span = createSpan(
       tracer,
-      "GET_QUERY_RESULTS",
+      'GET_QUERY_RESULTS',
       void 0,
       projectId,
-      config
+      config,
     );
     if (jobId) {
       span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_ID, jobId);
@@ -668,26 +728,28 @@ function instrumentJobGetQueryResults(Job) {
     return runWithSpan(span, () => {
       try {
         const result = originalGetQueryResults.call(this, options);
-        return Promise.resolve(result).then(([rows, nextQuery, apiResponse]) => {
-          if (Array.isArray(rows)) {
-            span.setAttribute(
-              SEMATTRS_GCP_BIGQUERY_ROWS_RETURNED,
-              rows.length
+        return Promise.resolve(result)
+          .then(([rows, nextQuery, apiResponse]) => {
+            if (Array.isArray(rows)) {
+              span.setAttribute(
+                SEMATTRS_GCP_BIGQUERY_ROWS_RETURNED,
+                rows.length,
+              );
+            }
+            finalizeSpan(span);
+            return [rows, nextQuery, apiResponse];
+          })
+          .catch((error) => {
+            finalizeSpan(
+              span,
+              error instanceof Error ? error : new Error(String(error)),
             );
-          }
-          finalizeSpan(span);
-          return [rows, nextQuery, apiResponse];
-        }).catch((error) => {
-          finalizeSpan(
-            span,
-            error instanceof Error ? error : new Error(String(error))
-          );
-          throw error;
-        });
+            throw error;
+          });
       } catch (error) {
         finalizeSpan(
           span,
-          error instanceof Error ? error : new Error(String(error))
+          error instanceof Error ? error : new Error(String(error)),
         );
         throw error;
       }
@@ -696,7 +758,7 @@ function instrumentJobGetQueryResults(Job) {
 }
 function instrumentDatasetCreate(Dataset) {
   const originalCreate = Dataset.prototype.create;
-  if (typeof originalCreate !== "function") {
+  if (typeof originalCreate !== 'function') {
     return;
   }
   Dataset.prototype.create = function instrumentedCreate(options) {
@@ -710,10 +772,10 @@ function instrumentDatasetCreate(Dataset) {
     const location = extractLocation(this.parent, options);
     const span = createSpan(
       tracer,
-      "CREATE_DATASET",
+      'CREATE_DATASET',
       datasetId,
       projectId,
-      config
+      config,
     );
     if (location) {
       span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_LOCATION, location);
@@ -724,20 +786,22 @@ function instrumentDatasetCreate(Dataset) {
     return runWithSpan(span, () => {
       try {
         const result = originalCreate.call(this, options);
-        return Promise.resolve(result).then((response) => {
-          finalizeSpan(span);
-          return response;
-        }).catch((error) => {
-          finalizeSpan(
-            span,
-            error instanceof Error ? error : new Error(String(error))
-          );
-          throw error;
-        });
+        return Promise.resolve(result)
+          .then((response) => {
+            finalizeSpan(span);
+            return response;
+          })
+          .catch((error) => {
+            finalizeSpan(
+              span,
+              error instanceof Error ? error : new Error(String(error)),
+            );
+            throw error;
+          });
       } catch (error) {
         finalizeSpan(
           span,
-          error instanceof Error ? error : new Error(String(error))
+          error instanceof Error ? error : new Error(String(error)),
         );
         throw error;
       }
@@ -746,7 +810,7 @@ function instrumentDatasetCreate(Dataset) {
 }
 function instrumentDatasetDelete(Dataset) {
   const originalDelete = Dataset.prototype.delete;
-  if (typeof originalDelete !== "function") {
+  if (typeof originalDelete !== 'function') {
     return;
   }
   Dataset.prototype.delete = function instrumentedDelete(options) {
@@ -759,10 +823,10 @@ function instrumentDatasetDelete(Dataset) {
     const projectId = extractProjectId(this.parent);
     const span = createSpan(
       tracer,
-      "DELETE_DATASET",
+      'DELETE_DATASET',
       datasetId,
       projectId,
-      config
+      config,
     );
     if (datasetId) {
       span.setAttribute(SEMATTRS_DB_NAMESPACE, datasetId);
@@ -770,20 +834,22 @@ function instrumentDatasetDelete(Dataset) {
     return runWithSpan(span, () => {
       try {
         const result = originalDelete.call(this, options);
-        return Promise.resolve(result).then((response) => {
-          finalizeSpan(span);
-          return response;
-        }).catch((error) => {
-          finalizeSpan(
-            span,
-            error instanceof Error ? error : new Error(String(error))
-          );
-          throw error;
-        });
+        return Promise.resolve(result)
+          .then((response) => {
+            finalizeSpan(span);
+            return response;
+          })
+          .catch((error) => {
+            finalizeSpan(
+              span,
+              error instanceof Error ? error : new Error(String(error)),
+            );
+            throw error;
+          });
       } catch (error) {
         finalizeSpan(
           span,
-          error instanceof Error ? error : new Error(String(error))
+          error instanceof Error ? error : new Error(String(error)),
         );
         throw error;
       }
@@ -792,10 +858,14 @@ function instrumentDatasetDelete(Dataset) {
 }
 function instrumentBigQueryCreateDataset(BigQuery) {
   const originalCreateDataset = BigQuery.prototype.createDataset;
-  if (typeof originalCreateDataset !== "function") {
+  if (typeof originalCreateDataset !== 'function') {
     return;
   }
-  BigQuery.prototype.createDataset = function instrumentedCreateDataset(id, options, callback) {
+  BigQuery.prototype.createDataset = function instrumentedCreateDataset(
+    id,
+    options,
+    callback,
+  ) {
     const config = getInstanceConfig(this);
     const tracer = getInstanceTracer(this);
     if (!config || !tracer || !config.instrumentAdminOps) {
@@ -806,10 +876,10 @@ function instrumentBigQueryCreateDataset(BigQuery) {
     const location = extractLocation(this, options);
     const span = createSpan(
       tracer,
-      "CREATE_DATASET",
+      'CREATE_DATASET',
       datasetId,
       projectId,
-      config
+      config,
     );
     if (location) {
       span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_LOCATION, location);
@@ -817,12 +887,12 @@ function instrumentBigQueryCreateDataset(BigQuery) {
     if (datasetId) {
       span.setAttribute(SEMATTRS_DB_NAMESPACE, datasetId);
     }
-    if (typeof callback === "function") {
+    if (typeof callback === 'function') {
       const wrappedCallback = (err, ...args) => {
         if (err) {
           finalizeSpan(
             span,
-            err instanceof Error ? err : new Error(String(err))
+            err instanceof Error ? err : new Error(String(err)),
           );
         } else {
           finalizeSpan(span);
@@ -834,20 +904,22 @@ function instrumentBigQueryCreateDataset(BigQuery) {
     return runWithSpan(span, () => {
       try {
         const result = originalCreateDataset.call(this, id, options);
-        return Promise.resolve(result).then((response) => {
-          finalizeSpan(span);
-          return response;
-        }).catch((error) => {
-          finalizeSpan(
-            span,
-            error instanceof Error ? error : new Error(String(error))
-          );
-          throw error;
-        });
+        return Promise.resolve(result)
+          .then((response) => {
+            finalizeSpan(span);
+            return response;
+          })
+          .catch((error) => {
+            finalizeSpan(
+              span,
+              error instanceof Error ? error : new Error(String(error)),
+            );
+            throw error;
+          });
       } catch (error) {
         finalizeSpan(
           span,
-          error instanceof Error ? error : new Error(String(error))
+          error instanceof Error ? error : new Error(String(error)),
         );
         throw error;
       }
@@ -856,7 +928,7 @@ function instrumentBigQueryCreateDataset(BigQuery) {
 }
 function instrumentTableCreate(Table) {
   const originalCreate = Table.prototype.create;
-  if (typeof originalCreate !== "function") {
+  if (typeof originalCreate !== 'function') {
     return;
   }
   Table.prototype.create = function instrumentedCreate(options) {
@@ -867,8 +939,12 @@ function instrumentTableCreate(Table) {
     }
     const tableRef = extractTableReference(this);
     const projectId = extractProjectId(this.dataset?.parent || this.parent);
-    const target = tableRef.tableId ? tableRef.datasetId ? `${tableRef.datasetId}.${tableRef.tableId}` : tableRef.tableId : void 0;
-    const span = createSpan(tracer, "CREATE_TABLE", target, projectId, config);
+    const target = tableRef.tableId
+      ? tableRef.datasetId
+        ? `${tableRef.datasetId}.${tableRef.tableId}`
+        : tableRef.tableId
+      : void 0;
+    const span = createSpan(tracer, 'CREATE_TABLE', target, projectId, config);
     if (tableRef.datasetId) {
       span.setAttribute(SEMATTRS_DB_NAMESPACE, tableRef.datasetId);
     }
@@ -878,20 +954,22 @@ function instrumentTableCreate(Table) {
     return runWithSpan(span, () => {
       try {
         const result = originalCreate.call(this, options);
-        return Promise.resolve(result).then((response) => {
-          finalizeSpan(span);
-          return response;
-        }).catch((error) => {
-          finalizeSpan(
-            span,
-            error instanceof Error ? error : new Error(String(error))
-          );
-          throw error;
-        });
+        return Promise.resolve(result)
+          .then((response) => {
+            finalizeSpan(span);
+            return response;
+          })
+          .catch((error) => {
+            finalizeSpan(
+              span,
+              error instanceof Error ? error : new Error(String(error)),
+            );
+            throw error;
+          });
       } catch (error) {
         finalizeSpan(
           span,
-          error instanceof Error ? error : new Error(String(error))
+          error instanceof Error ? error : new Error(String(error)),
         );
         throw error;
       }
@@ -900,7 +978,7 @@ function instrumentTableCreate(Table) {
 }
 function instrumentTableDelete(Table) {
   const originalDelete = Table.prototype.delete;
-  if (typeof originalDelete !== "function") {
+  if (typeof originalDelete !== 'function') {
     return;
   }
   Table.prototype.delete = function instrumentedDelete(options) {
@@ -911,8 +989,12 @@ function instrumentTableDelete(Table) {
     }
     const tableRef = extractTableReference(this);
     const projectId = extractProjectId(this.dataset?.parent || this.parent);
-    const target = tableRef.tableId ? tableRef.datasetId ? `${tableRef.datasetId}.${tableRef.tableId}` : tableRef.tableId : void 0;
-    const span = createSpan(tracer, "DELETE_TABLE", target, projectId, config);
+    const target = tableRef.tableId
+      ? tableRef.datasetId
+        ? `${tableRef.datasetId}.${tableRef.tableId}`
+        : tableRef.tableId
+      : void 0;
+    const span = createSpan(tracer, 'DELETE_TABLE', target, projectId, config);
     if (tableRef.datasetId) {
       span.setAttribute(SEMATTRS_DB_NAMESPACE, tableRef.datasetId);
     }
@@ -922,20 +1004,22 @@ function instrumentTableDelete(Table) {
     return runWithSpan(span, () => {
       try {
         const result = originalDelete.call(this, options);
-        return Promise.resolve(result).then((response) => {
-          finalizeSpan(span);
-          return response;
-        }).catch((error) => {
-          finalizeSpan(
-            span,
-            error instanceof Error ? error : new Error(String(error))
-          );
-          throw error;
-        });
+        return Promise.resolve(result)
+          .then((response) => {
+            finalizeSpan(span);
+            return response;
+          })
+          .catch((error) => {
+            finalizeSpan(
+              span,
+              error instanceof Error ? error : new Error(String(error)),
+            );
+            throw error;
+          });
       } catch (error) {
         finalizeSpan(
           span,
-          error instanceof Error ? error : new Error(String(error))
+          error instanceof Error ? error : new Error(String(error)),
         );
         throw error;
       }
@@ -951,15 +1035,15 @@ function instrumentBigQuery(bigquery, config) {
     return bigquery;
   }
   const finalConfig = {
-    projectId: config?.projectId || "",
-    location: config?.location || "",
+    projectId: config?.projectId || '',
+    location: config?.location || '',
     tracerName: config?.tracerName || DEFAULT_TRACER_NAME,
-    captureQueryText: config?.captureQueryText || "summary",
+    captureQueryText: config?.captureQueryText || 'summary',
     maxQueryTextLength: config?.maxQueryTextLength || 1e3,
     includeQueryHash: config?.includeQueryHash ?? true,
     instrumentAdminOps: config?.instrumentAdminOps ?? false,
     instrumentBqmlOps: config?.instrumentBqmlOps ?? false,
-    instrumentRoutineOps: config?.instrumentRoutineOps ?? false
+    instrumentRoutineOps: config?.instrumentRoutineOps ?? false,
   };
   const tracer = otelTrace.getTracer(finalConfig.tracerName);
   bq[CONFIG_STORAGE_KEY] = finalConfig;
@@ -971,18 +1055,17 @@ function instrumentBigQuery(bigquery, config) {
     instrumentBigQueryCreateDataset(BigQuery);
     if (bq.dataset) {
       try {
-        const tempDataset = bq.dataset("__temp__");
+        const tempDataset = bq.dataset('__temp__');
         const Dataset = tempDataset.constructor;
         instrumentDatasetCreate(Dataset);
         instrumentDatasetDelete(Dataset);
         tempDataset.dataset = null;
-      } catch {
-      }
+      } catch {}
     }
     if (bq.dataset) {
       try {
-        const tempDataset = bq.dataset("__temp__");
-        const tempTable = tempDataset.table("__temp__");
+        const tempDataset = bq.dataset('__temp__');
+        const tempTable = tempDataset.table('__temp__');
         const Table = tempTable.constructor;
         instrumentTableInsert(Table);
         instrumentTableGetRows(Table);
@@ -993,17 +1076,15 @@ function instrumentBigQuery(bigquery, config) {
         instrumentTableDelete(Table);
         tempTable.table = null;
         tempDataset.dataset = null;
-      } catch {
-      }
+      } catch {}
     }
     if (bq.job) {
       try {
-        const tempJob = bq.job("__temp__");
+        const tempJob = bq.job('__temp__');
         const Job = tempJob.constructor;
         instrumentJobGetQueryResults(Job);
         tempJob.job = null;
-      } catch {
-      }
+      } catch {}
     }
     BigQuery[PROTOTYPE_INSTRUMENTED_FLAG] = true;
   }
@@ -1022,4 +1103,4 @@ var BigQueryInstrumentation = class {
 
 export { BigQueryInstrumentation, instrumentBigQuery };
 //# sourceMappingURL=bigquery.js.map
-//# sourceMappingURL=bigquery.js.map
+//# sourceMappingURL=bigquery.js.map