为多租户应用程序轨道运行自定义 rake 任务

Question

我有一个自定义 rake 任务

namespace :abc do

  desc "seeds abc to a database" do

    task seed_abc: :environment do

      Tenant.find_each do |tenant|

        puts "Running task for tenant#{tenant.name}"
        Dir[File.join(Rails.root, 'db', 'seeds', '*.rb')].sort.each do |filename|

        p "Seeding #{filename}"
        load(filename) if File.exist?(filename)
      end
    end
  end
end

当我运行 rake 任务时，它只在默认租户中播种，但在播种时我可以看到

Running task for tenant abc "Seeding /path...../path/filename.rb" "Seeding /path...../path/filename.rb" Running task for tenant xyz "Seeding /path...../path/filename.rb" "Seeding /path...../path/filename.rb" Running task for tenant 123 "Seeding /path...../path/filename.rb" "Seeding /path...../path/filename.rb"

但是当我签入控制台时，它只为默认租户播种， 我如何为所有租户播种？

Answer 1

我知道这才几个月大，但我想我会为未来的人提供帮助。

对于公寓，确保租户在街区后切换回公共的更简洁的方法是执行以下操作：

Apartment::Tenant.switch(tenant.name) do
  #Your code in this block.
end

确保它切换回来很重要，因为如果没有，您最终可能会在错误的租户上运行代码。

如果您按照上述Apartment::Tenant.switch!(tenant.name) 的建议进行操作，则租户将保持选中状态，直到您再次切换它。如果下一次切换失败，它会停留在当前租户上，并在错误的 DB 上执行代码。