Class PostgresLikeDriver

src/drivers/postgres/postgres-driver.ts:64–438

}

export default class PostgresLikeDriver extends CommonSQLImplement {
  constructor(protected _db: QueryableBaseDriver) {
    super();
  }

  query(stmt: string): Promise<DatabaseResultSet> {
    return this._db.query(stmt);
  }

  transaction(stmts: string[]): Promise<DatabaseResultSet[]> {
    return this._db.transaction(stmts);
  }

  batch(stmts: string[]): Promise<DatabaseResultSet[]> {
    return this._db.batch ? this._db.batch(stmts) : super.batch(stmts);
  }

  close(): void {
    // Do nothing
  }

  columnTypeSelector: ColumnTypeSelector = POSTGRES_DATA_TYPE_SUGGESTION;

  escapeId(id: string) {
    return `"${id.replace(/"/g, '""')}"`;
  }

  escapeValue(value: unknown): string {
    return escapeSqlValue(value);
  }

  getFlags(): DriverFlags {
    return {
      defaultSchema: "public",
      dialect: "postgres",
      optionalSchema: false,
      supportRowId: false,
      supportBigInt: false,
      supportModifyColumn: true,
      supportCreateUpdateTable: true,
      supportCreateUpdateDatabase: false,
      supportInsertReturning: true,
      supportUpdateReturning: true,
      supportCreateUpdateTrigger: false,
      supportUseStatement: true,
    };
  }

  async getCurrentSchema(): Promise<string | null> {
    const result = (await this.query("SHOW search_path")) as unknown as {
      rows: { search_path?: string | null }[];
    };

    const db = result.rows[0].search_path!.split(",")[0];

    return db === this.escapeId("$user") ? "public" : db;
  }