@rootcat/node 0.0.0-stage → 0.2.0

package/dist/pg.js

@@ -0,0 +1,170 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PgInstrumentation = void 0;
+// pg (node-postgres): one span per query, nothing else. The upstream
+// instrumentation starts a span for every pool acquire beside the query,
+// another per connection, records metrics and a dozen attributes, and
+// costs about three times what the query span itself does (see the
+// benchmark); the agent counts operations and keeps error events, so
+// one client span with the statement, the database and the server is
+// what it needs. Client.prototype.query is patched (pool.query goes
+// through it); a submittable (a cursor or a stream) passes through
+// untraced; parameter values are never read.
+const api_1 = require("@opentelemetry/api");
+const instrumentation_1 = require("@opentelemetry/instrumentation");
+const callsite_js_1 = require("./callsite.js");
+const keys_js_1 = require("./keys.js");
+const privacy_js_1 = require("./privacy.js");
+const processor_js_1 = require("./processor.js");
+const maxStatementBytes = 4096;
+const supportedVersions = ['>=8.0.0 <9'];
+class PgInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(options) {
+        super('@rootcat/node/pg', '0.2.0', options);
+    }
+    init() {
+        return new instrumentation_1.InstrumentationNodeModuleDefinition('pg', supportedVersions, (moduleExports) => {
+            const proto = clientPrototype(moduleExports);
+            if (proto) {
+                if ((0, instrumentation_1.isWrapped)(proto.query) && (0, keys_js_1.mine)(proto.query)) {
+                    this._unwrap(proto, 'query');
+                }
+                const before = proto.query;
+                this._wrap(proto, 'query', this.wrapQuery);
+                if (proto.query !== before) {
+                    (0, keys_js_1.marked)(proto.query); // not when a guard kept another SDK's wrapper
+                }
+            }
+            return moduleExports;
+        }, (moduleExports) => {
+            const proto = clientPrototype(moduleExports);
+            if (proto && (0, instrumentation_1.isWrapped)(proto.query) && (0, keys_js_1.mine)(proto.query)) {
+                this._unwrap(proto, 'query');
+            }
+        });
+    }
+    wrapQuery = (original) => {
+        const instrumentation = this;
+        return function query(...args) {
+            const [config, values, callback] = args;
+            if (config && typeof config.submit === 'function') {
+                return original.apply(this, args); // a cursor or a stream: not a query with an end
+            }
+            const outer = api_1.context.active();
+            let span;
+            let parent = outer;
+            try {
+                const options = instrumentation.getConfig();
+                parent = options.sampling.carry(() => (0, callsite_js_1.captureCallSite)(options.depth, options.isApp, query)) ?? outer;
+                if ((0, processor_js_1.unsampled)(parent)) {
+                    return original.apply(this, args); // not recorded: nothing to start or carry
+                }
+                span = instrumentation.startQuerySpan(this, config, parent);
+            }
+            catch {
+                span = undefined;
+            }
+            if (!span) {
+                return original.apply(this, args);
+            }
+            const end = instrumentation.endQuerySpan.bind(instrumentation, span);
+            // The callback may be the second or third argument, or on the config.
+            // It continues in the caller's context, not the query's.
+            let callArgs = args;
+            const userCallback = typeof values === 'function' ? values : typeof callback === 'function' ? callback : config?.callback;
+            if (typeof userCallback === 'function') {
+                const wrapped = api_1.context.bind(outer, function (error, result) {
+                    end(error);
+                    return userCallback.call(this, error, result);
+                });
+                if (typeof values === 'function') {
+                    callArgs = [config, wrapped];
+                }
+                else if (typeof callback === 'function') {
+                    callArgs = [config, values, wrapped];
+                }
+                else {
+                    // A config may be a class instance whose text or values are
+                    // getters on its prototype: the callback goes on an object that
+                    // inherits from it, not on a copy of its own fields.
+                    const withCallback = Object.create(config);
+                    withCallback.callback = wrapped;
+                    callArgs = [withCallback, values, callback];
+                }
+            }
+            const active = api_1.trace.setSpan(parent, span);
+            let result;
+            try {
+                result = api_1.context.with(active, () => original.apply(this, callArgs));
+            }
+            catch (error) {
+                end(error);
+                throw error;
+            }
+            if (typeof userCallback !== 'function') {
+                if (result && typeof result.then === 'function') {
+                    result.then(() => end(undefined), (error) => end(error ?? new Error('query failed')));
+                }
+                else {
+                    end(undefined);
+                }
+            }
+            return result;
+        };
+    };
+    startQuerySpan(client, config, parent) {
+        const text = typeof config === 'string' ? config : typeof config?.text === 'string' ? config.text : '';
+        const keyword = (0, privacy_js_1.statementKeyword)(text);
+        const attributes = { 'db.system.name': 'postgresql' };
+        if (keyword !== '') {
+            attributes['db.operation.name'] = keyword;
+        }
+        if (text !== '') {
+            // Literals are masked (or the text reduced to its keyword) by the
+            // processor's statement rule as the span starts.
+            attributes['db.query.text'] = text.slice(0, maxStatementBytes);
+        }
+        if (typeof client.database === 'string') {
+            attributes['db.namespace'] = client.database;
+        }
+        if (typeof client.host === 'string') {
+            attributes['server.address'] = client.host;
+        }
+        if (typeof client.port === 'number') {
+            attributes['server.port'] = client.port;
+        }
+        return this.tracer.startSpan(keyword === '' ? 'pg.query' : `pg.query:${keyword}`, { kind: api_1.SpanKind.CLIENT, attributes }, parent);
+    }
+    endQuerySpan(span, error) {
+        try {
+            if (error !== undefined && error !== null) {
+                // The server's message can quote the input that failed; only the
+                // error's class and SQLSTATE travel.
+                const e = error;
+                const generic = typeof e.name !== 'string' || e.name === '' || e.name === 'Error' || e.name === 'error';
+                const name = (generic ? e.constructor?.name : e.name) || 'Error';
+                const code = typeof e.code === 'string' ? e.code.slice(0, 16) : '';
+                const message = code === '' ? `${name} from pg` : `${name} from pg (code: ${code})`;
+                span.setStatus({ code: api_1.SpanStatusCode.ERROR, message });
+                span.addEvent('exception', { 'exception.type': name.slice(0, 64), 'exception.message': message });
+                span.setAttribute('error.type', name.slice(0, 64));
+                if (code !== '') {
+                    span.setAttribute('db.response.status_code', code);
+                }
+            }
+        }
+        catch {
+            // the span still ends
+        }
+        finally {
+            span.end();
+        }
+    }
+}
+exports.PgInstrumentation = PgInstrumentation;
+function clientPrototype(moduleExports) {
+    const mod = moduleExports;
+    const proto = mod?.Client?.prototype;
+    return proto && typeof proto.query === 'function' ? proto : undefined;
+}
+//# sourceMappingURL=pg.js.map

