Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

// Nothing is listening on this port, so `connect()` fails and the connect span
// must be recorded with an errored status.
const client = new Client({ port: 5499, user: 'test', password: 'test', database: 'tests' });

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
await client.connect().catch(() => {
// expected: connection refused
});
},
);
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
() =>
// Chain off `connect()` with `.then()` instead of awaiting it: the query
// issued from the continuation must still be parented to the active
// transaction, proving the trace context survives the connect promise.
new Promise((resolve, reject) => {
const client = new Client({ port: 5494, user: 'test', password: 'test', database: 'tests' });
client
.connect()
.then(() => client.query('SELECT 1 AS connect_then'))
.then(() => client.end())
.then(resolve, reject);
}),
);
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

const client = new Client({ port: 5494, user: 'test', password: 'test', database: 'tests' });

async function run() {
// No active span here: `requireParentSpan` means the connect and this query
// must NOT produce spans.
await client.connect();
await client.query('SELECT 1 AS unparented');

// With an active span, the query is instrumented as a child span.
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
await client.query('SELECT 2 AS parented');
},
);

await client.end();
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
import * as Sentry from '@sentry/node';
import pg from 'pg';

const { Pool } = pg;

// The connection string carries credentials so we can assert they are masked
// out of the `db.connection_string` span attribute.
const pool = new Pool({ connectionString: 'postgresql://test:test@localhost:5494/tests' });

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
try {
// Callback-style query exercises the callback-patching path (the
// promise-based scenarios never hit it).
await new Promise((resolve, reject) => {
pool.query('SELECT 1 AS foo', (err, res) => (err ? reject(err) : resolve(res)));
});
} finally {
await pool.end();
}
},
);
}

run();
Original file line numberDiff line numberDiff line change
Expand Up@@ -23,6 +23,18 @@ async function run() {

await client.query('INSERT INTO "User" ("email", "name") VALUES ($1, $2)', ['tim', 'tim@domain.com']);
await client.query('SELECT * FROM "User"');

// A named (prepared) query records its name as the `db.postgresql.plan` attribute
await client.query({
name: 'select-user-by-email',
text: 'SELECT * FROM "User" WHERE "email" = $1',
values: ['tim'],
});

// A failing query should still produce an errored span
await client.query('SELECT * FROM "does_not_exist_table"').catch(() => {
// swallow: we only care about the span it produces
});
} finally {
await client.end();
}
Expand Down
182 changes: 182 additions & 0 deletions dev-packages/node-integration-tests/suites/tracing/postgres/test.ts
Original file line numberDiff line numberDiff line change
Expand Up@@ -48,6 +48,33 @@ describe('postgres auto instrumentation', () => {
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT * FROM "User" WHERE "email" = $1',
'db.postgresql.plan': 'select-user-by-email',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT * FROM "User" WHERE "email" = $1',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT * FROM "does_not_exist_table"',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT * FROM "does_not_exist_table"',
op: 'db',
status: 'internal_error',
origin: 'auto.db.otel.postgres',
}),
]),
};

Expand DownExpand Up@@ -114,6 +141,161 @@ describe('postgres auto instrumentation', () => {
});
});

describe('pool', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
// Pool connect span: no origin is set on connect spans, so it defaults
// to 'manual', and the connection-string credentials are masked out.
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.connection_string': 'postgresql://localhost:5494/tests',
'sentry.op': 'db',
}),
description: 'pg-pool.connect',
op: 'db',
status: 'ok',
origin: 'manual',
}),
// Callback-style query (no awaited promise returned to the caller).
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 1 AS foo',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 1 AS foo',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
};

createEsmAndCjsTests(__dirname, 'scenario-pool.mjs', 'instrument.mjs', (createTestRunner, test) => {
test(
'auto-instruments `pg.Pool`, masks connection-string credentials, and handles callback-style queries',
{ timeout: 90_000 },
async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({ transaction: EXPECTED_TRANSACTION })
.start()
.completed();
},
);
});
});

describe('connect error', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'sentry.op': 'db',
}),
description: 'pg.connect',
op: 'db',
status: 'internal_error',
origin: 'manual',
}),
]),
};

// No DB needed: the scenario connects to a port where nothing is listening.
createEsmAndCjsTests(__dirname, 'scenario-connect-error.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('records an errored connect span when the connection fails', { timeout: 90_000 }, async () => {
await createTestRunner().expect({ transaction: EXPECTED_TRANSACTION }).start().completed();
});
});
});

// A query chained off `connect()` with `.then()` (rather than awaited) must still
// be parented to the active transaction. Since the instrumentation requires a
// parent span, the query only produces a span if the trace context survives the
// connect promise's continuation.
describe('connect promise continuation', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 1 AS connect_then',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 1 AS connect_then',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
};

createEsmAndCjsTests(__dirname, 'scenario-connect-then.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('parents a query chained off connect() to the active transaction', { timeout: 90_000 }, async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({ transaction: EXPECTED_TRANSACTION })
.start()
.completed();
});
});
});

describe('requireParentSpan', () => {
createEsmAndCjsTests(__dirname, 'scenario-no-parent.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('does not instrument queries or connects without an active parent span', { timeout: 90_000 }, async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({
transaction: txn => {
const descriptions = txn.spans?.map(span => span.description) ?? [];
// The unparented connect + query must not have produced spans
expect(descriptions).not.toContain('SELECT 1 AS unparented');
expect(descriptions.find(name => name?.includes('connect'))).toBeUndefined();
// Only the parented query is instrumented
expect(txn).toMatchObject({
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 2 AS parented',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 2 AS parented',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
});
},
})
.start()
.completed();
});
});
});

conditionalTest({ max: 25 })('pg-native', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
Expand Down
6 changes: 1 addition & 5 deletions packages/node/src/integrations/tracing/postgres/index.ts
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
import { PgInstrumentation } from './vendored/instrumentation';
import type { IntegrationFn } from '@sentry/core';
import { defineIntegration } from '@sentry/core';
import { addOriginToSpan, generateInstrumentOnce } from '@sentry/node-core';
import { generateInstrumentOnce } from '@sentry/node-core';

interface PostgresIntegrationOptions {
ignoreConnectSpans?: boolean;
Expand All@@ -13,10 +13,6 @@ export const instrumentPostgres = generateInstrumentOnce(
INTEGRATION_NAME,
PgInstrumentation,
(options?: PostgresIntegrationOptions) => ({
requireParentSpan: true,
requestHook(span) {
addOriginToSpan(span, 'auto.db.otel.postgres');
},
ignoreConnectSpans: options?.ignoreConnectSpans ?? false,
}),
);
Expand Down
Original file line numberDiff line numberDiff line change
Expand Up@@ -6,11 +6,9 @@
* - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-pg
* - Upstream version: @opentelemetry/instrumentation-pg@0.70.0
*/
/* eslint-disable */

// Postgresql specific attributes not covered by semantic conventions
export enum AttributeNames {
PG_VALUES = 'db.postgresql.values',
PG_PLAN = 'db.postgresql.plan',
IDLE_TIMEOUT_MILLIS = 'db.postgresql.idle.timeout.millis',
MAX_CLIENT = 'db.postgresql.max.client',
Expand Down
Original file line numberDiff line numberDiff line change
Expand Up@@ -6,7 +6,6 @@
* - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-pg
* - Upstream version: @opentelemetry/instrumentation-pg@0.70.0
*/
/* eslint-disable */

// Contains span names produced by instrumentation
export enum SpanNames {
Expand Down
Loading
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Add copy buttons to all
 blocks\n(function() {\n function addCopyButtons() {\n document.querySelectorAll('pre code').forEach(function(codeBlock) {\n if (codeBlock.parentElement.hasAttribute('data-copy-added')) return;\n codeBlock.parentElement.setAttribute('data-copy-added', 'true');\n \n var btn = document.createElement('button');\n btn.textContent = 'Copy';\n btn.style.cssText = 'position:absolute;top:4px;right:4px;padding:2px 8px;font-size:11px;background:#4ecdc4;border:none;border-radius:4px;color:#1a1a2e;cursor:pointer;opacity:0.7;transition:opacity 0.2s;';\n btn.onmouseover = function() { this.style.opacity = '1'; };\n btn.onmouseout = function() { this.style.opacity = '0.7'; };\n btn.onclick = function() {\n navigator.clipboard.writeText(codeBlock.textContent).then(function() {\n btn.textContent = 'Copied!';\n setTimeout(function() { btn.textContent = 'Copy'; }, 1500);\n });\n };\n codeBlock.parentElement.style.position = 'relative';\n codeBlock.parentElement.appendChild(btn);\n });\n }\n \n addCopyButtons();\n \n // Re-run on dynamic content\n var observer = new MutationObserver(addCopyButtons);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Add Copy Buttons to Code Blocks");
}
} catch(__e) { console.warn('[Userscript:Add Copy Buttons to Code Blocks]', __e); }
})();
(function(){
try {
var __m = "github.com";
var __re = new RegExp('^' + "github\\.com" + '
Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

// Nothing is listening on this port, so `connect()` fails and the connect span
// must be recorded with an errored status.
const client = new Client({ port: 5499, user: 'test', password: 'test', database: 'tests' });

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
await client.connect().catch(() => {
// expected: connection refused
});
},
);
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
() =>
// Chain off `connect()` with `.then()` instead of awaiting it: the query
// issued from the continuation must still be parented to the active
// transaction, proving the trace context survives the connect promise.
new Promise((resolve, reject) => {
const client = new Client({ port: 5494, user: 'test', password: 'test', database: 'tests' });
client
.connect()
.then(() => client.query('SELECT 1 AS connect_then'))
.then(() => client.end())
.then(resolve, reject);
}),
);
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

const client = new Client({ port: 5494, user: 'test', password: 'test', database: 'tests' });

async function run() {
// No active span here: `requireParentSpan` means the connect and this query
// must NOT produce spans.
await client.connect();
await client.query('SELECT 1 AS unparented');

// With an active span, the query is instrumented as a child span.
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
await client.query('SELECT 2 AS parented');
},
);

await client.end();
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
import * as Sentry from '@sentry/node';
import pg from 'pg';

const { Pool } = pg;

// The connection string carries credentials so we can assert they are masked
// out of the `db.connection_string` span attribute.
const pool = new Pool({ connectionString: 'postgresql://test:test@localhost:5494/tests' });

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
try {
// Callback-style query exercises the callback-patching path (the
// promise-based scenarios never hit it).
await new Promise((resolve, reject) => {
pool.query('SELECT 1 AS foo', (err, res) => (err ? reject(err) : resolve(res)));
});
} finally {
await pool.end();
}
},
);
}

