autotel-plugins 0.19.2 → 0.19.4

package/dist/index.cjs

@@ -1,8 +1,8 @@
 'use strict';
 
-var api = require('@opentelemetry/api');
-var traceHelpers = require('autotel/trace-helpers');
 var autotel = require('autotel');
+var traceHelpers = require('autotel/trace-helpers');
+var api = require('@opentelemetry/api');
 
 // src/common/constants.ts
 var SEMATTRS_DB_SYSTEM = "db.system";
@@ -15,7 +15,6 @@ 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_DB_MONGODB_COLLECTION = "db.mongodb.collection";
 var SEMATTRS_NET_PEER_NAME = "net.peer.name";
 var SEMATTRS_NET_PEER_PORT = "net.peer.port";
 var SEMATTRS_MESSAGING_SYSTEM = "messaging.system";
@@ -52,847 +51,9 @@ var SEMATTRS_MESSAGING_KAFKA_BATCH_LAST_OFFSET = "messaging.kafka.batch.last_off
 var SEMATTRS_MESSAGING_KAFKA_BATCH_MESSAGES_PROCESSED = "messaging.kafka.batch.messages_processed";
 var SEMATTRS_MESSAGING_KAFKA_BATCH_MESSAGES_FAILED = "messaging.kafka.batch.messages_failed";
 var SEMATTRS_MESSAGING_KAFKA_BATCH_PROCESSING_TIME_MS = "messaging.kafka.batch.processing_time_ms";
