Sinatra项目重写ActiveRecord Rake任务：rollback功能异常求助

你的核心问题是db:rollback任务里递归调用了自身，导致执行时出现死循环或任务无法正常触发，而且没有像migrate那样直接调用ActiveRecord的rollback逻辑。以下是修正方案：

问题分析 #

• db:migrate任务的逻辑是对的：先执行原有的开发环境迁移，再切换测试环境用ActiveRecord::Tasks::DatabaseTasks.migrate执行迁移。
• 但db:rollback任务错误地调用了Rake::Task['db:rollback'].invoke，而这个任务就是你正在定义的任务，直接导致递归调用；同时clear操作会清空原有任务定义，反而破坏了默认的rollback逻辑。

修正后的Rakefile代码 #

namespace 'db' do
  task 'migrate' do
    # 先执行开发环境的迁移（使用默认任务）
    Rake::Task['db:migrate'].execute

    return if ENV.fetch('SINATRA_ENV', 'development') == 'production'

    # 保存原环境变量，执行完恢复
    original_env = ENV['RACK_ENV']
    ENV['RACK_ENV'] = 'test'

    # 建立测试环境数据库连接
    ActiveRecord::Base.establish_connection(
      adapter: 'postgresql',
      database: "sinatra_project_#{ENV['RACK_ENV']}"
    )
    # 执行测试环境迁移
    ActiveRecord::Tasks::DatabaseTasks.migrate

    # 恢复原环境
    ENV['RACK_ENV'] = original_env
  end

  task 'rollback' do
    # 先执行开发环境的回滚（使用默认任务）
    Rake::Task['db:rollback'].execute

    return if ENV.fetch('SINATRA_ENV', 'development') == 'production'

    # 保存原环境变量，执行完恢复
    original_env = ENV['RACK_ENV']
    ENV['RACK_ENV'] = 'test'

    # 建立测试环境数据库连接
    ActiveRecord::Base.establish_connection(
      adapter: 'postgresql',
      database: "sinatra_project_#{ENV['RACK_ENV']}"
    )
    # 执行测试环境回滚：调用ActiveRecord的回滚逻辑
    ActiveRecord::Migrator.rollback(ActiveRecord::Tasks::DatabaseTasks.migrations_paths, 1)

    # 恢复原环境
    ENV['RACK_ENV'] = original_env
  end
end

关键调整说明 #

1. 替换invoke为execute：invoke会标记任务为已执行，后续再次调用不会重复执行；而execute会直接执行任务逻辑，适合这里的场景。
2. 避免递归调用：rollback任务不再调用自身，而是直接使用ActiveRecord::Migrator.rollback执行回滚操作，和migrate的逻辑保持一致。
3. 环境变量恢复：保存并恢复原RACK_ENV，避免任务执行后环境被永久修改，影响后续操作。
4. 默认环境兜底：ENV.fetch添加默认值development，防止环境变量未设置时出错。

额外建议 #

• 确保你的数据库配置文件（比如config/database.yml）里已经正确配置了测试环境的连接信息，这样可以直接用ActiveRecord::Base.establish_connection(ENV['RACK_ENV'].to_sym)代替硬编码连接参数，更灵活。
• 可以把环境切换、数据库连接的逻辑抽成单独的方法，减少代码重复。

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