run();
Original file line numberDiff line numberDiff line change
Expand Up@@ -23,6 +23,18 @@ async function run() {

await client.query('INSERT INTO "User" ("email", "name") VALUES ($1, $2)', ['tim', 'tim@domain.com']);
await client.query('SELECT * FROM "User"');

// A named (prepared) query records its name as the `db.postgresql.plan` attribute
await client.query({
name: 'select-user-by-email',
text: 'SELECT * FROM "User" WHERE "email" = $1',
values: ['tim'],
});

// A failing query should still produce an errored span
await client.query('SELECT * FROM "does_not_exist_table"').catch(() => {
// swallow: we only care about the span it produces
});
} finally {
await client.end();
}
Expand Down
182 changes: 182 additions & 0 deletions dev-packages/node-integration-tests/suites/tracing/postgres/test.ts
Original file line numberDiff line numberDiff line change
Expand Up@@ -48,6 +48,33 @@ describe('postgres auto instrumentation', () => {
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT * FROM "User" WHERE "email" = $1',
'db.postgresql.plan': 'select-user-by-email',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT * FROM "User" WHERE "email" = $1',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT * FROM "does_not_exist_table"',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT * FROM "does_not_exist_table"',
op: 'db',
status: 'internal_error',
origin: 'auto.db.otel.postgres',
}),
]),
};

Expand DownExpand Up@@ -114,6 +141,161 @@ describe('postgres auto instrumentation', () => {
});
});

describe('pool', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
// Pool connect span: no origin is set on connect spans, so it defaults
// to 'manual', and the connection-string credentials are masked out.
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.connection_string': 'postgresql://localhost:5494/tests',
'sentry.op': 'db',
}),
description: 'pg-pool.connect',
op: 'db',
status: 'ok',
origin: 'manual',
}),
// Callback-style query (no awaited promise returned to the caller).
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 1 AS foo',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 1 AS foo',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
};

createEsmAndCjsTests(__dirname, 'scenario-pool.mjs', 'instrument.mjs', (createTestRunner, test) => {
test(
'auto-instruments `pg.Pool`, masks connection-string credentials, and handles callback-style queries',
{ timeout: 90_000 },
async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({ transaction: EXPECTED_TRANSACTION })
.start()
.completed();
},
);
});
});

describe('connect error', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'sentry.op': 'db',
}),
description: 'pg.connect',
op: 'db',
status: 'internal_error',
origin: 'manual',
}),
]),
};

// No DB needed: the scenario connects to a port where nothing is listening.
createEsmAndCjsTests(__dirname, 'scenario-connect-error.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('records an errored connect span when the connection fails', { timeout: 90_000 }, async () => {
await createTestRunner().expect({ transaction: EXPECTED_TRANSACTION }).start().completed();
});
});
});

// A query chained off `connect()` with `.then()` (rather than awaited) must still
// be parented to the active transaction. Since the instrumentation requires a
// parent span, the query only produces a span if the trace context survives the
// connect promise's continuation.
describe('connect promise continuation', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 1 AS connect_then',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 1 AS connect_then',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
};

createEsmAndCjsTests(__dirname, 'scenario-connect-then.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('parents a query chained off connect() to the active transaction', { timeout: 90_000 }, async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({ transaction: EXPECTED_TRANSACTION })
.start()
.completed();
});
});
});

describe('requireParentSpan', () => {
createEsmAndCjsTests(__dirname, 'scenario-no-parent.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('does not instrument queries or connects without an active parent span', { timeout: 90_000 }, async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({
transaction: txn => {
const descriptions = txn.spans?.map(span => span.description) ?? [];
// The unparented connect + query must not have produced spans
expect(descriptions).not.toContain('SELECT 1 AS unparented');
expect(descriptions.find(name => name?.includes('connect'))).toBeUndefined();
// Only the parented query is instrumented
expect(txn).toMatchObject({
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 2 AS parented',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 2 AS parented',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
});
},
})
.start()
.completed();
});
});
});

conditionalTest({ max: 25 })('pg-native', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
Expand Down
6 changes: 1 addition & 5 deletions packages/node/src/integrations/tracing/postgres/index.ts
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
import { PgInstrumentation } from './vendored/instrumentation';
import type { IntegrationFn } from '@sentry/core';
import { defineIntegration } from '@sentry/core';
import { addOriginToSpan, generateInstrumentOnce } from '@sentry/node-core';
import { generateInstrumentOnce } from '@sentry/node-core';

interface PostgresIntegrationOptions {
ignoreConnectSpans?: boolean;
Expand All@@ -13,10 +13,6 @@ export const instrumentPostgres = generateInstrumentOnce(
INTEGRATION_NAME,
PgInstrumentation,
(options?: PostgresIntegrationOptions) => ({
requireParentSpan: true,
requestHook(span) {
addOriginToSpan(span, 'auto.db.otel.postgres');
},
ignoreConnectSpans: options?.ignoreConnectSpans ?? false,
}),
);
Expand Down
Original file line numberDiff line numberDiff line change
Expand Up@@ -6,11 +6,9 @@
* - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-pg
* - Upstream version: @opentelemetry/instrumentation-pg@0.70.0
*/
/* eslint-disable */

// Postgresql specific attributes not covered by semantic conventions
export enum AttributeNames {
PG_VALUES = 'db.postgresql.values',
PG_PLAN = 'db.postgresql.plan',
IDLE_TIMEOUT_MILLIS = 'db.postgresql.idle.timeout.millis',
MAX_CLIENT = 'db.postgresql.max.client',
Expand Down
Original file line numberDiff line numberDiff line change
Expand Up@@ -6,7 +6,6 @@
* - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-pg
* - Upstream version: @opentelemetry/instrumentation-pg@0.70.0
*/
/* eslint-disable */

// Contains span names produced by instrumentation
export enum SpanNames {
Expand Down
Loading
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Force GitHub README to respect dark mode\n(function() {\n var style = document.createElement('style');\n style.textContent = '\n .markdown-body {\n color-scheme: dark light;\n }\n .markdown-body pre { background: #161b22 !important; }\n .markdown-body code { background: rgba(110, 118, 129, 0.4) !important; }\n .markdown-body table th, .markdown-body table td { border-color: #30363d !important; }\n .markdown-body img { background: #0d1117; }\n .markdown-body blockquote { border-left-color: #8b949e; }\n .markdown-body hr { border-color: #30363d; }\n ';\n document.head.appendChild(style);\n})();", "GitHub Dark Mode README Fix"); } } catch(__e) { console.warn('[Userscript:GitHub Dark Mode README Fix]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

// Nothing is listening on this port, so `connect()` fails and the connect span
// must be recorded with an errored status.
const client = new Client({ port: 5499, user: 'test', password: 'test', database: 'tests' });

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
await client.connect().catch(() => {
// expected: connection refused
});
},
);
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
() =>
// Chain off `connect()` with `.then()` instead of awaiting it: the query
// issued from the continuation must still be parented to the active
// transaction, proving the trace context survives the connect promise.
new Promise((resolve, reject) => {
const client = new Client({ port: 5494, user: 'test', password: 'test', database: 'tests' });
client
.connect()
.then(() => client.query('SELECT 1 AS connect_then'))
.then(() => client.end())
.then(resolve, reject);
}),
);
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

const client = new Client({ port: 5494, user: 'test', password: 'test', database: 'tests' });

async function run() {
// No active span here: `requireParentSpan` means the connect and this query
// must NOT produce spans.
await client.connect();
await client.query('SELECT 1 AS unparented');

// With an active span, the query is instrumented as a child span.
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
await client.query('SELECT 2 AS parented');
},
);

