@supabase/pg-delta 1.0.0-alpha.8 → 1.0.0-alpha.9

package/dist/cli/commands/declarative-apply.js

@@ -27,7 +27,7 @@ export const declarativeApplyCommand = buildCommand({
                 parse: Number,
                 optional: true,
             },
-            "no-validate-functions": {
+            "skip-function-validation": {
                 kind: "boolean",
                 brief: "Skip final function body validation pass",
                 optional: true,
@@ -110,7 +110,7 @@ Tip: Use DEBUG=pg-delta:declarative-apply for detailed defer/skip/fail logs (whi
                 content,
                 targetUrl: flags.target,
                 maxRounds: flags["max-rounds"],
-                validateFunctionBodies: !flags["no-validate-functions"],
+                validateFunctionBodies: !flags["skip-function-validation"],
                 onRoundComplete,
             });
         }

package/dist/core/postgres-config.js

@@ -107,8 +107,49 @@ export function createPool(connectionString, options) {
         connectionTimeoutMillis: DEFAULT_CONNECTION_TIMEOUT_MS,
         ...config,
     });
-    if (onConnect)
-        pool.on("connect", onConnect);
+    if (onConnect) {
+        const pendingClientSetup = new WeakMap();
+        const waitForClientSetup = async (client) => {
+            const setup = pendingClientSetup.get(client);
+            if (setup) {
+                await setup;
+                return;
+            }
+            throw new Error("Internal error: pool client was acquired before async onConnect setup was registered. This indicates a bug in the pool wrapper logic; please report it with reproduction steps.");
+        };
+        const originalConnect = pool.connect.bind(pool);
+        pool.on("connect", (client) => {
+            pendingClientSetup.set(client, Promise.resolve().then(() => onConnect(client)));
+        });
+        pool.connect = ((callback) => {
+            if (!callback) {
+                return originalConnect().then(async (client) => {
+                    try {
+                        await waitForClientSetup(client);
+                        return client;
+                    }
+                    catch (setupError) {
+                        client.release?.(setupError);
+                        throw setupError;
+                    }
+                });
+            }
+            return originalConnect(async (err, client, release) => {
+                if (err || !client) {
+                    callback(err, client, release);
+                    return;
+                }
+                try {
+                    await waitForClientSetup(client);
+                    callback(err, client, release);
+                }
+                catch (setupError) {
+                    release(setupError);
+                    callback(setupError, undefined, () => { });
+                }
+            });
+        });
+    }
     if (onError)
         pool.on("error", onError);
     if (onAcquire)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@supabase/pg-delta",
-  "version": "1.0.0-alpha.8",
+  "version": "1.0.0-alpha.9",
   "description": "PostgreSQL migrations made easy",
   "type": "module",
   "sideEffects": false,

package/src/cli/commands/declarative-apply.ts

@@ -42,7 +42,7 @@ export const declarativeApplyCommand = buildCommand({
         parse: Number,
         optional: true,
       },
-      "no-validate-functions": {
+      "skip-function-validation": {
         kind: "boolean",
         brief: "Skip final function body validation pass",
         optional: true,
@@ -93,7 +93,7 @@ Tip: Use DEBUG=pg-delta:declarative-apply for detailed defer/skip/fail logs (whi
       path: string;
       target: string;
       "max-rounds"?: number;
-      "no-validate-functions"?: boolean;
+      "skip-function-validation"?: boolean;
       verbose?: boolean;
       "ungroup-diagnostics"?: boolean;
     },
@@ -144,7 +144,7 @@ Tip: Use DEBUG=pg-delta:declarative-apply for detailed defer/skip/fail logs (whi
         content,
         targetUrl: flags.target,
         maxRounds: flags["max-rounds"],
-        validateFunctionBodies: !flags["no-validate-functions"],
+        validateFunctionBodies: !flags["skip-function-validation"],
         onRoundComplete,
       });
     } catch (error) {

package/src/core/postgres-config.ts

@@ -139,7 +139,62 @@ export function createPool(
     ...config,
   });
 
-  if (onConnect) pool.on("connect", onConnect);
+  if (onConnect) {
+    const pendingClientSetup = new WeakMap<PoolClient, Promise<void>>();
+    const waitForClientSetup = async (client: PoolClient) => {
+      const setup = pendingClientSetup.get(client);
+      if (setup) {
+        await setup;
+        return;
+      }
+      throw new Error(
+        "Internal error: pool client was acquired before async onConnect setup was registered. This indicates a bug in the pool wrapper logic; please report it with reproduction steps.",
+      );
+    };
+    const originalConnect = pool.connect.bind(pool);
+
+    pool.on("connect", (client) => {
+      pendingClientSetup.set(
+        client,
+        Promise.resolve().then(() => onConnect(client)),
+      );
+    });
+
+    pool.connect = ((
+      callback?: (
+        err: Error | undefined,
+        client: PoolClient | undefined,
+        release: (err?: Error | boolean) => void,
+      ) => void,
+    ) => {
+      if (!callback) {
+        return originalConnect().then(async (client) => {
+          try {
+            await waitForClientSetup(client);
+            return client;
+          } catch (setupError) {
+            (client as PoolClient).release?.(setupError as Error);
+            throw setupError;
+          }
+        });
+      }
+
+      return originalConnect(async (err, client, release) => {
+        if (err || !client) {
+          callback(err, client, release);
+          return;
+        }
+
+        try {
+          await waitForClientSetup(client);
+          callback(err, client, release);
+        } catch (setupError) {
+          release(setupError as Error);
+          callback(setupError as Error, undefined, () => {});
+        }
+      });
+    }) as Pool["connect"];
+  }
   if (onError) pool.on("error", onError);
   if (onAcquire) pool.on("acquire", onAcquire);
   if (onRemove) pool.on("remove", onRemove);