Function interactiveExecuteQuery

src/gcp/cloudsql/interactive.ts:27–53

(query: string, conn: pg.PoolClient)

// Pretty query execution display such as spinner and actual returned content for `SELECT` query.
export async function interactiveExecuteQuery(query: string, conn: pg.PoolClient) {
  const spinner = ora("Executing query...").start();
  try {
    const results = await conn.query(query);
    spinner.succeed(clc.green("Query executed successfully"));

    if (Array.isArray(results.rows) && results.rows.length > 0) {
      const table: any[] = new Table({
        head: Object.keys(results.rows[0]).map((key) => clc.cyan(key)),
        style: { head: [], border: [] },
      });

      for (const row of results.rows) {
        table.push(Object.values(row) as any);
      }

      logger.info(table.toString());
    } else {
      // If nothing is returned and the query was select, let the user know there was no results.
      if (query.toUpperCase().includes("SELECT")) {
        logger.info(clc.yellow("No results returned"));
      }
    }
  } catch (err) {
    spinner.fail(clc.red(`Failed executing query: ${err}`));
  }
}