await client.end();
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
import * as Sentry from '@sentry/node';
import pg from 'pg';

const { Pool } = pg;

// The connection string carries credentials so we can assert they are masked
// out of the `db.connection_string` span attribute.
const pool = new Pool({ connectionString: 'postgresql://test:test@localhost:5494/tests' });

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
try {
// Callback-style query exercises the callback-patching path (the
// promise-based scenarios never hit it).
await new Promise((resolve, reject) => {
pool.query('SELECT 1 AS foo', (err, res) => (err ? reject(err) : resolve(res)));
});
} finally {
await pool.end();
}
},
);
}

run();
Original file line numberDiff line numberDiff line change
Expand Up@@ -23,6 +23,18 @@ async function run() {

await client.query('INSERT INTO "User" ("email", "name") VALUES ($1, $2)', ['tim', 'tim@domain.com']);
await client.query('SELECT * FROM "User"');

// A named (prepared) query records its name as the `db.postgresql.plan` attribute
await client.query({
name: 'select-user-by-email',
text: 'SELECT * FROM "User" WHERE "email" = $1',
values: ['tim'],
});

// A failing query should still produce an errored span
await client.query('SELECT * FROM "does_not_exist_table"').catch(() => {
// swallow: we only care about the span it produces
});
} finally {
await client.end();
}
Expand Down
182 changes: 182 additions & 0 deletions dev-packages/node-integration-tests/suites/tracing/postgres/test.ts
Original file line numberDiff line numberDiff line change
Expand Up@@ -48,6 +48,33 @@ describe('postgres auto instrumentation', () => {
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT * FROM "User" WHERE "email" = $1',
'db.postgresql.plan': 'select-user-by-email',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT * FROM "User" WHERE "email" = $1',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT * FROM "does_not_exist_table"',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT * FROM "does_not_exist_table"',
op: 'db',
status: 'internal_error',
origin: 'auto.db.otel.postgres',
}),
]),
};

Expand DownExpand Up@@ -114,6 +141,161 @@ describe('postgres auto instrumentation', () => {
});
});

describe('pool', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
// Pool connect span: no origin is set on connect spans, so it defaults
// to 'manual', and the connection-string credentials are masked out.
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.connection_string': 'postgresql://localhost:5494/tests',
'sentry.op': 'db',
}),
description: 'pg-pool.connect',
op: 'db',
status: 'ok',
origin: 'manual',
}),
// Callback-style query (no awaited promise returned to the caller).
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 1 AS foo',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 1 AS foo',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
};

createEsmAndCjsTests(__dirname, 'scenario-pool.mjs', 'instrument.mjs', (createTestRunner, test) => {
test(
'auto-instruments `pg.Pool`, masks connection-string credentials, and handles callback-style queries',
{ timeout: 90_000 },
async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({ transaction: EXPECTED_TRANSACTION })
.start()
.completed();
},
);
});
});

describe('connect error', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'sentry.op': 'db',
}),
description: 'pg.connect',
op: 'db',
status: 'internal_error',
origin: 'manual',
}),
]),
};

// No DB needed: the scenario connects to a port where nothing is listening.
createEsmAndCjsTests(__dirname, 'scenario-connect-error.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('records an errored connect span when the connection fails', { timeout: 90_000 }, async () => {
await createTestRunner().expect({ transaction: EXPECTED_TRANSACTION }).start().completed();
});
});
});

// A query chained off `connect()` with `.then()` (rather than awaited) must still
// be parented to the active transaction. Since the instrumentation requires a
// parent span, the query only produces a span if the trace context survives the
// connect promise's continuation.
describe('connect promise continuation', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 1 AS connect_then',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 1 AS connect_then',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
};

createEsmAndCjsTests(__dirname, 'scenario-connect-then.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('parents a query chained off connect() to the active transaction', { timeout: 90_000 }, async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({ transaction: EXPECTED_TRANSACTION })
.start()
.completed();
});
});
});

describe('requireParentSpan', () => {
createEsmAndCjsTests(__dirname, 'scenario-no-parent.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('does not instrument queries or connects without an active parent span', { timeout: 90_000 }, async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({
transaction: txn => {
const descriptions = txn.spans?.map(span => span.description) ?? [];
// The unparented connect + query must not have produced spans
expect(descriptions).not.toContain('SELECT 1 AS unparented');
expect(descriptions.find(name => name?.includes('connect'))).toBeUndefined();
// Only the parented query is instrumented
expect(txn).toMatchObject({
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 2 AS parented',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 2 AS parented',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
});
},
})
.start()
.completed();
});
});
});

conditionalTest({ max: 25 })('pg-native', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
Expand Down
6 changes: 1 addition & 5 deletions packages/node/src/integrations/tracing/postgres/index.ts
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
import { PgInstrumentation } from './vendored/instrumentation';
import type { IntegrationFn } from '@sentry/core';
import { defineIntegration } from '@sentry/core';
import { addOriginToSpan, generateInstrumentOnce } from '@sentry/node-core';
import { generateInstrumentOnce } from '@sentry/node-core';

interface PostgresIntegrationOptions {
ignoreConnectSpans?: boolean;
Expand All@@ -13,10 +13,6 @@ export const instrumentPostgres = generateInstrumentOnce(
INTEGRATION_NAME,
PgInstrumentation,
(options?: PostgresIntegrationOptions) => ({
requireParentSpan: true,
requestHook(span) {
addOriginToSpan(span, 'auto.db.otel.postgres');
},
ignoreConnectSpans: options?.ignoreConnectSpans ?? false,
}),
);
Expand Down
Original file line numberDiff line numberDiff line change
Expand Up@@ -6,11 +6,9 @@
* - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-pg
* - Upstream version: @opentelemetry/instrumentation-pg@0.70.0
*/
/* eslint-disable */

// Postgresql specific attributes not covered by semantic conventions
export enum AttributeNames {
PG_VALUES = 'db.postgresql.values',
PG_PLAN = 'db.postgresql.plan',
IDLE_TIMEOUT_MILLIS = 'db.postgresql.idle.timeout.millis',
MAX_CLIENT = 'db.postgresql.max.client',
Expand Down
Original file line numberDiff line numberDiff line change
Expand Up@@ -6,7 +6,6 @@
* - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-pg
* - Upstream version: @opentelemetry/instrumentation-pg@0.70.0
*/
/* eslint-disable */

// Contains span names produced by instrumentation
export enum SpanNames {
Expand Down
Loading
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Highlight search terms from Google/DuckDuckGo/Bing referrer\n(function() {\n var ref = document.referrer;\n var terms = [];\n \n if (ref.includes('google.com') || ref.includes('duckduckgo.com') || ref.includes('bing.com')) {\n var url = new URL(ref);\n var q = url.searchParams.get('q') || url.searchParams.get('p');\n if (q) {\n terms = q.split(/\\s+/).filter(function(t) { return t.length > 2; });\n }\n }\n \n if (terms.length === 0) return;\n \n var style = document.createElement('style');\n style.textContent = '.userscript-highlight { background: #fbbf24; color: #1a1a2e; padding: 1px 3px; border-radius: 2px; }';\n document.head.appendChild(style);\n \n function highlight(node) {\n if (node.nodeType === 3) { // text node\n var text = node.textContent;\n var found = false;\n terms.forEach(function(term) {\n var regex = new RegExp('(' + term.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\') + ')', 'gi');\n if (regex.test(text)) {\n found = true;\n var frag = document.createDocumentFragment();\n var parts = text.split(regex);\n parts.forEach(function(part, i) {\n if (i % 2 === 0) {\n frag.appendChild(document.createTextNode(part));\n } else {\n var span = document.createElement('span');\n span.className = 'userscript-highlight';\n span.textContent = part;\n frag.appendChild(span);\n }\n });\n node.parentNode.replaceChild(frag, node);\n }\n });\n } else if (node.nodeType === 1 && node.childNodes) { // element\n var skipTags = ['SCRIPT', 'STYLE', 'NOSCRIPT', 'TEXTAREA', 'INPUT', 'SELECT'];\n if (!skipTags.includes(node.tagName)) {\n Array.from(node.childNodes).forEach(highlight);\n }\n }\n }\n \n highlight(document.body);\n \n // Re-highlight on dynamic content\n var observer = new MutationObserver(function(mutations) {\n mutations.forEach(function(m) {\n m.addedNodes.forEach(function(node) {\n if (node.nodeType === 1 || node.nodeType === 3) highlight(node);\n });\n });\n });\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Highlight Search Terms"); } } catch(__e) { console.warn('[Userscript:Highlight Search Terms]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

// Nothing is listening on this port, so `connect()` fails and the connect span
// must be recorded with an errored status.
const client = new Client({ port: 5499, user: 'test', password: 'test', database: 'tests' });

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
await client.connect().catch(() => {
// expected: connection refused
});
},
);
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
() =>
// Chain off `connect()` with `.then()` instead of awaiting it: the query
// issued from the continuation must still be parented to the active
// transaction, proving the trace context survives the connect promise.
new Promise((resolve, reject) => {
const client = new Client({ port: 5494, user: 'test', password: 'test', database: 'tests' });
client
.connect()
.then(() => client.query('SELECT 1 AS connect_then'))
.then(() => client.end())
.then(resolve, reject);
}),
);
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

const client = new Client({ port: 5494, user: 'test', password: 'test', database: 'tests' });

async function run() {
// No active span here: `requireParentSpan` means the connect and this query
// must NOT produce spans.
await client.connect();
await client.query('SELECT 1 AS unparented');

// With an active span, the query is instrumented as a child span.
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
await client.query('SELECT 2 AS parented');
},
);

await client.end();
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
import * as Sentry from '@sentry/node';
import pg from 'pg';

const { Pool } = pg;

// The connection string carries credentials so we can assert they are masked
// out of the `db.connection_string` span attribute.
const pool = new Pool({ connectionString: 'postgresql://test:test@localhost:5494/tests' });

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
try {
// Callback-style query exercises the callback-patching path (the
// promise-based scenarios never hit it).
await new Promise((resolve, reject) => {
pool.query('SELECT 1 AS foo', (err, res) => (err ? reject(err) : resolve(res)));
});
} finally {
await pool.end();
}
},
);
}

run();
Original file line numberDiff line numberDiff line change
Expand Up@@ -23,6 +23,18 @@ async function run() {

await client.query('INSERT INTO "User" ("email", "name") VALUES ($1, $2)', ['tim', 'tim@domain.com']);
await client.query('SELECT * FROM "User"');

// A named (prepared) query records its name as the `db.postgresql.plan` attribute
await client.query({
name: 'select-user-by-email',
text: 'SELECT * FROM "User" WHERE "email" = $1',
values: ['tim'],
});

// A failing query should still produce an errored span
await client.query('SELECT * FROM "does_not_exist_table"').catch(() => {
// swallow: we only care about the span it produces
});
} finally {
await client.end();
}
Expand Down
182 changes: 182 additions & 0 deletions dev-packages/node-integration-tests/suites/tracing/postgres/test.ts
Original file line numberDiff line numberDiff line change
Expand Up@@ -48,6 +48,33 @@ describe('postgres auto instrumentation', () => {
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT * FROM "User" WHERE "email" = $1',
'db.postgresql.plan': 'select-user-by-email',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT * FROM "User" WHERE "email" = $1',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT * FROM "does_not_exist_table"',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT * FROM "does_not_exist_table"',
op: 'db',
status: 'internal_error',
origin: 'auto.db.otel.postgres',
}),
]),
};

