Class PostgresDatabaseAdapter

packages/data-table-postgres/src/lib/adapter.ts:30–291

 * `DatabaseAdapter` implementation for postgres-compatible clients.
 */
export class PostgresDatabaseAdapter implements DatabaseAdapter {
  /**
   * The SQL dialect identifier reported by this adapter.
   */
  dialect = 'postgres'

  /**
   * Feature flags describing the postgres behaviors supported by this adapter.
   */
  capabilities

  #client: PostgresQueryable
  #transactions = new Map<string, TransactionState>()
  #transactionCounter = 0

  constructor(client: PostgresQueryable) {
    this.#client = client
    this.capabilities = {
      returning: true,
      savepoints: true,
      upsert: true,
      transactionalDdl: true,
      migrationLock: true,
    }
  }

  /**
   * Compiles a data-manipulation operation to postgres SQL statements.
   * @param operation Operation to compile.
   * @returns Compiled SQL statements.
   */
  compileSql(operation: DataManipulationOperation): SqlStatement[] {
    let compiled = compilePostgresOperation(operation)
    return [{ text: compiled.text, values: compiled.values }]
  }

  /**
   * Executes a postgres data-manipulation request.
   * @param request Request to execute.
   * @returns Execution result.
   */
  async execute(request: DataManipulationRequest): Promise<DataManipulationResult> {
    if (request.operation.kind === 'insertMany' && request.operation.values.length === 0) {
      return {
        affectedRows: 0,
        insertId: undefined,
        rows: request.operation.returning ? [] : undefined,
      }
    }

    let statement = compilePostgresOperation(request.operation)
    let client = this.#resolveClient(request.transaction)
    let result = await client.query(statement.text, statement.values)
    let rows = normalizeRows(result.rows)

    if (request.operation.kind === 'count' || request.operation.kind === 'exists') {
      rows = normalizeCountRows(rows)
    }