increment gradually the lock_timeout value

Author: MouadHilali

lib/safe-pg-migrations/configuration.rb

@@ -13,8 +13,9 @@ class Configuration
                     retry_delay
                     max_tries
                     sensitive_logger
+                    lock_timeout
+                    safe_timeout
                   ])
-    attr_reader :lock_timeout, :safe_timeout
 
     def initialize
       self.default_value_backfill_threshold = nil

lib/safe-pg-migrations/plugins/statement_retrier.rb

@@ -21,6 +21,15 @@ def retry_if_lock_timeout
         raise if transaction_open? # Retrying is useless if we're inside a transaction.
         raise unless remaining_tries > 0
 
+        number_of_retries = SafePgMigrations.config.max_tries - remaining_tries
+        if SafePgMigrations.config.max_tries - remaining_tries <= 20
+          SafePgMigrations.config.lock_timeout = SafePgMigrations.config.lock_timeout * number_of_retries
+          SafePgMigrations.config.safe_timeout = SafePgMigrations.config.safe_timeout * number_of_retries
+        else
+          SafePgMigrations.config.lock_timeout = SafePgMigrations.config.lock_timeout * 20
+          SafePgMigrations.config.safe_timeout = SafePgMigrations.config.safe_timeout * 20
+        end
+
         retry_delay = SafePgMigrations.config.retry_delay
         Helpers::Logger.say "Retrying in #{retry_delay} seconds...", sub_item: true
         sleep retry_delay