Expand DownExpand Up@@ -114,6 +141,161 @@ describe('postgres auto instrumentation', () => {
});
});

describe('pool', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
// Pool connect span: no origin is set on connect spans, so it defaults
// to 'manual', and the connection-string credentials are masked out.
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.connection_string': 'postgresql://localhost:5494/tests',
'sentry.op': 'db',
}),
description: 'pg-pool.connect',
op: 'db',
status: 'ok',
origin: 'manual',
}),
// Callback-style query (no awaited promise returned to the caller).
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 1 AS foo',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 1 AS foo',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
};

createEsmAndCjsTests(__dirname, 'scenario-pool.mjs', 'instrument.mjs', (createTestRunner, test) => {
test(
'auto-instruments `pg.Pool`, masks connection-string credentials, and handles callback-style queries',
{ timeout: 90_000 },
async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({ transaction: EXPECTED_TRANSACTION })
.start()
.completed();
},
);
});
});

describe('connect error', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'sentry.op': 'db',
}),
description: 'pg.connect',
op: 'db',
status: 'internal_error',
origin: 'manual',
}),
]),
};

// No DB needed: the scenario connects to a port where nothing is listening.
createEsmAndCjsTests(__dirname, 'scenario-connect-error.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('records an errored connect span when the connection fails', { timeout: 90_000 }, async () => {
await createTestRunner().expect({ transaction: EXPECTED_TRANSACTION }).start().completed();
});
});
});

// A query chained off `connect()` with `.then()` (rather than awaited) must still
// be parented to the active transaction. Since the instrumentation requires a
// parent span, the query only produces a span if the trace context survives the
// connect promise's continuation.
describe('connect promise continuation', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 1 AS connect_then',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 1 AS connect_then',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
};

createEsmAndCjsTests(__dirname, 'scenario-connect-then.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('parents a query chained off connect() to the active transaction', { timeout: 90_000 }, async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({ transaction: EXPECTED_TRANSACTION })
.start()
.completed();
});
});
});

describe('requireParentSpan', () => {
createEsmAndCjsTests(__dirname, 'scenario-no-parent.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('does not instrument queries or connects without an active parent span', { timeout: 90_000 }, async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({
transaction: txn => {
const descriptions = txn.spans?.map(span => span.description) ?? [];
// The unparented connect + query must not have produced spans
expect(descriptions).not.toContain('SELECT 1 AS unparented');
expect(descriptions.find(name => name?.includes('connect'))).toBeUndefined();
// Only the parented query is instrumented
expect(txn).toMatchObject({
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 2 AS parented',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 2 AS parented',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
});
},
})
.start()
.completed();
});
});
});

conditionalTest({ max: 25 })('pg-native', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
Expand Down
6 changes: 1 addition & 5 deletions packages/node/src/integrations/tracing/postgres/index.ts
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
import { PgInstrumentation } from './vendored/instrumentation';
import type { IntegrationFn } from '@sentry/core';
import { defineIntegration } from '@sentry/core';
import { addOriginToSpan, generateInstrumentOnce } from '@sentry/node-core';
import { generateInstrumentOnce } from '@sentry/node-core';

interface PostgresIntegrationOptions {
ignoreConnectSpans?: boolean;
Expand All@@ -13,10 +13,6 @@ export const instrumentPostgres = generateInstrumentOnce(
INTEGRATION_NAME,
PgInstrumentation,
(options?: PostgresIntegrationOptions) => ({
requireParentSpan: true,
requestHook(span) {
addOriginToSpan(span, 'auto.db.otel.postgres');
},
ignoreConnectSpans: options?.ignoreConnectSpans ?? false,
}),
);
Expand Down
Original file line numberDiff line numberDiff line change
Expand Up@@ -6,11 +6,9 @@
* - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-pg
* - Upstream version: @opentelemetry/instrumentation-pg@0.70.0
*/
/* eslint-disable */

// Postgresql specific attributes not covered by semantic conventions
export enum AttributeNames {
PG_VALUES = 'db.postgresql.values',
PG_PLAN = 'db.postgresql.plan',
IDLE_TIMEOUT_MILLIS = 'db.postgresql.idle.timeout.millis',
MAX_CLIENT = 'db.postgresql.max.client',
Expand Down
Original file line numberDiff line numberDiff line change
Expand Up@@ -6,7 +6,6 @@
* - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-pg
* - Upstream version: @opentelemetry/instrumentation-pg@0.70.0
*/
/* eslint-disable */

// Contains span names produced by instrumentation
export enum SpanNames {
Expand Down
Loading
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Strip utm_, fbclid, gclid, etc. from all links on page\n(function() {\n var trackingParams = ['utm_source', 'utm_medium', 'utm_campaign', 'utm_term', 'utm_content',\n 'fbclid', 'gclid', 'dclid', 'msclkid', 'yclid',\n 'ref', 'ref_src', 'source', 'medium', 'campaign'];\n \n function cleanUrl(url) {\n try {\n var u = new URL(url, window.location.origin);\n var changed = false;\n trackingParams.forEach(function(p) {\n if (u.searchParams.has(p)) {\n u.searchParams.delete(p);\n changed = true;\n }\n });\n return changed ? u.toString() : url;\n } catch (e) {\n return url;\n }\n }\n \n function cleanLinks() {\n document.querySelectorAll('a[href]').forEach(function(a) {\n var clean = cleanUrl(a.href);\n if (clean !== a.href) a.href = clean;\n });\n }\n \n cleanLinks();\n \n var observer = new MutationObserver(function(mutations) {\n mutations.forEach(function(m) {\n m.addedNodes.forEach(function(node) {\n if (node.nodeType === 1) {\n if (node.tagName === 'A') cleanLinks();\n node.querySelectorAll('a[href]').forEach(function(a) {\n var clean = cleanUrl(a.href);\n if (clean !== a.href) a.href = clean;\n });\n }\n });\n });\n });\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Remove Tracking Parameters from Links"); } } catch(__e) { console.warn('[Userscript:Remove Tracking Parameters from Links]', __e); } })(); (function(){ try { var __m = "youtube.com"; var __re = new RegExp('^' + "youtube\\.com" + '
Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

// Nothing is listening on this port, so `connect()` fails and the connect span
// must be recorded with an errored status.
const client = new Client({ port: 5499, user: 'test', password: 'test', database: 'tests' });

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
await client.connect().catch(() => {
// expected: connection refused
});
},
);
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
() =>
// Chain off `connect()` with `.then()` instead of awaiting it: the query
// issued from the continuation must still be parented to the active
// transaction, proving the trace context survives the connect promise.
new Promise((resolve, reject) => {
const client = new Client({ port: 5494, user: 'test', password: 'test', database: 'tests' });
client
.connect()
.then(() => client.query('SELECT 1 AS connect_then'))
.then(() => client.end())
.then(resolve, reject);
}),
);
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

const client = new Client({ port: 5494, user: 'test', password: 'test', database: 'tests' });

async function run() {
// No active span here: `requireParentSpan` means the connect and this query
// must NOT produce spans.
await client.connect();
await client.query('SELECT 1 AS unparented');

// With an active span, the query is instrumented as a child span.
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
await client.query('SELECT 2 AS parented');
},
);

await client.end();
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
import * as Sentry from '@sentry/node';
import pg from 'pg';

const { Pool } = pg;

// The connection string carries credentials so we can assert they are masked
// out of the `db.connection_string` span attribute.
const pool = new Pool({ connectionString: 'postgresql://test:test@localhost:5494/tests' });

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
try {
// Callback-style query exercises the callback-patching path (the
// promise-based scenarios never hit it).
await new Promise((resolve, reject) => {
pool.query('SELECT 1 AS foo', (err, res) => (err ? reject(err) : resolve(res)));
});
} finally {
await pool.end();
}
},
);
}

