20101104182107_add_unique_index_on_members.rb 1.34 KB
Newer Older
1
#-- encoding: UTF-8
2 3
#-- copyright
# ChiliProject is a project management system.
4
#
Holger Just's avatar
Holger Just committed
5
# Copyright (C) 2010-2012 the ChiliProject Team
6
#
7 8 9 10
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
11
#
12 13 14
# See doc/COPYRIGHT.rdoc for more details.
#++

15 16
class AddUniqueIndexOnMembers < ActiveRecord::Migration
  def self.up
17 18
    # Clean and reassign MemberRole rows if needed
    MemberRole.delete_all("member_id NOT IN (SELECT id FROM #{Member.table_name})")
19 20 21 22 23 24 25 26 27
    MemberRole.update_all("member_id =" +
      " (SELECT min(m2.id) FROM #{Member.table_name} m1, #{Member.table_name} m2" +
      " WHERE m1.user_id = m2.user_id AND m1.project_id = m2.project_id" +
      " AND m1.id = #{MemberRole.table_name}.member_id)")
    # Remove duplicates
    Member.connection.select_values("SELECT m.id FROM #{Member.table_name} m" +
      " WHERE m.id > (SELECT min(m1.id) FROM #{Member.table_name} m1 WHERE m1.user_id = m.user_id AND m1.project_id = m.project_id)").each do |i|
        Member.delete_all(["id = ?", i])
      end
28

29 30 31 32 33
    # Then add a unique index
    add_index :members, [:user_id, :project_id], :unique => true
  end

  def self.down
34
    remove_index :members, [:user_id, :project_id]
35 36
  end
end