@prisma-next/driver-postgres 0.9.0-dev.2 → 0.9.0-dev.4

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.
@@ -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
@@ -1 +1 @@
1
- {"version":3,"file":"runtime.d.mts","names":[],"sources":["../src/postgres-driver.ts","../src/exports/runtime.ts"],"mappings":";;;;;KAsBY,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;AAAA;AAAA,KAGR,2BAAA,GAA8B,IAAA,CAAK,qBAAA;;;KCrBnC,qBAAA,GAAwB,qBAAA,sBAClC,SAAA,CAAU,eAAA;AAAA,cAsKN,+BAAA,EAAiC,uBAAA,oBAGrC,2BAAA,EACA,qBAAA"}
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 cursor = client.query(new Cursor(sql, params));
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 result = await client.query(sql, params);
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
- constructor(cursorOpts) {
310
- this.#cursorOpts = cursorOpts;
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?.cursor);
507
+ return new PostgresUnboundDriverImpl(options);
396
508
  }
397
509
  };
398
510
  //#endregion
@@ -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.2",
3
+ "version": "0.9.0-dev.4",
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.2",
9
- "@prisma-next/errors": "0.9.0-dev.2",
10
- "@prisma-next/framework-components": "0.9.0-dev.2",
11
- "@prisma-next/sql-contract": "0.9.0-dev.2",
12
- "@prisma-next/sql-errors": "0.9.0-dev.2",
13
- "@prisma-next/sql-operations": "0.9.0-dev.2",
14
- "@prisma-next/sql-relational-core": "0.9.0-dev.2",
15
- "@prisma-next/utils": "0.9.0-dev.2",
8
+ "@prisma-next/contract": "0.9.0-dev.4",
9
+ "@prisma-next/errors": "0.9.0-dev.4",
10
+ "@prisma-next/framework-components": "0.9.0-dev.4",
11
+ "@prisma-next/sql-contract": "0.9.0-dev.4",
12
+ "@prisma-next/sql-errors": "0.9.0-dev.4",
13
+ "@prisma-next/sql-operations": "0.9.0-dev.4",
14
+ "@prisma-next/sql-relational-core": "0.9.0-dev.4",
15
+ "@prisma-next/utils": "0.9.0-dev.4",
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.2",
22
- "@prisma-next/tsconfig": "0.9.0-dev.2",
21
+ "@prisma-next/test-utils": "0.9.0-dev.4",
22
+ "@prisma-next/tsconfig": "0.9.0-dev.4",
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.2",
26
+ "@prisma-next/tsdown": "0.9.0-dev.4",
27
27
  "tsdown": "0.22.0",
28
28
  "typescript": "5.9.3",
29
29
  "vitest": "4.1.5"
@@ -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(cursorOpts?: PostgresDriverCreateOptions['cursor']) {
74
- this.#cursorOpts = 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?.cursor);
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
+ }
@@ -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 ConnectionOptions =
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
- if (!(cursorError instanceof Error)) {
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 cursor = client.query(new Cursor(sql, params as unknown[] | undefined));
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 result = await client.query(sql, params as unknown[] | undefined);
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
  }