Use LockWaitTimeout in lock retrier

lib/online_migrations/lock_retrier.rb

@@ -96,9 +96,8 @@ def with_lock_retries(&block)
         else
           yield
         end
-      # ActiveRecord::LockWaitTimeout can be used for Active Record 5.2+
-      rescue ActiveRecord::StatementInvalid => e
-        if lock_timeout_error?(e) && current_attempt <= attempts
+      rescue ActiveRecord::LockWaitTimeout
+        if current_attempt <= attempts
           current_delay = delay(current_attempt)
           Utils.say("Lock timeout. Retrying in #{current_delay} seconds...")
           sleep(current_delay)
@@ -122,10 +121,6 @@ def with_lock_timeout(value)
       ensure
         connection.execute("SET lock_timeout TO #{connection.quote(prev_value)}")
       end
-
-      def lock_timeout_error?(error)
-        error.message.include?("canceling statement due to lock timeout")
-      end
   end
 
   # `LockRetrier` implementation that has a constant delay between tries