allow customize number of reties and backoff policy for runJob

keboola · Dec 18, 2024 · f9eaebb · f9eaebb
1 parent 22a52bd
commit f9eaebb
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/packages/php-table-backend-utils/src/Connection/Bigquery/BigQueryClientWrapper.php b/packages/php-table-backend-utils/src/Connection/Bigquery/BigQueryClientWrapper.php
@@ -56,9 +56,12 @@ public function runQuery(JobConfigurationInterface $query, array $options = []):
      */
     public function runJob(JobConfigurationInterface $config, array $options = []): Job
     {
-        $retryPolicy = new SimpleRetryPolicy(5);
-        $backOffPolicy = new ExponentialRandomBackOffPolicy(10, 1.8, 300);
-        $proxy = new RetryProxy($retryPolicy, $backOffPolicy);
+        $options += [
+            'retryCount' => 5,
+            'backOffPolicy' => new ExponentialRandomBackOffPolicy(10, 1.8, 300),
+        ];
+        $retryPolicy = new SimpleRetryPolicy($options['retryCount']);
+        $proxy = new RetryProxy($retryPolicy, $options['backOffPolicy']);
         $job = $proxy->call(function () use ($config, $options): Job {
             return $this->startJob($config, $options);
         });