retry for ActiveRecord::StatementInvalid

retry.rb

module DbUtils
  def self.retry_with_reconnect
    retried = false
    begin
      yield
    rescue ActiveRecord::StatementInvalid => e
      raise if retried
      Rails.logger.info "Retrying after failed statement #{e}"
      ActiveRecord::Base.connection.reconnect!
      retried = true
      retry
    end
  end

  def self.retry_selects
    old = Thread.current[:retry_select_all]
    Thread.current[:retry_select_all] = true
    yield
  ensure
    Thread.current[:retry_select_all] = old
  end

  module ::ActiveRecord::ConnectionAdapters::DatabaseStatements

    alias_method :select_all_without_retry, :select_all

    def select_all(arel, name = nil, binds = [], preparable: nil, async: false)
      if Thread.current[:retry_select_all]
        ::DbUtils.retry_with_reconnect do
          select_all_without_retry(arel, name, binds, preparable:, async:)
        end
      else
        select_all_without_retry(arel, name, binds, preparable:, async:)
      end
    end
  end

end