run();
Original file line numberDiff line numberDiff line change
Expand Up@@ -23,6 +23,18 @@ async function run() {

await client.query('INSERT INTO "User" ("email", "name") VALUES ($1, $2)', ['tim', 'tim@domain.com']);
await client.query('SELECT * FROM "User"');

// A named (prepared) query records its name as the `db.postgresql.plan` attribute
await client.query({
name: 'select-user-by-email',
text: 'SELECT * FROM "User" WHERE "email" = $1',
values: ['tim'],
});

// A failing query should still produce an errored span
await client.query('SELECT * FROM "does_not_exist_table"').catch(() => {
// swallow: we only care about the span it produces
});
} finally {
await client.end();
}
Expand Down
182 changes: 182 additions & 0 deletions dev-packages/node-integration-tests/suites/tracing/postgres/test.ts
Original file line numberDiff line numberDiff line change
Expand Up@@ -48,6 +48,33 @@ describe('postgres auto instrumentation', () => {
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT * FROM "User" WHERE "email" = $1',
'db.postgresql.plan': 'select-user-by-email',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT * FROM "User" WHERE "email" = $1',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT * FROM "does_not_exist_table"',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT * FROM "does_not_exist_table"',
op: 'db',
status: 'internal_error',
origin: 'auto.db.otel.postgres',
}),
]),
};

Expand DownExpand Up@@ -114,6 +141,161 @@ describe('postgres auto instrumentation', () => {
});
});

describe('pool', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
// Pool connect span: no origin is set on connect spans, so it defaults
// to 'manual', and the connection-string credentials are masked out.
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.connection_string': 'postgresql://localhost:5494/tests',
'sentry.op': 'db',
}),
description: 'pg-pool.connect',
op: 'db',
status: 'ok',
origin: 'manual',
}),
// Callback-style query (no awaited promise returned to the caller).
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 1 AS foo',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 1 AS foo',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
};

createEsmAndCjsTests(__dirname, 'scenario-pool.mjs', 'instrument.mjs', (createTestRunner, test) => {
test(
'auto-instruments `pg.Pool`, masks connection-string credentials, and handles callback-style queries',
{ timeout: 90_000 },
async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({ transaction: EXPECTED_TRANSACTION })
.start()
.completed();
},
);
});
});

describe('connect error', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'sentry.op': 'db',
}),
description: 'pg.connect',
op: 'db',
status: 'internal_error',
origin: 'manual',
}),
]),
};

// No DB needed: the scenario connects to a port where nothing is listening.
createEsmAndCjsTests(__dirname, 'scenario-connect-error.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('records an errored connect span when the connection fails', { timeout: 90_000 }, async () => {
await createTestRunner().expect({ transaction: EXPECTED_TRANSACTION }).start().completed();
});
});
});

// A query chained off `connect()` with `.then()` (rather than awaited) must still
// be parented to the active transaction. Since the instrumentation requires a
// parent span, the query only produces a span if the trace context survives the
// connect promise's continuation.
describe('connect promise continuation', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 1 AS connect_then',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 1 AS connect_then',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
};

createEsmAndCjsTests(__dirname, 'scenario-connect-then.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('parents a query chained off connect() to the active transaction', { timeout: 90_000 }, async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({ transaction: EXPECTED_TRANSACTION })
.start()
.completed();
});
});
});

describe('requireParentSpan', () => {
createEsmAndCjsTests(__dirname, 'scenario-no-parent.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('does not instrument queries or connects without an active parent span', { timeout: 90_000 }, async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({
transaction: txn => {
const descriptions = txn.spans?.map(span => span.description) ?? [];
// The unparented connect + query must not have produced spans
expect(descriptions).not.toContain('SELECT 1 AS unparented');
expect(descriptions.find(name => name?.includes('connect'))).toBeUndefined();
// Only the parented query is instrumented
expect(txn).toMatchObject({
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 2 AS parented',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 2 AS parented',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
});
},
})
.start()
.completed();
});
});
});

conditionalTest({ max: 25 })('pg-native', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
Expand Down
6 changes: 1 addition & 5 deletions packages/node/src/integrations/tracing/postgres/index.ts
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
import { PgInstrumentation } from './vendored/instrumentation';
import type { IntegrationFn } from '@sentry/core';
import { defineIntegration } from '@sentry/core';
import { addOriginToSpan, generateInstrumentOnce } from '@sentry/node-core';
import { generateInstrumentOnce } from '@sentry/node-core';

interface PostgresIntegrationOptions {
ignoreConnectSpans?: boolean;
Expand All@@ -13,10 +13,6 @@ export const instrumentPostgres = generateInstrumentOnce(
INTEGRATION_NAME,
PgInstrumentation,
(options?: PostgresIntegrationOptions) => ({
requireParentSpan: true,
requestHook(span) {
addOriginToSpan(span, 'auto.db.otel.postgres');
},
ignoreConnectSpans: options?.ignoreConnectSpans ?? false,
}),
);
Expand Down
Original file line numberDiff line numberDiff line change
Expand Up@@ -6,11 +6,9 @@
* - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-pg
* - Upstream version: @opentelemetry/instrumentation-pg@0.70.0
*/
/* eslint-disable */

// Postgresql specific attributes not covered by semantic conventions
export enum AttributeNames {
PG_VALUES = 'db.postgresql.values',
PG_PLAN = 'db.postgresql.plan',
IDLE_TIMEOUT_MILLIS = 'db.postgresql.idle.timeout.millis',
MAX_CLIENT = 'db.postgresql.max.client',
Expand Down
Original file line numberDiff line numberDiff line change
Expand Up@@ -6,7 +6,6 @@
* - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-pg
* - Upstream version: @opentelemetry/instrumentation-pg@0.70.0
*/
/* eslint-disable */

// Contains span names produced by instrumentation
export enum SpanNames {
Expand Down
Loading
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Auto-enable theater mode on YouTube\n(function() {\n function tryTheater() {\n var btn = document.querySelector('button[aria-label=\"Theater mode\"], ytd-player #player button[title=\"Theater mode\"]');\n if (btn && !btn.classList.contains('activated')) {\n btn.click();\n }\n }\n \n // Try immediately\n tryTheater();\n \n // Try after navigation (SPA)\n var lastUrl = location.href;\n setInterval(function() {\n if (location.href !== lastUrl) {\n lastUrl = location.href;\n setTimeout(tryTheater, 500);\n }\n }, 1000);\n \n // Also try on player load\n var observer = new MutationObserver(tryTheater);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "YouTube Theater Mode Default"); } } catch(__e) { console.warn('[Userscript:YouTube Theater Mode Default]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

// Nothing is listening on this port, so `connect()` fails and the connect span
// must be recorded with an errored status.
const client = new Client({ port: 5499, user: 'test', password: 'test', database: 'tests' });

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
await client.connect().catch(() => {
// expected: connection refused
});
},
);
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
() =>
// Chain off `connect()` with `.then()` instead of awaiting it: the query
// issued from the continuation must still be parented to the active
// transaction, proving the trace context survives the connect promise.
new Promise((resolve, reject) => {
const client = new Client({ port: 5494, user: 'test', password: 'test', database: 'tests' });
client
.connect()
.then(() => client.query('SELECT 1 AS connect_then'))
.then(() => client.end())
.then(resolve, reject);
}),
);
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

const client = new Client({ port: 5494, user: 'test', password: 'test', database: 'tests' });

async function run() {
// No active span here: `requireParentSpan` means the connect and this query
// must NOT produce spans.
await client.connect();
await client.query('SELECT 1 AS unparented');

// With an active span, the query is instrumented as a child span.
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
await client.query('SELECT 2 AS parented');
},
);

await client.end();
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
import * as Sentry from '@sentry/node';
import pg from 'pg';

const { Pool } = pg;

// The connection string carries credentials so we can assert they are masked
// out of the `db.connection_string` span attribute.
const pool = new Pool({ connectionString: 'postgresql://test:test@localhost:5494/tests' });

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
try {
// Callback-style query exercises the callback-patching path (the
// promise-based scenarios never hit it).
await new Promise((resolve, reject) => {
pool.query('SELECT 1 AS foo', (err, res) => (err ? reject(err) : resolve(res)));
});
} finally {
await pool.end();
}
},
);
}

