Class SqliteDriver

packages/reltab-sqlite/src/reltab-sqlite.ts:44–150

);

export class SqliteDriver implements DbDriver {
  readonly displayName: string;
  readonly sourceId: DataSourceId;
  readonly dialect: SQLDialect = SQLiteDialect;
  dbfile: string;
  db: sqlite3.Database;
  private tableMap: LeafSchemaMap;

  constructor(dbfile: string, db: any) {
    this.dbfile = dbfile;
    this.displayName = dbfile;
    this.sourceId = { providerName: "sqlite", resourceId: dbfile };
    this.db = db;
    this.tableMap = {};
  }

  async getDisplayName(): Promise<string> {
    return this.displayName;
  }

  async runSqlQuery(sqlQuery: string): Promise<Row[]> {
    const dbRows = await dbAll(this.db, sqlQuery);
    const rows = dbRows as Row[];
    return rows;
  }

  // Get table info directly from sqlite db
  async getTableSchema(tableName: string): Promise<Schema> {
    const tiQuery = `PRAGMA table_info(${tableName})`;
    const dbRows = await this.runSqlQuery(tiQuery);
    // log.debug("getTableSchema: ", rows);

    const extendCMap = (
      cmm: ColumnMetaMap,
      row: any,
      idx: number
    ): ColumnMetaMap => {
      const cnm = row.name;
      const cType = row.type.toLocaleUpperCase();

      if (cType == null) {
        log.error(
          'mkTableInfo: No column type for "' + cnm + '", index: ' + idx
        );
      }
      const cmd = {
        displayName: cnm,
        columnType: cType,
      };
      cmm[cnm] = cmd;
      return cmm;
    };

    const cmMap = dbRows.reduce(extendCMap, {});
    const columnIds = dbRows.map((r) => r.name);
    const schema = new Schema(SQLiteDialect, columnIds as string[], cmMap);
    return schema;
  }