42 lines
1.2 KiB
TypeScript
42 lines
1.2 KiB
TypeScript
|
import postgres from "postgres";
|
||
|
|
import { OTEL_ENABLED, withSpan } from "./telemetry.ts";
|
||
|
|
|
||
|
|
const DATABASE_URL =
|
||
|
|
Deno.env.get("DATABASE_URL") ??
|
||
|
|
"postgres://driver:driver@localhost:5432/driver_db";
|
||
|
|
|
||
|
|
const _sql = postgres(DATABASE_URL, {
|
||
|
|
max: 10,
|
||
|
|
idle_timeout: 20,
|
||
|
|
connect_timeout: 10,
|
||
|
|
});
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Traced SQL tagged-template proxy.
|
||
|
|
*
|
||
|
|
* When OTEL is enabled every query is wrapped in a `db.query` span whose
|
||
|
|
* `db.statement` attribute contains the SQL template (parameters replaced
|
||
|
|
* with `$N` placeholders — no user data leaks into traces).
|
||
|
|
*
|
||
|
|
* When OTEL is disabled this is the raw postgres.js instance.
|
||
|
|
*/
|
||
|
|
// deno-lint-ignore no-explicit-any
|
||
|
|
const sql: typeof _sql = OTEL_ENABLED
|
||
|
|
? new Proxy(_sql, {
|
||
|
|
apply(_target, _thisArg, args) {
|
||
|
|
// Tagged-template call: sql`SELECT ...`
|
||
|
|
const [strings] = args as [TemplateStringsArray, ...unknown[]];
|
||
|
|
const statement = Array.isArray(strings)
|
||
|
|
? strings.join("$?")
|
||
|
|
: "unknown";
|
||
|
|
return withSpan(
|
||
|
|
"db.query",
|
||
|
|
{ "db.statement": statement, "db.system": "postgresql" },
|
||
|
|
() => Reflect.apply(_target, _thisArg, args),
|
||
|
|
);
|
||
|
|
},
|
||
|
|
})
|
||
|
|
: _sql;
|
||
|
|
|
||
|
|
export default sql;
|