run();
Original file line numberDiff line numberDiff line change
Expand Up@@ -23,6 +23,18 @@ async function run() {

await client.query('INSERT INTO "User" ("email", "name") VALUES ($1, $2)', ['tim', 'tim@domain.com']);
await client.query('SELECT * FROM "User"');

// A named (prepared) query records its name as the `db.postgresql.plan` attribute
await client.query({
name: 'select-user-by-email',
text: 'SELECT * FROM "User" WHERE "email" = $1',
values: ['tim'],
});

// A failing query should still produce an errored span
await client.query('SELECT * FROM "does_not_exist_table"').catch(() => {
// swallow: we only care about the span it produces
});
} finally {
await client.end();
}
Expand Down
182 changes: 182 additions & 0 deletions dev-packages/node-integration-tests/suites/tracing/postgres/test.ts
Original file line numberDiff line numberDiff line change
Expand Up@@ -48,6 +48,33 @@ describe('postgres auto instrumentation', () => {
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT * FROM "User" WHERE "email" = $1',
'db.postgresql.plan': 'select-user-by-email',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT * FROM "User" WHERE "email" = $1',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT * FROM "does_not_exist_table"',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT * FROM "does_not_exist_table"',
op: 'db',
status: 'internal_error',
origin: 'auto.db.otel.postgres',
}),
]),
};

Expand DownExpand Up@@ -114,6 +141,161 @@ describe('postgres auto instrumentation', () => {
});
});

describe('pool', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
// Pool connect span: no origin is set on connect spans, so it defaults
// to 'manual', and the connection-string credentials are masked out.
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.connection_string': 'postgresql://localhost:5494/tests',
'sentry.op': 'db',
}),
description: 'pg-pool.connect',
op: 'db',
status: 'ok',
origin: 'manual',
}),
// Callback-style query (no awaited promise returned to the caller).
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 1 AS foo',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 1 AS foo',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
};

createEsmAndCjsTests(__dirname, 'scenario-pool.mjs', 'instrument.mjs', (createTestRunner, test) => {
test(
'auto-instruments `pg.Pool`, masks connection-string credentials, and handles callback-style queries',
{ timeout: 90_000 },
async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({ transaction: EXPECTED_TRANSACTION })
.start()
.completed();
},
);
});
});

describe('connect error', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'sentry.op': 'db',
}),
description: 'pg.connect',
op: 'db',
status: 'internal_error',
origin: 'manual',
}),
]),
};

// No DB needed: the scenario connects to a port where nothing is listening.
createEsmAndCjsTests(__dirname, 'scenario-connect-error.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('records an errored connect span when the connection fails', { timeout: 90_000 }, async () => {
await createTestRunner().expect({ transaction: EXPECTED_TRANSACTION }).start().completed();
});
});
});

// A query chained off `connect()` with `.then()` (rather than awaited) must still
// be parented to the active transaction. Since the instrumentation requires a
// parent span, the query only produces a span if the trace context survives the
// connect promise's continuation.
describe('connect promise continuation', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 1 AS connect_then',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 1 AS connect_then',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
};

createEsmAndCjsTests(__dirname, 'scenario-connect-then.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('parents a query chained off connect() to the active transaction', { timeout: 90_000 }, async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({ transaction: EXPECTED_TRANSACTION })
.start()
.completed();
});
});
});

describe('requireParentSpan', () => {
createEsmAndCjsTests(__dirname, 'scenario-no-parent.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('does not instrument queries or connects without an active parent span', { timeout: 90_000 }, async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({
transaction: txn => {
const descriptions = txn.spans?.map(span => span.description) ?? [];
// The unparented connect + query must not have produced spans
expect(descriptions).not.toContain('SELECT 1 AS unparented');
expect(descriptions.find(name => name?.includes('connect'))).toBeUndefined();
// Only the parented query is instrumented
expect(txn).toMatchObject({
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 2 AS parented',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 2 AS parented',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
});
},
})
.start()
.completed();
});
});
});

conditionalTest({ max: 25 })('pg-native', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
Expand Down
6 changes: 1 addition & 5 deletions packages/node/src/integrations/tracing/postgres/index.ts
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
import { PgInstrumentation } from './vendored/instrumentation';
import type { IntegrationFn } from '@sentry/core';
import { defineIntegration } from '@sentry/core';
import { addOriginToSpan, generateInstrumentOnce } from '@sentry/node-core';
import { generateInstrumentOnce } from '@sentry/node-core';

interface PostgresIntegrationOptions {
ignoreConnectSpans?: boolean;
Expand All@@ -13,10 +13,6 @@ export const instrumentPostgres = generateInstrumentOnce(
INTEGRATION_NAME,
PgInstrumentation,
(options?: PostgresIntegrationOptions) => ({
requireParentSpan: true,
requestHook(span) {
addOriginToSpan(span, 'auto.db.otel.postgres');
},
ignoreConnectSpans: options?.ignoreConnectSpans ?? false,
}),
);
Expand Down
Original file line numberDiff line numberDiff line change
Expand Up@@ -6,11 +6,9 @@
* - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-pg
* - Upstream version: @opentelemetry/instrumentation-pg@0.70.0
*/
/* eslint-disable */

// Postgresql specific attributes not covered by semantic conventions
export enum AttributeNames {
PG_VALUES = 'db.postgresql.values',
PG_PLAN = 'db.postgresql.plan',
IDLE_TIMEOUT_MILLIS = 'db.postgresql.idle.timeout.millis',
MAX_CLIENT = 'db.postgresql.max.client',
Expand Down
Original file line numberDiff line numberDiff line change
Expand Up@@ -6,7 +6,6 @@
* - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-pg
* - Upstream version: @opentelemetry/instrumentation-pg@0.70.0
*/
/* eslint-disable */

// Contains span names produced by instrumentation
export enum SpanNames {
Expand Down
Loading
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Remove or un-stick sticky/fixed headers that block content\n(function() {\n function unstick() {\n document.querySelectorAll('header, nav, [role=\"banner\"], .header, .navbar, .sticky, .fixed-top, [style*=\"position: fixed\"], [style*=\"position:sticky\"]').forEach(function(el) {\n if (el.style.position === 'fixed' || el.style.position === 'sticky' || \n getComputedStyle(el).position === 'fixed' || getComputedStyle(el).position === 'sticky') {\n el.style.position = 'static';\n el.style.top = 'auto';\n el.style.zIndex = 'auto';\n }\n });\n }\n \n unstick();\n \n var observer = new MutationObserver(unstick);\n observer.observe(document.body, { childList: true, subtree: true, attributes: true, attributeFilter: ['style', 'class'] });\n})();", "Kill Sticky Headers"); } } catch(__e) { console.warn('[Userscript:Kill Sticky Headers]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

// Nothing is listening on this port, so `connect()` fails and the connect span
// must be recorded with an errored status.
const client = new Client({ port: 5499, user: 'test', password: 'test', database: 'tests' });

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
await client.connect().catch(() => {
// expected: connection refused
});
},
);
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
() =>
// Chain off `connect()` with `.then()` instead of awaiting it: the query
// issued from the continuation must still be parented to the active
// transaction, proving the trace context survives the connect promise.
new Promise((resolve, reject) => {
const client = new Client({ port: 5494, user: 'test', password: 'test', database: 'tests' });
client
.connect()
.then(() => client.query('SELECT 1 AS connect_then'))
.then(() => client.end())
.then(resolve, reject);
}),
);
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

const client = new Client({ port: 5494, user: 'test', password: 'test', database: 'tests' });

async function run() {
// No active span here: `requireParentSpan` means the connect and this query
// must NOT produce spans.
await client.connect();
await client.query('SELECT 1 AS unparented');

// With an active span, the query is instrumented as a child span.
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
await client.query('SELECT 2 AS parented');
},
);

