feat: add retryOptions passing to underlying Service class

src/bigquery.ts

@@ -43,7 +43,10 @@
   JobResponse,
   RowMetadata,
 } from './table';
-import {GoogleErrorBody} from '@google-cloud/common/build/src/util';
+import {
+  GoogleErrorBody,
+  RetryOptions,
+} from '@google-cloud/common/build/src/util';
 import bigquery from './types';
 import {logger, setLogFunction} from './logger';
 
@@ -216,16 +219,35 @@
    * We will exponentially backoff subsequent requests by default.
    *
    * Defaults to `true`.
+   *
+   * @deprecated Use retryOptions.
    */
   autoRetry?: boolean;
   /**
    * Maximum number of automatic retries
    * attempted before returning the error.
    *
    * Defaults to 3.
+   *
+   * @deprecated Use retryOptions.
    */
   maxRetries?: number;
 
+  /**
+   * Customize retry configuration for all requests in the SDK.
+   * By default, a request is retried if the response is related to rate limits
+   * or certain intermittent server errors.
+   * We will exponentially backoff subsequent requests by default.
+   *
+   * More on the default retry predicate on the `shouldRetryRequest` method:
+   * https://github.com/googleapis/nodejs-common/blob/main/src/util.ts
+   *
+   * Defaults:
+   * - retryOptions.autoRetry: true
+   * - retryOptions.maxRetries: 3
+   */
+  retryOptions?: RetryOptions;
+
   /**
    * The geographic location of all datasets and
    * jobs referenced and created through the client.
@@ -340,17 +362,17 @@
   private _universeDomain: string;
   private _enableQueryPreview: boolean;
 
   createQueryStream(options?: Query | string): ResourceStream<RowMetadata> {
     // placeholder body, overwritten in constructor
     return new ResourceStream<RowMetadata>({}, () => {});
   }
 
   getDatasetsStream(options?: GetDatasetsOptions): ResourceStream<Dataset> {
     // placeholder body, overwritten in constructor
     return new ResourceStream<Dataset>({}, () => {});
   }
 
   getJobsStream(options?: GetJobsOptions): ResourceStream<Job> {
     // placeholder body, overwritten in constructor
     return new ResourceStream<Job>({}, () => {});
   }
@@ -388,6 +410,7 @@
       packageJson: require('../../package.json'),
       autoRetry: options.autoRetry,
       maxRetries: options.maxRetries,
+      retryOptions: options.retryOptions,
     };
 
     if (options.scopes) {
@@ -1546,7 +1569,7 @@
     const parameterMode = is.array(params) ? 'positional' : 'named';
     const queryParameters: bigquery.IQueryParameter[] = [];
     if (parameterMode === 'named') {
       const namedParams = params as {[param: string]: any};
       for (const namedParameter of Object.getOwnPropertyNames(namedParams)) {
         const value = namedParams[namedParameter];
         let queryParameter;
@@ -2189,7 +2212,7 @@
 
       options = extend({job}, queryOpts, options);
       if (res && res.jobComplete) {
         let rows: any = [];
         if (res.schema && res.rows) {
           rows = BigQuery.mergeSchemaWithRows_(res.schema, res.rows, {
             wrapIntegers: options.wrapIntegers || false,

test/bigquery.ts

@@ -325,6 +325,19 @@ describe('BigQuery', () => {
       assert.deepStrictEqual(calledWith.maxRetries, retryVal);
     });
 
+    it('should pass retryOptions from options', () => {
+      const retryOptions = {
+        autoRetry: true,
+        maxRetries: 3,
+      };
+      const bq = new BigQuery({
+        retryOptions: retryOptions,
+      });
+
+      const calledWith = bq.calledWith_[0];
+      assert.deepStrictEqual(calledWith.retryOptions, retryOptions);
+    });
+
     it('should not modify options argument', () => {
       const options = {
         projectId: PROJECT_ID,