Class BigQueryDriver

packages/reltab-bigquery/src/reltab-bigquery.ts:52–200

}

export class BigQueryDriver implements DbDriver {
  readonly displayName: string;
  readonly sourceId: DataSourceId;
  readonly dialect: SQLDialect = BigQueryDialect;
  projectId: string;
  datasetName: string;
  bigquery: BigQuery;
  bigquery_meta: BigQuery;
  dataset: Dataset;
  tableMap: LeafSchemaMap;

  constructor(connectionInfo: BigQueryConnectionInfo) {
    const { projectId, datasetName } = connectionInfo;
    const resourceId = JSON.stringify(connectionInfo);
    this.displayName = `bigquery: ${projectId}`;
    this.sourceId = {
      providerName: "bigquery",
      resourceId,
    };
    this.projectId = projectId;
    this.datasetName = datasetName;

    this.bigquery = new BigQuery();
    this.bigquery_meta = new BigQuery({ projectId, location: LOCATION });
    this.dataset = this.bigquery_meta.dataset(datasetName);
    this.tableMap = {};
  }

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

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

  async importCsv(pathname: string, metadata: any): Promise<void> {
    const tableName = genTableName(pathname);
    const [job] = await this.dataset.table(tableName).load(pathname, metadata);
    console.log(
      "importCsv: load completed: ",
      JSON.stringify(job, undefined, 2)
    );
  }

  async getTableSchema(tableName: string): Promise<Schema> {
    const [projectId, datasetName, baseTableName] = tableName.split(".");
    const sqlQuery = `SELECT column_name, data_type FROM \`${projectId}.${datasetName}\`.INFORMATION_SCHEMA.COLUMNS WHERE table_name="${baseTableName}"`;

    const [dbRows] = await this.bigquery.query({
      query: sqlQuery,
      location: LOCATION,
    });