更新日: 2014年6月2日
Rails

db/seeds.rb（rake db:seed）でSQLからDBに初期データ追加

Rails では db/seeds.rb を作成して rake db:seed コマンドでDBに初期データを追加できますが、そのデータ追加を SQL ステートメントから行う方法。以下のページが大変参考になりました。

Seed your Rails App by Importing from SQL | panupan.com

【お知らせ】英単語を画像イメージで楽に暗記できる辞書サイトを作りました。英語学習中の方は、ぜひご利用ください！

画像付き英語辞書 Imagict | 英単語をイメージで暗記
【開発記録】
英単語を画像イメージで暗記できる英語辞書サービスを作って公開しました

— 環境 —
Rails 4.0.1

db/seeds.rb の作成

上記ページのコードを参考にして変更した以下のコードを書いた。

db/seeds.rb

unless Rails.env.production?
  connection = ActiveRecord::Base.connection
  tables = %w[table1 table2 table3]

  tables.each do |table|
    connection.execute("TRUNCATE #{table}") unless table == "schema_migrations"
  end
   
  # - IMPORTANT: SEED DATA ONLY
  # - DO NOT EXPORT TABLE STRUCTURES
  # - DO NOT EXPORT DATA FROM `schema_migrations`
  Dir.glob("db/sql_backup/*.sql").each do |sql_file|
    sql = File.read(sql_file)
    statements = sql.split(/;$/)
    statements.pop  # the last empty statement
   
    ActiveRecord::Base.transaction do
      statements.each do |statement|
        connection.execute(statement)
      end
    end
  end
end

unless Rails.env.production?

connection = ActiveRecord::Base.connection

tables = %w[table1 table2 table3]

tables.each do |table|

connection.execute("TRUNCATE #{table}") unless table == "schema_migrations"

end

# - IMPORTANT: SEED DATA ONLY

# - DO NOT EXPORT TABLE STRUCTURES

# - DO NOT EXPORT DATA FROM `schema_migrations`

Dir.glob("db/sql_backup/*.sql").each do |sql_file|

sql = File.read(sql_file)

statements = sql.split(/;$/)

statements.pop # the last empty statement

ActiveRecord::Base.transaction do

statements.each do |statement|

connection.execute(statement)

end

connection.tables は使わずに、tables 配列に手動で初期データを追加するテーブルを指定しました。また、Dir.glob で db/sql_backup/ 以下の、DB初期データ用の sql ファイルを全て読み込むようにしました。

DB に初期データ追加。


$ bundle exec rake db:seed

$ bundle exec rake db:seed

必要な初期データを db/sql_backup/*.sql と sql ファイルで保存すればOK。これでいつでも、rake db:seed で DB に初期データを投入できる。

外部キー制約のあるテーブルを truncate しようとするとエラー

最初、参考ページのコード通りに connection.tables を使って schema_migrations 以外の全てのテーブルに対して TRUNCATE しようとしたらエラーが発生しました。

connection.tables.each do |table|
  connection.execute("TRUNCATE #{table}") unless table == "schema_migrations"
end

connection.tables.each do |table|

connection.execute("TRUNCATE #{table}") unless table == "schema_migrations"

end

これだと以下のように、外部キー制約のあるテーブルの truncate ができない。


$ bundle exec rake db:seed
rake aborted!
ActiveRecord::StatementInvalid: Mysql2::Error: Cannot truncate a table referenced in a foreign key constraint

$ bundle exec rake db:seed

rake aborted!

ActiveRecord::StatementInvalid: Mysql2::Error: Cannot truncate a table referenced in a foreign key constraint

以下のように一旦外部キー制約をオフにして TRUNCATE して、テーブルを TRUNCATE した後、再び外部キー制約をオンにするという方法が使えるらしい。


mysql> SET FOREIGN_KEY_CHECKS=0;
mysql> TRUNCATE table1;
mysql> TRUNCATE table2;
mysql> SET FOREIGN_KEY_CHECKS=1;

mysql> SET FOREIGN_KEY_CHECKS=0;

mysql> TRUNCATE table1;

mysql> TRUNCATE table2;

mysql> SET FOREIGN_KEY_CHECKS=1;

db/seeds.rb に書くなら以下のような感じかな。

connection.execute("SET FOREIGN_KEY_CHECKS=0;")
connection.tables.each do |table|
  connection.execute("TRUNCATE #{table}") unless table == "schema_migrations"
end
connection.execute("SET FOREIGN_KEY_CHECKS=1;")

connection.execute("SET FOREIGN_KEY_CHECKS=0;")

connection.tables.each do |table|

connection.execute("TRUNCATE #{table}") unless table == "schema_migrations"

end

connection.execute("SET FOREIGN_KEY_CHECKS=1;")

しかし、私はこのやり方は若干怖い気がしたので、上記コードに書いたとおり初期データを追加するDBテーブル名を tables という配列で指定しました。外部キー制約のないテーブルに対する初期データ追加なら、これでOK。

パーフェクト Ruby on Rails は、最近読んだ Rails 本の中では一番役に立った本です。Chef や Capistrano など Rails と共によく使用される技術にも触れてあります。Ruby on Rails 4 アプリケーションプログラミングは、入門的な内容で Rails の機能全体を網羅されています。

パーフェクト Ruby on Rails

Ruby on Rails 4 アプリケーションプログラミング