如果扩展程序尚未存在,我该如何启用它?

时间:2016-08-29 19:27:41

标签: ruby-on-rails database postgresql migration rails-activerecord

我正在使用Rails 4.2.7。我想创建一个启用扩展的迁移,但前提是我正在运行的主机环境中不存在该扩展。我创建了

class EnableUuidOsspExtension < ActiveRecord::Migration
  def change
    enable_extension 'uuid-ossp'
  end
end

但是如果已经启用,我想禁止启用扩展程序。如何调整上述迁移才能实现此目的?这样做的动机是因为在我的本地机器上我必须运行它才能将它添加到PostGres,但是如果我迁移到Heroku,这个扩展可能已经到位,但是当我运行db迁移时我不想让事情崩溃脚本。

2 个答案:

答案 0 :(得分:5)

有一个extensions方法返回一个扩展名数组,所以你可以这样做:

def up
  enable_extension('uuid-ossp') unless extensions.include?('uuid-ossp')
end

def down
  disable_extension('uuid-ossp') if extensions.include?('uuid-ossp')
end

您也可以在SQL中手动执行此操作,您可以访问create extension if not exists

def up
  connection.execute('create extension if not exists "uuid-ossp"')
end

答案 1 :(得分:2)

通过Postgres文档,您可以明确标记IF NOT EXISTShttps://www.postgresql.org/docs/current/static/sql-createextension.html

这也用于PostgreSQLAdapter中的enable_extensionhttps://github.com/rails/rails/blob/master/activerecord/lib/active_record/connection_adapters/postgresql_adapter.rb#L332

  def enable_extension(name)
    exec_query("CREATE EXTENSION IF NOT EXISTS \"#{name}\"").tap {
      reload_type_map
    }
  end

此外,如果不使用它,Postgres如果已经安装了扩展程序,则不会重新创建或做一些魔术。它只会引发错误 - 在这种情况下,通常你的迁移不应该崩溃和烧毁:)