package/dist/pg.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pg.js","sourceRoot":"","sources":["../src/pg.ts"],"names":[],"mappings":";;;AAAA,qEAAqE;AACrE,yEAAyE;AACzE,sEAAsE;AACtE,mEAAmE;AACnE,qEAAqE;AACrE,qEAAqE;AACrE,oEAAoE;AACpE,mEAAmE;AACnE,6CAA6C;AAC7C,4CAAwH;AACxH,oEAAiJ;AACjJ,+CAAgD;AAChD,uCAAyC;AACzC,6CAAgD;AAChD,iDAAkE;AAUlE,MAAM,iBAAiB,GAAG,IAAI,CAAC;AAC/B,MAAM,iBAAiB,GAAG,CAAC,YAAY,CAAC,CAAC;AAiBzC,MAAa,iBAAkB,SAAQ,qCAA8B;IACnE,YAAY,OAAkB;QAC5B,KAAK,CAAC,kBAAkB,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IAC9C,CAAC;IAES,IAAI;QACZ,OAAO,IAAI,qDAAmC,CAC5C,IAAI,EACJ,iBAAiB,EACjB,CAAC,aAAsB,EAAE,EAAE;YACzB,MAAM,KAAK,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;YAC7C,IAAI,KAAK,EAAE,CAAC;gBACV,IAAI,IAAA,2BAAS,EAAC,KAAK,CAAC,KAAK,CAAC,IAAI,IAAA,cAAI,EAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;oBAChD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;gBAC/B,CAAC;gBACD,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC;gBAC3B,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC3C,IAAI,KAAK,CAAC,KAAK,KAAK,MAAM,EAAE,CAAC;oBAC3B,IAAA,gBAAM,EAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,8CAA8C;gBACrE,CAAC;YACH,CAAC;YACD,OAAO,aAAa,CAAC;QACvB,CAAC,EACD,CAAC,aAAsB,EAAE,EAAE;YACzB,MAAM,KAAK,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;YAC7C,IAAI,KAAK,IAAI,IAAA,2BAAS,EAAC,KAAK,CAAC,KAAK,CAAC,IAAI,IAAA,cAAI,EAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC,CACF,CAAC;IACJ,CAAC;IAEgB,SAAS,GAAG,CAAC,QAAuB,EAAiB,EAAE;QACtE,MAAM,eAAe,GAAG,IAAI,CAAC;QAC7B,OAAO,SAAS,KAAK,CAAmB,GAAG,IAAe;YACxD,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,GAAG,IAAI,CAAC;YACxC,IAAI,MAAM,IAAI,OAAQ,MAAsB,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;gBACnE,OAAO,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,gDAAgD;YACrF,CAAC;YACD,MAAM,KAAK,GAAG,aAAO,CAAC,MAAM,EAAE,CAAC;YAC/B,IAAI,IAAsB,CAAC;YAC3B,IAAI,MAAM,GAAG,KAAK,CAAC;YACnB,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,eAAe,CAAC,SAAS,EAAE,CAAC;gBAC5C,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAA,6BAAe,EAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC;gBACrG,IAAI,IAAA,wBAAS,EAAC,MAAM,CAAC,EAAE,CAAC;oBACtB,OAAO,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,0CAA0C;gBAC/E,CAAC;gBACD,IAAI,GAAG,eAAe,CAAC,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;YAC9D,CAAC;YAAC,MAAM,CAAC;gBACP,IAAI,GAAG,SAAS,CAAC;YACnB,CAAC;YACD,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAO,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACpC,CAAC;YACD,MAAM,GAAG,GAAG,eAAe,CAAC,YAAY,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;YACrE,sEAAsE;YACtE,yDAAyD;YACzD,IAAI,QAAQ,GAAG,IAAI,CAAC;YACpB,MAAM,YAAY,GAAG,OAAO,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,QAAQ,KAAK,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAE,MAA6B,EAAE,QAAQ,CAAC;YAClJ,IAAI,OAAO,YAAY,KAAK,UAAU,EAAE,CAAC;gBACvC,MAAM,OAAO,GAAa,aAAO,CAAC,IAAI,CAAC,KAAK,EAAE,UAAyB,KAAc,EAAE,MAAe;oBACpG,GAAG,CAAC,KAAK,CAAC,CAAC;oBACX,OAAQ,YAAyB,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;gBAC9D,CAAC,CAAC,CAAC;gBACH,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE,CAAC;oBACjC,QAAQ,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;gBAC/B,CAAC;qBAAM,IAAI,OAAO,QAAQ,KAAK,UAAU,EAAE,CAAC;oBAC1C,QAAQ,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;gBACvC,CAAC;qBAAM,CAAC;oBACN,4DAA4D;oBAC5D,gEAAgE;oBAChE,qDAAqD;oBACrD,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,MAAgB,CAAgB,CAAC;oBACpE,YAAY,CAAC,QAAQ,GAAG,OAAO,CAAC;oBAChC,QAAQ,GAAG,CAAC,YAAY,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;gBAC9C,CAAC;YACH,CAAC;YACD,MAAM,MAAM,GAAG,WAAK,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YAC3C,IAAI,MAAe,CAAC;YACpB,IAAI,CAAC;gBACH,MAAM,GAAG,aAAO,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;YACtE,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,KAAK,CAAC,CAAC;gBACX,MAAM,KAAK,CAAC;YACd,CAAC;YACD,IAAI,OAAO,YAAY,KAAK,UAAU,EAAE,CAAC;gBACvC,IAAI,MAAM,IAAI,OAAQ,MAA2B,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBACrE,MAA2B,CAAC,IAAI,CAC/B,GAAG,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,EACpB,CAAC,KAAc,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,IAAI,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC,CAC5D,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,GAAG,CAAC,SAAS,CAAC,CAAC;gBACjB,CAAC;YACH,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC;IACJ,CAAC,CAAC;IAEM,cAAc,CAAC,MAAkB,EAAE,MAAe,EAAE,MAAe;QACzE,MAAM,IAAI,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAQ,MAA6B,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAG,MAAsB,CAAC,IAAe,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5J,MAAM,OAAO,GAAG,IAAA,6BAAgB,EAAC,IAAI,CAAC,CAAC;QACvC,MAAM,UAAU,GAAe,EAAE,gBAAgB,EAAE,YAAY,EAAE,CAAC;QAClE,IAAI,OAAO,KAAK,EAAE,EAAE,CAAC;YACnB,UAAU,CAAC,mBAAmB,CAAC,GAAG,OAAO,CAAC;QAC5C,CAAC;QACD,IAAI,IAAI,KAAK,EAAE,EAAE,CAAC;YAChB,kEAAkE;YAClE,iDAAiD;YACjD,UAAU,CAAC,eAAe,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,iBAAiB,CAAC,CAAC;QACjE,CAAC;QACD,IAAI,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACxC,UAAU,CAAC,cAAc,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC;QAC/C,CAAC;QACD,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACpC,UAAU,CAAC,gBAAgB,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC;QAC7C,CAAC;QACD,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACpC,UAAU,CAAC,aAAa,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC;QAC1C,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,KAAK,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,YAAY,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,UAAU,EAAE,EAAE,MAAM,CAAC,CAAC;IACnI,CAAC;IAEO,YAAY,CAAC,IAAU,EAAE,KAAc;QAC7C,IAAI,CAAC;YACH,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;gBAC1C,iEAAiE;gBACjE,qCAAqC;gBACrC,MAAM,CAAC,GAAG,KAA4E,CAAC;gBACvF,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,KAAK,EAAE,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC;gBACxG,MAAM,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAE,CAAC,CAAC,IAAe,CAAC,IAAI,OAAO,CAAC;gBAC7E,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBACnE,MAAM,OAAO,GAAG,IAAI,KAAK,EAAE,CAAC,CAAC,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,GAAG,IAAI,mBAAmB,IAAI,GAAG,CAAC;gBACpF,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,oBAAc,CAAC,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;gBACxD,IAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,EAAE,gBAAgB,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,mBAAmB,EAAE,OAAO,EAAE,CAAC,CAAC;gBAClG,IAAI,CAAC,YAAY,CAAC,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;gBACnD,IAAI,IAAI,KAAK,EAAE,EAAE,CAAC;oBAChB,IAAI,CAAC,YAAY,CAAC,yBAAyB,EAAE,IAAI,CAAC,CAAC;gBACrD,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,sBAAsB;QACxB,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,GAAG,EAAE,CAAC;QACb,CAAC;IACH,CAAC;CACF;AAnJD,8CAmJC;AAED,SAAS,eAAe,CAAC,aAAsB;IAC7C,MAAM,GAAG,GAAG,aAAwE,CAAC;IACrF,MAAM,KAAK,GAAG,GAAG,EAAE,MAAM,EAAE,SAAS,CAAC;IACrC,OAAO,KAAK,IAAI,OAAO,KAAK,CAAC,KAAK,KAAK,UAAU,CAAC,CAAC,CAAE,KAAkC,CAAC,CAAC,CAAC,SAAS,CAAC;AACtG,CAAC"}

