feat(cubestore-driver): Introduce CUBEJS_CUBESTORE_NO_HEART_BEAT_TIMEOUT env

Author: paveltiunov

packages/cubejs-backend-shared/src/env.ts

@@ -1318,6 +1318,9 @@ const variables: Record<string, (...args: any) => any> = {
   cubeStoreMaxConnectRetries: () => get('CUBEJS_CUBESTORE_MAX_CONNECT_RETRIES')
     .default('5')
     .asInt(),
+  cubeStoreNoHeartBeatTimeout: () => get('CUBEJS_CUBESTORE_NO_HEART_BEAT_TIMEOUT')
+    .default('30')
+    .asInt(),
 
   // Redis
   redisPoolMin: () => get('CUBEJS_REDIS_POOL_MIN')

packages/cubejs-cubestore-driver/src/WebSocketConnection.ts

@@ -16,6 +16,8 @@ export class WebSocketConnection {
 
   protected maxConnectRetries: number;
 
+  protected noHeartBeatTimeout: number;
+
   protected currentConnectionTry: number;
 
   protected webSocket: any;
@@ -26,6 +28,7 @@ export class WebSocketConnection {
     this.url = url;
     this.messageCounter = 1;
     this.maxConnectRetries = getEnv('cubeStoreMaxConnectRetries');
+    this.noHeartBeatTimeout = getEnv('cubeStoreNoHeartBeatTimeout');
     this.currentConnectionTry = 0;
   }
 
@@ -39,7 +42,7 @@ export class WebSocketConnection {
             webSocket.ping();
           }
 
-          if (new Date().getTime() - webSocket.lastHeartBeat.getTime() > 30000) {
+          if (new Date().getTime() - webSocket.lastHeartBeat.getTime() > this.noHeartBeatTimeout * 1000) {
             webSocket.close();
           }
         }, 5000);