before_after_migration_callbacks.rb

if ActiveRecord::Base.connection_config[:adapter] == 'postgresql'
  ActiveRecord::Base.connection.execute("SET statement_timeout = '10000';") # 10 seconds

  # Register an before migration callback to reset the timeout to 0 which means no timeout
  ActiveRecord::Migration.before_migrate do
    ActiveRecord::Base.connection.execute("SET statement_timeout = '0';")
  end

  ActiveRecord::Migration.after_migrate do
    ActiveRecord::Base.connection.execute("SET statement_timeout = '10000';")
  end
end