-var DEFAULT_TRACER_NAME = "autotel-plugins/drizzle";
-var DEFAULT_DB_SYSTEM = "postgresql";
-var INSTRUMENTED_FLAG = "__autotelDrizzleInstrumented";
-function extractQueryText(queryArg) {
-  if (typeof queryArg === "string") {
-    return queryArg;
-  }
-  if (queryArg && typeof queryArg === "object") {
-    if (typeof queryArg.sql === "string") {
-      return queryArg.sql;
-    }
-    if (typeof queryArg.text === "string") {
-      return queryArg.text;
-    }
-    if (typeof queryArg.queryChunks === "object") {
-      const drizzleQuery = queryArg;
-      if (typeof drizzleQuery.sql === "string") {
-        return drizzleQuery.sql;
-      }
-    }
-  }
-  return void 0;
-}
-function sanitizeQueryText(queryText, maxLength) {
-  if (queryText.length <= maxLength) {
-    return queryText;
-  }
-  return `${queryText.slice(0, Math.max(0, maxLength))}...`;
-}
-function extractOperation(queryText) {
-  const trimmed = queryText.trimStart();
-  const match = /^(?<op>\w+)/u.exec(trimmed);
-  return match?.groups?.op?.toUpperCase();
-}
-function instrumentDrizzle(client, config) {
-  if (!client) {
-    return client;
-  }
-  const hasQuery = typeof client.query === "function";
-  const hasExecute = typeof client.execute === "function";
-  if (!hasQuery && !hasExecute) {
-    return client;
-  }
-  if (client[INSTRUMENTED_FLAG]) {
-    return client;
-  }
-  const {
-    tracerName = DEFAULT_TRACER_NAME,
-    dbSystem = DEFAULT_DB_SYSTEM,
-    dbName,
-    captureQueryText = true,
-    maxQueryTextLength = 1e3,
-    peerName,
-    peerPort
-  } = config ?? {};
-  const tracer = api.trace.getTracer(tracerName);
-  const originalMethod = hasQuery ? client.query : client.execute;
-  if (!originalMethod) {
-    return client;
-  }
-  const instrumentedMethod = function instrumented(...incomingArgs) {
-    const args = [...incomingArgs];
-    let callback;
-    if (typeof args.at(-1) === "function") {
-      callback = args.pop();
-    }
-    const queryText = extractQueryText(args[0]);
-    const operation = queryText ? extractOperation(queryText) : void 0;
-    const spanName = operation ? `drizzle.${operation.toLowerCase()}` : "drizzle.query";
-    const span = tracer.startSpan(spanName, { kind: api.SpanKind.CLIENT });
-    span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem);
-    if (operation) {
-      span.setAttribute(SEMATTRS_DB_OPERATION, operation);
-    }
-    if (dbName) {
-      span.setAttribute(SEMATTRS_DB_NAME, dbName);
-    }
-    if (captureQueryText && queryText !== void 0) {
-      const sanitized = sanitizeQueryText(queryText, maxQueryTextLength);
-      span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);
-    }
-    if (peerName) {
-      span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName);
-    }
-    if (peerPort) {
-      span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort);
-    }
-    if (callback) {
-      return traceHelpers.runWithSpan(span, () => {
-        const wrappedCallback = (err, result) => {
-          traceHelpers.finalizeSpan(span, err);
-          if (callback) {
-            callback(err, result);
-          }
-        };
-        try {
-          return Reflect.apply(originalMethod, this, [
-            ...args,
-            wrappedCallback
-          ]);
-        } catch (error) {
-          traceHelpers.finalizeSpan(span, error);
-          throw error;
-        }
-      });
-    }
-    return traceHelpers.runWithSpan(span, () => {
-      try {
-        const result = originalMethod.apply(this, args);
-        return Promise.resolve(result).then((value) => {
-          traceHelpers.finalizeSpan(span);
-          return value;
-        }).catch((error) => {
-          traceHelpers.finalizeSpan(span, error);
-          throw error;
-        });
-      } catch (error) {
-        traceHelpers.finalizeSpan(span, error);
-        throw error;
-      }
-    });
-  };
-  client[INSTRUMENTED_FLAG] = true;
-  if (hasQuery) {
-    client.query = instrumentedMethod;
-  } else {
-    client.execute = instrumentedMethod;
-  }
-  return client;
-}
-function instrumentDrizzleClient(db, config) {
-  if (!db) {
-    return db;
-  }
-  if (db[INSTRUMENTED_FLAG]) {
-    return db;
-  }
-  const {
-    tracerName = DEFAULT_TRACER_NAME,
-    dbSystem = DEFAULT_DB_SYSTEM,
-    dbName,
-    captureQueryText = true,
-    maxQueryTextLength = 1e3,
-    peerName,
-    peerPort
-  } = config ?? {};
-  const tracer = api.trace.getTracer(tracerName);
-  let instrumented = false;
-  if (db.session && !instrumented) {
-    const session = db.session;
-    if (typeof session.prepareQuery === "function" && !session[INSTRUMENTED_FLAG]) {
-      const originalPrepareQuery = session.prepareQuery;
-      session.prepareQuery = function(...args) {
-        const prepared = originalPrepareQuery.apply(this, args);
-        if (prepared && typeof prepared.execute === "function") {
-          const originalPreparedExecute = prepared.execute;
-          prepared.execute = function(...executeArgs) {
-            const queryObj = args[0];
-            const queryText = queryObj?.sql || queryObj?.queryString || extractQueryText(queryObj);
-            const operation = queryText ? extractOperation(queryText) : void 0;
-            const spanName = operation ? `drizzle.${operation.toLowerCase()}` : "drizzle.query";
-            const span = tracer.startSpan(spanName, { kind: api.SpanKind.CLIENT });
-            span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem);
-            if (operation) {
-              span.setAttribute(SEMATTRS_DB_OPERATION, operation);
-            }
-            if (dbName) {
-              span.setAttribute(SEMATTRS_DB_NAME, dbName);
-            }
-            if (captureQueryText && queryText !== void 0) {
-              const sanitized = sanitizeQueryText(
-                queryText,
-                maxQueryTextLength
-              );
-              span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);
-            }
-            if (peerName) {
-              span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName);
-            }
-            if (peerPort) {
-              span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort);
-            }
-            return traceHelpers.runWithSpan(span, () => {
-              try {
-                const result = originalPreparedExecute.apply(this, executeArgs);
-                return Promise.resolve(result).then((value) => {
-                  traceHelpers.finalizeSpan(span);
-                  return value;
-                }).catch((error) => {
-                  traceHelpers.finalizeSpan(span, error);
-                  throw error;
-                });
-              } catch (error) {
-                traceHelpers.finalizeSpan(span, error);
-                throw error;
-              }
-            });
-          };
-        }
-        return prepared;
-      };
-      session[INSTRUMENTED_FLAG] = true;
-      instrumented = true;
-    }
-    if (typeof session.query === "function" && !session[INSTRUMENTED_FLAG + "_query"]) {
-      const originalQuery = session.query;
-      session.query = function(queryString, params) {
-        const operation = queryString ? extractOperation(queryString) : void 0;
-        const spanName = operation ? `drizzle.${operation.toLowerCase()}` : "drizzle.query";
-        const span = tracer.startSpan(spanName, { kind: api.SpanKind.CLIENT });
-        span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem);
-        if (operation) {
-          span.setAttribute(SEMATTRS_DB_OPERATION, operation);
-        }
-        if (dbName) {
-          span.setAttribute(SEMATTRS_DB_NAME, dbName);
-        }
-        if (captureQueryText && queryString !== void 0) {
-          const sanitized = sanitizeQueryText(queryString, maxQueryTextLength);
-          span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);
-        }
-        if (peerName) {
-          span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName);
-        }
-        if (peerPort) {
-          span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort);
-        }
-        return traceHelpers.runWithSpan(span, () => {
-          try {
-            const result = Reflect.apply(originalQuery, this, [
-              queryString,
-              params
-            ]);
-            return Promise.resolve(result).then((value) => {
-              traceHelpers.finalizeSpan(span);
-              return value;
-            }).catch((error) => {
-              traceHelpers.finalizeSpan(span, error);
-              throw error;
-            });
-          } catch (error) {
-            traceHelpers.finalizeSpan(span, error);
-            throw error;
-          }
-        });
-      };
-      session[INSTRUMENTED_FLAG + "_query"] = true;
-      instrumented = true;
-    }
-    if (typeof session.transaction === "function" && !session[INSTRUMENTED_FLAG + "_transaction"]) {
-      const originalTransaction = session.transaction;
-      session.transaction = function(transactionCallback, ...restArgs) {
-        const wrappedCallback = async function(tx) {
-          if (tx && (tx.session || tx._?.session || tx)) {
-            const txSession = tx.session || tx._?.session || tx;
-            if (typeof tx.execute === "function" && !tx[INSTRUMENTED_FLAG + "_execute"]) {
-              const originalTxExecute = tx.execute;
-              tx.execute = function(...executeArgs) {
-                const queryText = extractQueryText(executeArgs[0]);
-                const operation = queryText ? extractOperation(queryText) : void 0;
-                const spanName = operation ? `drizzle.${operation.toLowerCase()}` : "drizzle.query";
-                const span = tracer.startSpan(spanName, {
-                  kind: api.SpanKind.CLIENT
-                });
-                span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem);
-                span.setAttribute("db.transaction", true);
-                if (operation) {
-                  span.setAttribute(SEMATTRS_DB_OPERATION, operation);
-                }
-                if (dbName) {
-                  span.setAttribute(SEMATTRS_DB_NAME, dbName);
-                }
-                if (captureQueryText && queryText !== void 0) {
-                  const sanitized = sanitizeQueryText(
-                    queryText,
-                    maxQueryTextLength
-                  );
-                  span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);
-                }
-                if (peerName) {
-                  span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName);
-                }
-                if (peerPort) {
-                  span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort);
-                }
-                return traceHelpers.runWithSpan(span, () => {
-                  try {
-                    const result = originalTxExecute.apply(this, executeArgs);
-                    return Promise.resolve(result).then((value) => {
-                      traceHelpers.finalizeSpan(span);
-                      return value;
-                    }).catch((error) => {
-                      traceHelpers.finalizeSpan(span, error);
-                      throw error;
-                    });
-                  } catch (error) {
-                    traceHelpers.finalizeSpan(span, error);
-                    throw error;
-                  }
-                });
-              };
-              tx[INSTRUMENTED_FLAG + "_execute"] = true;
-            }
-            if (typeof txSession.prepareQuery === "function" && !txSession[INSTRUMENTED_FLAG + "_tx"]) {
-              const originalTxPrepareQuery = txSession.prepareQuery;
-              txSession.prepareQuery = function(...prepareArgs) {
-                const prepared = originalTxPrepareQuery.apply(
-                  this,
-                  prepareArgs
-                );
-                if (prepared && typeof prepared.execute === "function") {
-                  const originalPreparedExecute = prepared.execute;
-                  prepared.execute = function(...executeArgs) {
-                    const queryObj = prepareArgs[0];
-                    const queryText = queryObj?.sql || queryObj?.queryString || extractQueryText(queryObj);
-                    const operation = queryText ? extractOperation(queryText) : void 0;
-                    const spanName = operation ? `drizzle.${operation.toLowerCase()}` : "drizzle.query";
-                    const span = tracer.startSpan(spanName, {
-                      kind: api.SpanKind.CLIENT
-                    });
-                    span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem);
-                    span.setAttribute("db.transaction", true);
-                    if (operation) {
-                      span.setAttribute(SEMATTRS_DB_OPERATION, operation);
-                    }
-                    if (dbName) {
-                      span.setAttribute(SEMATTRS_DB_NAME, dbName);
-                    }
-                    if (captureQueryText && queryText !== void 0) {
-                      const sanitized = sanitizeQueryText(
-                        queryText,
-                        maxQueryTextLength
-                      );
-                      span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);
-                    }
-                    if (peerName) {
-                      span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName);
-                    }
-                    if (peerPort) {
-                      span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort);
-                    }
-                    return traceHelpers.runWithSpan(span, () => {
-                      try {
-                        const result = originalPreparedExecute.apply(
-                          this,
-                          executeArgs
-                        );
-                        return Promise.resolve(result).then((value) => {
-                          traceHelpers.finalizeSpan(span);
-                          return value;
-                        }).catch((error) => {
-                          traceHelpers.finalizeSpan(span, error);
-                          throw error;
-                        });
-                      } catch (error) {
-                        traceHelpers.finalizeSpan(span, error);
-                        throw error;
-                      }
-                    });
-                  };
-                }
-                return prepared;
-              };
-              txSession[INSTRUMENTED_FLAG + "_tx"] = true;
-            }
-          }
-          return transactionCallback(tx);
-        };
-        return Reflect.apply(originalTransaction, this, [
-          wrappedCallback,
-          ...restArgs
-        ]);
-      };
-      session[INSTRUMENTED_FLAG + "_transaction"] = true;
-      instrumented = true;
-    }
-  }
-  if (db.$client && !instrumented) {
-    const client = db.$client;
-    if (typeof client.query === "function" || typeof client.execute === "function") {
-      instrumentDrizzle(client, config);
-      instrumented = true;
-    }
-  }
-  if (db._ && db._.session && typeof db._.session.execute === "function" && !instrumented) {
-    const session = db._.session;
-    if (session[INSTRUMENTED_FLAG]) {
-      return db;
-    }
-    const {
-      tracerName: tracerName2 = DEFAULT_TRACER_NAME,
-      dbSystem: dbSystem2 = DEFAULT_DB_SYSTEM,
-      dbName: dbName2,
-      captureQueryText: captureQueryText2 = true,
-      maxQueryTextLength: maxQueryTextLength2 = 1e3,
-      peerName: peerName2,
-      peerPort: peerPort2
-    } = config ?? {};
-    const tracer2 = api.trace.getTracer(tracerName2);
-    const originalExecute = session.execute;
-    if (!originalExecute) {
-      return db;
-    }
-    const instrumentedExecute = function instrumented2(...args) {
-      const queryText = extractQueryText(args[0]);
-      const operation = queryText ? extractOperation(queryText) : void 0;
-      const spanName = operation ? `drizzle.${operation.toLowerCase()}` : "drizzle.query";
-      const span = tracer2.startSpan(spanName, { kind: api.SpanKind.CLIENT });
-      span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem2);
-      if (operation) {
-        span.setAttribute(SEMATTRS_DB_OPERATION, operation);
-      }
-      if (dbName2) {
-        span.setAttribute(SEMATTRS_DB_NAME, dbName2);
-      }
-      if (captureQueryText2 && queryText !== void 0) {
-        const sanitized = sanitizeQueryText(queryText, maxQueryTextLength2);
-        span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);
-      }
-      if (peerName2) {
-        span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName2);
-      }
-      if (peerPort2) {
-        span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort2);
-      }
-      return traceHelpers.runWithSpan(span, () => {
-        try {
-          const result = originalExecute.apply(this, args);
-          return Promise.resolve(result).then((value) => {
-            traceHelpers.finalizeSpan(span);
-            return value;
-          }).catch((error) => {
-            traceHelpers.finalizeSpan(span, error);
-            throw error;
-          });
-        } catch (error) {
-          traceHelpers.finalizeSpan(span, error);
-          throw error;
-        }
-      });
-    };
-    session[INSTRUMENTED_FLAG] = true;
-    session.execute = instrumentedExecute;
-    instrumented = true;
-  }
-  if (instrumented) {
-    db[INSTRUMENTED_FLAG] = true;
-  }
-  return db;
-}
-var DEFAULT_TRACER_NAME2 = "autotel-plugins/mongoose";
-var DEFAULT_DB_SYSTEM2 = "mongoose";
-var INSTRUMENTED_FLAG2 = "__autotelMongooseInstrumented";
-var WRAPPED_HOOK_FLAG = "__autotelWrappedHook";
-var _STORED_PARENT_SPAN = /* @__PURE__ */ Symbol("stored-parent-span");
-function createSpan(tracer, operation, modelName, collectionName, config) {
-  const spanName = collectionName ? `mongoose.${collectionName}.${operation}` : modelName ? `mongoose.${modelName}.${operation}` : `mongoose.${operation}`;
-  const attributes = {
-    [SEMATTRS_DB_SYSTEM]: DEFAULT_DB_SYSTEM2,
-    [SEMATTRS_DB_OPERATION]: operation
-  };
-  if (collectionName && config.captureCollectionName) {
-    attributes[SEMATTRS_DB_MONGODB_COLLECTION] = collectionName;
-  }
-  if (config.dbName) {
-    attributes[SEMATTRS_DB_NAME] = config.dbName;
-  }
-  if (config.peerName) {
-    attributes[SEMATTRS_NET_PEER_NAME] = config.peerName;
-  }
-  if (config.peerPort) {
-    attributes[SEMATTRS_NET_PEER_PORT] = config.peerPort;
-  }
-  return tracer.startSpan(spanName, { kind: autotel.SpanKind.CLIENT, attributes });
-}
-function wrapQueryMethod(target, methodName, operation, getCollectionName, getModelName, tracer, config) {
-  const original = target[methodName];
-  if (typeof original !== "function") {
-    return;
-  }
-  target[methodName] = function instrumented(...args) {
-    const collectionName = getCollectionName(this);
-    const modelName = getModelName(this);
-    const span = createSpan(
-      tracer,
-      operation,
-      modelName,
-      collectionName,
-      config
-    );
-    return traceHelpers.runWithSpan(span, () => {
-      try {
-        const result = original.apply(this, args);
-        if (result && typeof result.exec === "function") {
-          const originalExec = result.exec.bind(result);
-          result.exec = function wrappedExec() {
-            try {
-              const execPromise = originalExec();
-              return Promise.resolve(execPromise).then((value) => {
-                traceHelpers.finalizeSpan(span);
-                return value;
-              }).catch((error) => {
-                traceHelpers.finalizeSpan(
-                  span,
-                  error instanceof Error ? error : new Error(String(error))
-                );
-                throw error;
-              });
-            } catch (error) {
-              traceHelpers.finalizeSpan(
-                span,
-                error instanceof Error ? error : new Error(String(error))
-              );
-              throw error;
-            }
-          };
-          return result;
-        }
-        if (result && typeof result.then === "function") {
-          return Promise.resolve(result).then((value) => {
-            traceHelpers.finalizeSpan(span);
-            return value;
-          }).catch((error) => {
-            traceHelpers.finalizeSpan(
-              span,
-              error instanceof Error ? error : new Error(String(error))
-            );
-            throw error;
-          });
-        }
-        traceHelpers.finalizeSpan(span);
-        return result;
-      } catch (error) {
-        traceHelpers.finalizeSpan(
-          span,
-          error instanceof Error ? error : new Error(String(error))
-        );
-        throw error;
-      }
-    });
-  };
-}
-function wrapChainableMethod(target, methodName) {
-  const original = target[methodName];
-  if (typeof original !== "function") {
-    return;
-  }
-  target[methodName] = function captureContext(...args) {
-    const currentSpan = traceHelpers.getActiveSpan();
-    const result = original.apply(this, args);
-    if (result && typeof result.exec === "function") {
-      result[_STORED_PARENT_SPAN] = currentSpan;
-    }
-    return result;
-  };
-}
-function patchSchemaHooks(Schema, tracer, config) {
-  if (!Schema?.prototype) {
-    return;
-  }
-  const HOOK_FLAG = "__autotelHookInstrumented";
-  if (Schema.prototype[HOOK_FLAG]) {
-    return;
-  }
-  const originalPre = Schema.prototype.pre;
-  if (typeof originalPre === "function") {
-    Schema.prototype.pre = function(hookName, ...args) {
-      const handler = typeof args[0] === "function" ? args[0] : typeof args[1] === "function" ? args[1] : null;
-      if (handler && !isMongooseInternalHook(handler)) {
-        const wrapped = wrapHookHandler(
-          handler,
-          hookName,
-          "pre",
-          tracer,
-          config
-        );
-        if (typeof args[0] === "function") {
-          args[0] = wrapped;
-        } else if (typeof args[1] === "function") {
-          args[1] = wrapped;
-        }
-      }
-      return Reflect.apply(originalPre, this, [hookName, ...args]);
-    };
-  }
-  const originalPost = Schema.prototype.post;
-  if (typeof originalPost === "function") {
-    Schema.prototype.post = function(hookName, ...args) {
-      const handler = typeof args[0] === "function" ? args[0] : typeof args[1] === "function" ? args[1] : null;
-      if (handler && !isMongooseInternalHook(handler)) {
-        const wrapped = wrapHookHandler(
-          handler,
-          hookName,
-          "post",
-          tracer,
-          config
-        );
-        if (typeof args[0] === "function") {
-          args[0] = wrapped;
-        } else if (typeof args[1] === "function") {
-          args[1] = wrapped;
-        }
-      }
-      return Reflect.apply(originalPost, this, [hookName, ...args]);
-    };
-  }
-  Schema.prototype[HOOK_FLAG] = true;
-}
-function isMongooseInternalHook(handler) {
-  if (typeof handler !== "function") {
-    return false;
-  }
-  const funcName = handler.name || "";
-  if (funcName.startsWith("_") || funcName.startsWith("$")) {
-    return true;
-  }
-  const mongooseInternalNamePatterns = [
-    "shardingPlugin",
-    "mongooseInternalHook",
-    "noop",
-    "wrapped",
-    "bound "
-  ];
-  if (mongooseInternalNamePatterns.some((pattern) => funcName.includes(pattern))) {
-    return true;
-  }
-  try {
-    const source = handler.toString();
-    const mongooseInternalSourcePatterns = [
-      "this.$__",
-      // Mongoose internal document methods
-      "this.$isValid",
-      // Mongoose validation
-      "this.$locals",
-      // Mongoose local properties
-      "_this.$__",
-      // Mongoose internal with closure
-      "schema.s.hooks",
-      // Mongoose hooks system
-      "kareem"
-      // Mongoose's hooks library
-    ];
-    if (mongooseInternalSourcePatterns.some((pattern) => source.includes(pattern))) {
-      return true;
-    }
-  } catch {
-  }
-  return false;
-}
-function wrapHookHandler(handler, hookName, hookType, tracer, config) {
-  if (typeof handler !== "function") {
-    return handler;
-  }
-  if (handler[WRAPPED_HOOK_FLAG]) {
-    return handler;
-  }
-  const wrappedHook = function wrappedHook2(...args) {
-    let modelName;
-    let collectionName;
-    try {
-      if (this.constructor?.modelName) {
-        modelName = this.constructor.modelName;
-        collectionName = this.constructor.collection?.collectionName || modelName;
-      } else if (this.model?.modelName) {
-        modelName = this.model.modelName;
-        collectionName = this.model.collection?.collectionName || modelName;
-      }
-    } catch {
-    }
-    const spanName = collectionName ? `mongoose.${collectionName}.${hookType}.${hookName}` : `mongoose.hook.${hookType}.${hookName}`;
-    const span = tracer.startSpan(spanName, { kind: autotel.SpanKind.INTERNAL });
-    span.setAttribute("hook.type", hookType);
-    span.setAttribute("hook.operation", hookName);
-    if (modelName) {
-      span.setAttribute("hook.model", modelName);
-    }
-    if (collectionName && config.captureCollectionName) {
-      span.setAttribute(SEMATTRS_DB_MONGODB_COLLECTION, collectionName);
-    }
-    span.setAttribute(SEMATTRS_DB_SYSTEM, DEFAULT_DB_SYSTEM2);
-    if (config.dbName) {
-      span.setAttribute(SEMATTRS_DB_NAME, config.dbName);
-    }
-    return traceHelpers.runWithSpan(span, () => {
-      try {
-        const result = handler.apply(this, args);
-        if (result && typeof result.then === "function") {
-          return Promise.resolve(result).then((value) => {
-            traceHelpers.finalizeSpan(span);
-            return value;
-          }).catch((error) => {
-            traceHelpers.finalizeSpan(
-              span,
-              error instanceof Error ? error : new Error(String(error))
-            );
-            throw error;
-          });
-        }
-        traceHelpers.finalizeSpan(span);
-        return result;
-      } catch (error) {
-        traceHelpers.finalizeSpan(
-          span,
-          error instanceof Error ? error : new Error(String(error))
-        );
-        throw error;
-      }
-    });
-  };
-  wrappedHook[WRAPPED_HOOK_FLAG] = true;
-  return wrappedHook;
-}
-function instrumentMongoose(mongoose, config) {
-  if (!mongoose?.Model) {
-    return mongoose;
-  }
-  const m = mongoose;
-  if (m[INSTRUMENTED_FLAG2]) {
-    return mongoose;
-  }
-  const finalConfig = {
-    dbName: config?.dbName || "",
-    peerName: config?.peerName || "",
-    peerPort: config?.peerPort || 27017,
-    tracerName: config?.tracerName || DEFAULT_TRACER_NAME2,
-    captureCollectionName: config?.captureCollectionName ?? true,
-    instrumentHooks: config?.instrumentHooks ?? false
-  };
-  const tracer = autotel.otelTrace.getTracer(finalConfig.tracerName);
-  if (m.Schema && finalConfig.instrumentHooks) {
-    patchSchemaHooks(m.Schema, tracer, finalConfig);
-  }
-  const getModelCollectionName = (model) => {
-    try {
-      return model.collection?.collectionName || model.modelName;
-    } catch {
-      return;
-    }
-  };
-  const queryMethods = [
-    { method: "find", operation: "find" },
-    { method: "findOne", operation: "findOne" },
-    { method: "findById", operation: "findById" },
-    { method: "findOneAndUpdate", operation: "findOneAndUpdate" },
-    { method: "findOneAndDelete", operation: "findOneAndDelete" },
-    { method: "findOneAndReplace", operation: "findOneAndReplace" },
-    { method: "deleteOne", operation: "deleteOne" },
-    { method: "deleteMany", operation: "deleteMany" },
-    { method: "updateOne", operation: "updateOne" },
-    { method: "updateMany", operation: "updateMany" },
-    { method: "countDocuments", operation: "countDocuments" },
-    { method: "estimatedDocumentCount", operation: "estimatedDocumentCount" }
-  ];
-  for (const { method, operation } of queryMethods) {
-    wrapQueryMethod(
-      m.Model,
-      method,
-      operation,
-      getModelCollectionName,
-      (model) => model.modelName,
-      tracer,
-      finalConfig
-    );
-    if (m.Query?.prototype?.[method]) {
-      wrapChainableMethod(m.Query.prototype, method);
-    }
-  }
-  const instanceMethods = ["save", "deleteOne"];
-  for (const method of instanceMethods) {
-    if (m.Model.prototype[method]) {
-      wrapQueryMethod(
-        m.Model.prototype,
-        method,
-        method,
-        (doc) => {
-          try {
-            return doc.constructor?.collection?.collectionName || doc.constructor?.modelName;
-          } catch {
-            return;
-          }
-        },
-        (doc) => {
-          try {
-            return doc.constructor?.modelName;
-          } catch {
-            return;
-          }
-        },
-        tracer,
-        finalConfig
-      );
-    }
-  }
-  const staticMethods = ["create", "insertMany", "aggregate", "bulkWrite"];
-  for (const method of staticMethods) {
-    if (m.Model[method]) {
-      wrapQueryMethod(
-        m.Model,
-        method,
-        method,
-        (model) => {
-          try {
-            return model.collection?.collectionName;
-          } catch {
-            return;
-          }
-        },
-        (model) => model.modelName,
-        tracer,
-        finalConfig
-      );
-    }
-  }
-  const chainableMethods = [
-    "populate",
-    "select",
-    "lean",
-    "where",
-    "sort",
-    "limit",
-    "skip"
-  ];
-  for (const method of chainableMethods) {
-    if (m.Query?.prototype?.[method]) {
-      wrapChainableMethod(m.Query.prototype, method);
-    }
-  }
-  m[INSTRUMENTED_FLAG2] = true;
-  return mongoose;
-}
-var MongooseInstrumentation = class {
-  constructor(config) {
-    this.config = config;
-  }
-  enable(mongoose) {
-    instrumentMongoose(mongoose, this.config);
-  }
-};
-var DEFAULT_TRACER_NAME3 = "autotel-plugins/bigquery";
+var DEFAULT_TRACER_NAME = "autotel-plugins/bigquery";
 var DEFAULT_DB_SYSTEM_NAME = "gcp.bigquery";
