shield360 1.0.0

package/dist/instrumentation/pg/index.d.ts

@@ -0,0 +1,20 @@
+import { InstrumentationBase, InstrumentationModuleDefinition, InstrumentationConfig } from '@opentelemetry/instrumentation';
+/**
+ * Configuration for the PostgreSQL (`pg`) instrumentation.
+ *
+ * `captureDbParameters` mirrors the Python psycopg instrumentor's
+ * `capture_db_parameters` flag. It is OFF by default because query parameters
+ * may contain PII / secrets. Enable only when you are certain parameters are safe.
+ */
+export interface PgInstrumentationConfig extends InstrumentationConfig {
+    captureDbParameters?: boolean;
+}
+export default class Shield360PgInstrumentation extends InstrumentationBase {
+    private _clientProto;
+    constructor(config?: PgInstrumentationConfig);
+    private get captureDbParameters();
+    protected init(): void | InstrumentationModuleDefinition | InstrumentationModuleDefinition[];
+    manualPatch(pg: any): void;
+    protected _patch(moduleExports: any, moduleVersion?: string): void;
+    protected _unpatch(moduleExports?: any): void;
+}

package/dist/instrumentation/pg/index.js

@@ -0,0 +1,74 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const instrumentation_1 = require("@opentelemetry/instrumentation");
+const api_1 = require("@opentelemetry/api");
+const constant_1 = require("../../constant");
+const wrapper_1 = __importDefault(require("./wrapper"));
+class Shield360PgInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-pg`, '1.0.0', config);
+        this._clientProto = null;
+    }
+    get captureDbParameters() {
+        return Boolean(this.getConfig()?.captureDbParameters);
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition('pg', ['>=7.0.0', '>=8.0.0'], (moduleExports, moduleVersion) => {
+            this._patch(moduleExports, moduleVersion);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined) {
+                this._unpatch(moduleExports);
+            }
+        });
+        return [module];
+    }
+    manualPatch(pg) {
+        this._patch(pg);
+    }
+    _patch(moduleExports, moduleVersion) {
+        try {
+            if (!moduleExports)
+                return;
+            const sdkVersion = moduleVersion ? String(moduleVersion) : undefined;
+            const wrapperConfig = { captureDbParameters: this.captureDbParameters, sdkVersion };
+            // Wrap ONLY Client.prototype.query.
+            //
+            // We deliberately do NOT wrap Pool.prototype.query. In `pg`, Pool.query()
+            // checks out a pooled Client and delegates to that Client's query(), so
+            // client-level wrapping already captures pool-routed queries. Wrapping both
+            // would emit two spans for a single logical pool query (a Pool span plus a
+            // child Client span). This matches the OTel community pg instrumentation,
+            // which also instruments only the Client. Result: exactly one span per query.
+            const Client = moduleExports.Client;
+            if (Client?.prototype && typeof Client.prototype.query === 'function') {
+                if ((0, instrumentation_1.isWrapped)(Client.prototype.query)) {
+                    this._unwrap(Client.prototype, 'query');
+                }
+                this._clientProto = Client.prototype;
+                this._wrap(Client.prototype, 'query', wrapper_1.default._patchQuery(this.tracer, wrapperConfig));
+            }
+        }
+        catch (e) {
+            api_1.diag.error('pg instrumentation: error in _patch method', e);
+        }
+    }
+    _unpatch(moduleExports) {
+        try {
+            if (moduleExports?.Client?.prototype && (0, instrumentation_1.isWrapped)(moduleExports.Client.prototype.query)) {
+                this._unwrap(moduleExports.Client.prototype, 'query');
+            }
+            else if (this._clientProto && (0, instrumentation_1.isWrapped)(this._clientProto.query)) {
+                this._unwrap(this._clientProto, 'query');
+            }
+        }
+        catch {
+            /* ignore */
+        }
+    }
+}
+exports.default = Shield360PgInstrumentation;
+//# sourceMappingURL=index.js.map

package/dist/instrumentation/pg/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/pg/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAMwC;AACxC,4CAA0C;AAC1C,6CAAwD;AACxD,wDAAkC;AAalC,MAAqB,0BAA2B,SAAQ,qCAAmB;IAGzE,YAAY,SAAkC,EAAE;QAC9C,KAAK,CAAC,GAAG,iCAAsB,qBAAqB,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QAHjE,iBAAY,GAAQ,IAAI,CAAC;IAIjC,CAAC;IAED,IAAY,mBAAmB;QAC7B,OAAO,OAAO,CAAE,IAAI,CAAC,SAAS,EAA8B,EAAE,mBAAmB,CAAC,CAAC;IACrF,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,IAAI,EACJ,CAAC,SAAS,EAAE,SAAS,CAAC,EACtB,CAAC,aAAa,EAAE,aAAa,EAAE,EAAE;YAC/B,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,aAAa,CAAC,CAAC;YAC1C,OAAO,aAAa,CAAC;QACvB,CAAC,EACD,CAAC,aAAa,EAAE,EAAE;YAChB,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;gBAChC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC,CACF,CAAC;QACF,OAAO,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAEM,WAAW,CAAC,EAAO;QACxB,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAClB,CAAC;IAES,MAAM,CAAC,aAAkB,EAAE,aAAsB;QACzD,IAAI,CAAC;YACH,IAAI,CAAC,aAAa;gBAAE,OAAO;YAC3B,MAAM,UAAU,GAAG,aAAa,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACrE,MAAM,aAAa,GAAG,EAAE,mBAAmB,EAAE,IAAI,CAAC,mBAAmB,EAAE,UAAU,EAAE,CAAC;YAEpF,oCAAoC;YACpC,EAAE;YACF,0EAA0E;YAC1E,wEAAwE;YACxE,4EAA4E;YAC5E,2EAA2E;YAC3E,0EAA0E;YAC1E,8EAA8E;YAC9E,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC;YACpC,IAAI,MAAM,EAAE,SAAS,IAAI,OAAO,MAAM,CAAC,SAAS,CAAC,KAAK,KAAK,UAAU,EAAE,CAAC;gBACtE,IAAI,IAAA,2BAAS,EAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;oBACtC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;gBAC1C,CAAC;gBACD,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,SAAS,CAAC;gBACrC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,EAAE,iBAAS,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,CAAC;YAC3F,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,UAAI,CAAC,KAAK,CAAC,4CAA4C,EAAE,CAAC,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;IAES,QAAQ,CAAC,aAAmB;QACpC,IAAI,CAAC;YACH,IAAI,aAAa,EAAE,MAAM,EAAE,SAAS,IAAI,IAAA,2BAAS,EAAC,aAAa,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;gBACxF,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACxD,CAAC;iBAAM,IAAI,IAAI,CAAC,YAAY,IAAI,IAAA,2BAAS,EAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;gBACnE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,YAAY;QACd,CAAC;IACH,CAAC;CACF;AAtED,6CAsEC"}

package/dist/instrumentation/pg/wrapper.d.ts

@@ -0,0 +1,66 @@
+import { Span, Tracer } from '@opentelemetry/api';
+import BaseWrapper from '../base-wrapper';
+export interface PgWrapperConfig {
+    captureDbParameters?: boolean;
+    sdkVersion?: string;
+}
+declare class PgWrapper extends BaseWrapper {
+    static dbSystem: string;
+    /** Detect the SQL operation verb from a query string (mirrors Python parse_sql_operation). */
+    static parseSqlOperation(query: unknown): string;
+    /** Extract the target table name for span naming (mirrors Python extract_table_name). */
+    static extractTableName(query: unknown): string;
+    /** Detect pgvector similarity operator → metric name (mirrors Python detect_special_features). */
+    static detectSimilarityMetric(query: unknown): string | undefined;
+    /** Detect full-text search keywords (mirrors Python detect_special_features). */
+    static detectFullTextSearch(query: unknown): boolean;
+    /** Resolve batch size when values is an array of parameter arrays (executemany-style). */
+    static resolveBatchSize(values: unknown): number | undefined;
+    /** Build the human-readable query summary (mirrors Python get_query_summary). */
+    static getQuerySummary(dbOperation: string, tableName: string, query: unknown): string;
+    /** Sanitize a parameter value for safe inclusion in a span (mirrors Python sanitize_parameter). */
+    static sanitizeParameter(value: unknown): string;
+    /**
+     * Extract { host, port, database } from a `pg` Client / Pool instance.
+     * `pg` stores connection params on the instance (and Pool on `options`).
+     * Defaults mirror the Python psycopg instrumentation (localhost:5432).
+     */
+    static extractConnectionInfo(instance: any): {
+        host: string;
+        port: number;
+        database?: string;
+    };
+    /**
+     * Resolve the SQL text and the parameter values from the `pg` query() arguments.
+     * Supports: (text, values?, cb?), ({ text, values, name }, cb?), and submittables.
+     */
+    static resolveQuery(args: any[]): {
+        queryText: unknown;
+        values: unknown;
+    };
+    /** Pull the last argument if it is a node-style callback. */
+    static extractCallback(args: any[]): ((...cbArgs: any[]) => void) | undefined;
+    /** Stamp all span attributes (mirrors Python common_psycopg_logic). */
+    static setSpanAttributes(span: Span, { dbOperation, tableName, queryText, values, connInfo, durationSeconds, rowCount, batchSize, captureDbParameters, sdkVersion, }: {
+        dbOperation: string;
+        tableName: string;
+        queryText: unknown;
+        values: unknown;
+        connInfo: {
+            host: string;
+            port: number;
+            database?: string;
+        };
+        durationSeconds: number;
+        rowCount?: number | null;
+        batchSize?: number;
+        captureDbParameters: boolean;
+        sdkVersion?: string;
+    }): void;
+    /**
+     * Build the patched `query()` replacement. Handles callback, promise, and
+     * submittable forms while always producing a single CLIENT span.
+     */
+    static _patchQuery(tracer: Tracer, config?: PgWrapperConfig): any;
+}
+export default PgWrapper;

package/dist/instrumentation/pg/wrapper.js

@@ -0,0 +1,441 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const api_1 = require("@opentelemetry/api");
+const core_1 = require("@opentelemetry/core");
+const semantic_conventions_1 = require("@opentelemetry/semantic-conventions");
+const config_1 = __importDefault(require("../../config"));
+const constant_1 = require("../../constant");
+const helpers_1 = __importStar(require("../../helpers"));
+const metrics_1 = __importDefault(require("../../otel/metrics"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+const base_wrapper_1 = __importDefault(require("../base-wrapper"));
+/**
+ * PostgreSQL (`pg`) instrumentation wrapper.
+ *
+ * Mirrors the Python SDK psycopg instrumentation
+ * (sdk/python/src/shield360/instrumentation/psycopg) for the Node `pg` driver:
+ *   - parses the SQL operation (SELECT/INSERT/...) and target table for span naming,
+ *   - detects pgvector similarity operators (`<=>`, `<->`, `<#>`),
+ *   - honors a `captureDbParameters` toggle to record query parameters
+ *     (off by default — parameters can contain PII/secrets),
+ *   - sets the same `db.*` semantic-convention attributes the Python SDK sets.
+ *
+ * The `pg` `query()` method has three call shapes which must all be handled:
+ *   1. callback:  client.query(text, values?, (err, res) => {})       → no return value
+ *   2. promise:   const res = await client.query(text, values?)        → returns a thenable
+ *   3. submittable: client.query(new Query(...)) / cursors             → returns the submittable
+ */
+// Maximum number of query parameters to capture (matches Python MAX_PARAMS_COUNT).
+const MAX_PARAMS_COUNT = 50;
+// Maximum captured parameter value length (matches Python MAX_PARAM_LENGTH).
+const MAX_PARAM_LENGTH = 256;
+// Map of leading SQL keyword → db.operation.name value (mirrors Python SQL_OPERATION_MAP).
+const SQL_OPERATION_MAP = {
+    SELECT: semantic_convention_1.default.DB_OPERATION_SELECT,
+    INSERT: semantic_convention_1.default.DB_OPERATION_INSERT,
+    UPDATE: semantic_convention_1.default.DB_OPERATION_UPDATE,
+    DELETE: semantic_convention_1.default.DB_OPERATION_DELETE,
+    COPY: semantic_convention_1.default.DB_OPERATION_COPY,
+    CREATE: semantic_convention_1.default.DB_OPERATION_CREATE,
+    ALTER: semantic_convention_1.default.DB_OPERATION_ALTER,
+    DROP: semantic_convention_1.default.DB_OPERATION_DROP,
+    TRUNCATE: semantic_convention_1.default.DB_OPERATION_TRUNCATE,
+    COMMIT: semantic_convention_1.default.DB_OPERATION_COMMIT,
+    ROLLBACK: semantic_convention_1.default.DB_OPERATION_ROLLBACK,
+};
+// Table-name extraction patterns (mirrors Python TABLE_PATTERNS).
+const TABLE_PATTERNS = [
+    /INSERT\s+INTO\s+["']?(\w+)["']?/i,
+    /UPDATE\s+["']?(\w+)["']?/i,
+    /DELETE\s+FROM\s+["']?(\w+)["']?/i,
+    /FROM\s+["']?(\w+)["']?/i,
+    /TRUNCATE\s+(?:TABLE\s+)?["']?(\w+)["']?/i,
+    /CREATE\s+(?:TABLE|INDEX)\s+(?:IF\s+NOT\s+EXISTS\s+)?["']?(\w+)["']?/i,
+    /ALTER\s+TABLE\s+["']?(\w+)["']?/i,
+    /DROP\s+(?:TABLE|INDEX)\s+(?:IF\s+EXISTS\s+)?["']?(\w+)["']?/i,
+    /COPY\s+["']?(\w+)["']?/i,
+];
+function recordDbMetrics(dbOperation, connInfo, durationSeconds) {
+    if (config_1.default.disableMetrics) {
+        return;
+    }
+    const applicationName = config_1.default.applicationName || '';
+    const environment = config_1.default.environment || '';
+    const attributes = {
+        [semantic_conventions_1.ATTR_TELEMETRY_SDK_NAME]: constant_1.SDK_NAME,
+        [semantic_conventions_1.ATTR_SERVICE_NAME]: applicationName,
+        [semantic_conventions_1.SEMRESATTRS_DEPLOYMENT_ENVIRONMENT]: environment,
+        [semantic_convention_1.default.DB_SYSTEM_NAME]: PgWrapper.dbSystem,
+        [semantic_convention_1.default.DB_OPERATION_NAME]: dbOperation,
+        [semantic_convention_1.default.SERVER_ADDRESS]: connInfo.host,
+        [semantic_convention_1.default.SERVER_PORT]: connInfo.port,
+    };
+    metrics_1.default.dbRequests?.add(1, attributes);
+    metrics_1.default.dbClientOperationDuration?.record(durationSeconds, attributes);
+}
+class PgWrapper extends base_wrapper_1.default {
+    /** Detect the SQL operation verb from a query string (mirrors Python parse_sql_operation). */
+    static parseSqlOperation(query) {
+        if (query === null || query === undefined) {
+            return semantic_convention_1.default.DB_OPERATION_QUERY;
+        }
+        const queryStr = String(query).trim();
+        if (!queryStr) {
+            return semantic_convention_1.default.DB_OPERATION_QUERY;
+        }
+        const queryUpper = queryStr.toUpperCase();
+        for (const [keyword, operation] of Object.entries(SQL_OPERATION_MAP)) {
+            if (queryUpper.startsWith(keyword)) {
+                return operation;
+            }
+        }
+        // Handle WITH ... SELECT/INSERT/UPDATE/DELETE (CTEs).
+        if (queryUpper.startsWith('WITH')) {
+            for (const [keyword, operation] of Object.entries(SQL_OPERATION_MAP)) {
+                if (queryUpper.includes(keyword)) {
+                    return operation;
+                }
+            }
+        }
+        return semantic_convention_1.default.DB_OPERATION_QUERY;
+    }
+    /** Extract the target table name for span naming (mirrors Python extract_table_name). */
+    static extractTableName(query) {
+        if (query === null || query === undefined) {
+            return 'unknown';
+        }
+        const queryStr = String(query);
+        for (const pattern of TABLE_PATTERNS) {
+            const match = pattern.exec(queryStr);
+            if (match) {
+                return match[1];
+            }
+        }
+        return 'unknown';
+    }
+    /** Detect pgvector similarity operator → metric name (mirrors Python detect_special_features). */
+    static detectSimilarityMetric(query) {
+        if (query === null || query === undefined) {
+            return undefined;
+        }
+        const queryStr = String(query);
+        if (queryStr.includes('<=>'))
+            return 'cosine';
+        if (queryStr.includes('<->'))
+            return 'l2';
+        if (queryStr.includes('<#>'))
+            return 'inner_product';
+        return undefined;
+    }
+    /** Detect full-text search keywords (mirrors Python detect_special_features). */
+    static detectFullTextSearch(query) {
+        if (query === null || query === undefined) {
+            return false;
+        }
+        const queryUpper = String(query).toUpperCase();
+        return (queryUpper.includes('TSVECTOR') ||
+            queryUpper.includes('TSQUERY') ||
+            queryUpper.includes('TS_RANK') ||
+            queryUpper.includes('WEBSEARCH_TO_TSQUERY'));
+    }
+    /** Resolve batch size when values is an array of parameter arrays (executemany-style). */
+    static resolveBatchSize(values) {
+        if (!Array.isArray(values) || values.length === 0) {
+            return undefined;
+        }
+        if (Array.isArray(values[0])) {
+            return values.length;
+        }
+        return undefined;
+    }
+    /** Build the human-readable query summary (mirrors Python get_query_summary). */
+    static getQuerySummary(dbOperation, tableName, query) {
+        let summary = `${dbOperation} ${tableName}`;
+        const metric = PgWrapper.detectSimilarityMetric(query);
+        if (metric) {
+            summary += ` (vector ${metric})`;
+        }
+        if (PgWrapper.detectFullTextSearch(query)) {
+            summary += ' (full-text)';
+        }
+        return summary;
+    }
+    /** Sanitize a parameter value for safe inclusion in a span (mirrors Python sanitize_parameter). */
+    static sanitizeParameter(value) {
+        if (value === null || value === undefined) {
+            return 'NULL';
+        }
+        if (value instanceof Uint8Array || Buffer.isBuffer(value)) {
+            return `<bytes:${value.length}>`;
+        }
+        let strValue;
+        try {
+            strValue = typeof value === 'object' ? JSON.stringify(value) : String(value);
+        }
+        catch {
+            return '<unrepresentable>';
+        }
+        if (strValue.length > MAX_PARAM_LENGTH) {
+            return `${strValue.slice(0, MAX_PARAM_LENGTH)}...<truncated:${strValue.length}>`;
+        }
+        return strValue;
+    }
+    /**
+     * Extract { host, port, database } from a `pg` Client / Pool instance.
+     * `pg` stores connection params on the instance (and Pool on `options`).
+     * Defaults mirror the Python psycopg instrumentation (localhost:5432).
+     */
+    static extractConnectionInfo(instance) {
+        const defaults = { host: 'localhost', port: 5432 };
+        try {
+            const params = instance?.connectionParameters || instance?.options || instance || {};
+            const host = params.host || defaults.host;
+            const port = params.port ? Number(params.port) : defaults.port;
+            const database = params.database;
+            return { host, port: Number.isFinite(port) ? port : defaults.port, database };
+        }
+        catch {
+            return defaults;
+        }
+    }
+    /**
+     * Resolve the SQL text and the parameter values from the `pg` query() arguments.
+     * Supports: (text, values?, cb?), ({ text, values, name }, cb?), and submittables.
+     */
+    static resolveQuery(args) {
+        const first = args[0];
+        if (typeof first === 'string') {
+            return { queryText: first, values: Array.isArray(args[1]) ? args[1] : undefined };
+        }
+        if (first && typeof first === 'object') {
+            // Query config object or a Submittable (Query/Cursor) — both may carry `text`/`values`.
+            const queryText = 'text' in first ? first.text : undefined;
+            const values = 'values' in first ? first.values : undefined;
+            return { queryText, values };
+        }
+        return { queryText: undefined, values: undefined };
+    }
+    /** Pull the last argument if it is a node-style callback. */
+    static extractCallback(args) {
+        const last = args[args.length - 1];
+        return typeof last === 'function' ? last : undefined;
+    }
+    /** Stamp all span attributes (mirrors Python common_psycopg_logic). */
+    static setSpanAttributes(span, { dbOperation, tableName, queryText, values, connInfo, durationSeconds, rowCount, batchSize, captureDbParameters, sdkVersion, }) {
+        const applicationName = config_1.default.applicationName || '';
+        const environment = config_1.default.environment || '';
+        span.setAttribute(semantic_conventions_1.ATTR_TELEMETRY_SDK_NAME, constant_1.SDK_NAME);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_OPERATION, semantic_convention_1.default.GEN_AI_OPERATION_TYPE_VECTORDB);
+        span.setAttribute(semantic_convention_1.default.DB_SYSTEM_NAME, PgWrapper.dbSystem);
+        span.setAttribute(semantic_convention_1.default.DB_OPERATION_NAME, dbOperation);
+        span.setAttribute(semantic_convention_1.default.SERVER_ADDRESS, connInfo.host);
+        span.setAttribute(semantic_convention_1.default.SERVER_PORT, connInfo.port);
+        span.setAttribute(semantic_convention_1.default.ATTR_DEPLOYMENT_ENVIRONMENT, environment);
+        span.setAttribute(semantic_conventions_1.ATTR_SERVICE_NAME, applicationName);
+        span.setAttribute(semantic_convention_1.default.DB_CLIENT_OPERATION_DURATION, durationSeconds);
+        if (sdkVersion) {
+            span.setAttribute(semantic_convention_1.default.DB_SDK_VERSION, sdkVersion);
+        }
+        if (connInfo.database && connInfo.database !== 'unknown') {
+            span.setAttribute(semantic_convention_1.default.DB_NAMESPACE_POSTGRESQL, connInfo.database);
+        }
+        if (tableName && tableName !== 'unknown') {
+            span.setAttribute(semantic_convention_1.default.DB_COLLECTION_NAME, tableName);
+        }
+        if (typeof batchSize === 'number' && batchSize > 0) {
+            span.setAttribute(semantic_convention_1.default.DB_BATCH_SIZE, batchSize);
+        }
+        // Capture query text when message-content capture is enabled (parity w/ Python).
+        if (config_1.default.captureMessageContent && queryText) {
+            let queryStr = String(queryText);
+            const maxLen = config_1.default.maxContentLength;
+            if (typeof maxLen === 'number' && maxLen > 0 && queryStr.length > maxLen) {
+                queryStr = queryStr.slice(0, maxLen);
+            }
+            span.setAttribute(semantic_convention_1.default.DB_QUERY_TEXT, queryStr);
+        }
+        // Capture query parameters (OTel per-index convention) only when explicitly enabled.
+        const paramsForCapture = typeof batchSize === 'number' && batchSize > 0 && Array.isArray(values)
+            ? values[0]
+            : values;
+        if (captureDbParameters && paramsForCapture !== null && paramsForCapture !== undefined) {
+            if (Array.isArray(paramsForCapture)) {
+                paramsForCapture.slice(0, MAX_PARAMS_COUNT).forEach((value, idx) => {
+                    span.setAttribute(`${semantic_convention_1.default.DB_QUERY_PARAMETER}.${idx}`, PgWrapper.sanitizeParameter(value));
+                });
+            }
+            else if (typeof paramsForCapture === 'object') {
+                Object.entries(paramsForCapture)
+                    .slice(0, MAX_PARAMS_COUNT)
+                    .forEach(([key, value]) => {
+                    span.setAttribute(`${semantic_convention_1.default.DB_QUERY_PARAMETER}.${key}`, PgWrapper.sanitizeParameter(value));
+                });
+            }
+        }
+        span.setAttribute(semantic_convention_1.default.DB_QUERY_SUMMARY, PgWrapper.getQuerySummary(dbOperation, tableName, queryText));
+        if (typeof rowCount === 'number' && rowCount >= 0) {
+            span.setAttribute(semantic_convention_1.default.DB_RESPONSE_RETURNED_ROWS, rowCount);
+        }
+        // pgvector similarity metric.
+        const metric = PgWrapper.detectSimilarityMetric(queryText);
+        if (metric) {
+            span.setAttribute(semantic_convention_1.default.DB_SEARCH_SIMILARITY_METRIC, metric);
+        }
+        (0, helpers_1.applyCustomSpanAttributes)(span);
+        span.setStatus({ code: api_1.SpanStatusCode.OK });
+        recordDbMetrics(dbOperation, connInfo, durationSeconds);
+    }
+    /**
+     * Build the patched `query()` replacement. Handles callback, promise, and
+     * submittable forms while always producing a single CLIENT span.
+     */
+    static _patchQuery(tracer, config = {}) {
+        const captureDbParameters = Boolean(config.captureDbParameters);
+        const sdkVersion = config.sdkVersion;
+        return (originalQuery) => {
+            return function (...args) {
+                if ((0, core_1.isTracingSuppressed)(api_1.context.active())) {
+                    return originalQuery.apply(this, args);
+                }
+                const { queryText, values } = PgWrapper.resolveQuery(args);
+                const dbOperation = PgWrapper.parseSqlOperation(queryText);
+                const tableName = PgWrapper.extractTableName(queryText);
+                const batchSize = PgWrapper.resolveBatchSize(values);
+                const connInfo = PgWrapper.extractConnectionInfo(this);
+                const spanName = typeof batchSize === 'number' && batchSize > 0
+                    ? `${dbOperation} ${tableName} (batch)`
+                    : `${dbOperation} ${tableName}`;
+                const span = tracer.startSpan(spanName, { kind: api_1.SpanKind.CLIENT });
+                const startTime = Date.now();
+                const finalize = (result) => {
+                    try {
+                        const durationSeconds = (Date.now() - startTime) / 1000;
+                        const rowCount = result && typeof result.rowCount === 'number' ? result.rowCount : undefined;
+                        PgWrapper.setSpanAttributes(span, {
+                            dbOperation,
+                            tableName,
+                            queryText,
+                            values,
+                            connInfo,
+                            durationSeconds,
+                            rowCount,
+                            batchSize,
+                            captureDbParameters,
+                            sdkVersion,
+                        });
+                    }
+                    catch (attrErr) {
+                        helpers_1.default.handleException(span, attrErr);
+                    }
+                    finally {
+                        span.end();
+                    }
+                };
+                const fail = (error) => {
+                    helpers_1.default.handleException(span, error);
+                    span.end();
+                };
+                return api_1.context.with(api_1.trace.setSpan(api_1.context.active(), span), () => {
+                    const callback = PgWrapper.extractCallback(args);
+                    // ── Callback form ────────────────────────────────────────────────
+                    if (callback) {
+                        const wrappedCallback = function (err, res) {
+                            if (err) {
+                                fail(err);
+                            }
+                            else {
+                                finalize(res);
+                            }
+                            return callback.apply(this, [err, res]);
+                        };
+                        const newArgs = args.slice(0, -1).concat(wrappedCallback);
+                        try {
+                            return originalQuery.apply(this, newArgs);
+                        }
+                        catch (syncErr) {
+                            fail(syncErr);
+                            throw syncErr;
+                        }
+                    }
+                    // ── Promise / submittable form ───────────────────────────────────
+                    let result;
+                    try {
+                        result = originalQuery.apply(this, args);
+                    }
+                    catch (syncErr) {
+                        fail(syncErr);
+                        throw syncErr;
+                    }
+                    if (result && typeof result.then === 'function') {
+                        return result.then((res) => {
+                            finalize(res);
+                            return res;
+                        }, (err) => {
+                            fail(err);
+                            throw err;
+                        });
+                    }
+                    // Submittable (Query/Cursor) — no promise, no callback. Listen for completion.
+                    if (result && typeof result.on === 'function') {
+                        let settled = false;
+                        result.on('end', () => {
+                            if (settled)
+                                return;
+                            settled = true;
+                            finalize(result);
+                        });
+                        result.on('error', (err) => {
+                            if (settled)
+                                return;
+                            settled = true;
+                            fail(err);
+                        });
+                        return result;
+                    }
+                    // Unknown shape — close the span without a row count.
+                    finalize(result);
+                    return result;
+                });
+            };
+        };
+    }
+}
+PgWrapper.dbSystem = semantic_convention_1.default.DB_SYSTEM_POSTGRESQL;
+exports.default = PgWrapper;
+//# sourceMappingURL=wrapper.js.map

package/dist/instrumentation/pg/wrapper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/instrumentation/pg/wrapper.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,4CAQ4B;AAC5B,8CAA0D;AAC1D,8EAI6C;AAC7C,0DAA2C;AAC3C,6CAA0C;AAC1C,yDAA2E;AAC3E,iEAAyC;AACzC,oFAA2D;AAC3D,mEAA0C;AAE1C;;;;;;;;;;;;;;;GAeG;AAEH,mFAAmF;AACnF,MAAM,gBAAgB,GAAG,EAAE,CAAC;AAC5B,6EAA6E;AAC7E,MAAM,gBAAgB,GAAG,GAAG,CAAC;AAE7B,2FAA2F;AAC3F,MAAM,iBAAiB,GAA2B;IAChD,MAAM,EAAE,6BAAkB,CAAC,mBAAmB;IAC9C,MAAM,EAAE,6BAAkB,CAAC,mBAAmB;IAC9C,MAAM,EAAE,6BAAkB,CAAC,mBAAmB;IAC9C,MAAM,EAAE,6BAAkB,CAAC,mBAAmB;IAC9C,IAAI,EAAE,6BAAkB,CAAC,iBAAiB;IAC1C,MAAM,EAAE,6BAAkB,CAAC,mBAAmB;IAC9C,KAAK,EAAE,6BAAkB,CAAC,kBAAkB;IAC5C,IAAI,EAAE,6BAAkB,CAAC,iBAAiB;IAC1C,QAAQ,EAAE,6BAAkB,CAAC,qBAAqB;IAClD,MAAM,EAAE,6BAAkB,CAAC,mBAAmB;IAC9C,QAAQ,EAAE,6BAAkB,CAAC,qBAAqB;CACnD,CAAC;AAEF,kEAAkE;AAClE,MAAM,cAAc,GAAa;IAC/B,kCAAkC;IAClC,2BAA2B;IAC3B,kCAAkC;IAClC,yBAAyB;IACzB,0CAA0C;IAC1C,sEAAsE;IACtE,kCAAkC;IAClC,8DAA8D;IAC9D,yBAAyB;CAC1B,CAAC;AAOF,SAAS,eAAe,CACtB,WAAmB,EACnB,QAAwC,EACxC,eAAuB;IAEvB,IAAI,gBAAe,CAAC,cAAc,EAAE,CAAC;QACnC,OAAO;IACT,CAAC;IAED,MAAM,eAAe,GAAG,gBAAe,CAAC,eAAe,IAAI,EAAE,CAAC;IAC9D,MAAM,WAAW,GAAG,gBAAe,CAAC,WAAW,IAAI,EAAE,CAAC;IAEtD,MAAM,UAAU,GAAe;QAC7B,CAAC,8CAAuB,CAAC,EAAE,mBAAQ;QACnC,CAAC,wCAAiB,CAAC,EAAE,eAAe;QACpC,CAAC,yDAAkC,CAAC,EAAE,WAAW;QACjD,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,SAAS,CAAC,QAAQ;QACvD,CAAC,6BAAkB,CAAC,iBAAiB,CAAC,EAAE,WAAW;QACnD,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,QAAQ,CAAC,IAAI;QAClD,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,QAAQ,CAAC,IAAI;KAChD,CAAC;IAEF,iBAAO,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IACvC,iBAAO,CAAC,yBAAyB,EAAE,MAAM,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;AACzE,CAAC;AAED,MAAM,SAAU,SAAQ,sBAAW;IAGjC,8FAA8F;IAC9F,MAAM,CAAC,iBAAiB,CAAC,KAAc;QACrC,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YAC1C,OAAO,6BAAkB,CAAC,kBAAkB,CAAC;QAC/C,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;QACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO,6BAAkB,CAAC,kBAAkB,CAAC;QAC/C,CAAC;QACD,MAAM,UAAU,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC;QAC1C,KAAK,MAAM,CAAC,OAAO,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACrE,IAAI,UAAU,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;gBACnC,OAAO,SAAS,CAAC;YACnB,CAAC;QACH,CAAC;QACD,sDAAsD;QACtD,IAAI,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAClC,KAAK,MAAM,CAAC,OAAO,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;gBACrE,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;oBACjC,OAAO,SAAS,CAAC;gBACnB,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,6BAAkB,CAAC,kBAAkB,CAAC;IAC/C,CAAC;IAED,yFAAyF;IACzF,MAAM,CAAC,gBAAgB,CAAC,KAAc;QACpC,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YAC1C,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,KAAK,MAAM,OAAO,IAAI,cAAc,EAAE,CAAC;YACrC,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACrC,IAAI,KAAK,EAAE,CAAC;gBACV,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC;YAClB,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,kGAAkG;IAClG,MAAM,CAAC,sBAAsB,CAAC,KAAc;QAC1C,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YAC1C,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,OAAO,QAAQ,CAAC;QAC9C,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAC1C,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,OAAO,eAAe,CAAC;QACrD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,iFAAiF;IACjF,MAAM,CAAC,oBAAoB,CAAC,KAAc;QACxC,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YAC1C,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAC/C,OAAO,CACL,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC;YAC/B,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC;YAC9B,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC;YAC9B,UAAU,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAC5C,CAAC;IACJ,CAAC;IAED,0FAA0F;IAC1F,MAAM,CAAC,gBAAgB,CAAC,MAAe;QACrC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClD,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7B,OAAO,MAAM,CAAC,MAAM,CAAC;QACvB,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,iFAAiF;IACjF,MAAM,CAAC,eAAe,CAAC,WAAmB,EAAE,SAAiB,EAAE,KAAc;QAC3E,IAAI,OAAO,GAAG,GAAG,WAAW,IAAI,SAAS,EAAE,CAAC;QAC5C,MAAM,MAAM,GAAG,SAAS,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;QACvD,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,IAAI,YAAY,MAAM,GAAG,CAAC;QACnC,CAAC;QACD,IAAI,SAAS,CAAC,oBAAoB,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1C,OAAO,IAAI,cAAc,CAAC;QAC5B,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,mGAAmG;IACnG,MAAM,CAAC,iBAAiB,CAAC,KAAc;QACrC,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YAC1C,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,IAAI,KAAK,YAAY,UAAU,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1D,OAAO,UAAW,KAAoB,CAAC,MAAM,GAAG,CAAC;QACnD,CAAC;QACD,IAAI,QAAgB,CAAC;QACrB,IAAI,CAAC;YACH,QAAQ,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/E,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,mBAAmB,CAAC;QAC7B,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,GAAG,gBAAgB,EAAE,CAAC;YACvC,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,gBAAgB,CAAC,iBAAiB,QAAQ,CAAC,MAAM,GAAG,CAAC;QACnF,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;;OAIG;IACH,MAAM,CAAC,qBAAqB,CAAC,QAAa;QACxC,MAAM,QAAQ,GAAG,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QACnD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,QAAQ,EAAE,oBAAoB,IAAI,QAAQ,EAAE,OAAO,IAAI,QAAQ,IAAI,EAAE,CAAC;YACrF,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,QAAQ,CAAC,IAAI,CAAC;YAC1C,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;YAC/D,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;YACjC,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,EAAE,CAAC;QAChF,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,QAAQ,CAAC;QAClB,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,YAAY,CAAC,IAAW;QAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACtB,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;QACpF,CAAC;QACD,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACvC,wFAAwF;YACxF,MAAM,SAAS,GAAG,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;YAC3D,MAAM,MAAM,GAAG,QAAQ,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;YAC5D,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;QAC/B,CAAC;QACD,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;IACrD,CAAC;IAED,6DAA6D;IAC7D,MAAM,CAAC,eAAe,CAAC,IAAW;QAChC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACnC,OAAO,OAAO,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;IACvD,CAAC;IAED,uEAAuE;IACvE,MAAM,CAAC,iBAAiB,CACtB,IAAU,EACV,EACE,WAAW,EACX,SAAS,EACT,SAAS,EACT,MAAM,EACN,QAAQ,EACR,eAAe,EACf,QAAQ,EACR,SAAS,EACT,mBAAmB,EACnB,UAAU,GAYX;QAED,MAAM,eAAe,GAAG,gBAAe,CAAC,eAAe,IAAI,EAAE,CAAC;QAC9D,MAAM,WAAW,GAAG,gBAAe,CAAC,WAAW,IAAI,EAAE,CAAC;QAEtD,IAAI,CAAC,YAAY,CAAC,8CAAuB,EAAE,mBAAQ,CAAC,CAAC;QACrD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,6BAAkB,CAAC,8BAA8B,CAAC,CAAC;QAC1G,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,SAAS,CAAC,QAAQ,CAAC,CAAC;QACzE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,WAAW,CAAC,CAAC;QACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;QACpE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;QACjE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,2BAA2B,EAAE,WAAW,CAAC,CAAC;QAC/E,IAAI,CAAC,YAAY,CAAC,wCAAiB,EAAE,eAAe,CAAC,CAAC;QACtD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,4BAA4B,EAAE,eAAe,CAAC,CAAC;QACpF,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;QACnE,CAAC;QAED,IAAI,QAAQ,CAAC,QAAQ,IAAI,QAAQ,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YACzD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,uBAAuB,EAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACnF,CAAC;QACD,IAAI,SAAS,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YACzC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;QACtE,CAAC;QACD,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YACnD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;QACjE,CAAC;QAED,iFAAiF;QACjF,IAAI,gBAAe,CAAC,qBAAqB,IAAI,SAAS,EAAE,CAAC;YACvD,IAAI,QAAQ,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC;YACjC,MAAM,MAAM,GAAG,gBAAe,CAAC,gBAAgB,CAAC;YAChD,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,GAAG,CAAC,IAAI,QAAQ,CAAC,MAAM,GAAG,MAAM,EAAE,CAAC;gBACzE,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;YACvC,CAAC;YACD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;QAChE,CAAC;QAED,qFAAqF;QACrF,MAAM,gBAAgB,GACpB,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,GAAG,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;YACrE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;YACX,CAAC,CAAC,MAAM,CAAC;QAEb,IAAI,mBAAmB,IAAI,gBAAgB,KAAK,IAAI,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;YACvF,IAAI,KAAK,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBACpC,gBAAgB,CAAC,KAAK,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;oBACjE,IAAI,CAAC,YAAY,CACf,GAAG,6BAAkB,CAAC,kBAAkB,IAAI,GAAG,EAAE,EACjD,SAAS,CAAC,iBAAiB,CAAC,KAAK,CAAC,CACnC,CAAC;gBACJ,CAAC,CAAC,CAAC;YACL,CAAC;iBAAM,IAAI,OAAO,gBAAgB,KAAK,QAAQ,EAAE,CAAC;gBAChD,MAAM,CAAC,OAAO,CAAC,gBAA2C,CAAC;qBACxD,KAAK,CAAC,CAAC,EAAE,gBAAgB,CAAC;qBAC1B,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;oBACxB,IAAI,CAAC,YAAY,CACf,GAAG,6BAAkB,CAAC,kBAAkB,IAAI,GAAG,EAAE,EACjD,SAAS,CAAC,iBAAiB,CAAC,KAAK,CAAC,CACnC,CAAC;gBACJ,CAAC,CAAC,CAAC;YACP,CAAC;QACH,CAAC;QAED,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,gBAAgB,EACnC,SAAS,CAAC,eAAe,CAAC,WAAW,EAAE,SAAS,EAAE,SAAS,CAAC,CAC7D,CAAC;QAEF,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,QAAQ,IAAI,CAAC,EAAE,CAAC;YAClD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,QAAQ,CAAC,CAAC;QAC5E,CAAC;QAED,8BAA8B;QAC9B,MAAM,MAAM,GAAG,SAAS,CAAC,sBAAsB,CAAC,SAAS,CAAC,CAAC;QAC3D,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,2BAA2B,EAAE,MAAM,CAAC,CAAC;QAC5E,CAAC;QAED,IAAA,mCAAyB,EAAC,IAAI,CAAC,CAAC;QAChC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,oBAAc,CAAC,EAAE,EAAE,CAAC,CAAC;QAC5C,eAAe,CAAC,WAAW,EAAE,QAAQ,EAAE,eAAe,CAAC,CAAC;IAC1D,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,WAAW,CAAC,MAAc,EAAE,SAA0B,EAAE;QAC7D,MAAM,mBAAmB,GAAG,OAAO,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;QAChE,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;QAErC,OAAO,CAAC,aAAsC,EAAE,EAAE;YAChD,OAAO,UAAqB,GAAG,IAAW;gBACxC,IAAI,IAAA,0BAAmB,EAAC,aAAO,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;oBAC1C,OAAO,aAAa,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBACzC,CAAC;gBAED,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,SAAS,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;gBAC3D,MAAM,WAAW,GAAG,SAAS,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;gBAC3D,MAAM,SAAS,GAAG,SAAS,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC;gBACxD,MAAM,SAAS,GAAG,SAAS,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;gBACrD,MAAM,QAAQ,GAAG,SAAS,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;gBACvD,MAAM,QAAQ,GACZ,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,GAAG,CAAC;oBAC5C,CAAC,CAAC,GAAG,WAAW,IAAI,SAAS,UAAU;oBACvC,CAAC,CAAC,GAAG,WAAW,IAAI,SAAS,EAAE,CAAC;gBACpC,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;gBACnE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAE7B,MAAM,QAAQ,GAAG,CAAC,MAAW,EAAE,EAAE;oBAC/B,IAAI,CAAC;wBACH,MAAM,eAAe,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;wBACxD,MAAM,QAAQ,GACZ,MAAM,IAAI,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;wBAC9E,SAAS,CAAC,iBAAiB,CAAC,IAAI,EAAE;4BAChC,WAAW;4BACX,SAAS;4BACT,SAAS;4BACT,MAAM;4BACN,QAAQ;4BACR,eAAe;4BACf,QAAQ;4BACR,SAAS;4BACT,mBAAmB;4BACnB,UAAU;yBACX,CAAC,CAAC;oBACL,CAAC;oBAAC,OAAO,OAAY,EAAE,CAAC;wBACtB,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;oBACjD,CAAC;4BAAS,CAAC;wBACT,IAAI,CAAC,GAAG,EAAE,CAAC;oBACb,CAAC;gBACH,CAAC,CAAC;gBAEF,MAAM,IAAI,GAAG,CAAC,KAAU,EAAE,EAAE;oBAC1B,iBAAe,CAAC,eAAe,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;oBAC7C,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,CAAC,CAAC;gBAEF,OAAO,aAAO,CAAC,IAAI,CAAC,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,EAAE,GAAG,EAAE;oBAC9D,MAAM,QAAQ,GAAG,SAAS,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;oBAEjD,oEAAoE;oBACpE,IAAI,QAAQ,EAAE,CAAC;wBACb,MAAM,eAAe,GAAG,UAAqB,GAAQ,EAAE,GAAQ;4BAC7D,IAAI,GAAG,EAAE,CAAC;gCACR,IAAI,CAAC,GAAG,CAAC,CAAC;4BACZ,CAAC;iCAAM,CAAC;gCACN,QAAQ,CAAC,GAAG,CAAC,CAAC;4BAChB,CAAC;4BACD,OAAO,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;wBAC1C,CAAC,CAAC;wBACF,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;wBAC1D,IAAI,CAAC;4BACH,OAAO,aAAa,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;wBAC5C,CAAC;wBAAC,OAAO,OAAY,EAAE,CAAC;4BACtB,IAAI,CAAC,OAAO,CAAC,CAAC;4BACd,MAAM,OAAO,CAAC;wBAChB,CAAC;oBACH,CAAC;oBAED,oEAAoE;oBACpE,IAAI,MAAW,CAAC;oBAChB,IAAI,CAAC;wBACH,MAAM,GAAG,aAAa,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;oBAC3C,CAAC;oBAAC,OAAO,OAAY,EAAE,CAAC;wBACtB,IAAI,CAAC,OAAO,CAAC,CAAC;wBACd,MAAM,OAAO,CAAC;oBAChB,CAAC;oBAED,IAAI,MAAM,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBAChD,OAAO,MAAM,CAAC,IAAI,CAChB,CAAC,GAAQ,EAAE,EAAE;4BACX,QAAQ,CAAC,GAAG,CAAC,CAAC;4BACd,OAAO,GAAG,CAAC;wBACb,CAAC,EACD,CAAC,GAAQ,EAAE,EAAE;4BACX,IAAI,CAAC,GAAG,CAAC,CAAC;4BACV,MAAM,GAAG,CAAC;wBACZ,CAAC,CACF,CAAC;oBACJ,CAAC;oBAED,+EAA+E;oBAC/E,IAAI,MAAM,IAAI,OAAO,MAAM,CAAC,EAAE,KAAK,UAAU,EAAE,CAAC;wBAC9C,IAAI,OAAO,GAAG,KAAK,CAAC;wBACpB,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;4BACpB,IAAI,OAAO;gCAAE,OAAO;4BACpB,OAAO,GAAG,IAAI,CAAC;4BACf,QAAQ,CAAC,MAAM,CAAC,CAAC;wBACnB,CAAC,CAAC,CAAC;wBACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAQ,EAAE,EAAE;4BAC9B,IAAI,OAAO;gCAAE,OAAO;4BACpB,OAAO,GAAG,IAAI,CAAC;4BACf,IAAI,CAAC,GAAG,CAAC,CAAC;wBACZ,CAAC,CAAC,CAAC;wBACH,OAAO,MAAM,CAAC;oBAChB,CAAC;oBAED,sDAAsD;oBACtD,QAAQ,CAAC,MAAM,CAAC,CAAC;oBACjB,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;;AAjYM,kBAAQ,GAAG,6BAAkB,CAAC,oBAAoB,CAAC;AAoY5D,kBAAe,SAAS,CAAC"}

package/dist/instrumentation/pinecone/index.d.ts

@@ -0,0 +1,11 @@
+import { InstrumentationBase, InstrumentationModuleDefinition } from '@opentelemetry/instrumentation';
+import { InstrumentationConfig } from '@opentelemetry/instrumentation';
+export interface PineconeInstrumentationConfig extends InstrumentationConfig {
+}
+export default class Shield360PineconeInstrumentation extends InstrumentationBase {
+    constructor(config?: PineconeInstrumentationConfig);
+    protected init(): void | InstrumentationModuleDefinition | InstrumentationModuleDefinition[];
+    manualPatch(pinecone: any): void;
+    protected _patch(moduleExports: any): void;
+    protected _unpatch(moduleExports: any): void;
+}