Rake 任务截断 Rails 3 中的所有表

Question

我想要一个 rake 任务来截断所有表。我找到了一个 在互联网上，但它只适用于 Rails 2，不适用于 Rails 3 （问题在于获取数据库连接）。

rake db:reset 不是一个选项，因为我使用的是 PostgreSQL，它也会删除用户。因此迁移失败。我只想清除数据。

你们有什么事情要给我吗？

Answer 1

我通过谷歌找到了这个，然后我得到了一个比批准的解决方案更简单的解决方案，所以这里是：使用 database_cleaner 宝石。这是步骤。

在 Gemfile 中（修改后执行包）：

gem 'database_cleaner' # you might want to limit this to the dev and staging group

使用该 gem 后，语句 DatabaseCleaner.clean_with :truncation 将截断数据库。将其添加到 rake 任务中很简单：

# tasks/db/clean.rake

namespace :db do

  desc "Truncate all existing data"
  task :truncate => "db:load_config" do
    DatabaseCleaner.clean_with :truncation
  end

end

就是这样。您还可以直接使用 db/seeds.rb 文件中的 DatabaseCleaner.clean_with :truncation 行，这样您就不会忘记在播种前截断数据库。

Answer 2

因此，我将链接的示例编辑为：

namespace :db do
  desc "Truncate all existing data"
  task :truncate => "db:load_config" do
   begin
    config = ActiveRecord::Base.configurations[::Rails.env]
    ActiveRecord::Base.establish_connection
    case config["adapter"]
      when "mysql", "postgresql"
        ActiveRecord::Base.connection.tables.each do |table|
          ActiveRecord::Base.connection.execute("TRUNCATE #{table}")
        end
      when "sqlite", "sqlite3"
        ActiveRecord::Base.connection.tables.each do |table|
          ActiveRecord::Base.connection.execute("DELETE FROM #{table}")
          ActiveRecord::Base.connection.execute("DELETE FROM sqlite_sequence where name='#{table}'")
        end                                                                                                                               
       ActiveRecord::Base.connection.execute("VACUUM")
     end
    end
  end
end

此示例基于 Chris Ledet 的以下代码（谢谢），并适用于 Rails 3.X。

感谢所有提示。

Answer 3

在 Rails 6 中截断数据库

rails db:truncate_all  

Answer 4

根据 Chris Ledet 的回答，这变得简单得多：

ActiveRecord::Base.connection.tables.each do |table|
    ActiveRecord::Base.connection.execute("TRUNCATE TABLE #{table};")
end

Answer 5

您始终可以迁移到版本 0，如下所示：

rake db:migrate VERSION=0

这样您甚至不必截断表，然后就可以再次迁移。唯一的问题是您需要down迁移才能正常工作。

尽管版本是基于时间戳的，但该解决方案确实适用于 Rails 3。

此解决方案如下所示： https://stackoverflow.com/a/1196822/241367

此外，您可以随时运行假设您的 schema.rb 是最新的，如下所示：

rake db:schema:load

正如@kikito 建议的那样，您可以运行database_cleaner（它是cucumber 和 rspec 喜欢在测试之间使用什么），如下所示：

DatabaseCleaner.clean_with :truncation

Answer 6

这将获取数据库中的所有表，找到与该表关联的模型并调用 #destroy_all。

tables = ActiveRecord::Base.connection.tables
tables.each do |tbl|
 # "users" => User
 tbl.classify.constantize.destroy_all
end

Answer 7

lzap给出的答案有一个具体问题。 Rails 希望再次运行所有迁移。以下代码是 Anthony Alberto 建议的并且它有效。此添加检查 schema_migrations 表

namespace :db do
  desc "Truncate all existing data"
  task :truncate => "db:load_config" do
   begin
    config = ActiveRecord::Base.configurations[::Rails.env]
    ActiveRecord::Base.establish_connection
    case config["adapter"]
      when "mysql", "postgresql"
        ActiveRecord::Base.connection.tables.each do |table|
          ActiveRecord::Base.connection.execute("TRUNCATE #{table}") if table != "schema_migrations"
        end
      when "sqlite", "sqlite3"
        ActiveRecord::Base.connection.tables.each do |table|
          ActiveRecord::Base.connection.execute("DELETE FROM #{table}") if table != "schema_migrations"
          ActiveRecord::Base.connection.execute("DELETE FROM sqlite_sequence where name='#{table}'") if table != "schema_migrations"
        end                                                                                                                               
       ActiveRecord::Base.connection.execute("VACUUM")
     end
    end
  end
end