ActiveRecord alter default timeout for PostgreSQL query

application_record.rb

class ApplicationRecord < ActiveRecord::Base
  self.abstract_class = true  

  class << self
    # Start a new transaction with a statement timeout.
    # The timeout is reset after the block is executed. Works even with pg_bouncer in transaction mode
    # This is useful for queries that are expected to be slower than the default timeout
    # timeout - the timeout in seconds
    # Usage:
    # with_statement_timeout(10) do
    #  Model.find_by_sql("SELECT pg_sleep(20)")
    # end
    def with_statement_timeout(timeout)
      timeout_ms = (timeout.seconds * 1000).to_i
      transaction(requires_new: true) do
        connection.exec_query(sanitize_sql_array(["SET LOCAL statement_timeout TO ?", timeout_ms]))

        yield
      end
    end
  end
end