-var INSTRUMENTED_FLAG3 = "__autotelBigQueryInstrumented";
+var INSTRUMENTED_FLAG = "__autotelBigQueryInstrumented";
 var PROTOTYPE_INSTRUMENTED_FLAG = "__autotelBigQueryPrototypeInstrumented";
 var CONFIG_STORAGE_KEY = "__autotelBigQueryConfig";
 var TRACER_STORAGE_KEY = "__autotelBigQueryTracer";
@@ -1006,7 +167,7 @@ function extractTableReference(obj) {
     return {};
   }
 }
-function createSpan2(tracer, operationName, target, projectId, config = {}) {
+function createSpan(tracer, operationName, target, projectId, config = {}) {
   const spanName = target ? `${operationName} ${target}` : operationName;
   const attributes = {
     [SEMATTRS_DB_SYSTEM_NAME]: DEFAULT_DB_SYSTEM_NAME,
@@ -1036,7 +197,7 @@ function instrumentQueryMethod(BigQuery) {
     const location = extractLocation(this, options);
     const operation = queryText ? extractOperationType(queryText) : "QUERY";
     const summary = queryText ? createQuerySummary(queryText) : "QUERY";
-    const span = createSpan2(
+    const span = createSpan(
       tracer,
       operation || "QUERY",
       summary,
@@ -1148,7 +309,7 @@ function instrumentCreateQueryJob(BigQuery) {
     const location = extractLocation(this, options);
     const operation = queryText ? extractOperationType(queryText) : "QUERY";
     const summary = queryText ? createQuerySummary(queryText) : "QUERY";
-    const span = createSpan2(
+    const span = createSpan(
       tracer,
       `${operation || "QUERY"}_JOB`,
       summary,
@@ -1234,7 +395,7 @@ function instrumentTableInsert(Table) {
       options
     );
     const target = tableRef.tableId ? tableRef.datasetId ? `${tableRef.datasetId}.${tableRef.tableId}` : tableRef.tableId : void 0;
-    const span = createSpan2(tracer, "INSERT", target, projectId, config);
+    const span = createSpan(tracer, "INSERT", target, projectId, config);
     if (location) {
       span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_LOCATION, location);
     }
@@ -1293,7 +454,7 @@ function instrumentTableGetRows(Table) {
       options
     );
     const target = tableRef.tableId ? tableRef.datasetId ? `${tableRef.datasetId}.${tableRef.tableId}` : tableRef.tableId : void 0;
-    const span = createSpan2(tracer, "SELECT", target, projectId, config);
+    const span = createSpan(tracer, "SELECT", target, projectId, config);
     if (location) {
       span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_LOCATION, location);
     }
@@ -1350,7 +511,7 @@ function instrumentTableCreateLoadJob(Table) {
       metadata
     );
     const target = tableRef.tableId ? tableRef.datasetId ? `${tableRef.datasetId}.${tableRef.tableId}` : tableRef.tableId : void 0;
-    const span = createSpan2(tracer, "LOAD", target, projectId, config);
+    const span = createSpan(tracer, "LOAD", target, projectId, config);
     if (location) {
       span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_LOCATION, location);
     }
@@ -1407,7 +568,7 @@ function instrumentTableCreateCopyJob(Table) {
       metadata
     );
     const source = sourceRef.tableId ? sourceRef.datasetId ? `${sourceRef.datasetId}.${sourceRef.tableId}` : sourceRef.tableId : "unknown";
-    const span = createSpan2(tracer, "COPY", source, projectId, config);
+    const span = createSpan(tracer, "COPY", source, projectId, config);
     if (location) {
       span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_LOCATION, location);
     }
@@ -1462,7 +623,7 @@ function instrumentTableCreateExtractJob(Table) {
       metadata
     );
     const source = sourceRef.tableId ? sourceRef.datasetId ? `${sourceRef.datasetId}.${sourceRef.tableId}` : sourceRef.tableId : "unknown";
-    const span = createSpan2(tracer, "EXTRACT", source, projectId, config);
+    const span = createSpan(tracer, "EXTRACT", source, projectId, config);
     if (location) {
       span.setAttribute(SEMATTRS_GCP_BIGQUERY_JOB_LOCATION, location);
     }
@@ -1525,7 +686,7 @@ function instrumentJobGetQueryResults(Job) {
     const jobId = this.id || this.metadata?.jobReference?.jobId;
     const projectId = this.parent?.projectId || this.projectId;
     const location = this.location || options?.location;
-    const span = createSpan2(
+    const span = createSpan(
       tracer,
       "GET_QUERY_RESULTS",
       void 0,
@@ -1581,7 +742,7 @@ function instrumentDatasetCreate(Dataset) {
     const datasetId = this.id;
     const projectId = extractProjectId(this.parent);
     const location = extractLocation(this.parent, options);
-    const span = createSpan2(
+    const span = createSpan(
       tracer,
       "CREATE_DATASET",
       datasetId,
@@ -1630,7 +791,7 @@ function instrumentDatasetDelete(Dataset) {
     }
     const datasetId = this.id;
     const projectId = extractProjectId(this.parent);
-    const span = createSpan2(
+    const span = createSpan(
       tracer,
       "DELETE_DATASET",
       datasetId,
@@ -1677,7 +838,7 @@ function instrumentBigQueryCreateDataset(BigQuery) {
     const datasetId = id;
     const projectId = extractProjectId(this);
     const location = extractLocation(this, options);
-    const span = createSpan2(
+    const span = createSpan(
       tracer,
       "CREATE_DATASET",
       datasetId,
@@ -1741,7 +902,7 @@ 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 = createSpan2(tracer, "CREATE_TABLE", target, projectId, config);
+    const span = createSpan(tracer, "CREATE_TABLE", target, projectId, config);
     if (tableRef.datasetId) {
       span.setAttribute(SEMATTRS_DB_NAMESPACE, tableRef.datasetId);
     }
@@ -1785,7 +946,7 @@ 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 = createSpan2(tracer, "DELETE_TABLE", target, projectId, config);
+    const span = createSpan(tracer, "DELETE_TABLE", target, projectId, config);
     if (tableRef.datasetId) {
       span.setAttribute(SEMATTRS_DB_NAMESPACE, tableRef.datasetId);
     }
@@ -1820,13 +981,13 @@ function instrumentBigQuery(bigquery, config) {
     return bigquery;
   }
   const bq = bigquery;
-  if (bq[INSTRUMENTED_FLAG3]) {
+  if (bq[INSTRUMENTED_FLAG]) {
     return bigquery;
   }
   const finalConfig = {
     projectId: config?.projectId || "",
     location: config?.location || "",
-    tracerName: config?.tracerName || DEFAULT_TRACER_NAME3,
+    tracerName: config?.tracerName || DEFAULT_TRACER_NAME,
     captureQueryText: config?.captureQueryText || "summary",
     maxQueryTextLength: config?.maxQueryTextLength || 1e3,
     includeQueryHash: config?.includeQueryHash ?? true,
@@ -1880,7 +1041,7 @@ function instrumentBigQuery(bigquery, config) {
     }
     BigQuery[PROTOTYPE_INSTRUMENTED_FLAG] = true;
   }
-  bq[INSTRUMENTED_FLAG3] = true;
+  bq[INSTRUMENTED_FLAG] = true;
   return bigquery;
 }
 var BigQueryInstrumentation = class {
@@ -1961,7 +1122,7 @@ function injectTraceHeaders(base = {}, options = {}) {
   }
   return carrier;
 }
-var DEFAULT_TRACER_NAME4 = "autotel-plugins/kafka";
+var DEFAULT_TRACER_NAME2 = "autotel-plugins/kafka";
 function isValidSpanContext(spanContext) {
   return !!(spanContext && spanContext.traceId && spanContext.spanId && autotel.otelTrace.isSpanContextValid(spanContext));
 }
@@ -1977,7 +1138,7 @@ async function withProcessingSpan(descriptor, fn) {
     offset,
     key
   } = descriptor;
-  const tracer = autotel.otelTrace.getTracer(DEFAULT_TRACER_NAME4);
+  const tracer = autotel.otelTrace.getTracer(DEFAULT_TRACER_NAME2);
   const normalizedHeaders = normalizeHeaders(headers);
   const extractedCtx = extractTraceContext(normalizedHeaders);
   const extractedSpanContext = autotel.otelTrace.getSpanContext(extractedCtx);
@@ -2081,10 +1242,10 @@ function setMessagingAttributes(span, attrs) {
     span.setAttribute(SEMATTRS_MESSAGING_KAFKA_MESSAGE_KEY, key);
   }
 }
-var DEFAULT_TRACER_NAME5 = "autotel-plugins/kafka";
+var DEFAULT_TRACER_NAME3 = "autotel-plugins/kafka";
 async function withProducerSpan(descriptor, fn) {
   const { name, topic, messageKey, system = "kafka" } = descriptor;
-  const tracer = autotel.otelTrace.getTracer(DEFAULT_TRACER_NAME5);
+  const tracer = autotel.otelTrace.getTracer(DEFAULT_TRACER_NAME3);
   const span = tracer.startSpan(name, {
     kind: autotel.SpanKind.PRODUCER
   });
@@ -2195,10 +1356,10 @@ async function extractBatchLineageAsync(items, options = {}) {
     ...includeTraceIds && { trace_ids: traceIds }
   };
 }
-var DEFAULT_TRACER_NAME6 = "autotel-plugins/kafka";
+var DEFAULT_TRACER_NAME4 = "autotel-plugins/kafka";
 function withBatchConsumer(config, handler) {
   const { name, consumerGroup, perMessageSpans = "none", onProgress } = config;
-  const tracer = autotel.otelTrace.getTracer(DEFAULT_TRACER_NAME6);
+  const tracer = autotel.otelTrace.getTracer(DEFAULT_TRACER_NAME4);
   return async (payload) => {
     const { batch } = payload;
     const startTime = Date.now();
@@ -2408,7 +1569,7 @@ function createWrappedPayload(original, perMessageSpans, tracer, parentContext,
   };
 }
 function createMessageErrorSpan(name, message, error, topic, partition) {
-  const tracer = autotel.otelTrace.getTracer(DEFAULT_TRACER_NAME6);
+  const tracer = autotel.otelTrace.getTracer(DEFAULT_TRACER_NAME4);
   const normalizedHeaders = normalizeHeaders(message.headers);
   const extractedCtx = extractTraceContext(normalizedHeaders);
   const span = tracer.startSpan(
@@ -2431,10 +1592,10 @@ function createMessageErrorSpan(name, message, error, topic, partition) {
   span.recordException(error);
   span.end();
 }
-var DEFAULT_TRACER_NAME7 = "autotel-plugins/kafka";
+var DEFAULT_TRACER_NAME5 = "autotel-plugins/kafka";
 function createStreamProcessor(config) {
   const { name } = config;
-  const tracer = autotel.otelTrace.getTracer(DEFAULT_TRACER_NAME7);
+  const tracer = autotel.otelTrace.getTracer(DEFAULT_TRACER_NAME5);
   return {
     async run(message, callback) {
       const normalizedHeaders = normalizeHeaders(message.headers);
@@ -2704,7 +1865,7 @@ var ConsumerMetrics = class {
     }
   }
 };
-var DEFAULT_TRACER_NAME8 = "autotel-plugins/kafka";
+var DEFAULT_TRACER_NAME6 = "autotel-plugins/kafka";
 var REBALANCE_EVENTS = [
   "consumer.group_join",
   "consumer.rebalancing",
@@ -2724,7 +1885,7 @@ function instrumentConsumerEvents(consumer, config = {}) {
     traceHeartbeats = false,
     lifecycleSpan
   } = config;
-  const tracer = autotel.otelTrace.getTracer(DEFAULT_TRACER_NAME8);
+  const tracer = autotel.otelTrace.getTracer(DEFAULT_TRACER_NAME6);
   const listeners = [];
   const addListener = (event, listener) => {
     consumer.on(event, listener);
@@ -2927,7 +2088,7 @@ function injectTraceHeaders2(base = {}, options = {}) {
   }
   return carrier;
 }
-var DEFAULT_TRACER_NAME9 = "autotel-plugins/rabbitmq";
+var DEFAULT_TRACER_NAME7 = "autotel-plugins/rabbitmq";
 function isValidSpanContext3(spanContext) {
   return !!(spanContext && spanContext.traceId && spanContext.spanId && autotel.otelTrace.isSpanContextValid(spanContext));
 }
@@ -2952,7 +2113,7 @@ async function withConsumeSpan(descriptor, fn) {
     deferSpanEnd = false,
     ackTimeoutMs
   } = descriptor;
-  const tracer = autotel.otelTrace.getTracer(DEFAULT_TRACER_NAME9);
+  const tracer = autotel.otelTrace.getTracer(DEFAULT_TRACER_NAME7);
   const normalizedHeaders = normalizeHeaders2(headers);
   const extractedCtx = extractTraceContext2(normalizedHeaders);
   const extractedSpanContext = autotel.otelTrace.getSpanContext(extractedCtx);
@@ -3126,7 +2287,7 @@ function setMessagingAttributes2(span, attrs) {
     span.setAttribute(SEMATTRS_MESSAGING_CONSUMER_ID, consumerTag);
   }
 }
-var DEFAULT_TRACER_NAME10 = "autotel-plugins/rabbitmq";
+var DEFAULT_TRACER_NAME8 = "autotel-plugins/rabbitmq";
 async function withPublishSpan(descriptor, fn) {
   const {
     name,
@@ -3136,7 +2297,7 @@ async function withPublishSpan(descriptor, fn) {
     correlationId,
     system = "rabbitmq"
   } = descriptor;
-  const tracer = autotel.otelTrace.getTracer(DEFAULT_TRACER_NAME10);
+  const tracer = autotel.otelTrace.getTracer(DEFAULT_TRACER_NAME8);
   const span = tracer.startSpan(name, {
     kind: autotel.SpanKind.PRODUCER
   });
@@ -3226,7 +2387,6 @@ function recordAckResult(span, result, options) {
 exports.BigQueryInstrumentation = BigQueryInstrumentation;
 exports.CORRELATION_ID_HEADER = CORRELATION_ID_HEADER;
 exports.ConsumerMetrics = ConsumerMetrics;
-exports.MongooseInstrumentation = MongooseInstrumentation;
 exports.SEMATTRS_DB_COLLECTION_NAME = SEMATTRS_DB_COLLECTION_NAME;
 exports.SEMATTRS_DB_NAME = SEMATTRS_DB_NAME;
 exports.SEMATTRS_DB_NAMESPACE = SEMATTRS_DB_NAMESPACE;
@@ -3287,9 +2447,6 @@ exports.injectRabbitMQTraceHeaders = injectTraceHeaders2;
 exports.injectTraceHeaders = injectTraceHeaders;
 exports.instrumentBigQuery = instrumentBigQuery;
 exports.instrumentConsumerEvents = instrumentConsumerEvents;
-exports.instrumentDrizzle = instrumentDrizzle;
-exports.instrumentDrizzleClient = instrumentDrizzleClient;
-exports.instrumentMongoose = instrumentMongoose;
 exports.normalizeHeaders = normalizeHeaders;
 exports.normalizeRabbitMQHeaders = normalizeHeaders2;
 exports.recordAckResult = recordAckResult;