package/dist/pgpool.d.ts

@@ -0,0 +1,20 @@
+import { InstrumentationBase, type InstrumentationConfig, InstrumentationNodeModuleDefinition } from '@opentelemetry/instrumentation';
+import { type CallSiteSampling } from './processor.js';
+/** PgPoolEntryOptions bounds and gates the capture. */
+export interface PgPoolEntryOptions extends InstrumentationConfig {
+    depth: number;
+    sampling: CallSiteSampling;
+    isApp?: (file: string) => boolean;
+    /**
+     * connect false leaves Pool.prototype.connect alone: inside a tracing
+     * Sentry, whose pg instrumentation spans the pool's connect under the
+     * caller's span (and drops it without one), the root context the
+     * package runs connect in would change Sentry's spans.
+     */
+    connect?: boolean;
+}
+export declare class PgPoolEntryInstrumentation extends InstrumentationBase<PgPoolEntryOptions> {
+    constructor(options: PgPoolEntryOptions);
+    protected init(): InstrumentationNodeModuleDefinition;
+    private readonly wrapQuery;
+}

package/dist/pgpool.js

@@ -0,0 +1,99 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PgPoolEntryInstrumentation = void 0;
+// pg-pool entry. Pool.prototype.query hands the query to a client on a
+// later tick, so the pg query span's stack never shows the caller: this
+// wrapper captures the call site where the application calls pool.query
+// and carries it in the active context to the span that starts later.
+//
+// Pool.prototype.connect queues its callback when every client is busy
+// and runs it from the release of another client, in that client's
+// query's context. Its callback is bound to the context of the connect
+// call, so a queued query keeps its own call site and its own parent
+// instead of borrowing the query that freed the client.
+const api_1 = require("@opentelemetry/api");
+const instrumentation_1 = require("@opentelemetry/instrumentation");
+const callsite_js_1 = require("./callsite.js");
+const keys_js_1 = require("./keys.js");
+const processor_js_1 = require("./processor.js");
+function poolPrototype(moduleExports) {
+    const proto = moduleExports?.prototype;
+    return proto && typeof proto.query === 'function' && typeof proto.connect === 'function' ? proto : undefined;
+}
+class PgPoolEntryInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(options) {
+        super('@rootcat/node/pg-pool-entry', '0.2.0', options);
+    }
+    init() {
+        return new instrumentation_1.InstrumentationNodeModuleDefinition('pg-pool', ['>=2.0.0 <4'], (moduleExports) => {
+            const proto = poolPrototype(moduleExports);
+            if (proto) {
+                for (const name of ['query', 'connect']) {
+                    if ((0, instrumentation_1.isWrapped)(proto[name]) && (0, keys_js_1.mine)(proto[name])) {
+                        this._unwrap(proto, name);
+                    }
+                }
+                const beforeQuery = proto.query;
+                this._wrap(proto, 'query', this.wrapQuery);
+                if (proto.query !== beforeQuery) {
+                    (0, keys_js_1.marked)(proto.query); // not when a guard kept another SDK's wrapper
+                }
+                if (this.getConfig().connect !== false) {
+                    const beforeConnect = proto.connect;
+                    this._wrap(proto, 'connect', wrapConnect);
+                    if (proto.connect !== beforeConnect) {
+                        (0, keys_js_1.marked)(proto.connect);
+                    }
+                }
+            }
+            return moduleExports;
+        }, (moduleExports) => {
+            const proto = poolPrototype(moduleExports);
+            if (proto) {
+                for (const name of ['query', 'connect']) {
+                    if ((0, instrumentation_1.isWrapped)(proto[name]) && (0, keys_js_1.mine)(proto[name])) {
+                        this._unwrap(proto, name);
+                    }
+                }
+            }
+        });
+    }
+    wrapQuery = (original) => {
+        const instrumentation = this;
+        return function query(...args) {
+            let carried;
+            try {
+                // A callback continues in the caller's context, never in the
+                // query's: a query issued from it captures its own site.
+                const last = args.length - 1;
+                if (last >= 0 && typeof args[last] === 'function') {
+                    args[last] = api_1.context.bind(api_1.context.active(), args[last]);
+                }
+                const config = instrumentation.getConfig();
+                carried = config.sampling.carry(() => (0, callsite_js_1.captureCallSite)(config.depth, config.isApp, query));
+            }
+            catch {
+                carried = undefined;
+            }
+            return (0, processor_js_1.runCarried)(carried, () => original.apply(this, args));
+        };
+    };
+}
+exports.PgPoolEntryInstrumentation = PgPoolEntryInstrumentation;
+/**
+ * wrapConnect binds a connect callback to the context of the call, and
+ * runs the pool's own work (opening a client and its socket) in the root
+ * context: a connection outlives the query that opened it, and the events
+ * it emits later (a LISTEN notification, an error) must not run in that
+ * query's context, with its span and call site.
+ */
+function wrapConnect(original) {
+    return function connect(...args) {
+        const callback = args[0];
+        if (typeof callback === 'function') {
+            args[0] = api_1.context.bind(api_1.context.active(), callback);
+        }
+        return api_1.context.with(api_1.ROOT_CONTEXT, () => original.apply(this, args));
+    };
+}
+//# sourceMappingURL=pgpool.js.map

