BigW Consortium Gitlab

20141121161704_add_identity_table.rb 1.22 KB
Newer Older
1 2 3 4 5 6 7 8 9 10
class AddIdentityTable < ActiveRecord::Migration
  def up
    create_table :identities do |t|
      t.string :extern_uid
      t.string :provider
      t.references :user
    end

    add_index :identities, :user_id

11 12 13 14 15
    execute <<eos
INSERT INTO identities (provider, extern_uid, user_id)
SELECT provider, extern_uid, id FROM users
WHERE provider IS NOT NULL
eos
16

17 18 19 20
    if index_exists?(:users, ["extern_uid", "provider"])
      remove_index :users, ["extern_uid", "provider"]
    end

21 22
    remove_column :users, :extern_uid
    remove_column :users, :provider
23 24 25
  end

  def down
26 27 28
    add_column :users, :extern_uid, :string
    add_column :users, :provider, :string

29 30 31 32 33 34 35 36 37 38
    if ActiveRecord::Base.connection.adapter_name == 'PostgreSQL'
      execute <<eos
UPDATE users u
SET provider = i.provider, extern_uid = i.extern_uid
FROM identities i
WHERE i.user_id = u.id
eos
      else
        execute "UPDATE users u, identities i SET u.provider = i.provider, u.extern_uid = i.extern_uid WHERE u.id = i.user_id"
      end
39 40

    drop_table :identities
41 42 43 44

    unless index_exists?(:users, ["extern_uid", "provider"])
      add_index "users", ["extern_uid", "provider"], name: "index_users_on_extern_uid_and_provider", unique: true, using: :btree
    end
45 46
  end
end