@prisma-next/driver-postgres 0.9.0-dev.2 → 0.9.0-dev.3
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/runtime.d.mts +7 -0
- package/dist/runtime.d.mts.map +1 -1
- package/dist/runtime.mjs +138 -26
- package/dist/runtime.mjs.map +1 -1
- package/package.json +12 -12
- package/src/exports/runtime.ts +20 -4
- package/src/named-cursor.ts +82 -0
- package/src/postgres-driver.ts +135 -21
package/dist/runtime.d.mts
CHANGED
|
@@ -25,6 +25,13 @@ interface PostgresDriverOptions {
|
|
|
25
25
|
pool: Pool;
|
|
26
26
|
};
|
|
27
27
|
readonly cursor?: PostgresCursorOptions | undefined;
|
|
28
|
+
/**
|
|
29
|
+
* Use server-side prepared statements for `executePrepared`. Default
|
|
30
|
+
* `true`. Set `false` when running behind a transaction-mode pooler that
|
|
31
|
+
* does not support session-scoped prepared statements (e.g. PgBouncer
|
|
32
|
+
* transaction mode).
|
|
33
|
+
*/
|
|
34
|
+
readonly preparedStatements?: boolean | undefined;
|
|
28
35
|
}
|
|
29
36
|
type PostgresDriverCreateOptions = Omit<PostgresDriverOptions, 'connect'>;
|
|
30
37
|
//#endregion
|
package/dist/runtime.d.mts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"runtime.d.mts","names":[],"sources":["../src/postgres-driver.ts","../src/exports/runtime.ts"],"mappings":";;;;;
|
|
1
|
+
{"version":3,"file":"runtime.d.mts","names":[],"sources":["../src/postgres-driver.ts","../src/exports/runtime.ts"],"mappings":";;;;;KAyBY,WAAA,WAAsB,cAAA,GAAiB,cAAA,IAAkB,aAAA,CAAc,CAAA;AAAA,KAEvE,eAAA;EAAA,SACG,IAAA;EAAA,SAAsB,GAAA;AAAA;EAAA,SACtB,IAAA;EAAA,SAAyB,IAAA,EAAM,IAAA;AAAA;EAAA,SAC/B,IAAA;EAAA,SAA2B,MAAA,EAAQ,MAAA;AAAA;AAAA,UAEjC,qBAAA;EAAA,SACN,SAAA;EAAA,SACA,QAAA;AAAA;AAAA,UAGD,qBAAA;EAAA,SACC,OAAA;IAAW,MAAA,EAAQ,MAAA;EAAA;IAAa,IAAA,EAAM,IAAA;EAAA;EAAA,SACtC,MAAA,GAAS,qBAAA;EAToC;;;;;;EAAA,SAgB7C,kBAAA;AAAA;AAAA,KAGC,2BAAA,GAA8B,IAAA,CAAK,qBAAA;;;KC9BnC,qBAAA,GAAwB,qBAAA,sBAClC,SAAA,CAAU,eAAA;AAAA,cAqLN,+BAAA,EAAiC,uBAAA,oBAGrC,2BAAA,EACA,qBAAA"}
|
package/dist/runtime.mjs
CHANGED
|
@@ -14,8 +14,69 @@ function callbackToPromise(fn) {
|
|
|
14
14
|
});
|
|
15
15
|
}
|
|
16
16
|
//#endregion
|
|
17
|
+
//#region src/named-cursor.ts
|
|
18
|
+
let nextNamedPortalId = 1;
|
|
19
|
+
/** Streaming cursor for a server-side named prepared statement. */
|
|
20
|
+
var NamedCursor = class extends Cursor {
|
|
21
|
+
name;
|
|
22
|
+
constructor(opts) {
|
|
23
|
+
super(opts.text, [...opts.values], opts.config);
|
|
24
|
+
this.name = opts.name;
|
|
25
|
+
this.submit = this.submitNamed;
|
|
26
|
+
}
|
|
27
|
+
submitNamed(connection) {
|
|
28
|
+
const self = this;
|
|
29
|
+
const conn = connection;
|
|
30
|
+
self.state = "submitted";
|
|
31
|
+
self.connection = conn;
|
|
32
|
+
self._portal = `np_${nextNamedPortalId++}`;
|
|
33
|
+
if (!conn.parsedStatements[this.name]) {
|
|
34
|
+
const parseMessage = {
|
|
35
|
+
text: self.text,
|
|
36
|
+
name: this.name,
|
|
37
|
+
types: []
|
|
38
|
+
};
|
|
39
|
+
conn.parse(parseMessage, true);
|
|
40
|
+
}
|
|
41
|
+
const bindMessage = {
|
|
42
|
+
portal: self._portal,
|
|
43
|
+
statement: this.name,
|
|
44
|
+
values: self.values ?? []
|
|
45
|
+
};
|
|
46
|
+
conn.bind(bindMessage, true);
|
|
47
|
+
conn.describe({
|
|
48
|
+
type: "P",
|
|
49
|
+
name: self._portal
|
|
50
|
+
}, true);
|
|
51
|
+
conn.flush();
|
|
52
|
+
if (self._conf.types) self._result._getTypeParser = self._conf.types.getTypeParser;
|
|
53
|
+
conn.once("noData", self._ifNoData);
|
|
54
|
+
conn.once("rowDescription", self._rowDescription);
|
|
55
|
+
}
|
|
56
|
+
};
|
|
57
|
+
//#endregion
|
|
17
58
|
//#region src/postgres-driver.ts
|
|
18
59
|
const DEFAULT_BATCH_SIZE = 100;
|
|
60
|
+
const DEFAULT_PREPARED_STATEMENTS = true;
|
|
61
|
+
function createHandleAllocator() {
|
|
62
|
+
let next = 1;
|
|
63
|
+
return { mint: () => `pn_${next++}` };
|
|
64
|
+
}
|
|
65
|
+
function buildConnectionOptions(options) {
|
|
66
|
+
return {
|
|
67
|
+
...options.cursor?.disabled ? { cursorDisabled: true } : {
|
|
68
|
+
cursorBatchSize: options.cursor?.batchSize ?? DEFAULT_BATCH_SIZE,
|
|
69
|
+
cursorDisabled: false
|
|
70
|
+
},
|
|
71
|
+
preparedStatementsEnabled: options.preparedStatements ?? DEFAULT_PREPARED_STATEMENTS,
|
|
72
|
+
handleAllocator: createHandleAllocator()
|
|
73
|
+
};
|
|
74
|
+
}
|
|
75
|
+
function isStalePreparedStatementError(error) {
|
|
76
|
+
if (!(error instanceof Error)) return false;
|
|
77
|
+
const code = error.code;
|
|
78
|
+
return code === "26000" || code === "0A000";
|
|
79
|
+
}
|
|
19
80
|
var AsyncMutex = class {
|
|
20
81
|
#queue = Promise.resolve();
|
|
21
82
|
async lock() {
|
|
@@ -37,18 +98,53 @@ var PostgresQueryable = class {
|
|
|
37
98
|
this.options = options;
|
|
38
99
|
}
|
|
39
100
|
async *execute(request) {
|
|
101
|
+
try {
|
|
102
|
+
yield* this.runQuery(request);
|
|
103
|
+
} catch (error) {
|
|
104
|
+
throw normalizePgError(error);
|
|
105
|
+
}
|
|
106
|
+
}
|
|
107
|
+
async *executePrepared(request) {
|
|
108
|
+
if (!this.options.preparedStatementsEnabled) {
|
|
109
|
+
yield* this.execute(request);
|
|
110
|
+
return;
|
|
111
|
+
}
|
|
112
|
+
let handle = request.handle.get();
|
|
113
|
+
if (handle === void 0) {
|
|
114
|
+
handle = this.options.handleAllocator.mint();
|
|
115
|
+
request.handle.set(handle);
|
|
116
|
+
}
|
|
117
|
+
yield* this.withStaleHandleRetry(request, handle, (h) => this.runQuery(request, h));
|
|
118
|
+
}
|
|
119
|
+
async *withStaleHandleRetry(request, handle, attempt) {
|
|
120
|
+
let yielded = false;
|
|
121
|
+
try {
|
|
122
|
+
for await (const row of attempt(handle)) {
|
|
123
|
+
yielded = true;
|
|
124
|
+
yield row;
|
|
125
|
+
}
|
|
126
|
+
return;
|
|
127
|
+
} catch (error) {
|
|
128
|
+
if (yielded || !isStalePreparedStatementError(error)) throw normalizePgError(error);
|
|
129
|
+
const retryHandle = this.options.handleAllocator.mint();
|
|
130
|
+
request.handle.set(retryHandle);
|
|
131
|
+
try {
|
|
132
|
+
yield* attempt(retryHandle);
|
|
133
|
+
} catch (retryError) {
|
|
134
|
+
throw normalizePgError(retryError);
|
|
135
|
+
}
|
|
136
|
+
}
|
|
137
|
+
}
|
|
138
|
+
async *runQuery(request, name) {
|
|
40
139
|
const client = await this.acquireClient();
|
|
41
140
|
try {
|
|
42
141
|
if (!this.options.cursorDisabled) try {
|
|
43
|
-
for await (const row of this.executeWithCursor(client, request.sql, request.params, this.options.cursorBatchSize)) yield row;
|
|
142
|
+
for await (const row of this.executeWithCursor(client, request.sql, request.params, this.options.cursorBatchSize, name)) yield row;
|
|
44
143
|
return;
|
|
45
144
|
} catch (cursorError) {
|
|
46
|
-
if (!(cursorError instanceof Error)) throw cursorError;
|
|
47
|
-
if (isPostgresError(cursorError)) throw normalizePgError(cursorError);
|
|
145
|
+
if (!(cursorError instanceof Error) || isPostgresError(cursorError)) throw cursorError;
|
|
48
146
|
}
|
|
49
|
-
for await (const row of this.executeBuffered(client, request.sql, request.params)) yield row;
|
|
50
|
-
} catch (error) {
|
|
51
|
-
throw normalizePgError(error);
|
|
147
|
+
for await (const row of this.executeBuffered(client, request.sql, request.params, name)) yield row;
|
|
52
148
|
} finally {
|
|
53
149
|
await this.releaseClient(client);
|
|
54
150
|
}
|
|
@@ -70,8 +166,13 @@ var PostgresQueryable = class {
|
|
|
70
166
|
await this.releaseClient(client);
|
|
71
167
|
}
|
|
72
168
|
}
|
|
73
|
-
async *executeWithCursor(client, sql, params, cursorBatchSize) {
|
|
74
|
-
const
|
|
169
|
+
async *executeWithCursor(client, sql, params, cursorBatchSize, name) {
|
|
170
|
+
const values = params ?? [];
|
|
171
|
+
const cursor = client.query(name === void 0 ? new Cursor(sql, values) : new NamedCursor({
|
|
172
|
+
name,
|
|
173
|
+
text: sql,
|
|
174
|
+
values
|
|
175
|
+
}));
|
|
75
176
|
try {
|
|
76
177
|
while (true) {
|
|
77
178
|
const rows = await readCursor(cursor, cursorBatchSize);
|
|
@@ -82,8 +183,13 @@ var PostgresQueryable = class {
|
|
|
82
183
|
await closeCursor(cursor);
|
|
83
184
|
}
|
|
84
185
|
}
|
|
85
|
-
async *executeBuffered(client, sql, params) {
|
|
86
|
-
const
|
|
186
|
+
async *executeBuffered(client, sql, params, name) {
|
|
187
|
+
const config = {
|
|
188
|
+
name,
|
|
189
|
+
text: sql,
|
|
190
|
+
values: params ?? []
|
|
191
|
+
};
|
|
192
|
+
const result = await client.query(config);
|
|
87
193
|
for (const row of result.rows) yield row;
|
|
88
194
|
}
|
|
89
195
|
};
|
|
@@ -152,10 +258,7 @@ var PostgresPoolDriverImpl = class extends PostgresQueryable {
|
|
|
152
258
|
pool;
|
|
153
259
|
#closed = false;
|
|
154
260
|
constructor(options) {
|
|
155
|
-
super(
|
|
156
|
-
cursorBatchSize: options.cursor?.batchSize ?? DEFAULT_BATCH_SIZE,
|
|
157
|
-
cursorDisabled: options.cursor?.disabled ?? false
|
|
158
|
-
});
|
|
261
|
+
super(buildConnectionOptions(options));
|
|
159
262
|
this.pool = options.connect.pool;
|
|
160
263
|
}
|
|
161
264
|
get state() {
|
|
@@ -184,10 +287,7 @@ var PostgresDirectDriverImpl = class extends PostgresQueryable {
|
|
|
184
287
|
#connected = false;
|
|
185
288
|
#connectPromise;
|
|
186
289
|
constructor(options) {
|
|
187
|
-
super(
|
|
188
|
-
cursorBatchSize: options.cursor?.batchSize ?? DEFAULT_BATCH_SIZE,
|
|
189
|
-
cursorDisabled: options.cursor?.disabled ?? false
|
|
190
|
-
});
|
|
290
|
+
super(buildConnectionOptions(options));
|
|
191
291
|
this.directClient = options.connect.client;
|
|
192
292
|
}
|
|
193
293
|
get state() {
|
|
@@ -244,7 +344,8 @@ var PostgresDirectDriverImpl = class extends PostgresQueryable {
|
|
|
244
344
|
}
|
|
245
345
|
async releaseClient(_client) {}
|
|
246
346
|
};
|
|
247
|
-
function createBoundDriverFromBinding(binding, cursorOpts) {
|
|
347
|
+
function createBoundDriverFromBinding(binding, cursorOpts, extraOpts) {
|
|
348
|
+
const preparedStatements = extraOpts?.preparedStatements;
|
|
248
349
|
switch (binding.kind) {
|
|
249
350
|
case "url": return new PostgresPoolDriverImpl({
|
|
250
351
|
connect: { pool: new Pool({
|
|
@@ -252,15 +353,18 @@ function createBoundDriverFromBinding(binding, cursorOpts) {
|
|
|
252
353
|
connectionTimeoutMillis: 2e4,
|
|
253
354
|
idleTimeoutMillis: 3e4
|
|
254
355
|
}) },
|
|
255
|
-
cursor: cursorOpts
|
|
356
|
+
cursor: cursorOpts,
|
|
357
|
+
preparedStatements
|
|
256
358
|
});
|
|
257
359
|
case "pgPool": return new PostgresPoolDriverImpl({
|
|
258
360
|
connect: { pool: binding.pool },
|
|
259
|
-
cursor: cursorOpts
|
|
361
|
+
cursor: cursorOpts,
|
|
362
|
+
preparedStatements
|
|
260
363
|
});
|
|
261
364
|
case "pgClient": return new PostgresDirectDriverImpl({
|
|
262
365
|
connect: { client: binding.client },
|
|
263
|
-
cursor: cursorOpts
|
|
366
|
+
cursor: cursorOpts,
|
|
367
|
+
preparedStatements
|
|
264
368
|
});
|
|
265
369
|
}
|
|
266
370
|
}
|
|
@@ -306,8 +410,10 @@ var PostgresUnboundDriverImpl = class {
|
|
|
306
410
|
#delegate = null;
|
|
307
411
|
#closed = false;
|
|
308
412
|
#cursorOpts;
|
|
309
|
-
|
|
310
|
-
|
|
413
|
+
#preparedStatements;
|
|
414
|
+
constructor(options) {
|
|
415
|
+
this.#cursorOpts = options?.cursor;
|
|
416
|
+
this.#preparedStatements = options?.preparedStatements;
|
|
311
417
|
}
|
|
312
418
|
get state() {
|
|
313
419
|
if (this.#delegate !== null) return "connected";
|
|
@@ -321,7 +427,7 @@ var PostgresUnboundDriverImpl = class {
|
|
|
321
427
|
}
|
|
322
428
|
async connect(binding) {
|
|
323
429
|
if (this.#delegate !== null) throw driverError("DRIVER.ALREADY_CONNECTED", ALREADY_CONNECTED_MESSAGE, { bindingKind: binding.kind });
|
|
324
|
-
this.#delegate = createBoundDriverFromBinding(binding, this.#cursorOpts);
|
|
430
|
+
this.#delegate = createBoundDriverFromBinding(binding, this.#cursorOpts, { preparedStatements: this.#preparedStatements });
|
|
325
431
|
this.#closed = false;
|
|
326
432
|
}
|
|
327
433
|
async acquireConnection() {
|
|
@@ -347,6 +453,7 @@ var PostgresUnboundDriverImpl = class {
|
|
|
347
453
|
const wrapped = {
|
|
348
454
|
beginTransaction: connection.beginTransaction.bind(connection),
|
|
349
455
|
execute: connection.execute.bind(connection),
|
|
456
|
+
executePrepared: connection.executePrepared.bind(connection),
|
|
350
457
|
query: connection.query.bind(connection),
|
|
351
458
|
release: async () => {
|
|
352
459
|
try {
|
|
@@ -379,6 +486,11 @@ var PostgresUnboundDriverImpl = class {
|
|
|
379
486
|
if (delegate === null) return unboundExecute();
|
|
380
487
|
return delegate.execute(request);
|
|
381
488
|
}
|
|
489
|
+
executePrepared(request) {
|
|
490
|
+
const delegate = this.#delegate;
|
|
491
|
+
if (delegate === null) return unboundExecute();
|
|
492
|
+
return delegate.executePrepared(request);
|
|
493
|
+
}
|
|
382
494
|
async explain(request) {
|
|
383
495
|
const delegate = this.#requireDelegate();
|
|
384
496
|
const explain = delegate.explain;
|
|
@@ -392,7 +504,7 @@ var PostgresUnboundDriverImpl = class {
|
|
|
392
504
|
const postgresRuntimeDriverDescriptor = {
|
|
393
505
|
...postgresDriverDescriptorMeta,
|
|
394
506
|
create(options) {
|
|
395
|
-
return new PostgresUnboundDriverImpl(options
|
|
507
|
+
return new PostgresUnboundDriverImpl(options);
|
|
396
508
|
}
|
|
397
509
|
};
|
|
398
510
|
//#endregion
|
package/dist/runtime.mjs.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"runtime.mjs","names":["#queue","#connection","#onRelease","#onDestroy","#closed","#connectionMutex","#closeWhileHoldingLease","#connected","#connectPromise","#cursorOpts","#delegate","#closed","#requireDelegate","#wrapConnection"],"sources":["../src/callback-to-promise.ts","../src/postgres-driver.ts","../src/exports/runtime.ts"],"sourcesContent":["/**\n * Wraps a Node-style callback function into a Promise.\n * Supports both (err) => void and (err, result) => void patterns.\n */\nexport function callbackToPromise(\n fn: (callback: (err: Error | null | undefined) => void) => void,\n): Promise<void>;\nexport function callbackToPromise<T>(\n fn: (callback: (err: Error | null | undefined, result: T) => void) => void,\n): Promise<T>;\nexport function callbackToPromise<T = void>(\n fn: (callback: (err: Error | null | undefined, result?: T) => void) => void,\n): Promise<T> {\n return new Promise<T>((resolve, reject) => {\n fn((err, result) => {\n if (err) {\n reject(err);\n return;\n }\n resolve(result as T);\n });\n });\n}\n","import type {\n SqlConnection,\n SqlDriver,\n SqlDriverState,\n SqlExecuteRequest,\n SqlExplainResult,\n SqlQueryable,\n SqlQueryResult,\n SqlTransaction,\n} from '@prisma-next/sql-relational-core/ast';\nimport type {\n Client,\n QueryResult as PgQueryResult,\n PoolClient,\n Pool as PoolType,\n QueryResultRow,\n} from 'pg';\nimport { Pool } from 'pg';\nimport Cursor from 'pg-cursor';\nimport { callbackToPromise } from './callback-to-promise';\nimport { isAlreadyConnectedError, isPostgresError, normalizePgError } from './normalize-error';\n\nexport type QueryResult<T extends QueryResultRow = QueryResultRow> = PgQueryResult<T>;\n\nexport type PostgresBinding =\n | { readonly kind: 'url'; readonly url: string }\n | { readonly kind: 'pgPool'; readonly pool: PoolType }\n | { readonly kind: 'pgClient'; readonly client: Client };\n\nexport interface PostgresCursorOptions {\n readonly batchSize?: number;\n readonly disabled?: boolean;\n}\n\ninterface PostgresDriverOptions {\n readonly connect: { client: Client } | { pool: PoolType };\n readonly cursor?: PostgresCursorOptions | undefined;\n}\n\nexport type PostgresDriverCreateOptions = Omit<PostgresDriverOptions, 'connect'>;\n\nconst DEFAULT_BATCH_SIZE = 100;\n\ntype ConnectionOptions =\n | { readonly cursorDisabled: true }\n | { readonly cursorBatchSize: number; readonly cursorDisabled: false };\n\nclass AsyncMutex {\n #queue = Promise.resolve();\n\n async lock(): Promise<() => void> {\n const previous = this.#queue;\n let releaseLock: (() => void) | undefined;\n this.#queue = new Promise<void>((resolve) => {\n releaseLock = resolve;\n });\n await previous;\n return () => {\n releaseLock?.();\n releaseLock = undefined;\n };\n }\n}\n\nabstract class PostgresQueryable<C extends PoolClient | Client = PoolClient | Client>\n implements SqlQueryable\n{\n abstract acquireClient(): Promise<C>;\n abstract releaseClient(client: C): Promise<void>;\n\n protected readonly options: ConnectionOptions;\n\n constructor(options: ConnectionOptions) {\n this.options = options;\n }\n\n async *execute<Row = Record<string, unknown>>(request: SqlExecuteRequest): AsyncIterable<Row> {\n const client = await this.acquireClient();\n try {\n if (!this.options.cursorDisabled) {\n try {\n for await (const row of this.executeWithCursor(\n client,\n request.sql,\n request.params,\n this.options.cursorBatchSize,\n )) {\n yield row as Row;\n }\n return;\n } catch (cursorError) {\n if (!(cursorError instanceof Error)) {\n throw cursorError;\n }\n // Check if this is a pg error - if so, normalize and throw\n // Otherwise, fall back to buffered mode for cursor-specific errors\n if (isPostgresError(cursorError)) {\n throw normalizePgError(cursorError);\n }\n // Not a pg error - cursor-specific error, fall back to buffered mode\n }\n }\n\n for await (const row of this.executeBuffered(client, request.sql, request.params)) {\n yield row as Row;\n }\n } catch (error) {\n throw normalizePgError(error);\n } finally {\n await this.releaseClient(client);\n }\n }\n\n async explain(request: SqlExecuteRequest): Promise<SqlExplainResult> {\n // SQL is generated by Prisma Next planners (or validated raw paths), so\n // EXPLAIN prefixing preserves the same statement semantics.\n const text = `EXPLAIN (FORMAT JSON) ${request.sql}`;\n const client = await this.acquireClient();\n try {\n const result = await client\n .query(text, request.params as unknown[] | undefined)\n .catch(rethrowNormalizedError);\n return { rows: result.rows as ReadonlyArray<Record<string, unknown>> };\n } finally {\n await this.releaseClient(client);\n }\n }\n\n async query<Row = Record<string, unknown>>(\n sql: string,\n params?: readonly unknown[],\n ): Promise<SqlQueryResult<Row>> {\n const client = await this.acquireClient();\n try {\n const result = await client\n .query(sql, params as unknown[] | undefined)\n .catch(rethrowNormalizedError);\n return result as unknown as SqlQueryResult<Row>;\n } finally {\n await this.releaseClient(client);\n }\n }\n\n private async *executeWithCursor(\n client: PoolClient | Client,\n sql: string,\n params: readonly unknown[] | undefined,\n cursorBatchSize: number,\n ): AsyncIterable<Record<string, unknown>> {\n const cursor = client.query(new Cursor(sql, params as unknown[] | undefined));\n\n try {\n while (true) {\n const rows = await readCursor(cursor, cursorBatchSize);\n if (rows.length === 0) {\n break;\n }\n\n for (const row of rows) {\n yield row;\n }\n }\n } finally {\n await closeCursor(cursor);\n }\n }\n\n private async *executeBuffered(\n client: PoolClient | Client,\n sql: string,\n params: readonly unknown[] | undefined,\n ): AsyncIterable<Record<string, unknown>> {\n const result = await client.query(sql, params as unknown[] | undefined);\n for (const row of result.rows as Record<string, unknown>[]) {\n yield row;\n }\n }\n}\n\nclass PostgresConnectionImpl extends PostgresQueryable implements SqlConnection {\n #connection: PoolClient | Client;\n #onRelease: (() => void) | undefined;\n #onDestroy: ((reason: unknown) => Promise<void> | void) | undefined;\n\n constructor(\n connection: PoolClient | Client,\n options: ConnectionOptions,\n onRelease?: () => void,\n onDestroy?: (reason: unknown) => Promise<void> | void,\n ) {\n super(options);\n this.#connection = connection;\n this.#onRelease = onRelease;\n this.#onDestroy = onDestroy;\n }\n\n override acquireClient(): Promise<PoolClient | Client> {\n return Promise.resolve(this.#connection);\n }\n\n override releaseClient(_client: PoolClient | Client): Promise<void> {\n return Promise.resolve();\n }\n\n async beginTransaction(): Promise<SqlTransaction> {\n await this.#connection.query('BEGIN').catch(rethrowNormalizedError);\n return new PostgresTransactionImpl(this.#connection, this.options);\n }\n\n async release(): Promise<void> {\n const conn = this.#connection;\n if ('release' in conn) {\n conn.release();\n }\n const onRelease = this.#onRelease;\n this.#onRelease = undefined;\n this.#onDestroy = undefined;\n onRelease?.();\n }\n\n async destroy(reason?: unknown): Promise<void> {\n const onDestroy = this.#onDestroy;\n const onRelease = this.#onRelease;\n this.#onDestroy = undefined;\n this.#onRelease = undefined;\n\n const conn = this.#connection;\n if ('release' in conn) {\n // Pass a truthy Error to pg's PoolClient.release so the pool evicts the\n // client instead of returning it for reuse. A connection that reaches\n // destroy() is in an indeterminate state (failed rollback/commit, etc.)\n // and must not be handed back to another caller. The Error value is\n // surfaced on pg-pool's 'release' event as advisory context; pg-pool\n // itself only uses its truthiness for the eviction decision.\n const releaseArg: Error =\n reason instanceof Error ? reason : new Error('Connection destroyed');\n conn.release(releaseArg);\n }\n\n if (onDestroy) {\n await onDestroy(reason);\n } else {\n onRelease?.();\n }\n }\n}\n\nclass PostgresTransactionImpl extends PostgresQueryable implements SqlTransaction {\n #connection: PoolClient | Client;\n\n constructor(connection: PoolClient | Client, options: ConnectionOptions) {\n super(options);\n this.#connection = connection;\n }\n\n override acquireClient(): Promise<PoolClient | Client> {\n return Promise.resolve(this.#connection);\n }\n\n override releaseClient(_client: PoolClient | Client): Promise<void> {\n return Promise.resolve();\n }\n\n async commit(): Promise<void> {\n await this.#connection.query('COMMIT').catch(rethrowNormalizedError);\n }\n\n async rollback(): Promise<void> {\n await this.#connection.query('ROLLBACK').catch(rethrowNormalizedError);\n }\n}\n\nclass PostgresPoolDriverImpl\n extends PostgresQueryable<PoolClient>\n implements SqlDriver<PostgresBinding>\n{\n private readonly pool: PoolType;\n #closed = false;\n\n constructor(options: PostgresDriverOptions & { connect: { pool: PoolType } }) {\n super({\n cursorBatchSize: options.cursor?.batchSize ?? DEFAULT_BATCH_SIZE,\n cursorDisabled: options.cursor?.disabled ?? false,\n });\n this.pool = options.connect.pool;\n }\n\n get state(): SqlDriverState {\n return this.#closed ? 'closed' : 'connected';\n }\n\n // Bound drivers are created via createBoundDriverFromBinding with pool already\n // configured; connect is intentionally no-op.\n async connect(_binding: PostgresBinding): Promise<void> {}\n\n async acquireConnection(): Promise<SqlConnection> {\n const client = await this.acquireClient();\n return new PostgresConnectionImpl(client, this.options);\n }\n\n async close(): Promise<void> {\n if (this.#closed) {\n return;\n }\n this.#closed = true;\n await this.pool.end();\n }\n\n async acquireClient(): Promise<PoolClient> {\n return this.pool.connect();\n }\n\n async releaseClient(client: PoolClient): Promise<void> {\n client.release();\n }\n}\n\nclass PostgresDirectDriverImpl\n extends PostgresQueryable<Client>\n implements SqlDriver<PostgresBinding>\n{\n private readonly directClient: Client;\n readonly #connectionMutex = new AsyncMutex();\n #closed = false;\n #connected = false;\n #connectPromise: Promise<void> | undefined;\n\n constructor(options: PostgresDriverOptions & { connect: { client: Client } }) {\n super({\n cursorBatchSize: options.cursor?.batchSize ?? DEFAULT_BATCH_SIZE,\n cursorDisabled: options.cursor?.disabled ?? false,\n });\n this.directClient = options.connect.client;\n }\n\n get state(): SqlDriverState {\n return this.#closed ? 'closed' : 'connected';\n }\n\n // Bound drivers are created via createBoundDriverFromBinding with client already\n // configured; connect is intentionally no-op.\n async connect(_binding: PostgresBinding): Promise<void> {}\n\n async acquireConnection(): Promise<SqlConnection> {\n const releaseLease = await this.#connectionMutex.lock();\n try {\n const client = await this.acquireClient();\n return new PostgresConnectionImpl(\n client,\n this.options,\n releaseLease,\n // A direct driver has a single underlying socket, so a destroyed\n // connection means the driver itself is no longer usable. Tear down\n // the driver while still holding the lease so that any\n // acquireConnection() queued on #connectionMutex only observes the\n // driver after .end() has completed and #closed is set — preventing\n // a concurrent caller from reusing a socket that is already being\n // closed. The lease is released in a finally so a failing .end()\n // cannot leave the mutex permanently held.\n async () => {\n try {\n await this.#closeWhileHoldingLease();\n } finally {\n releaseLease();\n }\n },\n );\n } catch (error) {\n releaseLease();\n throw error;\n }\n }\n\n async close(): Promise<void> {\n const releaseLease = await this.#connectionMutex.lock();\n try {\n await this.#closeWhileHoldingLease();\n } finally {\n releaseLease();\n }\n }\n\n async #closeWhileHoldingLease(): Promise<void> {\n if (this.#closed) {\n return;\n }\n this.#closed = true;\n await this.directClient.end();\n this.#connected = false;\n }\n\n async acquireClient(): Promise<Client> {\n if (this.#connected) {\n return this.directClient;\n }\n if (this.#connectPromise !== undefined) {\n await this.#connectPromise;\n return this.directClient;\n }\n\n this.#connectPromise = (async () => {\n try {\n await this.directClient.connect();\n } catch (error: unknown) {\n if (!isAlreadyConnectedError(error)) {\n throw error;\n }\n } finally {\n this.#connectPromise = undefined;\n }\n this.#connected = true;\n })();\n await this.#connectPromise;\n\n return this.directClient;\n }\n\n async releaseClient(_client: Client): Promise<void> {}\n}\n\nexport function createBoundDriverFromBinding(\n binding: PostgresBinding,\n cursorOpts: PostgresDriverCreateOptions['cursor'],\n): SqlDriver<PostgresBinding> {\n switch (binding.kind) {\n case 'url': {\n const pool = new Pool({\n connectionString: binding.url,\n connectionTimeoutMillis: 20_000,\n idleTimeoutMillis: 30_000,\n });\n return new PostgresPoolDriverImpl({\n connect: { pool },\n cursor: cursorOpts,\n });\n }\n case 'pgPool':\n return new PostgresPoolDriverImpl({\n connect: { pool: binding.pool },\n cursor: cursorOpts,\n });\n case 'pgClient':\n return new PostgresDirectDriverImpl({\n connect: { client: binding.client },\n cursor: cursorOpts,\n });\n }\n}\n\nfunction readCursor<Row>(cursor: Cursor<Row>, size: number): Promise<Row[]> {\n return callbackToPromise<Row[]>((cb) => {\n cursor.read(size, (err, rows) => cb(err, rows));\n });\n}\n\nfunction closeCursor(cursor: Cursor<unknown>): Promise<void> {\n return callbackToPromise((cb) => cursor.close(cb));\n}\n\nfunction rethrowNormalizedError(error: unknown): never {\n throw normalizePgError(error);\n}\n","import type {\n RuntimeDriverDescriptor,\n RuntimeDriverInstance,\n} from '@prisma-next/framework-components/execution';\nimport type {\n SqlConnection,\n SqlDriver,\n SqlExecuteRequest,\n SqlExplainResult,\n SqlQueryResult,\n} from '@prisma-next/sql-relational-core/ast';\nimport { postgresDriverDescriptorMeta } from '../core/descriptor-meta';\nimport {\n createBoundDriverFromBinding,\n type PostgresBinding,\n type PostgresDriverCreateOptions,\n} from '../postgres-driver';\n\nexport type PostgresRuntimeDriver = RuntimeDriverInstance<'sql', 'postgres'> &\n SqlDriver<PostgresBinding>;\n\nconst USE_BEFORE_CONNECT_MESSAGE =\n 'Postgres driver not connected. Call connect(binding) before acquireConnection or execute.';\nconst ALREADY_CONNECTED_MESSAGE =\n 'Postgres driver already connected. Call close() before reconnecting with a new binding.';\n\ninterface DriverRuntimeError extends Error {\n readonly code: 'DRIVER.NOT_CONNECTED' | 'DRIVER.ALREADY_CONNECTED';\n readonly category: 'RUNTIME';\n readonly severity: 'error';\n readonly details?: Record<string, unknown>;\n}\n\nfunction driverError(\n code: DriverRuntimeError['code'],\n message: string,\n details?: Record<string, unknown>,\n): DriverRuntimeError {\n const error = new Error(message) as DriverRuntimeError;\n Object.defineProperty(error, 'name', {\n value: 'RuntimeError',\n configurable: true,\n });\n return Object.assign(error, {\n code,\n category: 'RUNTIME' as const,\n severity: 'error' as const,\n message,\n details,\n });\n}\n\nfunction unboundExecute<Row>(): AsyncIterable<Row> {\n return {\n [Symbol.asyncIterator]() {\n return {\n async next() {\n throw driverError('DRIVER.NOT_CONNECTED', USE_BEFORE_CONNECT_MESSAGE);\n },\n };\n },\n };\n}\n\nclass PostgresUnboundDriverImpl implements PostgresRuntimeDriver {\n readonly familyId = 'sql' as const;\n readonly targetId = 'postgres' as const;\n\n #delegate: SqlDriver<PostgresBinding> | null = null;\n #closed = false;\n #cursorOpts: PostgresDriverCreateOptions['cursor'];\n\n constructor(cursorOpts?: PostgresDriverCreateOptions['cursor']) {\n this.#cursorOpts = cursorOpts;\n }\n\n get state(): 'unbound' | 'connected' | 'closed' {\n if (this.#delegate !== null) {\n return 'connected';\n }\n if (this.#closed) {\n return 'closed';\n }\n return 'unbound';\n }\n\n #requireDelegate(): SqlDriver<PostgresBinding> {\n const delegate = this.#delegate;\n if (delegate === null) {\n throw driverError('DRIVER.NOT_CONNECTED', USE_BEFORE_CONNECT_MESSAGE);\n }\n return delegate;\n }\n\n async connect(binding: PostgresBinding): Promise<void> {\n if (this.#delegate !== null) {\n throw driverError('DRIVER.ALREADY_CONNECTED', ALREADY_CONNECTED_MESSAGE, {\n bindingKind: binding.kind,\n });\n }\n this.#delegate = createBoundDriverFromBinding(binding, this.#cursorOpts);\n this.#closed = false;\n }\n\n async acquireConnection(): Promise<SqlConnection> {\n const delegate = this.#requireDelegate();\n const connection = await delegate.acquireConnection();\n return this.#wrapConnection(connection, delegate);\n }\n\n /**\n * Wraps an acquired connection so that teardown paths which close the\n * underlying delegate (notably `destroy()` on a pgClient binding, where\n * the single socket means a destroyed connection invalidates the driver)\n * also reset our own `#delegate` reference. Without this, a failed\n * transaction rollback would leave the outer unbound wrapper reporting\n * `connected` while routing subsequent work to an already-ended delegate.\n */\n #wrapConnection(connection: SqlConnection, delegate: SqlDriver<PostgresBinding>): SqlConnection {\n const syncDelegateState = (): void => {\n if (this.#delegate === delegate && delegate.state === 'closed') {\n this.#delegate = null;\n this.#closed = true;\n }\n };\n const wrapped: SqlConnection = {\n beginTransaction: connection.beginTransaction.bind(connection),\n execute: connection.execute.bind(connection),\n query: connection.query.bind(connection),\n release: async () => {\n try {\n await connection.release();\n } finally {\n syncDelegateState();\n }\n },\n destroy: async (reason?: unknown) => {\n try {\n await connection.destroy(reason);\n } finally {\n syncDelegateState();\n }\n },\n };\n if (connection.explain) {\n wrapped.explain = connection.explain.bind(connection);\n }\n return wrapped;\n }\n\n async close(): Promise<void> {\n const delegate = this.#delegate;\n if (delegate !== null) {\n this.#delegate = null;\n await delegate.close();\n }\n this.#closed = true;\n }\n\n execute<Row = Record<string, unknown>>(request: SqlExecuteRequest): AsyncIterable<Row> {\n const delegate = this.#delegate;\n if (delegate === null) {\n return unboundExecute<Row>();\n }\n return delegate.execute<Row>(request);\n }\n\n async explain(request: SqlExecuteRequest): Promise<SqlExplainResult> {\n const delegate = this.#requireDelegate();\n const explain = delegate.explain;\n if (explain === undefined) {\n throw driverError('DRIVER.NOT_CONNECTED', USE_BEFORE_CONNECT_MESSAGE);\n }\n return explain.call(delegate, request);\n }\n\n async query<Row = Record<string, unknown>>(\n sql: string,\n params?: readonly unknown[],\n ): Promise<SqlQueryResult<Row>> {\n const delegate = this.#requireDelegate();\n return delegate.query<Row>(sql, params);\n }\n}\n\nconst postgresRuntimeDriverDescriptor: RuntimeDriverDescriptor<\n 'sql',\n 'postgres',\n PostgresDriverCreateOptions,\n PostgresRuntimeDriver\n> = {\n ...postgresDriverDescriptorMeta,\n create(options?: PostgresDriverCreateOptions): PostgresRuntimeDriver {\n return new PostgresUnboundDriverImpl(options?.cursor);\n },\n};\n\nexport default postgresRuntimeDriverDescriptor;\nexport type {\n PostgresBinding,\n PostgresDriverCreateOptions,\n QueryResult,\n} from '../postgres-driver';\n"],"mappings":";;;;AAUA,SAAgB,kBACd,IACY;CACZ,OAAO,IAAI,SAAY,SAAS,WAAW;EACzC,IAAI,KAAK,WAAW;GAClB,IAAI,KAAK;IACP,OAAO,IAAI;IACX;;GAEF,QAAQ,OAAY;IACpB;GACF;;;;ACoBJ,MAAM,qBAAqB;AAM3B,IAAM,aAAN,MAAiB;CACf,SAAS,QAAQ,SAAS;CAE1B,MAAM,OAA4B;EAChC,MAAM,WAAW,KAAKA;EACtB,IAAI;EACJ,KAAKA,SAAS,IAAI,SAAe,YAAY;GAC3C,cAAc;IACd;EACF,MAAM;EACN,aAAa;GACX,eAAe;GACf,cAAc,KAAA;;;;AAKpB,IAAe,oBAAf,MAEA;CAIE;CAEA,YAAY,SAA4B;EACtC,KAAK,UAAU;;CAGjB,OAAO,QAAuC,SAAgD;EAC5F,MAAM,SAAS,MAAM,KAAK,eAAe;EACzC,IAAI;GACF,IAAI,CAAC,KAAK,QAAQ,gBAChB,IAAI;IACF,WAAW,MAAM,OAAO,KAAK,kBAC3B,QACA,QAAQ,KACR,QAAQ,QACR,KAAK,QAAQ,gBACd,EACC,MAAM;IAER;YACO,aAAa;IACpB,IAAI,EAAE,uBAAuB,QAC3B,MAAM;IAIR,IAAI,gBAAgB,YAAY,EAC9B,MAAM,iBAAiB,YAAY;;GAMzC,WAAW,MAAM,OAAO,KAAK,gBAAgB,QAAQ,QAAQ,KAAK,QAAQ,OAAO,EAC/E,MAAM;WAED,OAAO;GACd,MAAM,iBAAiB,MAAM;YACrB;GACR,MAAM,KAAK,cAAc,OAAO;;;CAIpC,MAAM,QAAQ,SAAuD;EAGnE,MAAM,OAAO,yBAAyB,QAAQ;EAC9C,MAAM,SAAS,MAAM,KAAK,eAAe;EACzC,IAAI;GAIF,OAAO,EAAE,OAAM,MAHM,OAClB,MAAM,MAAM,QAAQ,OAAgC,CACpD,MAAM,uBAAuB,EACV,MAAgD;YAC9D;GACR,MAAM,KAAK,cAAc,OAAO;;;CAIpC,MAAM,MACJ,KACA,QAC8B;EAC9B,MAAM,SAAS,MAAM,KAAK,eAAe;EACzC,IAAI;GAIF,OAAO,MAHc,OAClB,MAAM,KAAK,OAAgC,CAC3C,MAAM,uBAAuB;YAExB;GACR,MAAM,KAAK,cAAc,OAAO;;;CAIpC,OAAe,kBACb,QACA,KACA,QACA,iBACwC;EACxC,MAAM,SAAS,OAAO,MAAM,IAAI,OAAO,KAAK,OAAgC,CAAC;EAE7E,IAAI;GACF,OAAO,MAAM;IACX,MAAM,OAAO,MAAM,WAAW,QAAQ,gBAAgB;IACtD,IAAI,KAAK,WAAW,GAClB;IAGF,KAAK,MAAM,OAAO,MAChB,MAAM;;YAGF;GACR,MAAM,YAAY,OAAO;;;CAI7B,OAAe,gBACb,QACA,KACA,QACwC;EACxC,MAAM,SAAS,MAAM,OAAO,MAAM,KAAK,OAAgC;EACvE,KAAK,MAAM,OAAO,OAAO,MACvB,MAAM;;;AAKZ,IAAM,yBAAN,cAAqC,kBAA2C;CAC9E;CACA;CACA;CAEA,YACE,YACA,SACA,WACA,WACA;EACA,MAAM,QAAQ;EACd,KAAKC,cAAc;EACnB,KAAKC,aAAa;EAClB,KAAKC,aAAa;;CAGpB,gBAAuD;EACrD,OAAO,QAAQ,QAAQ,KAAKF,YAAY;;CAG1C,cAAuB,SAA6C;EAClE,OAAO,QAAQ,SAAS;;CAG1B,MAAM,mBAA4C;EAChD,MAAM,KAAKA,YAAY,MAAM,QAAQ,CAAC,MAAM,uBAAuB;EACnE,OAAO,IAAI,wBAAwB,KAAKA,aAAa,KAAK,QAAQ;;CAGpE,MAAM,UAAyB;EAC7B,MAAM,OAAO,KAAKA;EAClB,IAAI,aAAa,MACf,KAAK,SAAS;EAEhB,MAAM,YAAY,KAAKC;EACvB,KAAKA,aAAa,KAAA;EAClB,KAAKC,aAAa,KAAA;EAClB,aAAa;;CAGf,MAAM,QAAQ,QAAiC;EAC7C,MAAM,YAAY,KAAKA;EACvB,MAAM,YAAY,KAAKD;EACvB,KAAKC,aAAa,KAAA;EAClB,KAAKD,aAAa,KAAA;EAElB,MAAM,OAAO,KAAKD;EAClB,IAAI,aAAa,MAAM;GAOrB,MAAM,aACJ,kBAAkB,QAAQ,yBAAS,IAAI,MAAM,uBAAuB;GACtE,KAAK,QAAQ,WAAW;;EAG1B,IAAI,WACF,MAAM,UAAU,OAAO;OAEvB,aAAa;;;AAKnB,IAAM,0BAAN,cAAsC,kBAA4C;CAChF;CAEA,YAAY,YAAiC,SAA4B;EACvE,MAAM,QAAQ;EACd,KAAKA,cAAc;;CAGrB,gBAAuD;EACrD,OAAO,QAAQ,QAAQ,KAAKA,YAAY;;CAG1C,cAAuB,SAA6C;EAClE,OAAO,QAAQ,SAAS;;CAG1B,MAAM,SAAwB;EAC5B,MAAM,KAAKA,YAAY,MAAM,SAAS,CAAC,MAAM,uBAAuB;;CAGtE,MAAM,WAA0B;EAC9B,MAAM,KAAKA,YAAY,MAAM,WAAW,CAAC,MAAM,uBAAuB;;;AAI1E,IAAM,yBAAN,cACU,kBAEV;CACE;CACA,UAAU;CAEV,YAAY,SAAkE;EAC5E,MAAM;GACJ,iBAAiB,QAAQ,QAAQ,aAAa;GAC9C,gBAAgB,QAAQ,QAAQ,YAAY;GAC7C,CAAC;EACF,KAAK,OAAO,QAAQ,QAAQ;;CAG9B,IAAI,QAAwB;EAC1B,OAAO,KAAKG,UAAU,WAAW;;CAKnC,MAAM,QAAQ,UAA0C;CAExD,MAAM,oBAA4C;EAEhD,OAAO,IAAI,uBAAuB,MADb,KAAK,eAAe,EACC,KAAK,QAAQ;;CAGzD,MAAM,QAAuB;EAC3B,IAAI,KAAKA,SACP;EAEF,KAAKA,UAAU;EACf,MAAM,KAAK,KAAK,KAAK;;CAGvB,MAAM,gBAAqC;EACzC,OAAO,KAAK,KAAK,SAAS;;CAG5B,MAAM,cAAc,QAAmC;EACrD,OAAO,SAAS;;;AAIpB,IAAM,2BAAN,cACU,kBAEV;CACE;CACA,mBAA4B,IAAI,YAAY;CAC5C,UAAU;CACV,aAAa;CACb;CAEA,YAAY,SAAkE;EAC5E,MAAM;GACJ,iBAAiB,QAAQ,QAAQ,aAAa;GAC9C,gBAAgB,QAAQ,QAAQ,YAAY;GAC7C,CAAC;EACF,KAAK,eAAe,QAAQ,QAAQ;;CAGtC,IAAI,QAAwB;EAC1B,OAAO,KAAKA,UAAU,WAAW;;CAKnC,MAAM,QAAQ,UAA0C;CAExD,MAAM,oBAA4C;EAChD,MAAM,eAAe,MAAM,KAAKC,iBAAiB,MAAM;EACvD,IAAI;GAEF,OAAO,IAAI,uBACT,MAFmB,KAAK,eAAe,EAGvC,KAAK,SACL,cASA,YAAY;IACV,IAAI;KACF,MAAM,KAAKC,yBAAyB;cAC5B;KACR,cAAc;;KAGnB;WACM,OAAO;GACd,cAAc;GACd,MAAM;;;CAIV,MAAM,QAAuB;EAC3B,MAAM,eAAe,MAAM,KAAKD,iBAAiB,MAAM;EACvD,IAAI;GACF,MAAM,KAAKC,yBAAyB;YAC5B;GACR,cAAc;;;CAIlB,MAAMA,0BAAyC;EAC7C,IAAI,KAAKF,SACP;EAEF,KAAKA,UAAU;EACf,MAAM,KAAK,aAAa,KAAK;EAC7B,KAAKG,aAAa;;CAGpB,MAAM,gBAAiC;EACrC,IAAI,KAAKA,YACP,OAAO,KAAK;EAEd,IAAI,KAAKC,oBAAoB,KAAA,GAAW;GACtC,MAAM,KAAKA;GACX,OAAO,KAAK;;EAGd,KAAKA,mBAAmB,YAAY;GAClC,IAAI;IACF,MAAM,KAAK,aAAa,SAAS;YAC1B,OAAgB;IACvB,IAAI,CAAC,wBAAwB,MAAM,EACjC,MAAM;aAEA;IACR,KAAKA,kBAAkB,KAAA;;GAEzB,KAAKD,aAAa;MAChB;EACJ,MAAM,KAAKC;EAEX,OAAO,KAAK;;CAGd,MAAM,cAAc,SAAgC;;AAGtD,SAAgB,6BACd,SACA,YAC4B;CAC5B,QAAQ,QAAQ,MAAhB;EACE,KAAK,OAMH,OAAO,IAAI,uBAAuB;GAChC,SAAS,EAAE,MAAA,IANI,KAAK;IACpB,kBAAkB,QAAQ;IAC1B,yBAAyB;IACzB,mBAAmB;IACpB,CAEgB,EAAE;GACjB,QAAQ;GACT,CAAC;EAEJ,KAAK,UACH,OAAO,IAAI,uBAAuB;GAChC,SAAS,EAAE,MAAM,QAAQ,MAAM;GAC/B,QAAQ;GACT,CAAC;EACJ,KAAK,YACH,OAAO,IAAI,yBAAyB;GAClC,SAAS,EAAE,QAAQ,QAAQ,QAAQ;GACnC,QAAQ;GACT,CAAC;;;AAIR,SAAS,WAAgB,QAAqB,MAA8B;CAC1E,OAAO,mBAA0B,OAAO;EACtC,OAAO,KAAK,OAAO,KAAK,SAAS,GAAG,KAAK,KAAK,CAAC;GAC/C;;AAGJ,SAAS,YAAY,QAAwC;CAC3D,OAAO,mBAAmB,OAAO,OAAO,MAAM,GAAG,CAAC;;AAGpD,SAAS,uBAAuB,OAAuB;CACrD,MAAM,iBAAiB,MAAM;;;;ACvb/B,MAAM,6BACJ;AACF,MAAM,4BACJ;AASF,SAAS,YACP,MACA,SACA,SACoB;CACpB,MAAM,QAAQ,IAAI,MAAM,QAAQ;CAChC,OAAO,eAAe,OAAO,QAAQ;EACnC,OAAO;EACP,cAAc;EACf,CAAC;CACF,OAAO,OAAO,OAAO,OAAO;EAC1B;EACA,UAAU;EACV,UAAU;EACV;EACA;EACD,CAAC;;AAGJ,SAAS,iBAA0C;CACjD,OAAO,EACL,CAAC,OAAO,iBAAiB;EACvB,OAAO,EACL,MAAM,OAAO;GACX,MAAM,YAAY,wBAAwB,2BAA2B;KAExE;IAEJ;;AAGH,IAAM,4BAAN,MAAiE;CAC/D,WAAoB;CACpB,WAAoB;CAEpB,YAA+C;CAC/C,UAAU;CACV;CAEA,YAAY,YAAoD;EAC9D,KAAKC,cAAc;;CAGrB,IAAI,QAA4C;EAC9C,IAAI,KAAKC,cAAc,MACrB,OAAO;EAET,IAAI,KAAKC,SACP,OAAO;EAET,OAAO;;CAGT,mBAA+C;EAC7C,MAAM,WAAW,KAAKD;EACtB,IAAI,aAAa,MACf,MAAM,YAAY,wBAAwB,2BAA2B;EAEvE,OAAO;;CAGT,MAAM,QAAQ,SAAyC;EACrD,IAAI,KAAKA,cAAc,MACrB,MAAM,YAAY,4BAA4B,2BAA2B,EACvE,aAAa,QAAQ,MACtB,CAAC;EAEJ,KAAKA,YAAY,6BAA6B,SAAS,KAAKD,YAAY;EACxE,KAAKE,UAAU;;CAGjB,MAAM,oBAA4C;EAChD,MAAM,WAAW,KAAKC,kBAAkB;EACxC,MAAM,aAAa,MAAM,SAAS,mBAAmB;EACrD,OAAO,KAAKC,gBAAgB,YAAY,SAAS;;;;;;;;;;CAWnD,gBAAgB,YAA2B,UAAqD;EAC9F,MAAM,0BAAgC;GACpC,IAAI,KAAKH,cAAc,YAAY,SAAS,UAAU,UAAU;IAC9D,KAAKA,YAAY;IACjB,KAAKC,UAAU;;;EAGnB,MAAM,UAAyB;GAC7B,kBAAkB,WAAW,iBAAiB,KAAK,WAAW;GAC9D,SAAS,WAAW,QAAQ,KAAK,WAAW;GAC5C,OAAO,WAAW,MAAM,KAAK,WAAW;GACxC,SAAS,YAAY;IACnB,IAAI;KACF,MAAM,WAAW,SAAS;cAClB;KACR,mBAAmB;;;GAGvB,SAAS,OAAO,WAAqB;IACnC,IAAI;KACF,MAAM,WAAW,QAAQ,OAAO;cACxB;KACR,mBAAmB;;;GAGxB;EACD,IAAI,WAAW,SACb,QAAQ,UAAU,WAAW,QAAQ,KAAK,WAAW;EAEvD,OAAO;;CAGT,MAAM,QAAuB;EAC3B,MAAM,WAAW,KAAKD;EACtB,IAAI,aAAa,MAAM;GACrB,KAAKA,YAAY;GACjB,MAAM,SAAS,OAAO;;EAExB,KAAKC,UAAU;;CAGjB,QAAuC,SAAgD;EACrF,MAAM,WAAW,KAAKD;EACtB,IAAI,aAAa,MACf,OAAO,gBAAqB;EAE9B,OAAO,SAAS,QAAa,QAAQ;;CAGvC,MAAM,QAAQ,SAAuD;EACnE,MAAM,WAAW,KAAKE,kBAAkB;EACxC,MAAM,UAAU,SAAS;EACzB,IAAI,YAAY,KAAA,GACd,MAAM,YAAY,wBAAwB,2BAA2B;EAEvE,OAAO,QAAQ,KAAK,UAAU,QAAQ;;CAGxC,MAAM,MACJ,KACA,QAC8B;EAE9B,OADiB,KAAKA,kBACP,CAAC,MAAW,KAAK,OAAO;;;AAI3C,MAAM,kCAKF;CACF,GAAG;CACH,OAAO,SAA8D;EACnE,OAAO,IAAI,0BAA0B,SAAS,OAAO;;CAExD"}
|
|
1
|
+
{"version":3,"file":"runtime.mjs","names":["#queue","#connection","#onRelease","#onDestroy","#closed","#connectionMutex","#closeWhileHoldingLease","#connected","#connectPromise","#cursorOpts","#preparedStatements","#delegate","#closed","#requireDelegate","#wrapConnection"],"sources":["../src/callback-to-promise.ts","../src/named-cursor.ts","../src/postgres-driver.ts","../src/exports/runtime.ts"],"sourcesContent":["/**\n * Wraps a Node-style callback function into a Promise.\n * Supports both (err) => void and (err, result) => void patterns.\n */\nexport function callbackToPromise(\n fn: (callback: (err: Error | null | undefined) => void) => void,\n): Promise<void>;\nexport function callbackToPromise<T>(\n fn: (callback: (err: Error | null | undefined, result: T) => void) => void,\n): Promise<T>;\nexport function callbackToPromise<T = void>(\n fn: (callback: (err: Error | null | undefined, result?: T) => void) => void,\n): Promise<T> {\n return new Promise<T>((resolve, reject) => {\n fn((err, result) => {\n if (err) {\n reject(err);\n return;\n }\n resolve(result as T);\n });\n });\n}\n","import type { BindConfig, Connection, QueryParse } from 'pg';\nimport Cursor from 'pg-cursor';\n\nlet nextNamedPortalId = 1;\n\nexport interface NamedCursorOptions<Row> {\n readonly name: string;\n readonly text: string;\n readonly values: readonly unknown[];\n readonly config?: ConstructorParameters<typeof Cursor<Row>>[2];\n}\n\n// `parsedStatements` is set on the Connection by pg.Client, not by Connection\n// itself, so it isn't declared on @types/pg's `Connection`.\ntype ConnectionWithStatementCache = Connection & {\n readonly parsedStatements: Record<string, string>;\n};\n\n// pg-cursor's runtime instance shape. @types/pg-cursor only declares the\n// constructor + submit/read/close, so the fields pg-cursor sets on `this` and\n// its handlers consume have to be re-declared here.\ninterface CursorInstanceFields {\n state: 'initialized' | 'submitted' | 'idle' | 'busy' | 'done' | 'error';\n connection: Connection | null;\n text: string;\n values: ReadonlyArray<unknown> | null;\n _portal: string;\n _conf: { types?: { getTypeParser: (oid: number, format?: string) => unknown } };\n _result: { _getTypeParser?: unknown };\n _ifNoData: () => void;\n _rowDescription: () => void;\n}\n\n/** Streaming cursor for a server-side named prepared statement. */\nexport class NamedCursor<Row = Record<string, unknown>> extends Cursor<Row> {\n readonly name: string;\n\n constructor(opts: NamedCursorOptions<Row>) {\n super(opts.text, [...opts.values], opts.config);\n this.name = opts.name;\n // Cursor declares `submit` as a property, not a class method, so\n // `override submit() {}` at the class level conflicts. Replace the\n // field on this instance instead.\n this.submit = this.submitNamed;\n }\n\n private submitNamed(connection: Connection): void {\n const self = this as unknown as CursorInstanceFields;\n const conn = connection as ConnectionWithStatementCache;\n\n self.state = 'submitted';\n self.connection = conn;\n self._portal = `np_${nextNamedPortalId++}`;\n\n if (!conn.parsedStatements[this.name]) {\n // QueryParse types `types` as required; pg's runtime accepts an empty\n // array (server infers parameter types).\n const parseMessage: QueryParse = { text: self.text, name: this.name, types: [] };\n conn.parse(parseMessage, true);\n }\n\n // pg-cursor's constructor pre-maps values via prepareValue, so the array\n // here is already wire-ready; the cast bridges pg-cursor's loose `any[]`\n // to BindConfig's stricter union.\n const bindMessage: BindConfig = {\n portal: self._portal,\n statement: this.name,\n values: (self.values ?? []) as BindConfig['values'],\n };\n conn.bind(bindMessage, true);\n\n conn.describe({ type: 'P', name: self._portal }, true);\n conn.flush();\n\n if (self._conf.types) {\n self._result._getTypeParser = self._conf.types.getTypeParser;\n }\n\n conn.once('noData', self._ifNoData);\n conn.once('rowDescription', self._rowDescription);\n }\n}\n","import type {\n PreparedExecuteRequest,\n SqlConnection,\n SqlDriver,\n SqlDriverState,\n SqlExecuteRequest,\n SqlExplainResult,\n SqlQueryable,\n SqlQueryResult,\n SqlTransaction,\n} from '@prisma-next/sql-relational-core/ast';\nimport type {\n Client,\n QueryResult as PgQueryResult,\n PoolClient,\n Pool as PoolType,\n QueryConfig,\n QueryResultRow,\n} from 'pg';\nimport { Pool } from 'pg';\nimport Cursor from 'pg-cursor';\nimport { callbackToPromise } from './callback-to-promise';\nimport { NamedCursor } from './named-cursor';\nimport { isAlreadyConnectedError, isPostgresError, normalizePgError } from './normalize-error';\n\nexport type QueryResult<T extends QueryResultRow = QueryResultRow> = PgQueryResult<T>;\n\nexport type PostgresBinding =\n | { readonly kind: 'url'; readonly url: string }\n | { readonly kind: 'pgPool'; readonly pool: PoolType }\n | { readonly kind: 'pgClient'; readonly client: Client };\n\nexport interface PostgresCursorOptions {\n readonly batchSize?: number;\n readonly disabled?: boolean;\n}\n\ninterface PostgresDriverOptions {\n readonly connect: { client: Client } | { pool: PoolType };\n readonly cursor?: PostgresCursorOptions | undefined;\n /**\n * Use server-side prepared statements for `executePrepared`. Default\n * `true`. Set `false` when running behind a transaction-mode pooler that\n * does not support session-scoped prepared statements (e.g. PgBouncer\n * transaction mode).\n */\n readonly preparedStatements?: boolean | undefined;\n}\n\nexport type PostgresDriverCreateOptions = Omit<PostgresDriverOptions, 'connect'>;\n\nconst DEFAULT_BATCH_SIZE = 100;\nconst DEFAULT_PREPARED_STATEMENTS = true;\n\ntype CursorOptions =\n | { readonly cursorDisabled: true }\n | { readonly cursorBatchSize: number; readonly cursorDisabled: false };\n\ninterface HandleAllocator {\n mint(): string;\n}\n\nfunction createHandleAllocator(): HandleAllocator {\n // Driver-scoped so a single PreparedStatement reused across multiple\n // Connections/Transactions of the same driver always sees one handle.\n let next = 1;\n return {\n mint: () => `pn_${next++}`,\n };\n}\n\ntype ConnectionOptions = CursorOptions & {\n readonly preparedStatementsEnabled: boolean;\n readonly handleAllocator: HandleAllocator;\n};\n\nfunction buildConnectionOptions(options: PostgresDriverOptions): ConnectionOptions {\n const cursorOptions: CursorOptions = options.cursor?.disabled\n ? { cursorDisabled: true }\n : {\n cursorBatchSize: options.cursor?.batchSize ?? DEFAULT_BATCH_SIZE,\n cursorDisabled: false,\n };\n return {\n ...cursorOptions,\n preparedStatementsEnabled: options.preparedStatements ?? DEFAULT_PREPARED_STATEMENTS,\n handleAllocator: createHandleAllocator(),\n };\n}\n\nfunction isStalePreparedStatementError(error: unknown): boolean {\n if (!(error instanceof Error)) {\n return false;\n }\n const code = (error as { code?: string }).code;\n // 26000 — invalid_sql_statement_name (server lost the named statement,\n // e.g. after DEALLOCATE ALL).\n // 0A000 — cached plan invalidated by DDL (row shape changed).\n return code === '26000' || code === '0A000';\n}\n\nclass AsyncMutex {\n #queue = Promise.resolve();\n\n async lock(): Promise<() => void> {\n const previous = this.#queue;\n let releaseLock: (() => void) | undefined;\n this.#queue = new Promise<void>((resolve) => {\n releaseLock = resolve;\n });\n await previous;\n return () => {\n releaseLock?.();\n releaseLock = undefined;\n };\n }\n}\n\nabstract class PostgresQueryable<C extends PoolClient | Client = PoolClient | Client>\n implements SqlQueryable\n{\n abstract acquireClient(): Promise<C>;\n abstract releaseClient(client: C): Promise<void>;\n\n protected readonly options: ConnectionOptions;\n\n constructor(options: ConnectionOptions) {\n this.options = options;\n }\n\n async *execute<Row = Record<string, unknown>>(request: SqlExecuteRequest): AsyncIterable<Row> {\n try {\n yield* this.runQuery<Row>(request);\n } catch (error) {\n throw normalizePgError(error);\n }\n }\n\n async *executePrepared<Row = Record<string, unknown>>(\n request: PreparedExecuteRequest,\n ): AsyncIterable<Row> {\n if (!this.options.preparedStatementsEnabled) {\n // Skip server-side prepare entirely; route as a regular ad-hoc execute.\n yield* this.execute<Row>(request);\n return;\n }\n\n let handle = request.handle.get();\n if (handle === undefined) {\n handle = this.options.handleAllocator.mint();\n request.handle.set(handle);\n }\n\n yield* this.withStaleHandleRetry<Row>(request, handle as string, (h) =>\n this.runQuery<Row>(request, h),\n );\n }\n\n private async *withStaleHandleRetry<Row>(\n request: PreparedExecuteRequest,\n handle: string,\n attempt: (handle: string) => AsyncIterable<Row>,\n ): AsyncIterable<Row> {\n let yielded = false;\n try {\n for await (const row of attempt(handle)) {\n yielded = true;\n yield row;\n }\n return;\n } catch (error) {\n // If a row was already yielded, the error came from mid-stream and a\n // retry would re-yield prior rows to the consumer.\n if (yielded || !isStalePreparedStatementError(error)) {\n throw normalizePgError(error);\n }\n // pg's parsedStatements still records the old name; only a fresh name\n // forces pg to re-Parse on this Client.\n const retryHandle = this.options.handleAllocator.mint();\n request.handle.set(retryHandle);\n try {\n yield* attempt(retryHandle);\n } catch (retryError) {\n throw normalizePgError(retryError);\n }\n }\n }\n\n // Errors propagate as raw pg errors so the caller can read SQLSTATE before\n // normalising (the prepared-statement retry layer needs the original code).\n private async *runQuery<Row>(request: SqlExecuteRequest, name?: string): AsyncIterable<Row> {\n const client = await this.acquireClient();\n try {\n if (!this.options.cursorDisabled) {\n try {\n for await (const row of this.executeWithCursor(\n client,\n request.sql,\n request.params,\n this.options.cursorBatchSize,\n name,\n )) {\n yield row as Row;\n }\n return;\n } catch (cursorError) {\n // Non-pg cursor-specific errors fall through to the buffered path;\n // real pg errors propagate.\n if (!(cursorError instanceof Error) || isPostgresError(cursorError)) {\n throw cursorError;\n }\n }\n }\n\n for await (const row of this.executeBuffered(client, request.sql, request.params, name)) {\n yield row as Row;\n }\n } finally {\n await this.releaseClient(client);\n }\n }\n\n async explain(request: SqlExecuteRequest): Promise<SqlExplainResult> {\n // SQL is generated by Prisma Next planners (or validated raw paths), so\n // EXPLAIN prefixing preserves the same statement semantics.\n const text = `EXPLAIN (FORMAT JSON) ${request.sql}`;\n const client = await this.acquireClient();\n try {\n const result = await client\n .query(text, request.params as unknown[] | undefined)\n .catch(rethrowNormalizedError);\n return { rows: result.rows as ReadonlyArray<Record<string, unknown>> };\n } finally {\n await this.releaseClient(client);\n }\n }\n\n async query<Row = Record<string, unknown>>(\n sql: string,\n params?: readonly unknown[],\n ): Promise<SqlQueryResult<Row>> {\n const client = await this.acquireClient();\n try {\n const result = await client\n .query(sql, params as unknown[] | undefined)\n .catch(rethrowNormalizedError);\n return result as unknown as SqlQueryResult<Row>;\n } finally {\n await this.releaseClient(client);\n }\n }\n\n private async *executeWithCursor(\n client: PoolClient | Client,\n sql: string,\n params: readonly unknown[] | undefined,\n cursorBatchSize: number,\n name?: string,\n ): AsyncIterable<Record<string, unknown>> {\n const values = (params ?? []) as unknown[];\n const cursor = client.query(\n name === undefined ? new Cursor(sql, values) : new NamedCursor({ name, text: sql, values }),\n );\n\n try {\n while (true) {\n const rows = await readCursor(cursor, cursorBatchSize);\n if (rows.length === 0) {\n break;\n }\n\n for (const row of rows) {\n yield row;\n }\n }\n } finally {\n await closeCursor(cursor);\n }\n }\n\n private async *executeBuffered(\n client: PoolClient | Client,\n sql: string,\n params: readonly unknown[] | undefined,\n name?: string,\n ): AsyncIterable<Record<string, unknown>> {\n const config: QueryConfig = { name, text: sql, values: (params ?? []) as unknown[] };\n const result = await client.query(config);\n for (const row of result.rows as Record<string, unknown>[]) {\n yield row;\n }\n }\n}\n\nclass PostgresConnectionImpl extends PostgresQueryable implements SqlConnection {\n #connection: PoolClient | Client;\n #onRelease: (() => void) | undefined;\n #onDestroy: ((reason: unknown) => Promise<void> | void) | undefined;\n\n constructor(\n connection: PoolClient | Client,\n options: ConnectionOptions,\n onRelease?: () => void,\n onDestroy?: (reason: unknown) => Promise<void> | void,\n ) {\n super(options);\n this.#connection = connection;\n this.#onRelease = onRelease;\n this.#onDestroy = onDestroy;\n }\n\n override acquireClient(): Promise<PoolClient | Client> {\n return Promise.resolve(this.#connection);\n }\n\n override releaseClient(_client: PoolClient | Client): Promise<void> {\n return Promise.resolve();\n }\n\n async beginTransaction(): Promise<SqlTransaction> {\n await this.#connection.query('BEGIN').catch(rethrowNormalizedError);\n return new PostgresTransactionImpl(this.#connection, this.options);\n }\n\n async release(): Promise<void> {\n const conn = this.#connection;\n if ('release' in conn) {\n conn.release();\n }\n const onRelease = this.#onRelease;\n this.#onRelease = undefined;\n this.#onDestroy = undefined;\n onRelease?.();\n }\n\n async destroy(reason?: unknown): Promise<void> {\n const onDestroy = this.#onDestroy;\n const onRelease = this.#onRelease;\n this.#onDestroy = undefined;\n this.#onRelease = undefined;\n\n const conn = this.#connection;\n if ('release' in conn) {\n // Pass a truthy Error to pg's PoolClient.release so the pool evicts the\n // client instead of returning it for reuse. A connection that reaches\n // destroy() is in an indeterminate state (failed rollback/commit, etc.)\n // and must not be handed back to another caller. The Error value is\n // surfaced on pg-pool's 'release' event as advisory context; pg-pool\n // itself only uses its truthiness for the eviction decision.\n const releaseArg: Error =\n reason instanceof Error ? reason : new Error('Connection destroyed');\n conn.release(releaseArg);\n }\n\n if (onDestroy) {\n await onDestroy(reason);\n } else {\n onRelease?.();\n }\n }\n}\n\nclass PostgresTransactionImpl extends PostgresQueryable implements SqlTransaction {\n #connection: PoolClient | Client;\n\n constructor(connection: PoolClient | Client, options: ConnectionOptions) {\n super(options);\n this.#connection = connection;\n }\n\n override acquireClient(): Promise<PoolClient | Client> {\n return Promise.resolve(this.#connection);\n }\n\n override releaseClient(_client: PoolClient | Client): Promise<void> {\n return Promise.resolve();\n }\n\n async commit(): Promise<void> {\n await this.#connection.query('COMMIT').catch(rethrowNormalizedError);\n }\n\n async rollback(): Promise<void> {\n await this.#connection.query('ROLLBACK').catch(rethrowNormalizedError);\n }\n}\n\nclass PostgresPoolDriverImpl\n extends PostgresQueryable<PoolClient>\n implements SqlDriver<PostgresBinding>\n{\n private readonly pool: PoolType;\n #closed = false;\n\n constructor(options: PostgresDriverOptions & { connect: { pool: PoolType } }) {\n super(buildConnectionOptions(options));\n this.pool = options.connect.pool;\n }\n\n get state(): SqlDriverState {\n return this.#closed ? 'closed' : 'connected';\n }\n\n // Bound drivers are created via createBoundDriverFromBinding with pool already\n // configured; connect is intentionally no-op.\n async connect(_binding: PostgresBinding): Promise<void> {}\n\n async acquireConnection(): Promise<SqlConnection> {\n const client = await this.acquireClient();\n return new PostgresConnectionImpl(client, this.options);\n }\n\n async close(): Promise<void> {\n if (this.#closed) {\n return;\n }\n this.#closed = true;\n await this.pool.end();\n }\n\n async acquireClient(): Promise<PoolClient> {\n return this.pool.connect();\n }\n\n async releaseClient(client: PoolClient): Promise<void> {\n client.release();\n }\n}\n\nclass PostgresDirectDriverImpl\n extends PostgresQueryable<Client>\n implements SqlDriver<PostgresBinding>\n{\n private readonly directClient: Client;\n readonly #connectionMutex = new AsyncMutex();\n #closed = false;\n #connected = false;\n #connectPromise: Promise<void> | undefined;\n\n constructor(options: PostgresDriverOptions & { connect: { client: Client } }) {\n super(buildConnectionOptions(options));\n this.directClient = options.connect.client;\n }\n\n get state(): SqlDriverState {\n return this.#closed ? 'closed' : 'connected';\n }\n\n // Bound drivers are created via createBoundDriverFromBinding with client already\n // configured; connect is intentionally no-op.\n async connect(_binding: PostgresBinding): Promise<void> {}\n\n async acquireConnection(): Promise<SqlConnection> {\n const releaseLease = await this.#connectionMutex.lock();\n try {\n const client = await this.acquireClient();\n return new PostgresConnectionImpl(\n client,\n this.options,\n releaseLease,\n // A direct driver has a single underlying socket, so a destroyed\n // connection means the driver itself is no longer usable. Tear down\n // the driver while still holding the lease so that any\n // acquireConnection() queued on #connectionMutex only observes the\n // driver after .end() has completed and #closed is set — preventing\n // a concurrent caller from reusing a socket that is already being\n // closed. The lease is released in a finally so a failing .end()\n // cannot leave the mutex permanently held.\n async () => {\n try {\n await this.#closeWhileHoldingLease();\n } finally {\n releaseLease();\n }\n },\n );\n } catch (error) {\n releaseLease();\n throw error;\n }\n }\n\n async close(): Promise<void> {\n const releaseLease = await this.#connectionMutex.lock();\n try {\n await this.#closeWhileHoldingLease();\n } finally {\n releaseLease();\n }\n }\n\n async #closeWhileHoldingLease(): Promise<void> {\n if (this.#closed) {\n return;\n }\n this.#closed = true;\n await this.directClient.end();\n this.#connected = false;\n }\n\n async acquireClient(): Promise<Client> {\n if (this.#connected) {\n return this.directClient;\n }\n if (this.#connectPromise !== undefined) {\n await this.#connectPromise;\n return this.directClient;\n }\n\n this.#connectPromise = (async () => {\n try {\n await this.directClient.connect();\n } catch (error: unknown) {\n if (!isAlreadyConnectedError(error)) {\n throw error;\n }\n } finally {\n this.#connectPromise = undefined;\n }\n this.#connected = true;\n })();\n await this.#connectPromise;\n\n return this.directClient;\n }\n\n async releaseClient(_client: Client): Promise<void> {}\n}\n\nexport function createBoundDriverFromBinding(\n binding: PostgresBinding,\n cursorOpts: PostgresDriverCreateOptions['cursor'],\n extraOpts?: Pick<PostgresDriverCreateOptions, 'preparedStatements'>,\n): SqlDriver<PostgresBinding> {\n const preparedStatements = extraOpts?.preparedStatements;\n switch (binding.kind) {\n case 'url': {\n const pool = new Pool({\n connectionString: binding.url,\n connectionTimeoutMillis: 20_000,\n idleTimeoutMillis: 30_000,\n });\n return new PostgresPoolDriverImpl({\n connect: { pool },\n cursor: cursorOpts,\n preparedStatements,\n });\n }\n case 'pgPool':\n return new PostgresPoolDriverImpl({\n connect: { pool: binding.pool },\n cursor: cursorOpts,\n preparedStatements,\n });\n case 'pgClient':\n return new PostgresDirectDriverImpl({\n connect: { client: binding.client },\n cursor: cursorOpts,\n preparedStatements,\n });\n }\n}\n\nfunction readCursor<Row>(cursor: Cursor<Row>, size: number): Promise<Row[]> {\n return callbackToPromise<Row[]>((cb) => {\n cursor.read(size, (err, rows) => cb(err, rows));\n });\n}\n\nfunction closeCursor(cursor: Cursor<unknown>): Promise<void> {\n return callbackToPromise((cb) => cursor.close(cb));\n}\n\nfunction rethrowNormalizedError(error: unknown): never {\n throw normalizePgError(error);\n}\n","import type {\n RuntimeDriverDescriptor,\n RuntimeDriverInstance,\n} from '@prisma-next/framework-components/execution';\nimport type {\n PreparedExecuteRequest,\n SqlConnection,\n SqlDriver,\n SqlExecuteRequest,\n SqlExplainResult,\n SqlQueryResult,\n} from '@prisma-next/sql-relational-core/ast';\nimport { postgresDriverDescriptorMeta } from '../core/descriptor-meta';\nimport {\n createBoundDriverFromBinding,\n type PostgresBinding,\n type PostgresDriverCreateOptions,\n} from '../postgres-driver';\n\nexport type PostgresRuntimeDriver = RuntimeDriverInstance<'sql', 'postgres'> &\n SqlDriver<PostgresBinding>;\n\nconst USE_BEFORE_CONNECT_MESSAGE =\n 'Postgres driver not connected. Call connect(binding) before acquireConnection or execute.';\nconst ALREADY_CONNECTED_MESSAGE =\n 'Postgres driver already connected. Call close() before reconnecting with a new binding.';\n\ninterface DriverRuntimeError extends Error {\n readonly code: 'DRIVER.NOT_CONNECTED' | 'DRIVER.ALREADY_CONNECTED';\n readonly category: 'RUNTIME';\n readonly severity: 'error';\n readonly details?: Record<string, unknown>;\n}\n\nfunction driverError(\n code: DriverRuntimeError['code'],\n message: string,\n details?: Record<string, unknown>,\n): DriverRuntimeError {\n const error = new Error(message) as DriverRuntimeError;\n Object.defineProperty(error, 'name', {\n value: 'RuntimeError',\n configurable: true,\n });\n return Object.assign(error, {\n code,\n category: 'RUNTIME' as const,\n severity: 'error' as const,\n message,\n details,\n });\n}\n\nfunction unboundExecute<Row>(): AsyncIterable<Row> {\n return {\n [Symbol.asyncIterator]() {\n return {\n async next() {\n throw driverError('DRIVER.NOT_CONNECTED', USE_BEFORE_CONNECT_MESSAGE);\n },\n };\n },\n };\n}\n\nclass PostgresUnboundDriverImpl implements PostgresRuntimeDriver {\n readonly familyId = 'sql' as const;\n readonly targetId = 'postgres' as const;\n\n #delegate: SqlDriver<PostgresBinding> | null = null;\n #closed = false;\n #cursorOpts: PostgresDriverCreateOptions['cursor'];\n #preparedStatements: PostgresDriverCreateOptions['preparedStatements'];\n\n constructor(options?: PostgresDriverCreateOptions) {\n this.#cursorOpts = options?.cursor;\n this.#preparedStatements = options?.preparedStatements;\n }\n\n get state(): 'unbound' | 'connected' | 'closed' {\n if (this.#delegate !== null) {\n return 'connected';\n }\n if (this.#closed) {\n return 'closed';\n }\n return 'unbound';\n }\n\n #requireDelegate(): SqlDriver<PostgresBinding> {\n const delegate = this.#delegate;\n if (delegate === null) {\n throw driverError('DRIVER.NOT_CONNECTED', USE_BEFORE_CONNECT_MESSAGE);\n }\n return delegate;\n }\n\n async connect(binding: PostgresBinding): Promise<void> {\n if (this.#delegate !== null) {\n throw driverError('DRIVER.ALREADY_CONNECTED', ALREADY_CONNECTED_MESSAGE, {\n bindingKind: binding.kind,\n });\n }\n this.#delegate = createBoundDriverFromBinding(binding, this.#cursorOpts, {\n preparedStatements: this.#preparedStatements,\n });\n this.#closed = false;\n }\n\n async acquireConnection(): Promise<SqlConnection> {\n const delegate = this.#requireDelegate();\n const connection = await delegate.acquireConnection();\n return this.#wrapConnection(connection, delegate);\n }\n\n /**\n * Wraps an acquired connection so that teardown paths which close the\n * underlying delegate (notably `destroy()` on a pgClient binding, where\n * the single socket means a destroyed connection invalidates the driver)\n * also reset our own `#delegate` reference. Without this, a failed\n * transaction rollback would leave the outer unbound wrapper reporting\n * `connected` while routing subsequent work to an already-ended delegate.\n */\n #wrapConnection(connection: SqlConnection, delegate: SqlDriver<PostgresBinding>): SqlConnection {\n const syncDelegateState = (): void => {\n if (this.#delegate === delegate && delegate.state === 'closed') {\n this.#delegate = null;\n this.#closed = true;\n }\n };\n const wrapped: SqlConnection = {\n beginTransaction: connection.beginTransaction.bind(connection),\n execute: connection.execute.bind(connection),\n executePrepared: connection.executePrepared.bind(connection),\n query: connection.query.bind(connection),\n release: async () => {\n try {\n await connection.release();\n } finally {\n syncDelegateState();\n }\n },\n destroy: async (reason?: unknown) => {\n try {\n await connection.destroy(reason);\n } finally {\n syncDelegateState();\n }\n },\n };\n if (connection.explain) {\n wrapped.explain = connection.explain.bind(connection);\n }\n return wrapped;\n }\n\n async close(): Promise<void> {\n const delegate = this.#delegate;\n if (delegate !== null) {\n this.#delegate = null;\n await delegate.close();\n }\n this.#closed = true;\n }\n\n execute<Row = Record<string, unknown>>(request: SqlExecuteRequest): AsyncIterable<Row> {\n const delegate = this.#delegate;\n if (delegate === null) {\n return unboundExecute<Row>();\n }\n return delegate.execute<Row>(request);\n }\n\n executePrepared<Row = Record<string, unknown>>(\n request: PreparedExecuteRequest,\n ): AsyncIterable<Row> {\n const delegate = this.#delegate;\n if (delegate === null) {\n return unboundExecute<Row>();\n }\n return delegate.executePrepared<Row>(request);\n }\n\n async explain(request: SqlExecuteRequest): Promise<SqlExplainResult> {\n const delegate = this.#requireDelegate();\n const explain = delegate.explain;\n if (explain === undefined) {\n throw driverError('DRIVER.NOT_CONNECTED', USE_BEFORE_CONNECT_MESSAGE);\n }\n return explain.call(delegate, request);\n }\n\n async query<Row = Record<string, unknown>>(\n sql: string,\n params?: readonly unknown[],\n ): Promise<SqlQueryResult<Row>> {\n const delegate = this.#requireDelegate();\n return delegate.query<Row>(sql, params);\n }\n}\n\nconst postgresRuntimeDriverDescriptor: RuntimeDriverDescriptor<\n 'sql',\n 'postgres',\n PostgresDriverCreateOptions,\n PostgresRuntimeDriver\n> = {\n ...postgresDriverDescriptorMeta,\n create(options?: PostgresDriverCreateOptions): PostgresRuntimeDriver {\n return new PostgresUnboundDriverImpl(options);\n },\n};\n\nexport default postgresRuntimeDriverDescriptor;\nexport type {\n PostgresBinding,\n PostgresDriverCreateOptions,\n QueryResult,\n} from '../postgres-driver';\n"],"mappings":";;;;AAUA,SAAgB,kBACd,IACY;CACZ,OAAO,IAAI,SAAY,SAAS,WAAW;EACzC,IAAI,KAAK,WAAW;GAClB,IAAI,KAAK;IACP,OAAO,IAAI;IACX;;GAEF,QAAQ,OAAY;IACpB;GACF;;;;AClBJ,IAAI,oBAAoB;;AA+BxB,IAAa,cAAb,cAAgE,OAAY;CAC1E;CAEA,YAAY,MAA+B;EACzC,MAAM,KAAK,MAAM,CAAC,GAAG,KAAK,OAAO,EAAE,KAAK,OAAO;EAC/C,KAAK,OAAO,KAAK;EAIjB,KAAK,SAAS,KAAK;;CAGrB,YAAoB,YAA8B;EAChD,MAAM,OAAO;EACb,MAAM,OAAO;EAEb,KAAK,QAAQ;EACb,KAAK,aAAa;EAClB,KAAK,UAAU,MAAM;EAErB,IAAI,CAAC,KAAK,iBAAiB,KAAK,OAAO;GAGrC,MAAM,eAA2B;IAAE,MAAM,KAAK;IAAM,MAAM,KAAK;IAAM,OAAO,EAAE;IAAE;GAChF,KAAK,MAAM,cAAc,KAAK;;EAMhC,MAAM,cAA0B;GAC9B,QAAQ,KAAK;GACb,WAAW,KAAK;GAChB,QAAS,KAAK,UAAU,EAAE;GAC3B;EACD,KAAK,KAAK,aAAa,KAAK;EAE5B,KAAK,SAAS;GAAE,MAAM;GAAK,MAAM,KAAK;GAAS,EAAE,KAAK;EACtD,KAAK,OAAO;EAEZ,IAAI,KAAK,MAAM,OACb,KAAK,QAAQ,iBAAiB,KAAK,MAAM,MAAM;EAGjD,KAAK,KAAK,UAAU,KAAK,UAAU;EACnC,KAAK,KAAK,kBAAkB,KAAK,gBAAgB;;;;;AC5BrD,MAAM,qBAAqB;AAC3B,MAAM,8BAA8B;AAUpC,SAAS,wBAAyC;CAGhD,IAAI,OAAO;CACX,OAAO,EACL,YAAY,MAAM,UACnB;;AAQH,SAAS,uBAAuB,SAAmD;CAOjF,OAAO;EACL,GAPmC,QAAQ,QAAQ,WACjD,EAAE,gBAAgB,MAAM,GACxB;GACE,iBAAiB,QAAQ,QAAQ,aAAa;GAC9C,gBAAgB;GACjB;EAGH,2BAA2B,QAAQ,sBAAsB;EACzD,iBAAiB,uBAAuB;EACzC;;AAGH,SAAS,8BAA8B,OAAyB;CAC9D,IAAI,EAAE,iBAAiB,QACrB,OAAO;CAET,MAAM,OAAQ,MAA4B;CAI1C,OAAO,SAAS,WAAW,SAAS;;AAGtC,IAAM,aAAN,MAAiB;CACf,SAAS,QAAQ,SAAS;CAE1B,MAAM,OAA4B;EAChC,MAAM,WAAW,KAAKA;EACtB,IAAI;EACJ,KAAKA,SAAS,IAAI,SAAe,YAAY;GAC3C,cAAc;IACd;EACF,MAAM;EACN,aAAa;GACX,eAAe;GACf,cAAc,KAAA;;;;AAKpB,IAAe,oBAAf,MAEA;CAIE;CAEA,YAAY,SAA4B;EACtC,KAAK,UAAU;;CAGjB,OAAO,QAAuC,SAAgD;EAC5F,IAAI;GACF,OAAO,KAAK,SAAc,QAAQ;WAC3B,OAAO;GACd,MAAM,iBAAiB,MAAM;;;CAIjC,OAAO,gBACL,SACoB;EACpB,IAAI,CAAC,KAAK,QAAQ,2BAA2B;GAE3C,OAAO,KAAK,QAAa,QAAQ;GACjC;;EAGF,IAAI,SAAS,QAAQ,OAAO,KAAK;EACjC,IAAI,WAAW,KAAA,GAAW;GACxB,SAAS,KAAK,QAAQ,gBAAgB,MAAM;GAC5C,QAAQ,OAAO,IAAI,OAAO;;EAG5B,OAAO,KAAK,qBAA0B,SAAS,SAAmB,MAChE,KAAK,SAAc,SAAS,EAAE,CAC/B;;CAGH,OAAe,qBACb,SACA,QACA,SACoB;EACpB,IAAI,UAAU;EACd,IAAI;GACF,WAAW,MAAM,OAAO,QAAQ,OAAO,EAAE;IACvC,UAAU;IACV,MAAM;;GAER;WACO,OAAO;GAGd,IAAI,WAAW,CAAC,8BAA8B,MAAM,EAClD,MAAM,iBAAiB,MAAM;GAI/B,MAAM,cAAc,KAAK,QAAQ,gBAAgB,MAAM;GACvD,QAAQ,OAAO,IAAI,YAAY;GAC/B,IAAI;IACF,OAAO,QAAQ,YAAY;YACpB,YAAY;IACnB,MAAM,iBAAiB,WAAW;;;;CAOxC,OAAe,SAAc,SAA4B,MAAmC;EAC1F,MAAM,SAAS,MAAM,KAAK,eAAe;EACzC,IAAI;GACF,IAAI,CAAC,KAAK,QAAQ,gBAChB,IAAI;IACF,WAAW,MAAM,OAAO,KAAK,kBAC3B,QACA,QAAQ,KACR,QAAQ,QACR,KAAK,QAAQ,iBACb,KACD,EACC,MAAM;IAER;YACO,aAAa;IAGpB,IAAI,EAAE,uBAAuB,UAAU,gBAAgB,YAAY,EACjE,MAAM;;GAKZ,WAAW,MAAM,OAAO,KAAK,gBAAgB,QAAQ,QAAQ,KAAK,QAAQ,QAAQ,KAAK,EACrF,MAAM;YAEA;GACR,MAAM,KAAK,cAAc,OAAO;;;CAIpC,MAAM,QAAQ,SAAuD;EAGnE,MAAM,OAAO,yBAAyB,QAAQ;EAC9C,MAAM,SAAS,MAAM,KAAK,eAAe;EACzC,IAAI;GAIF,OAAO,EAAE,OAAM,MAHM,OAClB,MAAM,MAAM,QAAQ,OAAgC,CACpD,MAAM,uBAAuB,EACV,MAAgD;YAC9D;GACR,MAAM,KAAK,cAAc,OAAO;;;CAIpC,MAAM,MACJ,KACA,QAC8B;EAC9B,MAAM,SAAS,MAAM,KAAK,eAAe;EACzC,IAAI;GAIF,OAAO,MAHc,OAClB,MAAM,KAAK,OAAgC,CAC3C,MAAM,uBAAuB;YAExB;GACR,MAAM,KAAK,cAAc,OAAO;;;CAIpC,OAAe,kBACb,QACA,KACA,QACA,iBACA,MACwC;EACxC,MAAM,SAAU,UAAU,EAAE;EAC5B,MAAM,SAAS,OAAO,MACpB,SAAS,KAAA,IAAY,IAAI,OAAO,KAAK,OAAO,GAAG,IAAI,YAAY;GAAE;GAAM,MAAM;GAAK;GAAQ,CAAC,CAC5F;EAED,IAAI;GACF,OAAO,MAAM;IACX,MAAM,OAAO,MAAM,WAAW,QAAQ,gBAAgB;IACtD,IAAI,KAAK,WAAW,GAClB;IAGF,KAAK,MAAM,OAAO,MAChB,MAAM;;YAGF;GACR,MAAM,YAAY,OAAO;;;CAI7B,OAAe,gBACb,QACA,KACA,QACA,MACwC;EACxC,MAAM,SAAsB;GAAE;GAAM,MAAM;GAAK,QAAS,UAAU,EAAE;GAAgB;EACpF,MAAM,SAAS,MAAM,OAAO,MAAM,OAAO;EACzC,KAAK,MAAM,OAAO,OAAO,MACvB,MAAM;;;AAKZ,IAAM,yBAAN,cAAqC,kBAA2C;CAC9E;CACA;CACA;CAEA,YACE,YACA,SACA,WACA,WACA;EACA,MAAM,QAAQ;EACd,KAAKC,cAAc;EACnB,KAAKC,aAAa;EAClB,KAAKC,aAAa;;CAGpB,gBAAuD;EACrD,OAAO,QAAQ,QAAQ,KAAKF,YAAY;;CAG1C,cAAuB,SAA6C;EAClE,OAAO,QAAQ,SAAS;;CAG1B,MAAM,mBAA4C;EAChD,MAAM,KAAKA,YAAY,MAAM,QAAQ,CAAC,MAAM,uBAAuB;EACnE,OAAO,IAAI,wBAAwB,KAAKA,aAAa,KAAK,QAAQ;;CAGpE,MAAM,UAAyB;EAC7B,MAAM,OAAO,KAAKA;EAClB,IAAI,aAAa,MACf,KAAK,SAAS;EAEhB,MAAM,YAAY,KAAKC;EACvB,KAAKA,aAAa,KAAA;EAClB,KAAKC,aAAa,KAAA;EAClB,aAAa;;CAGf,MAAM,QAAQ,QAAiC;EAC7C,MAAM,YAAY,KAAKA;EACvB,MAAM,YAAY,KAAKD;EACvB,KAAKC,aAAa,KAAA;EAClB,KAAKD,aAAa,KAAA;EAElB,MAAM,OAAO,KAAKD;EAClB,IAAI,aAAa,MAAM;GAOrB,MAAM,aACJ,kBAAkB,QAAQ,yBAAS,IAAI,MAAM,uBAAuB;GACtE,KAAK,QAAQ,WAAW;;EAG1B,IAAI,WACF,MAAM,UAAU,OAAO;OAEvB,aAAa;;;AAKnB,IAAM,0BAAN,cAAsC,kBAA4C;CAChF;CAEA,YAAY,YAAiC,SAA4B;EACvE,MAAM,QAAQ;EACd,KAAKA,cAAc;;CAGrB,gBAAuD;EACrD,OAAO,QAAQ,QAAQ,KAAKA,YAAY;;CAG1C,cAAuB,SAA6C;EAClE,OAAO,QAAQ,SAAS;;CAG1B,MAAM,SAAwB;EAC5B,MAAM,KAAKA,YAAY,MAAM,SAAS,CAAC,MAAM,uBAAuB;;CAGtE,MAAM,WAA0B;EAC9B,MAAM,KAAKA,YAAY,MAAM,WAAW,CAAC,MAAM,uBAAuB;;;AAI1E,IAAM,yBAAN,cACU,kBAEV;CACE;CACA,UAAU;CAEV,YAAY,SAAkE;EAC5E,MAAM,uBAAuB,QAAQ,CAAC;EACtC,KAAK,OAAO,QAAQ,QAAQ;;CAG9B,IAAI,QAAwB;EAC1B,OAAO,KAAKG,UAAU,WAAW;;CAKnC,MAAM,QAAQ,UAA0C;CAExD,MAAM,oBAA4C;EAEhD,OAAO,IAAI,uBAAuB,MADb,KAAK,eAAe,EACC,KAAK,QAAQ;;CAGzD,MAAM,QAAuB;EAC3B,IAAI,KAAKA,SACP;EAEF,KAAKA,UAAU;EACf,MAAM,KAAK,KAAK,KAAK;;CAGvB,MAAM,gBAAqC;EACzC,OAAO,KAAK,KAAK,SAAS;;CAG5B,MAAM,cAAc,QAAmC;EACrD,OAAO,SAAS;;;AAIpB,IAAM,2BAAN,cACU,kBAEV;CACE;CACA,mBAA4B,IAAI,YAAY;CAC5C,UAAU;CACV,aAAa;CACb;CAEA,YAAY,SAAkE;EAC5E,MAAM,uBAAuB,QAAQ,CAAC;EACtC,KAAK,eAAe,QAAQ,QAAQ;;CAGtC,IAAI,QAAwB;EAC1B,OAAO,KAAKA,UAAU,WAAW;;CAKnC,MAAM,QAAQ,UAA0C;CAExD,MAAM,oBAA4C;EAChD,MAAM,eAAe,MAAM,KAAKC,iBAAiB,MAAM;EACvD,IAAI;GAEF,OAAO,IAAI,uBACT,MAFmB,KAAK,eAAe,EAGvC,KAAK,SACL,cASA,YAAY;IACV,IAAI;KACF,MAAM,KAAKC,yBAAyB;cAC5B;KACR,cAAc;;KAGnB;WACM,OAAO;GACd,cAAc;GACd,MAAM;;;CAIV,MAAM,QAAuB;EAC3B,MAAM,eAAe,MAAM,KAAKD,iBAAiB,MAAM;EACvD,IAAI;GACF,MAAM,KAAKC,yBAAyB;YAC5B;GACR,cAAc;;;CAIlB,MAAMA,0BAAyC;EAC7C,IAAI,KAAKF,SACP;EAEF,KAAKA,UAAU;EACf,MAAM,KAAK,aAAa,KAAK;EAC7B,KAAKG,aAAa;;CAGpB,MAAM,gBAAiC;EACrC,IAAI,KAAKA,YACP,OAAO,KAAK;EAEd,IAAI,KAAKC,oBAAoB,KAAA,GAAW;GACtC,MAAM,KAAKA;GACX,OAAO,KAAK;;EAGd,KAAKA,mBAAmB,YAAY;GAClC,IAAI;IACF,MAAM,KAAK,aAAa,SAAS;YAC1B,OAAgB;IACvB,IAAI,CAAC,wBAAwB,MAAM,EACjC,MAAM;aAEA;IACR,KAAKA,kBAAkB,KAAA;;GAEzB,KAAKD,aAAa;MAChB;EACJ,MAAM,KAAKC;EAEX,OAAO,KAAK;;CAGd,MAAM,cAAc,SAAgC;;AAGtD,SAAgB,6BACd,SACA,YACA,WAC4B;CAC5B,MAAM,qBAAqB,WAAW;CACtC,QAAQ,QAAQ,MAAhB;EACE,KAAK,OAMH,OAAO,IAAI,uBAAuB;GAChC,SAAS,EAAE,MAAA,IANI,KAAK;IACpB,kBAAkB,QAAQ;IAC1B,yBAAyB;IACzB,mBAAmB;IACpB,CAEgB,EAAE;GACjB,QAAQ;GACR;GACD,CAAC;EAEJ,KAAK,UACH,OAAO,IAAI,uBAAuB;GAChC,SAAS,EAAE,MAAM,QAAQ,MAAM;GAC/B,QAAQ;GACR;GACD,CAAC;EACJ,KAAK,YACH,OAAO,IAAI,yBAAyB;GAClC,SAAS,EAAE,QAAQ,QAAQ,QAAQ;GACnC,QAAQ;GACR;GACD,CAAC;;;AAIR,SAAS,WAAgB,QAAqB,MAA8B;CAC1E,OAAO,mBAA0B,OAAO;EACtC,OAAO,KAAK,OAAO,KAAK,SAAS,GAAG,KAAK,KAAK,CAAC;GAC/C;;AAGJ,SAAS,YAAY,QAAwC;CAC3D,OAAO,mBAAmB,OAAO,OAAO,MAAM,GAAG,CAAC;;AAGpD,SAAS,uBAAuB,OAAuB;CACrD,MAAM,iBAAiB,MAAM;;;;ACxiB/B,MAAM,6BACJ;AACF,MAAM,4BACJ;AASF,SAAS,YACP,MACA,SACA,SACoB;CACpB,MAAM,QAAQ,IAAI,MAAM,QAAQ;CAChC,OAAO,eAAe,OAAO,QAAQ;EACnC,OAAO;EACP,cAAc;EACf,CAAC;CACF,OAAO,OAAO,OAAO,OAAO;EAC1B;EACA,UAAU;EACV,UAAU;EACV;EACA;EACD,CAAC;;AAGJ,SAAS,iBAA0C;CACjD,OAAO,EACL,CAAC,OAAO,iBAAiB;EACvB,OAAO,EACL,MAAM,OAAO;GACX,MAAM,YAAY,wBAAwB,2BAA2B;KAExE;IAEJ;;AAGH,IAAM,4BAAN,MAAiE;CAC/D,WAAoB;CACpB,WAAoB;CAEpB,YAA+C;CAC/C,UAAU;CACV;CACA;CAEA,YAAY,SAAuC;EACjD,KAAKC,cAAc,SAAS;EAC5B,KAAKC,sBAAsB,SAAS;;CAGtC,IAAI,QAA4C;EAC9C,IAAI,KAAKC,cAAc,MACrB,OAAO;EAET,IAAI,KAAKC,SACP,OAAO;EAET,OAAO;;CAGT,mBAA+C;EAC7C,MAAM,WAAW,KAAKD;EACtB,IAAI,aAAa,MACf,MAAM,YAAY,wBAAwB,2BAA2B;EAEvE,OAAO;;CAGT,MAAM,QAAQ,SAAyC;EACrD,IAAI,KAAKA,cAAc,MACrB,MAAM,YAAY,4BAA4B,2BAA2B,EACvE,aAAa,QAAQ,MACtB,CAAC;EAEJ,KAAKA,YAAY,6BAA6B,SAAS,KAAKF,aAAa,EACvE,oBAAoB,KAAKC,qBAC1B,CAAC;EACF,KAAKE,UAAU;;CAGjB,MAAM,oBAA4C;EAChD,MAAM,WAAW,KAAKC,kBAAkB;EACxC,MAAM,aAAa,MAAM,SAAS,mBAAmB;EACrD,OAAO,KAAKC,gBAAgB,YAAY,SAAS;;;;;;;;;;CAWnD,gBAAgB,YAA2B,UAAqD;EAC9F,MAAM,0BAAgC;GACpC,IAAI,KAAKH,cAAc,YAAY,SAAS,UAAU,UAAU;IAC9D,KAAKA,YAAY;IACjB,KAAKC,UAAU;;;EAGnB,MAAM,UAAyB;GAC7B,kBAAkB,WAAW,iBAAiB,KAAK,WAAW;GAC9D,SAAS,WAAW,QAAQ,KAAK,WAAW;GAC5C,iBAAiB,WAAW,gBAAgB,KAAK,WAAW;GAC5D,OAAO,WAAW,MAAM,KAAK,WAAW;GACxC,SAAS,YAAY;IACnB,IAAI;KACF,MAAM,WAAW,SAAS;cAClB;KACR,mBAAmB;;;GAGvB,SAAS,OAAO,WAAqB;IACnC,IAAI;KACF,MAAM,WAAW,QAAQ,OAAO;cACxB;KACR,mBAAmB;;;GAGxB;EACD,IAAI,WAAW,SACb,QAAQ,UAAU,WAAW,QAAQ,KAAK,WAAW;EAEvD,OAAO;;CAGT,MAAM,QAAuB;EAC3B,MAAM,WAAW,KAAKD;EACtB,IAAI,aAAa,MAAM;GACrB,KAAKA,YAAY;GACjB,MAAM,SAAS,OAAO;;EAExB,KAAKC,UAAU;;CAGjB,QAAuC,SAAgD;EACrF,MAAM,WAAW,KAAKD;EACtB,IAAI,aAAa,MACf,OAAO,gBAAqB;EAE9B,OAAO,SAAS,QAAa,QAAQ;;CAGvC,gBACE,SACoB;EACpB,MAAM,WAAW,KAAKA;EACtB,IAAI,aAAa,MACf,OAAO,gBAAqB;EAE9B,OAAO,SAAS,gBAAqB,QAAQ;;CAG/C,MAAM,QAAQ,SAAuD;EACnE,MAAM,WAAW,KAAKE,kBAAkB;EACxC,MAAM,UAAU,SAAS;EACzB,IAAI,YAAY,KAAA,GACd,MAAM,YAAY,wBAAwB,2BAA2B;EAEvE,OAAO,QAAQ,KAAK,UAAU,QAAQ;;CAGxC,MAAM,MACJ,KACA,QAC8B;EAE9B,OADiB,KAAKA,kBACP,CAAC,MAAW,KAAK,OAAO;;;AAI3C,MAAM,kCAKF;CACF,GAAG;CACH,OAAO,SAA8D;EACnE,OAAO,IAAI,0BAA0B,QAAQ;;CAEhD"}
|
package/package.json
CHANGED
|
@@ -1,29 +1,29 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@prisma-next/driver-postgres",
|
|
3
|
-
"version": "0.9.0-dev.
|
|
3
|
+
"version": "0.9.0-dev.3",
|
|
4
4
|
"license": "Apache-2.0",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"sideEffects": false,
|
|
7
7
|
"dependencies": {
|
|
8
|
-
"@prisma-next/contract": "0.9.0-dev.
|
|
9
|
-
"@prisma-next/errors": "0.9.0-dev.
|
|
10
|
-
"@prisma-next/framework-components": "0.9.0-dev.
|
|
11
|
-
"@prisma-next/sql-contract": "0.9.0-dev.
|
|
12
|
-
"@prisma-next/sql-errors": "0.9.0-dev.
|
|
13
|
-
"@prisma-next/sql-operations": "0.9.0-dev.
|
|
14
|
-
"@prisma-next/sql-relational-core": "0.9.0-dev.
|
|
15
|
-
"@prisma-next/utils": "0.9.0-dev.
|
|
8
|
+
"@prisma-next/contract": "0.9.0-dev.3",
|
|
9
|
+
"@prisma-next/errors": "0.9.0-dev.3",
|
|
10
|
+
"@prisma-next/framework-components": "0.9.0-dev.3",
|
|
11
|
+
"@prisma-next/sql-contract": "0.9.0-dev.3",
|
|
12
|
+
"@prisma-next/sql-errors": "0.9.0-dev.3",
|
|
13
|
+
"@prisma-next/sql-operations": "0.9.0-dev.3",
|
|
14
|
+
"@prisma-next/sql-relational-core": "0.9.0-dev.3",
|
|
15
|
+
"@prisma-next/utils": "0.9.0-dev.3",
|
|
16
16
|
"arktype": "^2.1.29",
|
|
17
17
|
"pg": "8.20.0",
|
|
18
18
|
"pg-cursor": "^2.19.0"
|
|
19
19
|
},
|
|
20
20
|
"devDependencies": {
|
|
21
|
-
"@prisma-next/test-utils": "0.9.0-dev.
|
|
22
|
-
"@prisma-next/tsconfig": "0.9.0-dev.
|
|
21
|
+
"@prisma-next/test-utils": "0.9.0-dev.3",
|
|
22
|
+
"@prisma-next/tsconfig": "0.9.0-dev.3",
|
|
23
23
|
"@types/pg": "8.20.0",
|
|
24
24
|
"@types/pg-cursor": "^2.4.6",
|
|
25
25
|
"pg-mem": "^3.0.5",
|
|
26
|
-
"@prisma-next/tsdown": "0.9.0-dev.
|
|
26
|
+
"@prisma-next/tsdown": "0.9.0-dev.3",
|
|
27
27
|
"tsdown": "0.22.0",
|
|
28
28
|
"typescript": "5.9.3",
|
|
29
29
|
"vitest": "4.1.5"
|
package/src/exports/runtime.ts
CHANGED
|
@@ -3,6 +3,7 @@ import type {
|
|
|
3
3
|
RuntimeDriverInstance,
|
|
4
4
|
} from '@prisma-next/framework-components/execution';
|
|
5
5
|
import type {
|
|
6
|
+
PreparedExecuteRequest,
|
|
6
7
|
SqlConnection,
|
|
7
8
|
SqlDriver,
|
|
8
9
|
SqlExecuteRequest,
|
|
@@ -69,9 +70,11 @@ class PostgresUnboundDriverImpl implements PostgresRuntimeDriver {
|
|
|
69
70
|
#delegate: SqlDriver<PostgresBinding> | null = null;
|
|
70
71
|
#closed = false;
|
|
71
72
|
#cursorOpts: PostgresDriverCreateOptions['cursor'];
|
|
73
|
+
#preparedStatements: PostgresDriverCreateOptions['preparedStatements'];
|
|
72
74
|
|
|
73
|
-
constructor(
|
|
74
|
-
this.#cursorOpts =
|
|
75
|
+
constructor(options?: PostgresDriverCreateOptions) {
|
|
76
|
+
this.#cursorOpts = options?.cursor;
|
|
77
|
+
this.#preparedStatements = options?.preparedStatements;
|
|
75
78
|
}
|
|
76
79
|
|
|
77
80
|
get state(): 'unbound' | 'connected' | 'closed' {
|
|
@@ -98,7 +101,9 @@ class PostgresUnboundDriverImpl implements PostgresRuntimeDriver {
|
|
|
98
101
|
bindingKind: binding.kind,
|
|
99
102
|
});
|
|
100
103
|
}
|
|
101
|
-
this.#delegate = createBoundDriverFromBinding(binding, this.#cursorOpts
|
|
104
|
+
this.#delegate = createBoundDriverFromBinding(binding, this.#cursorOpts, {
|
|
105
|
+
preparedStatements: this.#preparedStatements,
|
|
106
|
+
});
|
|
102
107
|
this.#closed = false;
|
|
103
108
|
}
|
|
104
109
|
|
|
@@ -126,6 +131,7 @@ class PostgresUnboundDriverImpl implements PostgresRuntimeDriver {
|
|
|
126
131
|
const wrapped: SqlConnection = {
|
|
127
132
|
beginTransaction: connection.beginTransaction.bind(connection),
|
|
128
133
|
execute: connection.execute.bind(connection),
|
|
134
|
+
executePrepared: connection.executePrepared.bind(connection),
|
|
129
135
|
query: connection.query.bind(connection),
|
|
130
136
|
release: async () => {
|
|
131
137
|
try {
|
|
@@ -165,6 +171,16 @@ class PostgresUnboundDriverImpl implements PostgresRuntimeDriver {
|
|
|
165
171
|
return delegate.execute<Row>(request);
|
|
166
172
|
}
|
|
167
173
|
|
|
174
|
+
executePrepared<Row = Record<string, unknown>>(
|
|
175
|
+
request: PreparedExecuteRequest,
|
|
176
|
+
): AsyncIterable<Row> {
|
|
177
|
+
const delegate = this.#delegate;
|
|
178
|
+
if (delegate === null) {
|
|
179
|
+
return unboundExecute<Row>();
|
|
180
|
+
}
|
|
181
|
+
return delegate.executePrepared<Row>(request);
|
|
182
|
+
}
|
|
183
|
+
|
|
168
184
|
async explain(request: SqlExecuteRequest): Promise<SqlExplainResult> {
|
|
169
185
|
const delegate = this.#requireDelegate();
|
|
170
186
|
const explain = delegate.explain;
|
|
@@ -191,7 +207,7 @@ const postgresRuntimeDriverDescriptor: RuntimeDriverDescriptor<
|
|
|
191
207
|
> = {
|
|
192
208
|
...postgresDriverDescriptorMeta,
|
|
193
209
|
create(options?: PostgresDriverCreateOptions): PostgresRuntimeDriver {
|
|
194
|
-
return new PostgresUnboundDriverImpl(options
|
|
210
|
+
return new PostgresUnboundDriverImpl(options);
|
|
195
211
|
},
|
|
196
212
|
};
|
|
197
213
|
|
|
@@ -0,0 +1,82 @@
|
|
|
1
|
+
import type { BindConfig, Connection, QueryParse } from 'pg';
|
|
2
|
+
import Cursor from 'pg-cursor';
|
|
3
|
+
|
|
4
|
+
let nextNamedPortalId = 1;
|
|
5
|
+
|
|
6
|
+
export interface NamedCursorOptions<Row> {
|
|
7
|
+
readonly name: string;
|
|
8
|
+
readonly text: string;
|
|
9
|
+
readonly values: readonly unknown[];
|
|
10
|
+
readonly config?: ConstructorParameters<typeof Cursor<Row>>[2];
|
|
11
|
+
}
|
|
12
|
+
|
|
13
|
+
// `parsedStatements` is set on the Connection by pg.Client, not by Connection
|
|
14
|
+
// itself, so it isn't declared on @types/pg's `Connection`.
|
|
15
|
+
type ConnectionWithStatementCache = Connection & {
|
|
16
|
+
readonly parsedStatements: Record<string, string>;
|
|
17
|
+
};
|
|
18
|
+
|
|
19
|
+
// pg-cursor's runtime instance shape. @types/pg-cursor only declares the
|
|
20
|
+
// constructor + submit/read/close, so the fields pg-cursor sets on `this` and
|
|
21
|
+
// its handlers consume have to be re-declared here.
|
|
22
|
+
interface CursorInstanceFields {
|
|
23
|
+
state: 'initialized' | 'submitted' | 'idle' | 'busy' | 'done' | 'error';
|
|
24
|
+
connection: Connection | null;
|
|
25
|
+
text: string;
|
|
26
|
+
values: ReadonlyArray<unknown> | null;
|
|
27
|
+
_portal: string;
|
|
28
|
+
_conf: { types?: { getTypeParser: (oid: number, format?: string) => unknown } };
|
|
29
|
+
_result: { _getTypeParser?: unknown };
|
|
30
|
+
_ifNoData: () => void;
|
|
31
|
+
_rowDescription: () => void;
|
|
32
|
+
}
|
|
33
|
+
|
|
34
|
+
/** Streaming cursor for a server-side named prepared statement. */
|
|
35
|
+
export class NamedCursor<Row = Record<string, unknown>> extends Cursor<Row> {
|
|
36
|
+
readonly name: string;
|
|
37
|
+
|
|
38
|
+
constructor(opts: NamedCursorOptions<Row>) {
|
|
39
|
+
super(opts.text, [...opts.values], opts.config);
|
|
40
|
+
this.name = opts.name;
|
|
41
|
+
// Cursor declares `submit` as a property, not a class method, so
|
|
42
|
+
// `override submit() {}` at the class level conflicts. Replace the
|
|
43
|
+
// field on this instance instead.
|
|
44
|
+
this.submit = this.submitNamed;
|
|
45
|
+
}
|
|
46
|
+
|
|
47
|
+
private submitNamed(connection: Connection): void {
|
|
48
|
+
const self = this as unknown as CursorInstanceFields;
|
|
49
|
+
const conn = connection as ConnectionWithStatementCache;
|
|
50
|
+
|
|
51
|
+
self.state = 'submitted';
|
|
52
|
+
self.connection = conn;
|
|
53
|
+
self._portal = `np_${nextNamedPortalId++}`;
|
|
54
|
+
|
|
55
|
+
if (!conn.parsedStatements[this.name]) {
|
|
56
|
+
// QueryParse types `types` as required; pg's runtime accepts an empty
|
|
57
|
+
// array (server infers parameter types).
|
|
58
|
+
const parseMessage: QueryParse = { text: self.text, name: this.name, types: [] };
|
|
59
|
+
conn.parse(parseMessage, true);
|
|
60
|
+
}
|
|
61
|
+
|
|
62
|
+
// pg-cursor's constructor pre-maps values via prepareValue, so the array
|
|
63
|
+
// here is already wire-ready; the cast bridges pg-cursor's loose `any[]`
|
|
64
|
+
// to BindConfig's stricter union.
|
|
65
|
+
const bindMessage: BindConfig = {
|
|
66
|
+
portal: self._portal,
|
|
67
|
+
statement: this.name,
|
|
68
|
+
values: (self.values ?? []) as BindConfig['values'],
|
|
69
|
+
};
|
|
70
|
+
conn.bind(bindMessage, true);
|
|
71
|
+
|
|
72
|
+
conn.describe({ type: 'P', name: self._portal }, true);
|
|
73
|
+
conn.flush();
|
|
74
|
+
|
|
75
|
+
if (self._conf.types) {
|
|
76
|
+
self._result._getTypeParser = self._conf.types.getTypeParser;
|
|
77
|
+
}
|
|
78
|
+
|
|
79
|
+
conn.once('noData', self._ifNoData);
|
|
80
|
+
conn.once('rowDescription', self._rowDescription);
|
|
81
|
+
}
|
|
82
|
+
}
|
package/src/postgres-driver.ts
CHANGED
|
@@ -1,4 +1,5 @@
|
|
|
1
1
|
import type {
|
|
2
|
+
PreparedExecuteRequest,
|
|
2
3
|
SqlConnection,
|
|
3
4
|
SqlDriver,
|
|
4
5
|
SqlDriverState,
|
|
@@ -13,11 +14,13 @@ import type {
|
|
|
13
14
|
QueryResult as PgQueryResult,
|
|
14
15
|
PoolClient,
|
|
15
16
|
Pool as PoolType,
|
|
17
|
+
QueryConfig,
|
|
16
18
|
QueryResultRow,
|
|
17
19
|
} from 'pg';
|
|
18
20
|
import { Pool } from 'pg';
|
|
19
21
|
import Cursor from 'pg-cursor';
|
|
20
22
|
import { callbackToPromise } from './callback-to-promise';
|
|
23
|
+
import { NamedCursor } from './named-cursor';
|
|
21
24
|
import { isAlreadyConnectedError, isPostgresError, normalizePgError } from './normalize-error';
|
|
22
25
|
|
|
23
26
|
export type QueryResult<T extends QueryResultRow = QueryResultRow> = PgQueryResult<T>;
|
|
@@ -35,16 +38,67 @@ export interface PostgresCursorOptions {
|
|
|
35
38
|
interface PostgresDriverOptions {
|
|
36
39
|
readonly connect: { client: Client } | { pool: PoolType };
|
|
37
40
|
readonly cursor?: PostgresCursorOptions | undefined;
|
|
41
|
+
/**
|
|
42
|
+
* Use server-side prepared statements for `executePrepared`. Default
|
|
43
|
+
* `true`. Set `false` when running behind a transaction-mode pooler that
|
|
44
|
+
* does not support session-scoped prepared statements (e.g. PgBouncer
|
|
45
|
+
* transaction mode).
|
|
46
|
+
*/
|
|
47
|
+
readonly preparedStatements?: boolean | undefined;
|
|
38
48
|
}
|
|
39
49
|
|
|
40
50
|
export type PostgresDriverCreateOptions = Omit<PostgresDriverOptions, 'connect'>;
|
|
41
51
|
|
|
42
52
|
const DEFAULT_BATCH_SIZE = 100;
|
|
53
|
+
const DEFAULT_PREPARED_STATEMENTS = true;
|
|
43
54
|
|
|
44
|
-
type
|
|
55
|
+
type CursorOptions =
|
|
45
56
|
| { readonly cursorDisabled: true }
|
|
46
57
|
| { readonly cursorBatchSize: number; readonly cursorDisabled: false };
|
|
47
58
|
|
|
59
|
+
interface HandleAllocator {
|
|
60
|
+
mint(): string;
|
|
61
|
+
}
|
|
62
|
+
|
|
63
|
+
function createHandleAllocator(): HandleAllocator {
|
|
64
|
+
// Driver-scoped so a single PreparedStatement reused across multiple
|
|
65
|
+
// Connections/Transactions of the same driver always sees one handle.
|
|
66
|
+
let next = 1;
|
|
67
|
+
return {
|
|
68
|
+
mint: () => `pn_${next++}`,
|
|
69
|
+
};
|
|
70
|
+
}
|
|
71
|
+
|
|
72
|
+
type ConnectionOptions = CursorOptions & {
|
|
73
|
+
readonly preparedStatementsEnabled: boolean;
|
|
74
|
+
readonly handleAllocator: HandleAllocator;
|
|
75
|
+
};
|
|
76
|
+
|
|
77
|
+
function buildConnectionOptions(options: PostgresDriverOptions): ConnectionOptions {
|
|
78
|
+
const cursorOptions: CursorOptions = options.cursor?.disabled
|
|
79
|
+
? { cursorDisabled: true }
|
|
80
|
+
: {
|
|
81
|
+
cursorBatchSize: options.cursor?.batchSize ?? DEFAULT_BATCH_SIZE,
|
|
82
|
+
cursorDisabled: false,
|
|
83
|
+
};
|
|
84
|
+
return {
|
|
85
|
+
...cursorOptions,
|
|
86
|
+
preparedStatementsEnabled: options.preparedStatements ?? DEFAULT_PREPARED_STATEMENTS,
|
|
87
|
+
handleAllocator: createHandleAllocator(),
|
|
88
|
+
};
|
|
89
|
+
}
|
|
90
|
+
|
|
91
|
+
function isStalePreparedStatementError(error: unknown): boolean {
|
|
92
|
+
if (!(error instanceof Error)) {
|
|
93
|
+
return false;
|
|
94
|
+
}
|
|
95
|
+
const code = (error as { code?: string }).code;
|
|
96
|
+
// 26000 — invalid_sql_statement_name (server lost the named statement,
|
|
97
|
+
// e.g. after DEALLOCATE ALL).
|
|
98
|
+
// 0A000 — cached plan invalidated by DDL (row shape changed).
|
|
99
|
+
return code === '26000' || code === '0A000';
|
|
100
|
+
}
|
|
101
|
+
|
|
48
102
|
class AsyncMutex {
|
|
49
103
|
#queue = Promise.resolve();
|
|
50
104
|
|
|
@@ -75,6 +129,66 @@ abstract class PostgresQueryable<C extends PoolClient | Client = PoolClient | Cl
|
|
|
75
129
|
}
|
|
76
130
|
|
|
77
131
|
async *execute<Row = Record<string, unknown>>(request: SqlExecuteRequest): AsyncIterable<Row> {
|
|
132
|
+
try {
|
|
133
|
+
yield* this.runQuery<Row>(request);
|
|
134
|
+
} catch (error) {
|
|
135
|
+
throw normalizePgError(error);
|
|
136
|
+
}
|
|
137
|
+
}
|
|
138
|
+
|
|
139
|
+
async *executePrepared<Row = Record<string, unknown>>(
|
|
140
|
+
request: PreparedExecuteRequest,
|
|
141
|
+
): AsyncIterable<Row> {
|
|
142
|
+
if (!this.options.preparedStatementsEnabled) {
|
|
143
|
+
// Skip server-side prepare entirely; route as a regular ad-hoc execute.
|
|
144
|
+
yield* this.execute<Row>(request);
|
|
145
|
+
return;
|
|
146
|
+
}
|
|
147
|
+
|
|
148
|
+
let handle = request.handle.get();
|
|
149
|
+
if (handle === undefined) {
|
|
150
|
+
handle = this.options.handleAllocator.mint();
|
|
151
|
+
request.handle.set(handle);
|
|
152
|
+
}
|
|
153
|
+
|
|
154
|
+
yield* this.withStaleHandleRetry<Row>(request, handle as string, (h) =>
|
|
155
|
+
this.runQuery<Row>(request, h),
|
|
156
|
+
);
|
|
157
|
+
}
|
|
158
|
+
|
|
159
|
+
private async *withStaleHandleRetry<Row>(
|
|
160
|
+
request: PreparedExecuteRequest,
|
|
161
|
+
handle: string,
|
|
162
|
+
attempt: (handle: string) => AsyncIterable<Row>,
|
|
163
|
+
): AsyncIterable<Row> {
|
|
164
|
+
let yielded = false;
|
|
165
|
+
try {
|
|
166
|
+
for await (const row of attempt(handle)) {
|
|
167
|
+
yielded = true;
|
|
168
|
+
yield row;
|
|
169
|
+
}
|
|
170
|
+
return;
|
|
171
|
+
} catch (error) {
|
|
172
|
+
// If a row was already yielded, the error came from mid-stream and a
|
|
173
|
+
// retry would re-yield prior rows to the consumer.
|
|
174
|
+
if (yielded || !isStalePreparedStatementError(error)) {
|
|
175
|
+
throw normalizePgError(error);
|
|
176
|
+
}
|
|
177
|
+
// pg's parsedStatements still records the old name; only a fresh name
|
|
178
|
+
// forces pg to re-Parse on this Client.
|
|
179
|
+
const retryHandle = this.options.handleAllocator.mint();
|
|
180
|
+
request.handle.set(retryHandle);
|
|
181
|
+
try {
|
|
182
|
+
yield* attempt(retryHandle);
|
|
183
|
+
} catch (retryError) {
|
|
184
|
+
throw normalizePgError(retryError);
|
|
185
|
+
}
|
|
186
|
+
}
|
|
187
|
+
}
|
|
188
|
+
|
|
189
|
+
// Errors propagate as raw pg errors so the caller can read SQLSTATE before
|
|
190
|
+
// normalising (the prepared-statement retry layer needs the original code).
|
|
191
|
+
private async *runQuery<Row>(request: SqlExecuteRequest, name?: string): AsyncIterable<Row> {
|
|
78
192
|
const client = await this.acquireClient();
|
|
79
193
|
try {
|
|
80
194
|
if (!this.options.cursorDisabled) {
|
|
@@ -84,28 +198,23 @@ abstract class PostgresQueryable<C extends PoolClient | Client = PoolClient | Cl
|
|
|
84
198
|
request.sql,
|
|
85
199
|
request.params,
|
|
86
200
|
this.options.cursorBatchSize,
|
|
201
|
+
name,
|
|
87
202
|
)) {
|
|
88
203
|
yield row as Row;
|
|
89
204
|
}
|
|
90
205
|
return;
|
|
91
206
|
} catch (cursorError) {
|
|
92
|
-
|
|
207
|
+
// Non-pg cursor-specific errors fall through to the buffered path;
|
|
208
|
+
// real pg errors propagate.
|
|
209
|
+
if (!(cursorError instanceof Error) || isPostgresError(cursorError)) {
|
|
93
210
|
throw cursorError;
|
|
94
211
|
}
|
|
95
|
-
// Check if this is a pg error - if so, normalize and throw
|
|
96
|
-
// Otherwise, fall back to buffered mode for cursor-specific errors
|
|
97
|
-
if (isPostgresError(cursorError)) {
|
|
98
|
-
throw normalizePgError(cursorError);
|
|
99
|
-
}
|
|
100
|
-
// Not a pg error - cursor-specific error, fall back to buffered mode
|
|
101
212
|
}
|
|
102
213
|
}
|
|
103
214
|
|
|
104
|
-
for await (const row of this.executeBuffered(client, request.sql, request.params)) {
|
|
215
|
+
for await (const row of this.executeBuffered(client, request.sql, request.params, name)) {
|
|
105
216
|
yield row as Row;
|
|
106
217
|
}
|
|
107
|
-
} catch (error) {
|
|
108
|
-
throw normalizePgError(error);
|
|
109
218
|
} finally {
|
|
110
219
|
await this.releaseClient(client);
|
|
111
220
|
}
|
|
@@ -146,8 +255,12 @@ abstract class PostgresQueryable<C extends PoolClient | Client = PoolClient | Cl
|
|
|
146
255
|
sql: string,
|
|
147
256
|
params: readonly unknown[] | undefined,
|
|
148
257
|
cursorBatchSize: number,
|
|
258
|
+
name?: string,
|
|
149
259
|
): AsyncIterable<Record<string, unknown>> {
|
|
150
|
-
const
|
|
260
|
+
const values = (params ?? []) as unknown[];
|
|
261
|
+
const cursor = client.query(
|
|
262
|
+
name === undefined ? new Cursor(sql, values) : new NamedCursor({ name, text: sql, values }),
|
|
263
|
+
);
|
|
151
264
|
|
|
152
265
|
try {
|
|
153
266
|
while (true) {
|
|
@@ -169,8 +282,10 @@ abstract class PostgresQueryable<C extends PoolClient | Client = PoolClient | Cl
|
|
|
169
282
|
client: PoolClient | Client,
|
|
170
283
|
sql: string,
|
|
171
284
|
params: readonly unknown[] | undefined,
|
|
285
|
+
name?: string,
|
|
172
286
|
): AsyncIterable<Record<string, unknown>> {
|
|
173
|
-
const
|
|
287
|
+
const config: QueryConfig = { name, text: sql, values: (params ?? []) as unknown[] };
|
|
288
|
+
const result = await client.query(config);
|
|
174
289
|
for (const row of result.rows as Record<string, unknown>[]) {
|
|
175
290
|
yield row;
|
|
176
291
|
}
|
|
@@ -278,10 +393,7 @@ class PostgresPoolDriverImpl
|
|
|
278
393
|
#closed = false;
|
|
279
394
|
|
|
280
395
|
constructor(options: PostgresDriverOptions & { connect: { pool: PoolType } }) {
|
|
281
|
-
super(
|
|
282
|
-
cursorBatchSize: options.cursor?.batchSize ?? DEFAULT_BATCH_SIZE,
|
|
283
|
-
cursorDisabled: options.cursor?.disabled ?? false,
|
|
284
|
-
});
|
|
396
|
+
super(buildConnectionOptions(options));
|
|
285
397
|
this.pool = options.connect.pool;
|
|
286
398
|
}
|
|
287
399
|
|
|
@@ -326,10 +438,7 @@ class PostgresDirectDriverImpl
|
|
|
326
438
|
#connectPromise: Promise<void> | undefined;
|
|
327
439
|
|
|
328
440
|
constructor(options: PostgresDriverOptions & { connect: { client: Client } }) {
|
|
329
|
-
super(
|
|
330
|
-
cursorBatchSize: options.cursor?.batchSize ?? DEFAULT_BATCH_SIZE,
|
|
331
|
-
cursorDisabled: options.cursor?.disabled ?? false,
|
|
332
|
-
});
|
|
441
|
+
super(buildConnectionOptions(options));
|
|
333
442
|
this.directClient = options.connect.client;
|
|
334
443
|
}
|
|
335
444
|
|
|
@@ -421,7 +530,9 @@ class PostgresDirectDriverImpl
|
|
|
421
530
|
export function createBoundDriverFromBinding(
|
|
422
531
|
binding: PostgresBinding,
|
|
423
532
|
cursorOpts: PostgresDriverCreateOptions['cursor'],
|
|
533
|
+
extraOpts?: Pick<PostgresDriverCreateOptions, 'preparedStatements'>,
|
|
424
534
|
): SqlDriver<PostgresBinding> {
|
|
535
|
+
const preparedStatements = extraOpts?.preparedStatements;
|
|
425
536
|
switch (binding.kind) {
|
|
426
537
|
case 'url': {
|
|
427
538
|
const pool = new Pool({
|
|
@@ -432,17 +543,20 @@ export function createBoundDriverFromBinding(
|
|
|
432
543
|
return new PostgresPoolDriverImpl({
|
|
433
544
|
connect: { pool },
|
|
434
545
|
cursor: cursorOpts,
|
|
546
|
+
preparedStatements,
|
|
435
547
|
});
|
|
436
548
|
}
|
|
437
549
|
case 'pgPool':
|
|
438
550
|
return new PostgresPoolDriverImpl({
|
|
439
551
|
connect: { pool: binding.pool },
|
|
440
552
|
cursor: cursorOpts,
|
|
553
|
+
preparedStatements,
|
|
441
554
|
});
|
|
442
555
|
case 'pgClient':
|
|
443
556
|
return new PostgresDirectDriverImpl({
|
|
444
557
|
connect: { client: binding.client },
|
|
445
558
|
cursor: cursorOpts,
|
|
559
|
+
preparedStatements,
|
|
446
560
|
});
|
|
447
561
|
}
|
|
448
562
|
}
|