Heroku Rails控制台执行复杂查询被强制退出的解决求助

一、延长Heroku控制台超时时间 #

Heroku的heroku run命令默认存在超时限制，你可以通过以下方式调整：

• 临时延长单次控制台超时：

  heroku run rails console -t --timeout 3600
  

  其中3600为秒数（即1小时），可根据实际需求修改数值。
• 全局设置控制台超时（对后续所有heroku run命令生效）：

  heroku config:set HEROKU_RUN_TIMEOUT=3600
  

二、优化查询避免内存过载 #

你的查询使用includes会一次性将所有匹配的Author及关联Books加载到内存，数据量过大时会导致Heroku dyno内存不足直接崩溃，建议采用以下分批/轻量化处理方式：

1. 分批获取记录 #

用find_each分批加载数据（默认每批1000条），避免一次性占用大量内存：

Author.includes(:books).where(books: {book_release_date: ('01/01/2020'.to_date.beginning_of_day..'30/12/2022'.to_date.end_of_day)}).find_each do |author|
  # 在这里处理单个author的逻辑，比如打印、导出数据等
  puts author.name
end

2. 只查询必要字段 #

如果不需要模型的所有字段，用select指定所需字段，进一步降低内存占用：

Author.select(:id, :name).includes(:books).where(books: {book_release_date: ('01/01/2020'.to_date.beginning_of_day..'30/12/2022'.to_date.end_of_day)}).find_each do |author|
  # 处理逻辑
end

3. 使用纯SQL查询（无需ActiveRecord对象时） #

如果仅需要原始数据而非ActiveRecord实例，直接执行SQL查询，内存占用更低：

sql = <<-SQL
  SELECT authors.id, authors.name, books.title, books.book_release_date
  FROM authors
  JOIN books ON authors.id = books.author_id
  WHERE books.book_release_date BETWEEN ? AND ?
SQL
results = ActiveRecord::Base.connection.execute(sql, '2020-01-01 00:00:00', '2022-12-30 23:59:59')
results.each do |row|
  # 处理每行数据，row为哈希结构
  puts "#{row['name']} - #{row['title']}"
end

三、添加数据库索引提升查询速度 #

确保books表的book_release_date和author_id字段有索引，减少查询执行时间：

# 本地创建索引迁移文件
class AddIndexesToBooks < ActiveRecord::Migration[7.0]
  def change
    add_index :books, :book_release_date
    add_index :books, :author_id
  end
end

部署迁移到Heroku：

heroku run rails db:migrate

内容的提问来源于stack exchange，提问作者bubbaspaarx