await client.end();
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
import * as Sentry from '@sentry/node';
import pg from 'pg';

const { Pool } = pg;

// The connection string carries credentials so we can assert they are masked
// out of the `db.connection_string` span attribute.
const pool = new Pool({ connectionString: 'postgresql://test:test@localhost:5494/tests' });

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
try {
// Callback-style query exercises the callback-patching path (the
// promise-based scenarios never hit it).
await new Promise((resolve, reject) => {
pool.query('SELECT 1 AS foo', (err, res) => (err ? reject(err) : resolve(res)));
});
} finally {
await pool.end();
}
},
);
}

run();
Original file line numberDiff line numberDiff line change
Expand Up@@ -23,6 +23,18 @@ async function run() {

await client.query('INSERT INTO "User" ("email", "name") VALUES ($1, $2)', ['tim', 'tim@domain.com']);
await client.query('SELECT * FROM "User"');

// A named (prepared) query records its name as the `db.postgresql.plan` attribute
await client.query({
name: 'select-user-by-email',
text: 'SELECT * FROM "User" WHERE "email" = $1',
values: ['tim'],
});

// A failing query should still produce an errored span
await client.query('SELECT * FROM "does_not_exist_table"').catch(() => {
// swallow: we only care about the span it produces
});
} finally {
await client.end();
}
Expand Down
182 changes: 182 additions & 0 deletions dev-packages/node-integration-tests/suites/tracing/postgres/test.ts
Original file line numberDiff line numberDiff line change
Expand Up@@ -48,6 +48,33 @@ describe('postgres auto instrumentation', () => {
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT * FROM "User" WHERE "email" = $1',
'db.postgresql.plan': 'select-user-by-email',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT * FROM "User" WHERE "email" = $1',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT * FROM "does_not_exist_table"',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT * FROM "does_not_exist_table"',
op: 'db',
status: 'internal_error',
origin: 'auto.db.otel.postgres',
}),
]),
};

Expand DownExpand Up@@ -114,6 +141,161 @@ describe('postgres auto instrumentation', () => {
});
});

describe('pool', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
// Pool connect span: no origin is set on connect spans, so it defaults
// to 'manual', and the connection-string credentials are masked out.
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.connection_string': 'postgresql://localhost:5494/tests',
'sentry.op': 'db',
}),
description: 'pg-pool.connect',
op: 'db',
status: 'ok',
origin: 'manual',
}),
// Callback-style query (no awaited promise returned to the caller).
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 1 AS foo',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 1 AS foo',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
};

createEsmAndCjsTests(__dirname, 'scenario-pool.mjs', 'instrument.mjs', (createTestRunner, test) => {
test(
'auto-instruments `pg.Pool`, masks connection-string credentials, and handles callback-style queries',
{ timeout: 90_000 },
async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({ transaction: EXPECTED_TRANSACTION })
.start()
.completed();
},
);
});
});

describe('connect error', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'sentry.op': 'db',
}),
description: 'pg.connect',
op: 'db',
status: 'internal_error',
origin: 'manual',
}),
]),
};

// No DB needed: the scenario connects to a port where nothing is listening.
createEsmAndCjsTests(__dirname, 'scenario-connect-error.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('records an errored connect span when the connection fails', { timeout: 90_000 }, async () => {
await createTestRunner().expect({ transaction: EXPECTED_TRANSACTION }).start().completed();
});
});
});

// A query chained off `connect()` with `.then()` (rather than awaited) must still
// be parented to the active transaction. Since the instrumentation requires a
// parent span, the query only produces a span if the trace context survives the
// connect promise's continuation.
describe('connect promise continuation', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 1 AS connect_then',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 1 AS connect_then',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
};

createEsmAndCjsTests(__dirname, 'scenario-connect-then.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('parents a query chained off connect() to the active transaction', { timeout: 90_000 }, async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({ transaction: EXPECTED_TRANSACTION })
.start()
.completed();
});
});
});

describe('requireParentSpan', () => {
createEsmAndCjsTests(__dirname, 'scenario-no-parent.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('does not instrument queries or connects without an active parent span', { timeout: 90_000 }, async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({
transaction: txn => {
const descriptions = txn.spans?.map(span => span.description) ?? [];
// The unparented connect + query must not have produced spans
expect(descriptions).not.toContain('SELECT 1 AS unparented');
expect(descriptions.find(name => name?.includes('connect'))).toBeUndefined();
// Only the parented query is instrumented
expect(txn).toMatchObject({
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 2 AS parented',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 2 AS parented',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
});
},
})
.start()
.completed();
});
});
});

conditionalTest({ max: 25 })('pg-native', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
Expand Down
6 changes: 1 addition & 5 deletions packages/node/src/integrations/tracing/postgres/index.ts
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
import { PgInstrumentation } from './vendored/instrumentation';
import type { IntegrationFn } from '@sentry/core';
import { defineIntegration } from '@sentry/core';
import { addOriginToSpan, generateInstrumentOnce } from '@sentry/node-core';
import { generateInstrumentOnce } from '@sentry/node-core';

interface PostgresIntegrationOptions {
ignoreConnectSpans?: boolean;
Expand All@@ -13,10 +13,6 @@ export const instrumentPostgres = generateInstrumentOnce(
INTEGRATION_NAME,
PgInstrumentation,
(options?: PostgresIntegrationOptions) => ({
requireParentSpan: true,
requestHook(span) {
addOriginToSpan(span, 'auto.db.otel.postgres');
},
ignoreConnectSpans: options?.ignoreConnectSpans ?? false,
}),
);
Expand Down
Original file line numberDiff line numberDiff line change
Expand Up@@ -6,11 +6,9 @@
* - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-pg
* - Upstream version: @opentelemetry/instrumentation-pg@0.70.0
*/
/* eslint-disable */

// Postgresql specific attributes not covered by semantic conventions
export enum AttributeNames {
PG_VALUES = 'db.postgresql.values',
PG_PLAN = 'db.postgresql.plan',
IDLE_TIMEOUT_MILLIS = 'db.postgresql.idle.timeout.millis',
MAX_CLIENT = 'db.postgresql.max.client',
Expand Down
Original file line numberDiff line numberDiff line change
Expand Up@@ -6,7 +6,6 @@
* - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-pg
* - Upstream version: @opentelemetry/instrumentation-pg@0.70.0
*/
/* eslint-disable */

// Contains span names produced by instrumentation
export enum SpanNames {
Expand Down
Loading
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Universal Dark Mode - works on any site\n(function() {\n var enabled = true;\n \n function applyDarkMode() {\n if (!enabled) return;\n \n // Create style element if it doesn't exist\n var style = document.getElementById('universal-dark-mode-style');\n if (!style) {\n style = document.createElement('style');\n style.id = 'universal-dark-mode-style';\n document.head.appendChild(style);\n }\n \n // Dark mode CSS - inverts colors but preserves images/video\n style.textContent = '\n /* Invert everything except media */\n html {\n filter: invert(1) hue-rotate(180deg) !important;\n background: #1a1a2e !important;\n }\n \n /* Restore images, videos, iframes, canvas */\n img, video, iframe, canvas, svg, picture, [style*=\"background-image\"] {\n filter: invert(1) hue-rotate(180deg) !important;\n }\n \n /* Preserve specific elements that should not be inverted */\n .no-dark-mode, .no-dark-mode *,\n [data-theme=\"light\"], [data-theme=\"light\"],\n .ace_editor, .ace_editor *,\n .CodeMirror, .CodeMirror *,\n .monaco-editor, .monaco-editor *,\n .markdown-body pre, .markdown-body pre *,\n .highlight, .highlight *,\n pre code, pre code * {\n filter: none !important;\n }\n \n /* Fix common UI elements */\n .modal, .popup, .dropdown-menu, .tooltip, .popover {\n filter: invert(1) hue-rotate(180deg) !important;\n background: #2d2d44 !important;\n border-color: #444 !important;\n }\n \n /* Scrollbars */\n ::-webkit-scrollbar { background: #1a1a2e !important; }\n ::-webkit-scrollbar-thumb { background: #444 !important; }\n ::-webkit-scrollbar-thumb:hover { background: #555 !important; }\n \n /* Selection */\n ::selection { background: #4ecdc4 !important; color: #1a1a2e !important; }\n ::-moz-selection { background: #4ecdc4 !important; color: #1a1a2e !important; }\n ';\n }\n \n function removeDarkMode() {\n var style = document.getElementById('universal-dark-mode-style');\n if (style) style.remove();\n }\n \n // Toggle with Alt+Shift+D\n document.addEventListener('keydown', function(e) {\n if (e.altKey && e.shiftKey && e.key === 'D') {\n e.preventDefault();\n enabled = !enabled;\n if (enabled) {\n applyDarkMode();\n console.log('[Universal Dark Mode] Enabled');\n } else {\n removeDarkMode();\n console.log('[Universal Dark Mode] Disabled');\n }\n }\n });\n \n // Apply on load\n applyDarkMode();\n \n // Re-apply on dynamic content\n var observer = new MutationObserver(function(mutations) {\n if (enabled && !document.getElementById('universal-dark-mode-style')) {\n applyDarkMode();\n }\n });\n observer.observe(document.head, { childList: true });\n \n console.log('[Universal Dark Mode] Loaded - Press Alt+Shift+D to toggle');\n})();", "Universal Dark Mode"); } } catch(__e) { console.warn('[Userscript:Universal Dark Mode]', __e); } })(); })();
Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

// Nothing is listening on this port, so `connect()` fails and the connect span
// must be recorded with an errored status.
const client = new Client({ port: 5499, user: 'test', password: 'test', database: 'tests' });

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
await client.connect().catch(() => {
// expected: connection refused
});
},
);
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
() =>
// Chain off `connect()` with `.then()` instead of awaiting it: the query
// issued from the continuation must still be parented to the active
// transaction, proving the trace context survives the connect promise.
new Promise((resolve, reject) => {
const client = new Client({ port: 5494, user: 'test', password: 'test', database: 'tests' });
client
.connect()
.then(() => client.query('SELECT 1 AS connect_then'))
.then(() => client.end())
.then(resolve, reject);
}),
);
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
import * as Sentry from '@sentry/node';
import { Client } from 'pg';

const client = new Client({ port: 5494, user: 'test', password: 'test', database: 'tests' });

async function run() {
// No active span here: `requireParentSpan` means the connect and this query
// must NOT produce spans.
await client.connect();
await client.query('SELECT 1 AS unparented');

// With an active span, the query is instrumented as a child span.
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
await client.query('SELECT 2 AS parented');
},
);

await client.end();
}

run();
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
import * as Sentry from '@sentry/node';
import pg from 'pg';

const { Pool } = pg;

// The connection string carries credentials so we can assert they are masked
// out of the `db.connection_string` span attribute.
const pool = new Pool({ connectionString: 'postgresql://test:test@localhost:5494/tests' });

async function run() {
await Sentry.startSpan(
{
name: 'Test Transaction',
op: 'transaction',
},
async () => {
try {
// Callback-style query exercises the callback-patching path (the
// promise-based scenarios never hit it).
await new Promise((resolve, reject) => {
pool.query('SELECT 1 AS foo', (err, res) => (err ? reject(err) : resolve(res)));
});
} finally {
await pool.end();
}
},
);
}

run();
Original file line numberDiff line numberDiff line change
Expand Up@@ -23,6 +23,18 @@ async function run() {

await client.query('INSERT INTO "User" ("email", "name") VALUES ($1, $2)', ['tim', 'tim@domain.com']);
await client.query('SELECT * FROM "User"');

// A named (prepared) query records its name as the `db.postgresql.plan` attribute
await client.query({
name: 'select-user-by-email',
text: 'SELECT * FROM "User" WHERE "email" = $1',
values: ['tim'],
});

// A failing query should still produce an errored span
await client.query('SELECT * FROM "does_not_exist_table"').catch(() => {
// swallow: we only care about the span it produces
});
} finally {
await client.end();
}
Expand Down
182 changes: 182 additions & 0 deletions dev-packages/node-integration-tests/suites/tracing/postgres/test.ts
Original file line numberDiff line numberDiff line change
Expand Up@@ -48,6 +48,33 @@ describe('postgres auto instrumentation', () => {
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT * FROM "User" WHERE "email" = $1',
'db.postgresql.plan': 'select-user-by-email',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT * FROM "User" WHERE "email" = $1',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT * FROM "does_not_exist_table"',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT * FROM "does_not_exist_table"',
op: 'db',
status: 'internal_error',
origin: 'auto.db.otel.postgres',
}),
]),
};

