switch back to pg.Pool

firebase · May 21, 2024 · ea68811 · ea68811
1 parent fcc84ce
commit ea68811
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/src/gcp/cloudsql/connect.ts b/src/gcp/cloudsql/connect.ts
@@ -32,7 +32,7 @@ export async function execute(
     );
   }
   let connector: Connector;
-  let client: pg.Client;
+  let pool: pg.Pool;
   switch (user.type) {
     case "CLOUD_IAM_USER": {
       connector = new Connector({
@@ -43,7 +43,7 @@ export async function execute(
         ipType: IpAddressTypes.PUBLIC,
         authType: AuthTypes.IAM,
       });
-      client = new pg.Client({
+      pool = new pg.Pool({
         ...clientOpts,
         user: opts.username,
         database: opts.databaseId,
@@ -60,7 +60,8 @@ export async function execute(
         ipType: IpAddressTypes.PUBLIC,
         authType: AuthTypes.IAM,
       });
-      client = new pg.Client({
+      console.log("CLOUD_IAM_SERVICE_ACCOUNT", user, "clientOpts", clientOpts);
+      pool = new pg.Pool({
         ...clientOpts,
         user: opts.username,
         database: opts.databaseId,
@@ -79,7 +80,7 @@ export async function execute(
         instanceConnectionName: connectionName,
         ipType: IpAddressTypes.PUBLIC,
       });
-      client = new pg.Client({
+      pool = new pg.Pool({
         ...clientOpts,
         user: opts.username,
         password: opts.password,
@@ -89,6 +90,7 @@ export async function execute(
     }
   }
 
+  const client = await pool.connect();
   logFn(`Logged in as ${opts.username}`);
   for (const s of sqlStatements) {
     logFn(`Executing: '${s}'`);
@@ -99,7 +101,8 @@ export async function execute(
     }
   }
 
-  await client.end();
+  // This hangs somehow.
+  // await pool.end();
   connector.close();
 }