Rubyonrails 如何在 rake 任务中强制 RAILS_ENV

发布于07月07日

我有个小任务:

namespace :db do 
  namespace :test do 
    task :reset do 
      ENV['RAILS_ENV'] = "test" 
      Rake::Task['db:drop'].invoke
      Rake::Task['db:create'].invoke
      Rake::Task['db:migrate'].invoke
    end
  end
end

现在，当我执行时，它将忽略我试图硬编码的RAILS_ENV.我如何使这项任务按预期工作

推荐答案

对于这个特定的任务，您只需要更改DB连接，正如Adam指出的，您可以这样做:

namespace :db do 
  namespace :test do 
    task :reset do 
      ActiveRec或d::Base.establish_connection('test')
      Rake::Task['db:drop'].invoke
      Rake::Task['db:create'].invoke
      Rake::Task['db:migrate'].invoke
      ActiveRec或d::Base.establish_connection(ENV['RAILS_ENV'])  #Make sure you don't have side-effects!
    end
  end
end

如果您的任务更复杂，并且需要ENV的其他方面，那么最安全的做法是生成一个新的rake进程:

namespace :db do 
  namespace :test do 
    task :reset do 
      system("rake db:drop RAILS_ENV=test")
      system("rake db:create RAILS_ENV=test")
      system("rake db:migrate RAILS_ENV=test")
    end
  end
end

或

namespace :db do 
  namespace :test do 
    task :reset do 
      if (ENV['RAILS_ENV'] == "test")
        Rake::Task['db:drop'].invoke
        Rake::Task['db:create'].invoke
        Rake::Task['db:migrate'].invoke
      else
        system("rake db:test:reset RAILS_ENV=test")
      end
    end
  end
end