ruby-on-rails - 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 数据库。

如果您在任务中做很多更复杂的事情，那也是可能的。参见 How do I force RAILS_ENV in a rake task?获取一些提示。