2009-08-04 67 views
7

我必須創建一個Rails遷移,它會創建許多觸發器和存儲過程。是否可以在Rails遷移中使用外部SQL文件?

通常情況下,人們會使用execute方法來做到這一點,但由於語句的大小,我寧願將它們保存在外部文件中並從遷移中引用它。

我該怎麼做?它甚至有可能嗎?

回答

10

您可以將它們存儲在文本文件中並通過File對象讀取它們。

sql = "" 
source = File.new("./sql/procedures.sql", "r") 
while (line = source.gets) 
    sql << line 
end 
source.close 
execute sql 

這是醜陋的,但工程。我強烈建議保留遷移中的存儲過程/觸發器以便輕鬆回滾。

如果你做的「外部文件」的方法,你需要保持每遷移兩個文件,一個是將所有的東西,一個是在下降的情況下的在:

rake db:rollback 
+2

的ActiveRecord :: Base的.establish_connection ActiveRecord :: Base.connection.execute(sql) – drhenner 2012-08-23 19:22:48

2

如果文件中只有一條語句,Mike的答案沒有問題,但如果有更多語句(例如多個插入和更新),ActiveRecord將會失敗,因爲它默認情況下不支持具有一個調用的多個語句。

一個解決方案是修改ActiveRecord以支持多條語句,如here指示。

其他的解決辦法是,以確保您的SQL文件包含每行只有一個語句,並使用循環像

source = File.open "db/foo.sql", "r" 
source.readlines.each do |line| 
    line.strip! 
    next if line.empty? # ensure that rows that contains newlines and nothing else does not get processed 
    execute line 
end 
source.close 
2

我做了以下內容,我們需要:

class RawSqlMigration < ActiveRecord::Migration 
    def up 
    execute File.read(File.expand_path('../../sql_migrations/file.sql', __FILE__)) 
    end 
end 
相關問題