如何根据查询插入多行？

Question

我正在开发一个具有文件夹、组和权限的系统。权限决定了不同组可以在每个文件夹中执行的操作。因此，每当我创建一个新组时，我都想在每个文件夹的权限表中添加一条记录，描述新组在该文件夹中可以做什么。

目前我只是循环遍历系统中的所有文件夹并为每个文件夹添加权限记录：

group = Group.create(params)

Folder.all.each do |folder|
  Permission.create! do |permission|
    permission.folder = folder
    permission.group = group
    permission.can_create = true
    permission.can_read = true
    permission.can_update = true
    permission.can_delete = true
  end
end

我不喜欢每次创建新组时都必须遍历所有记录的事实。所以基本上我正在寻找一种优雅的方式来使用 ActiveRecord 执行以下 SQL。

INSERT INTO permissions (folder_id, group_id, can_creat, can_read, can_update, can_delete)
SELECT id, #{group.id}, true, true, true, true
FROM folders

我想我可以使用find_by_sql 运行上述查询，但感觉不对，因为我是INSERTing，而不是SELECTing。

或者我应该忘记这一点并像上面的示例一样继续循环浏览我的文件夹记录吗？

提前致谢。

Answer 1

你要找的是ar-extensions

---

使用安装gem

sudo gem install ar-extensions

在你的 environment.rb 中包含 gem（或者直接在你想要插入的模型中）

require 'ar-extensions'

并在一个INSERT查询中插入多条记录

fields = [:first_name, :last_name, :email]
data = [["glenn", "gillen", "foo@bar.com"],
       ["john", "jones", "jim@bar.com"],
       ["steve", "smith", "bar@foo.com"]]

User.import fields, data

您也可以使用 ActiveRecord 对象来做到这一点。

data = [ 
         User.new(:first_name => 'glenn', :last_name => 'gillen', :email => 'foo@bar.com'),
         User.new(:first_name => 'john', :last_name => 'jones', :email => 'jim@bar.com'),
         User.new(:first_name => 'steve', :last_name => 'smith', :email => 'bar@foo.com')
       ]

User.import fields, data

已将 3 个新行插入到 users 表中，只需 单个查询！

更多关于它的信息here、here 和 here。

Answer 2

这就是我在 rails/activerecord 中处理自定义 sql 的方式（包括 after_create 技巧！）

class Group < ActiveRecord::Base
  after_create :create_default_folder_permissions

  def create_default_folder_permissions
    sql = <<-SQL
     INSERT INTO permissions (folder_id, group_id, can_creat, can_read, can_update, can_delete)
        SELECT id, #{id}, true, true, true, true FROM folders
    SQL
    connection.execute(sql)
  end
end

但是，为每个组和每个文件夹添加权限很快就会成为瓶颈，因为您在权限表中获得了number_of_groups * number_of_folders 行。但是，如果您的查询很简单并且索引正确，则可以轻松扩展到数百万行。

Answer 3

您是在寻找最“优雅”的做事方式，还是最快的方式？为什么不直接从activerecord 调用你的SQL 查询呢？另一种解决方案是将您的查询放入数据库函数中，并从您的网络应用程序中调用该函数。维护起来可能有点困难，但性能提升令人难以置信，尤其是当您的网络应用程序中开始有大量文件夹时。