Class SqljsQueryable

src/drivers/database/sqljs.ts:11–81

import { SqliteLikeBaseDriver } from "../sqlite-base-driver";

class SqljsQueryable implements QueryableBaseDriver {
  public hasRowsChanged: boolean = false;

  constructor(protected db: Database) {}

  async transaction(stmts: InStatement[]): Promise<DatabaseResultSet[]> {
    const r: DatabaseResultSet[] = [];

    for (const s of stmts) {
      r.push(await this.query(s));
    }

    return r;
  }

  async query(stmt: InStatement): Promise<DatabaseResultSet> {
    const sql = typeof stmt === "string" ? stmt : stmt.sql;
    const bind =
      typeof stmt === "string" ? undefined : (stmt.args as BindParams);

    const startTime = Date.now();
    const s = this.db.prepare(sql, bind);
    const endTime = Date.now();

    // Do the transform result here
    const headerName = s.getColumnNames();
    const headerSet = new Set();

    const headers: DatabaseHeader[] = headerName.map((colName) => {
      let renameColName = colName;

      for (let i = 0; i < 20; i++) {
        if (!headerSet.has(renameColName)) break;
        renameColName = `__${colName}_${i}`;
      }

      return {
        name: renameColName,
        displayName: colName,
        originalType: null,
        type: undefined,
      };
    });

    const rows: DatabaseRow[] = [];
    while (s.step()) {
      const r = s.get();
      rows.push(
        headers.reduce((a, b, idx) => {
          a[b.name] = r[idx];
          return a;
        }, {} as DatabaseRow)
      );
    }

    if (this.db.getRowsModified() > 0) {
      this.hasRowsChanged = true;
    }