feat(drivers): add an option to skip connection test probes in bigquery, aurora, postgres(redshift), snowflake

feat(bigquery-driver): add an option to skip connection test probes
feat(aurora-serverless-driver): add an option to skip connection test probes
feat(postgres-driver): add an option to skip connection test probes
feat(snowflake-driver): add an option to skip connection test probes

Author: KSDaemon

packages/cubejs-bigquery-driver/src/BigQueryDriver.ts

@@ -93,14 +93,15 @@ export class BigQueryDriver extends BaseDriver implements DriverInterface {
       testConnectionTimeout?: number,
     } = {}
   ) {
-    super({
-      testConnectionTimeout: config.testConnectionTimeout,
-    });
-
     const dataSource =
       config.dataSource ||
       assertDataSource('default');
 
+    super({
+      testConnectionTimeout: config.testConnectionTimeout,
+      isTestConnectionDisabled: getEnv('dbDisableTestConnection', { dataSource }),
+    });
+
     this.options = {
       scopes: [
         'https://www.googleapis.com/auth/bigquery',
@@ -155,6 +156,10 @@ export class BigQueryDriver extends BaseDriver implements DriverInterface {
   }
 
   public async testConnection() {
+    if (this.isTestConnectionDisabled()) {
+      return;
+    }
+
     await this.bigquery.query({
       query: 'SELECT ? AS number',
       params: ['1'],

packages/cubejs-mysql-aurora-serverless-driver/driver/AuroraServerlessMySqlDriver.js

@@ -33,14 +33,15 @@ class AuroraServerlessMySqlDriver extends BaseDriver {
    * Class constructor.
    */
   constructor(config = {}) {
-    super({
-      testConnectionTimeout: config.testConnectionTimeout,
-    });
-    
     const dataSource =
       config.dataSource ||
       assertDataSource('default');
-    
+
+    super({
+      testConnectionTimeout: config.testConnectionTimeout,
+      isTestConnectionDisabled: getEnv('dbDisableTestConnection', { dataSource }),
+    });
+
     this.config = {
       secretArn:
         config.secretArn ||
@@ -66,7 +67,11 @@ class AuroraServerlessMySqlDriver extends BaseDriver {
   }
 
   async testConnection() {
-    return this.dataApi.query('SELECT 1');
+    if (this.isTestConnectionDisabled()) {
+      return;
+    }
+
+    await this.dataApi.query('SELECT 1');
   }
 
   positionBindings(sql) {

packages/cubejs-postgres-driver/src/PostgresDriver.ts

@@ -111,14 +111,15 @@ export class PostgresDriver<Config extends PostgresDriverConfiguration = Postgre
       testConnectionTimeout?: number,
     } = {}
   ) {
+    const dataSource =
+      config.dataSource ||
+      assertDataSource('default');
+
     super({
       testConnectionTimeout: config.testConnectionTimeout,
+      isTestConnectionDisabled: getEnv('dbDisableTestConnection', { dataSource }),
     });
 
-    const dataSource =
-      config.dataSource ||
-      assertDataSource('default');
-    
     this.pool = new Pool({
       idleTimeoutMillis: 30000,
       max:
@@ -146,9 +147,9 @@ export class PostgresDriver<Config extends PostgresDriverConfiguration = Postgre
   }
 
   protected primaryKeysQuery(conditionString?: string): string | null {
-    return `SELECT 
+    return `SELECT
       columns.table_schema as ${this.quoteIdentifier('table_schema')},
-      columns.table_name as ${this.quoteIdentifier('table_name')}, 
+      columns.table_name as ${this.quoteIdentifier('table_name')},
       columns.column_name as ${this.quoteIdentifier('column_name')}
     FROM information_schema.table_constraints tc
     JOIN information_schema.constraint_column_usage AS ccu USING (constraint_schema, constraint_name)
@@ -225,6 +226,10 @@ export class PostgresDriver<Config extends PostgresDriverConfiguration = Postgre
   }
 
   public async testConnection(): Promise<void> {
+    if (this.isTestConnectionDisabled()) {
+      return;
+    }
+
     try {
       await this.pool.query('SELECT $1::int AS number', ['1']);
     } catch (e) {

packages/cubejs-snowflake-driver/src/SnowflakeDriver.ts

@@ -241,14 +241,15 @@ export class SnowflakeDriver extends BaseDriver implements DriverInterface {
       testConnectionTimeout?: number,
     } = {}
   ) {
-    super({
-      testConnectionTimeout: config.testConnectionTimeout,
-    });
-
     const dataSource =
       config.dataSource ||
       assertDataSource('default');
 
+    super({
+      testConnectionTimeout: config.testConnectionTimeout,
+      isTestConnectionDisabled: getEnv('dbDisableTestConnection', { dataSource }),
+    });
+
     let privateKey = getEnv('snowflakePrivateKey', { dataSource });
     if (privateKey && !privateKey.endsWith('\n')) {
       privateKey += '\n';
@@ -420,6 +421,10 @@ export class SnowflakeDriver extends BaseDriver implements DriverInterface {
    * Test driver's connection.
    */
   public async testConnection() {
+    if (this.isTestConnectionDisabled()) {
+      return;
+    }
+
     const connection = await this.createConnection();
 
     await new Promise(