Expand DownExpand Up@@ -114,6 +141,161 @@ describe('postgres auto instrumentation', () => {
});
});

describe('pool', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
// Pool connect span: no origin is set on connect spans, so it defaults
// to 'manual', and the connection-string credentials are masked out.
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.connection_string': 'postgresql://localhost:5494/tests',
'sentry.op': 'db',
}),
description: 'pg-pool.connect',
op: 'db',
status: 'ok',
origin: 'manual',
}),
// Callback-style query (no awaited promise returned to the caller).
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 1 AS foo',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 1 AS foo',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
};

createEsmAndCjsTests(__dirname, 'scenario-pool.mjs', 'instrument.mjs', (createTestRunner, test) => {
test(
'auto-instruments `pg.Pool`, masks connection-string credentials, and handles callback-style queries',
{ timeout: 90_000 },
async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({ transaction: EXPECTED_TRANSACTION })
.start()
.completed();
},
);
});
});

describe('connect error', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'sentry.op': 'db',
}),
description: 'pg.connect',
op: 'db',
status: 'internal_error',
origin: 'manual',
}),
]),
};

// No DB needed: the scenario connects to a port where nothing is listening.
createEsmAndCjsTests(__dirname, 'scenario-connect-error.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('records an errored connect span when the connection fails', { timeout: 90_000 }, async () => {
await createTestRunner().expect({ transaction: EXPECTED_TRANSACTION }).start().completed();
});
});
});

// A query chained off `connect()` with `.then()` (rather than awaited) must still
// be parented to the active transaction. Since the instrumentation requires a
// parent span, the query only produces a span if the trace context survives the
// connect promise's continuation.
describe('connect promise continuation', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 1 AS connect_then',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 1 AS connect_then',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
};

createEsmAndCjsTests(__dirname, 'scenario-connect-then.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('parents a query chained off connect() to the active transaction', { timeout: 90_000 }, async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({ transaction: EXPECTED_TRANSACTION })
.start()
.completed();
});
});
});

describe('requireParentSpan', () => {
createEsmAndCjsTests(__dirname, 'scenario-no-parent.mjs', 'instrument.mjs', (createTestRunner, test) => {
test('does not instrument queries or connects without an active parent span', { timeout: 90_000 }, async () => {
await createTestRunner()
.withDockerCompose({
workingDirectory: [__dirname],
})
.expect({
transaction: txn => {
const descriptions = txn.spans?.map(span => span.description) ?? [];
// The unparented connect + query must not have produced spans
expect(descriptions).not.toContain('SELECT 1 AS unparented');
expect(descriptions.find(name => name?.includes('connect'))).toBeUndefined();
// Only the parented query is instrumented
expect(txn).toMatchObject({
transaction: 'Test Transaction',
spans: expect.arrayContaining([
expect.objectContaining({
data: expect.objectContaining({
'db.system': 'postgresql',
'db.name': 'tests',
'db.statement': 'SELECT 2 AS parented',
'sentry.origin': 'auto.db.otel.postgres',
'sentry.op': 'db',
}),
description: 'SELECT 2 AS parented',
op: 'db',
status: 'ok',
origin: 'auto.db.otel.postgres',
}),
]),
});
},
})
.start()
.completed();
});
});
});

conditionalTest({ max: 25 })('pg-native', () => {
const EXPECTED_TRANSACTION = {
transaction: 'Test Transaction',
Expand Down
6 changes: 1 addition & 5 deletions packages/node/src/integrations/tracing/postgres/index.ts
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
import { PgInstrumentation } from './vendored/instrumentation';
import type { IntegrationFn } from '@sentry/core';
import { defineIntegration } from '@sentry/core';
import { addOriginToSpan, generateInstrumentOnce } from '@sentry/node-core';
import { generateInstrumentOnce } from '@sentry/node-core';

interface PostgresIntegrationOptions {
ignoreConnectSpans?: boolean;
Expand All@@ -13,10 +13,6 @@ export const instrumentPostgres = generateInstrumentOnce(
INTEGRATION_NAME,
PgInstrumentation,
(options?: PostgresIntegrationOptions) => ({
requireParentSpan: true,
requestHook(span) {
addOriginToSpan(span, 'auto.db.otel.postgres');
},
ignoreConnectSpans: options?.ignoreConnectSpans ?? false,
}),
);
Expand Down
Original file line numberDiff line numberDiff line change
Expand Up@@ -6,11 +6,9 @@
* - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-pg
* - Upstream version: @opentelemetry/instrumentation-pg@0.70.0
*/
/* eslint-disable */

// Postgresql specific attributes not covered by semantic conventions
export enum AttributeNames {
PG_VALUES = 'db.postgresql.values',
PG_PLAN = 'db.postgresql.plan',
IDLE_TIMEOUT_MILLIS = 'db.postgresql.idle.timeout.millis',
MAX_CLIENT = 'db.postgresql.max.client',
Expand Down
Original file line numberDiff line numberDiff line change
Expand Up@@ -6,7 +6,6 @@
* - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-pg
* - Upstream version: @opentelemetry/instrumentation-pg@0.70.0
*/
/* eslint-disable */

// Contains span names produced by instrumentation
export enum SpanNames {
Expand Down
Loading
Loading