Rake задача для усечения всех таблиц в Rails 3

Question

Я бы хотел получить грабли для усечения всех таблиц. Я нашел один в Интернете, но он предназначен только для Rails 2 и не работает для Rails 3 (проблема в получении соединения с базой данных).

rake db:reset не вариант, потому что я использую PostgreSQL, и он также удаляет пользователя. Поэтому миграция не удалась. Я только хочу очистить данные.

Ребята, у вас есть что-нибудь для меня?

Answer 1

Я нашел это через Google, и затем я нашел намного более простое решение, чем одобренное, поэтому вот оно: Используйте database_cleaner gem. Вот шаги.

В вашем Gemfile (выполнить связку после изменения):

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

С этим гемом оператор 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

Вот и все. Вы также можете напрямую использовать строку DatabaseCleaner.clean_with :truncation внутри файла db/seeds.rb, чтобы не забыть обрезать базу данных перед заполнением.

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

Этот пример основан на приведенном ниже коде Криса Леде (спасибо) и работает с Rails 3.X .

Спасибо за все подсказки.

Answer 3

Согласно ответу Криса Леде, это становится намного проще:

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

Answer 4

Вы всегда можете перейти на версию 0, например, так:

rake db:migrate VERSION=0

Таким образом, вам даже не нужно обрезать свои таблицы, и вы можете выполнить миграцию снова.Единственный улов в том, что вам нужно, чтобы ваши миграции down работали правильно.

Это решение работает в рельсах 3, несмотря на тот факт, что версии основаны на отметках времени.

Это решение выглядит так: https://stackoverflow.com/a/1196822/241367

Кроме того, вы всегда можете выполнить следующее, если ваш schema.rb обновлен:

rake db:schema:load

И как @Кикито предлагает, вы можете запустить database_cleaner (это то, что cucumber и rspec нравится использовать между тестами) примерно так:

DatabaseCleaner.clean_with :truncation

Answer 5

Это позволит получить все таблицы в вашей базе данных, найти модель, связанную с этой таблицей, и вызвать # destroy_all .

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

Answer 6

Ответ, данный lzap, имеет одну конкретную проблему. Rails хочет снова запустить все миграции. Следующий код соответствует предложению Энтони Альберто, и он работает. Это дополнение сверяется с таблицей 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