Ruby on rails在rake任務中指定環境

Question

我想創建一個自定義db/seeds文件並將其傳遞給特定環境

desc "Select for bonus"
task :bonus => :environment do
  puts "Bonus for: #{pick(User).name}"
end

我的rake任務的任務部分是否可以指定只說測試環境而不必在我的命令行中運行RAILS_ENV=test rake ？

Answer 1

如果您所做的只是使用此特定任務為數據庫設定種子，則只需要在正確的數據庫中建立連接作為任務的一部分。

desc "Select for bonus"
task :bonus => :environment do
  puts "Bonus for: #{pick(User).name}"
  ActiveRecord::Base.establish_connection('test')
  ....
end

ActiveRecord::Base.establish_connection('test')在運行該數據庫上的其余步驟之前連接到test數據庫。

如果你做了很多復雜的事情作為任務的一部分，那也是可能的。 請參閱如何在rake任務中強制執行RAILS_ENV？ 一些提示。