从控制器/助手重置数据库，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

当您的连接处于活动状态时，您不能删除数据库。相反，您可以做的是发出 TRUNCATE 命令，从表中删除所有行并重置 ID 列。

如果您想要更精细地控制哪些表被截断，您可以向模型添加一个类方法：

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

Answer 2

我从以下 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