activerecord-import(multi-level)の使い方

Question

activerecord-import(multi-level)の使い方で悩んでいます。
実験用のコードを書いてみましたが、思うようにデータベースが更新されません。
コードの書き方や考え方で間違っているところをご指摘いただけると幸いです。

リンク : activerecord-import(multi-level)

$ ruby -v
ruby 1.9.3p551 (2014-11-13 revision 48407) [x86_64-darwin14.0.0]

# https://github.com/zdennis/activerecord-import/wiki#multi-level-example

require "pp"
require "rubygems"
require "active_record"
require "activerecord-import"

# 環境設定
ActiveRecord::Base.establish_connection(
  adapter:  "sqlite3",
  database: "main.db"
)

# クラス宣言
class Book   < ActiveRecord::Base 
  has_many :reviews
end
class Review < ActiveRecord::Base 
  belongs_to :book
end

# テーブルの作成 
class CreateBook < ActiveRecord::Migration
  def self.up
    create_table :books, force: true do|t|
      t.string :name
    end
  end
end
CreateBook.new.up
class CreateReview < ActiveRecord::Migration
  def self.up
    create_table :reviews, force: true do|t|
      t.integer :book_id
      t.string :title
    end
  end
end
CreateReview.new.up

# エントリーの追加
books = []
10.times do |i|
  book = Book.new(:name => "book #{i}")
  book.reviews.build(:title => "Excellent")
  books << book
end
Book.import books, recursive: true

# レコード取得
pp Book.all   # OK : [#<Book id: ...
pp Review.all # NG : []

Answer 1

んー、この機能は、sqliteやMySQLはサポートしてないようです。PostgresSQLでは有効です。
おそらく、primary_keyがNULLの状態でもセットできるようなデータベース側でのサポートが必要です。
機能を持っているかは、アダプターの
support_setting_primary_key_of_imported_objects?
で判断しています。これを以下で無理やりtrueにしても、動かないので対応していないのかと思います。
importは、オブジェクトごとにbulk-insertすれば同じことはできそうです。

#無理やり機能をtrueにしたが失敗記録、正しくないかもしれません。
# 環境設定
require "pp"
require "rubygems"
require "active_record"
require "activerecord-import"

module SQLite3Adapter
  def support_setting_primary_key_of_imported_objects?
      true
  end
end

con = ActiveRecord::Base.establish_connection(
  adapter:  "sqlite3",
  database: "main.db"
)
p con.extend(SQLite3Adapter)
p con.respond_to?(:support_setting_primary_key_of_imported_objects?)

# クラス宣言
class Book   < ActiveRecord::Base
  has_many :reviews
end
class Review < ActiveRecord::Base
  belongs_to :book
end

# テーブルの作成
class CreateBook < ActiveRecord::Migration
  def self.up
    create_table :books, force: true do|t|
      t.string :name
    end
  end
end
CreateBook.new.up
class CreateReview < ActiveRecord::Migration
  def self.up
    create_table :reviews, force: true do|t|
      t.integer :book_id
      t.string :title
    end
  end
end
CreateReview.new.up

# エントリーの追加
books = []
10.times do |i|
  book = Book.new(:name => "book #{i}")
  book.reviews.build(:title => "Excellent")
  books << book
end
pp books
pp books.first.reviews
Book.import books, recursive: true

# レコード取得
pp Book.all   # OK : [#<Book id: ...
pp Book.first.reviews
pp Review.all # NG : []

Comments

[kanedayo]

ありがとうございます。
PostgressSQLでは動くという情報は貴重でした。

ご指摘の通り、bulk-insertを検討したところ、
そこから「activerecord-import」に辿り着きました。

> importは、オブジェクトごとにbulk-insertすれば同じことはできそうです。
この場合、オブジェクトの関連(この例だとbook_id)の管理で悩みが生じます。

uniqなbook_idをアプリ側で管理できれば、book_idを事前に詰めこんでbulk-insertできそうだと気付いたのですが、適切な実現方法がわかりません。

[kanedayo]

その後、postgres特有のreturning句というのを知り、
「データベースに依存する」という意味を理解できた気がします。

また、その解決方法として、
採番テーブルやLAST_INSERT_ID()を用いた
uniqueなIDの管理方法があることを知りました。

気付きを与えてくださりありがとうございました。

Postgresで利用できるRETURNING句の例：
INSERT INTO books(name) VALUES('~') RETURNING id;