Rails和PostgreSQL -- 为指定的schema写migration文件

淡淡回忆

　　估计很多人看到标题都不知道是什么意思
。我也实在想不到比较清晰的标题……
　　当Rails和PostgreSQL碰到一起时，schema的意思就变得非常模糊了。对Rails而言，schema就是数据库的结构，实际上对大多数的数据库而言也是这样。而PostgreSQL中的schema指的是数据库和数据库对象（表，序列等）中间的一层，我也不知道翻译过来该叫什么，但pgAdmin中把这东西叫“模式”。用过PostgreSQL的同学应该不会陌生。下面的内容中，我会统一地把PostgreSQL中的schema叫做“模式”。
　　这篇文章，说的就是如何写这么一个迁移文件 -- 在指定的模式中进行操作，如创建表。比如以下代码的作用，就是在project模式下创建users表。

# config/initializers/migration_ext.rb
class CreateProjectTables < ActiveRecord::Migration
def self.up
run_in_schema :project do
create_table :users do |t|
t.string      :login
end
end
end
end
　　这段代码和普通的migration文件没有什么区别，只是在create_table的外层加了一个run_in_schema而已。这样，users表就不是创建在默认的public模式，而是project模式。
　　run_in_schema在这里做了两件事：

• 如果数据库中没有project模式，则创建project模式。
  
• 让block中执行的所有操作（建表，执行指定sql语句等）都只影响project模式（这不是绝对的，下面会说明）。
  

　　run_in_schema的代码如下：

class ActiveRecord::Migration
def self.run_in_schema(schema, &block)
schema = schema.to_s
# 如果没有该schema，就创建一个
schemas = select_values "SELECT * FROM pg_catalog.pg_namespace WHERE nspname <> 'information_schema' AND nspname NOT LIKE 'pg%'"
execute "CREATE SCHEMA #{schema}" unless schemas.include?(schema)
# 设置一下search_path，让建表操作优先在指定的schema内执行
re = execute "SHOW search_path"
execute "SET search_path TO #{schema}, #{re.result}"
yield
execute "SET search_path TO #{re.result}"
end
end
　　代码很简单，关键就在于search_path的设置。它把指定的模式（本文例子中的project）放在前面，后面才是原来的search_path。
　　但这样做有个问题。比如这段代码：

run_in_schema :project do
drop_table :categories
end
　　这是一个删除categories表的操作。如果此时project模式中没有该表，但public模式中有的话，则会删除public模式下的categories表。原因就在于search_path中还是有public模式，只是搜索顺序靠后了一点。
　　那修改search_path的时候，只加入project模式不就完了？之所以不这样做，完全是因为schema_migrations这个表。Rails在执行完每一个migration文件后，都会去更新schema_migrations表。如果找不到，就会抛出异常。所以我不得不把原来的search_path加在进去。这个小bug只要稍加注意，还是没什么问题的。如果有同学知道这个问题的解决方法，请赐教。

参考资料
　　Creating a multi-tenant application using PostgreSQL's schemas and Rails

　　本来我是想找找Rails中怎么判断数据库中一个模式是否存在的方法的，然后在这篇文章中找到了解决办法，然后发现他设置search_path的思路和我一样，直接取消了我发到原创博客的想法……