Сбросить DB из контроллера/помощника, Ruby on Rails

Question

Я создал демоверсию моей программы. Демонстрация используется для представления программы клиентам. Перед каждой презентации я хочу сбросить БД, и вставить некоторые данные, действия, которые я должен сделать это:

rake db:drop 
rake db:setup 
rake db:seed:demo 

Но они не всегда мне представить его, а другие не использовать консоль, так что я хотел поставить на страницу настроек возможность сброса с помощью кнопки.

Я пытался таким образом:

module SettingsHelper 
require 'rake' 
    def reset 
     Rake::Task[db:drop].invoke 
     Rake::Task[db:setup].invoke 
     Rake::Task[db:seeds:demo].invoke 
    end 
end 

Я положил в форме:

<%=button_to "Reset", reset, class:"btn red" %> 

Но я не могу удалить базу данных с активным сервером ... как я могу сделать?

Answer 1

Я истреблю код из следующих SO отвечает

• https://stackoverflow.com/a/1196202/4758119
• https://stackoverflow.com/a/5212588/4758119

Вы можете добавить что-то вроде этого в контроллере

def reset_db 
    conn = ActiveRecord::Base.connection 
    tables = ActiveRecord::Base.connection.tables 
    tables.each { |t| conn.execute("TRUNCATE #{t}") } 

    Rails.application.load_seed 
end 

Answer 2

Вы не можете оставить базу данных, пока у вас активное соединение. Вместо этого вы можете сделать команду TRUNCATE, которая удалит все строки из таблицы и сбросит столбцы идентификаторов.

Incase вы хотите более мелкозернистого контроль над точно, какие таблицы усечены вы можете добавить метод класса модели:

class User < ActiveRecord::Base 
    self.truncate! 
    self.connection.execute(
     "TRUNCATE TABLE #{ self.table_name };" 
    ) 
    self.connection.reset_pk_sequence!(self.table_name) 
    end 
end 

Но позволяет DRY его в модуль таким образом, мы не должны повторять это во всех моделях, которые должны иметь такое поведение:

# app/models/concerns/resetable.rb 
module Resetable 
    extend ActiveSupport::Concern 
    class_methods do 
    def truncate! 
     self.connection.execute(
     "TRUNCATE TABLE #{ self.table_name };" 
    ) 
     self.connection.reset_pk_sequence!(self.table_name) 
    end 
    end 
end 

# app/models/user.rb 
class User < ActiveRecord::Base 
    include Resetable 
    # ... 
end 

# app/models/book.rb 
class Book < ActiveRecord::Base 
    include Resetable 
    # ... 
end 
# ... 

Вы можете сбросить данные для каждой модели по:

def reset_db 
    [User, Book, Thing].each { |model| model.truncate! } 
    Rails.application.load_seed 
end 

Если вы просто хотите, чтобы ядерное оружие всех таблиц, вы можете просто сделать:

ActiveRecord::Base.connection.tables.each do |t| 
    conn = ActiveRecord::Base.connection 
    conn.execute("TRUNCATE TABLE #{t} CASCADE;") 
    conn.reset_pk_sequence!(t) 
end