Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 14 additions & 7 deletions packages/cubejs-athena-driver/src/AthenaDriver.ts
Original file line number Diff line number Diff line change
@@ -1,3 +1,16 @@
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `AthenaDriver` and related types declaration.
*/

import {
getEnv,
assertDataSource,
checkNonNullable,
pausePromise,
Required,
} from '@cubejs-backend/shared';
import { Athena, GetQueryResultsCommandOutput } from '@aws-sdk/client-athena';
import { S3, GetObjectCommand } from '@aws-sdk/client-s3';
import { getSignedUrl } from '@aws-sdk/s3-request-presigner';
Expand All @@ -9,13 +22,6 @@ import {
QueryOptions, StreamOptions,
StreamTableData, TableName
} from '@cubejs-backend/base-driver';
import {
getEnv,
assertDataSource,
checkNonNullable,
pausePromise,
Required,
} from '@cubejs-backend/shared';
import * as SqlString from 'sqlstring';
import { AthenaClientConfig } from '@aws-sdk/client-athena/dist-types/AthenaClient';
import { URL } from 'url';
Expand Down Expand Up @@ -125,6 +131,7 @@ export class AthenaDriver extends BaseDriver implements DriverInterface {
});
}

// eslint-disable-next-line @typescript-eslint/no-unused-vars
public async query<R = unknown>(query: string, values: unknown[], options?: QueryOptions): Promise<R[]> {
const qid = await this.startQuery(query, values);
await this.waitForSuccess(qid);
Expand Down
15 changes: 13 additions & 2 deletions packages/cubejs-base-driver/src/BaseDriver.ts
Original file line number Diff line number Diff line change
@@ -1,7 +1,18 @@
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `BaseDriver` and related types declaration.
*/

import {
getEnv,
keyByDataSource,
isFilePath,
isSslKey,
isSslCert,
} from '@cubejs-backend/shared';
import { reduce } from 'ramda';
import fs from 'fs';
import { getEnv, keyByDataSource, isFilePath, isSslKey, isSslCert } from '@cubejs-backend/shared';

import { cancelCombinator } from './utils';
import {
CreateTableIndex,
Expand Down
19 changes: 12 additions & 7 deletions packages/cubejs-bigquery-driver/src/BigQueryDriver.ts
Original file line number Diff line number Diff line change
@@ -1,4 +1,15 @@
/* eslint-disable no-underscore-dangle */
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `BigQueryDriver` and related types declaration.
*/

import {
getEnv,
assertDataSource,
pausePromise,
Required,
} from '@cubejs-backend/shared';
import R from 'ramda';
import {
BigQuery,
Expand All @@ -12,12 +23,6 @@ import {
BaseDriver, DownloadTableCSVData,
DriverInterface, QueryOptions, StreamTableData,
} from '@cubejs-backend/base-driver';
import {
getEnv,
assertDataSource,
pausePromise,
Required,
} from '@cubejs-backend/shared';
import { Query } from '@google-cloud/bigquery/build/src/bigquery';
import { HydrationStream } from './HydrationStream';

Expand Down
12 changes: 10 additions & 2 deletions packages/cubejs-clickhouse-driver/src/ClickHouseDriver.ts
Original file line number Diff line number Diff line change
@@ -1,4 +1,13 @@
/* eslint-disable no-restricted-syntax */
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `ClickHouseDriver` and related types declaration.
*/

import {
getEnv,
assertDataSource,
} from '@cubejs-backend/shared';
import {
BaseDriver,
DownloadQueryResultsOptions,
Expand All @@ -7,7 +16,6 @@ import {
StreamOptions,
StreamTableDataWithTypes,
} from '@cubejs-backend/base-driver';
import { getEnv, assertDataSource } from '@cubejs-backend/shared';
import genericPool, { Pool } from 'generic-pool';
import { v4 as uuidv4 } from 'uuid';
import sqlstring from 'sqlstring';
Expand Down
6 changes: 6 additions & 0 deletions packages/cubejs-crate-driver/src/CrateDriver.ts
Original file line number Diff line number Diff line change
@@ -1,3 +1,9 @@
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `CrateDriver` and related types declaration.
*/

import { PostgresDriver } from '@cubejs-backend/postgres-driver';
import { PoolClient } from 'pg';

Expand Down
12 changes: 10 additions & 2 deletions packages/cubejs-databricks-jdbc-driver/src/DatabricksDriver.ts
Original file line number Diff line number Diff line change
@@ -1,4 +1,13 @@
/* eslint-disable no-restricted-syntax */
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `DatabricksDriver` and related types declaration.
*/

import {
getEnv,
assertDataSource,
} from '@cubejs-backend/shared';
import fs from 'fs';
import path from 'path';
import { S3, GetObjectCommand } from '@aws-sdk/client-s3';
Expand All @@ -15,7 +24,6 @@ import {
JDBCDriver,
JDBCDriverConfiguration,
} from '@cubejs-backend/jdbc-driver';
import { getEnv, assertDataSource } from '@cubejs-backend/shared';
import { DatabricksQuery } from './DatabricksQuery';
import { downloadJDBCDriver } from './installer';

Expand Down
1 change: 1 addition & 0 deletions packages/cubejs-databricks-jdbc-driver/src/index.ts
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
import { DatabricksDriver } from './DatabricksDriver';

export default DatabricksDriver;
export { DatabricksDriver };
13 changes: 11 additions & 2 deletions packages/cubejs-dremio-driver/driver/DremioDriver.js
Original file line number Diff line number Diff line change
@@ -1,8 +1,17 @@
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `DremioDriver` and related types declaration.
*/

const {
getEnv,
assertDataSource,
pausePromise,
} = require('@cubejs-backend/shared');
const axios = require('axios');
const SqlString = require('sqlstring');
const { BaseDriver } = require('@cubejs-backend/base-driver');
const { getEnv, assertDataSource, pausePromise } = require('@cubejs-backend/shared');

const DremioQuery = require('./DremioQuery');

// limit - Determines how many rows are returned (maximum of 500). Default: 100
Expand Down
11 changes: 10 additions & 1 deletion packages/cubejs-druid-driver/src/DruidDriver.ts
Original file line number Diff line number Diff line change
@@ -1,5 +1,14 @@
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `DruidDriver` and related types declaration.
*/

import {
getEnv,
assertDataSource,
} from '@cubejs-backend/shared';
import { BaseDriver, TableQueryResult } from '@cubejs-backend/base-driver';
import { getEnv, assertDataSource } from '@cubejs-backend/shared';
import { DruidClient, DruidClientBaseConfiguration, DruidClientConfiguration } from './DruidClient';
import { DruidQuery } from './DruidQuery';

Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,15 @@
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `ElasticSearchDriver` and related types declaration.
*/

const {
getEnv,
assertDataSource,
} = require('@cubejs-backend/shared');
const { Client } = require('@elastic/elasticsearch');
const SqlString = require('sqlstring');
const { getEnv, assertDataSource } = require('@cubejs-backend/shared');
const { BaseDriver } = require('@cubejs-backend/base-driver');

/**
Expand Down
11 changes: 10 additions & 1 deletion packages/cubejs-firebolt-driver/src/FireboltDriver.ts
Original file line number Diff line number Diff line change
@@ -1,4 +1,13 @@
import { getEnv, assertDataSource } from '@cubejs-backend/shared';
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `FireboltDriver` and related types declaration.
*/

import {
getEnv,
assertDataSource,
} from '@cubejs-backend/shared';
import {
BaseDriver,
DriverInterface,
Expand Down
22 changes: 16 additions & 6 deletions packages/cubejs-hive-driver/driver/HiveDriver.js
Original file line number Diff line number Diff line change
@@ -1,15 +1,25 @@
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `HiveDriver` and related types declaration.
*/

const {
getEnv,
assertDataSource,
} = require('@cubejs-backend/shared');
const jshs2 = require('jshs2');
const SqlString = require('sqlstring');
const genericPool = require('generic-pool');
const { BaseDriver } = require('@cubejs-backend/base-driver');
const { getEnv, assertDataSource } = require('@cubejs-backend/shared');

const {
HS2Util, IDLContainer, HiveConnection, Configuration
} = jshs2;
const Connection = require('jshs2/lib/Connection');

const IDLFactory = require('jshs2/lib/common/IDLFactory');
const {
HS2Util,
IDLContainer,
HiveConnection,
Configuration,
} = jshs2;

const newIDL = [
"2.1.1",
Expand Down
6 changes: 6 additions & 0 deletions packages/cubejs-jdbc-driver/src/JDBCDriver.ts
Original file line number Diff line number Diff line change
@@ -1,3 +1,9 @@
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `JDBCDriver` and related types declaration.
*/

/* eslint-disable no-restricted-syntax,import/no-extraneous-dependencies */
import {
getEnv,
Expand Down
10 changes: 9 additions & 1 deletion packages/cubejs-ksql-driver/src/KsqlDriver.ts
Original file line number Diff line number Diff line change
@@ -1,4 +1,9 @@
/* eslint-disable no-restricted-syntax */
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `KsqlDriver` and related types declaration.
*/

import {
getEnv,
assertDataSource,
Expand Down Expand Up @@ -202,14 +207,17 @@ export class KsqlDriver extends BaseDriver implements DriverInterface {
return fields.map(c => ({ name: c.name, type: this.toGenericType(c.schema.type) }));
}

// eslint-disable-next-line @typescript-eslint/no-unused-vars
public loadPreAggregationIntoTable(preAggregationTableName: string, loadSql: string, params: any[], options: any): Promise<any> {
return this.query(loadSql.replace(preAggregationTableName, this.tableDashName(preAggregationTableName)), params);
}

// eslint-disable-next-line @typescript-eslint/no-unused-vars
public async downloadTable(table: string, options: any): Promise<any> {
return this.getStreamingTableData(this.tableDashName(table));
}

// eslint-disable-next-line @typescript-eslint/no-unused-vars
public async downloadQueryResults(query: string, params: any, options: any) {
const table = KsqlQuery.extractTableFromSimpleSelectAsteriskQuery(query);
if (!table) {
Expand Down
6 changes: 6 additions & 0 deletions packages/cubejs-materialize-driver/src/MaterializeDriver.ts
Original file line number Diff line number Diff line change
@@ -1,3 +1,9 @@
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `MaterializeDriver` and related types declaration.
*/

import { PostgresDriver, PostgresDriverConfiguration } from '@cubejs-backend/postgres-driver';
import { BaseDriver, DownloadTableMemoryData, IndexesSQL, StreamOptions, StreamTableDataWithTypes, TableStructure } from '@cubejs-backend/base-driver';
import { PoolClient, QueryResult } from 'pg';
Expand Down
12 changes: 9 additions & 3 deletions packages/cubejs-mongobi-driver/src/MongoBIDriver.ts
Original file line number Diff line number Diff line change
@@ -1,10 +1,16 @@
import { createConnection, Connection, ConnectionOptions, RowDataPacket, Field } from 'mysql2';
import genericPool, { Pool } from 'generic-pool';
import { Readable } from 'stream';
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `MongoBIDriver` and related types declaration.
*/

import {
getEnv,
assertDataSource,
} from '@cubejs-backend/shared';
import { createConnection, Connection, ConnectionOptions, RowDataPacket, Field } from 'mysql2';
import genericPool, { Pool } from 'generic-pool';
import { Readable } from 'stream';
import {
BaseDriver,
DownloadQueryResultsOptions,
Expand Down
15 changes: 11 additions & 4 deletions packages/cubejs-mssql-driver/driver/MSSqlDriver.js
Original file line number Diff line number Diff line change
@@ -1,9 +1,16 @@
const sql = require('mssql');
const { BaseDriver } = require('@cubejs-backend/base-driver');
import {
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `MSSqlDriver` and related types declaration.
*/

const {
getEnv,
assertDataSource,
} from '@cubejs-backend/shared';
} = require('@cubejs-backend/shared');
const sql = require('mssql');
const { BaseDriver } = require('@cubejs-backend/base-driver');


const GenericTypeToMSSql = {
boolean: 'bit',
Expand Down
Original file line number Diff line number Diff line change
@@ -1,10 +1,17 @@
const crypto = require('crypto');
const dataApi = require('data-api-client');
const { BaseDriver } = require('@cubejs-backend/base-driver');
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `AuroraServerlessMySqlDriver` and related types
* declaration.
*/

const {
getEnv,
assertDataSource,
} = require('@cubejs-backend/shared');
const crypto = require('crypto');
const dataApi = require('data-api-client');
const { BaseDriver } = require('@cubejs-backend/base-driver');

const GenericTypeToMySql = {
string: 'varchar(255) CHARACTER SET utf8mb4',
Expand Down
12 changes: 9 additions & 3 deletions packages/cubejs-mysql-driver/src/MySqlDriver.ts
Original file line number Diff line number Diff line change
@@ -1,10 +1,16 @@
import mysql, { Connection, ConnectionConfig, FieldInfo, QueryOptions } from 'mysql';
import genericPool from 'generic-pool';
import { promisify } from 'util';
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `MySqlDriver` and related types declaration.
*/

import {
getEnv,
assertDataSource,
} from '@cubejs-backend/shared';
import mysql, { Connection, ConnectionConfig, FieldInfo, QueryOptions } from 'mysql';
import genericPool from 'generic-pool';
import { promisify } from 'util';
import {
BaseDriver,
GenericDataBaseType,
Expand Down
12 changes: 9 additions & 3 deletions packages/cubejs-oracle-driver/driver/OracleDriver.js
Original file line number Diff line number Diff line change
@@ -1,8 +1,14 @@
const { BaseDriver } = require('@cubejs-backend/base-driver');
import {
/**
* @copyright Cube Dev, Inc.
* @license Apache-2.0
* @fileoverview The `OracleDriver` and related types declaration.
*/

const {
getEnv,
assertDataSource,
} from '@cubejs-backend/shared';
} = require('@cubejs-backend/shared');
const { BaseDriver } = require('@cubejs-backend/base-driver');
const oracledb = require('oracledb');
const { reduce } = require('ramda');

Expand Down
Loading