package/dist/pgpool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pgpool.js","sourceRoot":"","sources":["../src/pgpool.ts"],"names":[],"mappings":";;;AAAA,uEAAuE;AACvE,wEAAwE;AACxE,wEAAwE;AACxE,sEAAsE;AACtE,EAAE;AACF,uEAAuE;AACvE,mEAAmE;AACnE,uEAAuE;AACvE,qEAAqE;AACrE,wDAAwD;AACxD,4CAA2D;AAC3D,oEAAiJ;AACjJ,+CAAgD;AAChD,uCAAyC;AACzC,iDAAmE;AAmBnE,SAAS,aAAa,CAAC,aAAsB;IAC3C,MAAM,KAAK,GAAI,aAA+D,EAAE,SAAS,CAAC;IAC1F,OAAO,KAAK,IAAI,OAAO,KAAK,CAAC,KAAK,KAAK,UAAU,IAAI,OAAO,KAAK,CAAC,OAAO,KAAK,UAAU,CAAC,CAAC,CAAE,KAAuB,CAAC,CAAC,CAAC,SAAS,CAAC;AAClI,CAAC;AAED,MAAa,0BAA2B,SAAQ,qCAAuC;IACrF,YAAY,OAA2B;QACrC,KAAK,CAAC,6BAA6B,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IACzD,CAAC;IAES,IAAI;QACZ,OAAO,IAAI,qDAAmC,CAC5C,SAAS,EACT,CAAC,YAAY,CAAC,EACd,CAAC,aAAsB,EAAE,EAAE;YACzB,MAAM,KAAK,GAAG,aAAa,CAAC,aAAa,CAAC,CAAC;YAC3C,IAAI,KAAK,EAAE,CAAC;gBACV,KAAK,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,SAAS,CAAU,EAAE,CAAC;oBACjD,IAAI,IAAA,2BAAS,EAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,IAAA,cAAI,EAAC,KAAK,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;wBAChD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;oBAC5B,CAAC;gBACH,CAAC;gBACD,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC;gBAChC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC3C,IAAI,KAAK,CAAC,KAAK,KAAK,WAAW,EAAE,CAAC;oBAChC,IAAA,gBAAM,EAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,8CAA8C;gBACrE,CAAC;gBACD,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC,OAAO,KAAK,KAAK,EAAE,CAAC;oBACvC,MAAM,aAAa,GAAG,KAAK,CAAC,OAAO,CAAC;oBACpC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;oBAC1C,IAAI,KAAK,CAAC,OAAO,KAAK,aAAa,EAAE,CAAC;wBACpC,IAAA,gBAAM,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBACxB,CAAC;gBACH,CAAC;YACH,CAAC;YACD,OAAO,aAAa,CAAC;QACvB,CAAC,EACD,CAAC,aAAsB,EAAE,EAAE;YACzB,MAAM,KAAK,GAAG,aAAa,CAAC,aAAa,CAAC,CAAC;YAC3C,IAAI,KAAK,EAAE,CAAC;gBACV,KAAK,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,SAAS,CAAU,EAAE,CAAC;oBACjD,IAAI,IAAA,2BAAS,EAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,IAAA,cAAI,EAAC,KAAK,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;wBAChD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;oBAC5B,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC,CACF,CAAC;IACJ,CAAC;IAEgB,SAAS,GAAG,CAAC,QAAgB,EAAU,EAAE;QACxD,MAAM,eAAe,GAAG,IAAI,CAAC;QAC7B,OAAO,SAAS,KAAK,CAAgB,GAAG,IAAe;YACrD,IAAI,OAAO,CAAC;YACZ,IAAI,CAAC;gBACH,6DAA6D;gBAC7D,yDAAyD;gBACzD,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;gBAC7B,IAAI,IAAI,IAAI,CAAC,IAAI,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;oBAClD,IAAI,CAAC,IAAI,CAAC,GAAG,aAAO,CAAC,IAAI,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,IAAI,CAAW,CAAC,CAAC;gBACpE,CAAC;gBACD,MAAM,MAAM,GAAG,eAAe,CAAC,SAAS,EAAE,CAAC;gBAC3C,OAAO,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAA,6BAAe,EAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;YAC5F,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,GAAG,SAAS,CAAC;YACtB,CAAC;YACD,OAAO,IAAA,yBAAU,EAAC,OAAO,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;QAC/D,CAAC,CAAC;IACJ,CAAC,CAAC;CACH;AAhED,gEAgEC;AAED;;;;;;GAMG;AACH,SAAS,WAAW,CAAC,QAAgB;IACnC,OAAO,SAAS,OAAO,CAAgB,GAAG,IAAe;QACvD,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACzB,IAAI,OAAO,QAAQ,KAAK,UAAU,EAAE,CAAC;YACnC,IAAI,CAAC,CAAC,CAAC,GAAG,aAAO,CAAC,IAAI,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,QAAkB,CAAC,CAAC;QAC/D,CAAC;QACD,OAAO,aAAO,CAAC,IAAI,CAAC,kBAAY,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;IACtE,CAAC,CAAC;AACJ,CAAC"}

