automatically start the engine after connection

Author: stepansergeevitch

packages/cubejs-firebolt-driver/src/FireboltDriver.ts

@@ -77,7 +77,8 @@ export class FireboltDriver extends BaseDriver implements DriverInterface {
       testConnectionTimeout?: number,
     } = {},
   ) {
-    super(config);
+    // Set connection timeout to 2 minutes to allow the engine to start if it's stopped
+    super({testConnectionTimeout: 120000, ...config});
 
     const dataSource =
       config.dataSource ||
@@ -122,6 +123,7 @@ export class FireboltDriver extends BaseDriver implements DriverInterface {
   private async initConnection() {
     try {
       const connection = await this.firebolt.connect(this.config.connection);
+      await this.ensureEngineRunning();
       return connection;
     } catch (e) {
       this.connection = null;
@@ -174,7 +176,7 @@ export class FireboltDriver extends BaseDriver implements DriverInterface {
       await connection.testConnection();
     } catch (error) {
       console.log(error);
-      throw new Error('Unable to connect');
+      throw error;
     }
   }