feat(client-core): Add fetch timeout and network retry options (cube-js#9484)

Author: paveltiunov

packages/cubejs-client-core/index.d.ts

@@ -23,6 +23,10 @@ declare module '@cubejs-client/core' {
     headers?: Record<string, string>;
     credentials?: 'omit' | 'same-origin' | 'include';
     method?: 'GET' | 'PUT' | 'POST' | 'PATCH';
+    /**
+     * Fetch timeout in milliseconds. Would be passed as AbortSignal.timeout()
+     */
+    fetchTimeout?: number;
   };
 
   export interface ITransportResponse<R> {
@@ -81,6 +85,10 @@ declare module '@cubejs-client/core' {
     parseDateMeasures?: boolean;
     resType?: 'default' | 'compact';
     castNumerics?: boolean;
+    /**
+     * How many network errors would be retried before returning to users. Default to 0.
+     */
+    networkErrorRetries?: number;
   };
 
   export type LoadMethodOptions = {

packages/cubejs-client-core/src/HttpTransport.js

@@ -2,12 +2,13 @@ import fetch from 'cross-fetch';
 import 'url-search-params-polyfill';
 
 class HttpTransport {
-  constructor({ authorization, apiUrl, method, headers = {}, credentials }) {
+  constructor({ authorization, apiUrl, method, headers = {}, credentials, fetchTimeout }) {
     this.authorization = authorization;
     this.apiUrl = apiUrl;
     this.method = method;
     this.headers = headers;
     this.credentials = credentials;
+    this.fetchTimeout = fetchTimeout;
   }
 
   request(method, { baseRequestId, ...params }) {
@@ -36,7 +37,8 @@ class HttpTransport {
         ...this.headers
       },
       credentials: this.credentials,
-      body: requestMethod === 'POST' ? JSON.stringify(params) : null
+      body: requestMethod === 'POST' ? JSON.stringify(params) : null,
+      signal: this.fetchTimeout ? AbortSignal.timeout(this.fetchTimeout) : undefined,
     });
 
     return {

packages/cubejs-client-core/src/index.js

@@ -57,6 +57,7 @@ class CubeApi {
     this.pollInterval = options.pollInterval || 5;
     this.parseDateMeasures = options.parseDateMeasures;
     this.castNumerics = typeof options.castNumerics === 'boolean' ? options.castNumerics : false;
+    this.networkErrorRetries = options.networkErrorRetries || 0;
 
     this.updateAuthorizationPromise = null;
   }
@@ -104,6 +105,8 @@ class CubeApi {
       }
     };
 
+    let networkRetries = this.networkErrorRetries;
+
     const loadImpl = async (response, next) => {
       const requestInstance = await requestPromise;
 
@@ -135,7 +138,11 @@ class CubeApi {
 
       skipAuthorizationUpdate = false;
 
-      if (response.status === 502) {
+      if (response.status === 502 ||
+        response.error &&
+        response.error.toLowerCase() === 'network error' &&
+        --networkRetries >= 0
+      ) {
         await checkMutex();
         return continueWait(true);
       }