package/dist/postgresjs.d.ts

@@ -0,0 +1,60 @@
+import { InstrumentationBase, type InstrumentationConfig, InstrumentationNodeModuleDefinition } from '@opentelemetry/instrumentation';
+import { type CallSiteLocation } from './callsite.js';
+import { type CallSiteSampling } from './processor.js';
+/** PostgresJsOptions configures the instrumentation. */
+export interface PostgresJsOptions extends InstrumentationConfig {
+    /** sampling gates the origin parsing, like any capture. */
+    sampling: CallSiteSampling;
+    /**
+     * spans false starts no span: another SDK (Sentry, tracing on) has
+     * already started one for the query when handle() runs, and onSite is
+     * handed the query's origin to give that span its call site.
+     */
+    spans?: boolean;
+    /** onSite receives a query's call site when spans is false. */
+    onSite?: (site: CallSiteLocation) => void;
+}
+export declare class PostgresJsInstrumentation extends InstrumentationBase<PostgresJsOptions> {
+    private readonly patched;
+    /** origins remembers the parsed call site per template, null for none. */
+    private readonly origins;
+    /** recording maps a sql function to its site-recording proxy (spans false). */
+    private readonly recording;
+    /** templateSites remembers the captured call site per template (spans false). */
+    private readonly templateSites;
+    /** querySites holds the call site captured for a query until it is handled (spans false). */
+    private readonly querySites;
+    constructor(options: PostgresJsOptions);
+    protected init(): InstrumentationNodeModuleDefinition;
+    /**
+     * patchModule wraps the factory. CommonJS exports the function itself
+     * (the wrapper is returned in its place); an ES module namespace has it
+     * as default (replaced in place).
+     */
+    private patchModule;
+    private unpatchModule;
+    private wrapFactory;
+    /**
+     * siteRecording is the sql function as the application gets it when
+     * another SDK (Sentry) spans the queries: the driver's own origin then
+     * cannot name the application, as it keeps four frames and Sentry's
+     * wrapper of the sql function takes the last one. The call site is
+     * captured here instead, where the application calls: once per tagged
+     * template (a template is one place in the source, as the driver's own
+     * origin cache assumes), and per unsafe() call unless an outer layer
+     * (Drizzle) already carries one; queries inside sql.begin() get the
+     * same treatment through the transaction's sql function.
+     */
+    private siteRecording;
+    /** patchQueryClass finds Query.prototype through an unexecuted query. */
+    private patchQueryClass;
+    private readonly wrapHandle;
+    /**
+     * originSite is the application frame of the driver's own origin stack
+     * for a tagged query, parsed once per template; undefined for unsafe()
+     * and file() queries, which hold no origin.
+     */
+    private originSite;
+    private startQuerySpan;
+    private endQuerySpan;
+}