如何使用一个 ActiveRecord 连接截断 n 个表答案

【问题标题】：How to truncate n tables with one single ActiveRecord connection如何使用一个 ActiveRecord 连接截断 n 个表
【发布时间】：2017-01-26 05:38:36
【问题描述】：

我想创建一个 rake 任务，以便从某个 n 数量的表中删除所有记录并重置 id 列的值，这样当我创建新记录时它的 id 为 1。上一个问题Truncate table(s) with rails console 对于完成单个表的工作很有用（如果可以提高任务的性能，我会使用delete_all 代替destroy_all）：

Model.delete_all
ActiveRecord::Base.connection.execute("TRUNCATE #{table_name} RESTART IDENTITY")

如果我需要截断多个表，例如表table1、table2、table3，是否有机会使用单个 ActiveRecord 连接，例如：

Model1.delete_all
Model2.delete_all
Model3.delete_all
ActiveRecord::Base.connection.execute("TRUNCATE #{table1, table2, table3} RESTART IDENTITY")

因此避免像下面的示例那样建立多个 ActiveRecord 连接？

Model1.delete_all
Model2.delete_all
Model3.delete_all
ActiveRecord::Base.connection.execute("TRUNCATE #{table1} RESTART IDENTITY")
ActiveRecord::Base.connection.execute("TRUNCATE #{table2} RESTART IDENTITY")
ActiveRecord::Base.connection.execute("TRUNCATE #{table3} RESTART IDENTITY")

【问题讨论】：

即使你尝试使用原始 SQL，它也需要复杂的查询，检查这个 SO 线程stackoverflow.com/questions/15926826/truncate-multi-tables
我可能在 Postgresql 文档中找到了解决方案：postgresql.org/docs/9.1/static/sql-truncate.html

标签： ruby-on-rails postgresql sql-delete

【解决方案1】：

Rails 6.0 添加新方法：truncate_tables

ActiveRecord::Base.connection.truncate_tables(:table1_name, :table2_name)

truncate_tables还没有提供api文档，它本质上是基于truncate方法

source code

  def truncate(table_name, name = nil)
    execute(build_truncate_statements(table_name), name)
  end

  def truncate_tables(*table_names) # :nodoc:
    return if table_names.empty?

    with_multi_statements do
      disable_referential_integrity do
        Array(build_truncate_statements(*table_names)).each do |sql|
          execute_batch(sql, "Truncate Tables")
        end
      end
    end
  end

Rails 6.1 进行了调整以防止错误删除与迁移相关的表source code

  def truncate_tables(*table_names) # :nodoc:
    table_names -= [schema_migration.table_name, InternalMetadata.table_name]

    return if table_names.empty?

    with_multi_statements do
      disable_referential_integrity do
        statements = build_truncate_statements(table_names)
        execute_batch(statements, "Truncate Tables")
      end
    end
  end

【讨论】：