有几次,我想重构某些模型的设计并最终将更新逻辑放入迁移中。但是,据我了解,这不是一个好的做法(特别是因为鼓励您使用架构文件进行部署,而不是迁移)。你如何处理这类问题?
为了澄清我的意思,假设我有一个用户模型。因为我认为只有两种用户,即“普通”用户和管理员,所以我选择使用一个简单的布尔字段来判断用户是否是管理员。
然而,在我想我需要第三类用户之后,也许是版主或类似的东西。在这种情况下,我添加了一个 UserType 模型(和相应的迁移),以及从用户表中删除“admin”标志的第二个迁移。问题来了。在“add_user_type_to_users”迁移中,我必须将管理标志值映射到用户类型。此外,为了做到这一点,用户类型必须存在,这意味着我不能使用种子文件,而是在迁移中创建用户类型(也被认为是不好的做法)。这里有一些代表这种情况的虚构代码:
class CreateUserTypes < ActiveRecord::Migration
def self.up
create_table :user_types do |t|
t.string :name, :nil => false, :unique => true
end
#Create basic types (can not put in seed, because of future migration dependency)
UserType.create!(:name => "BASIC")
UserType.create!(:name => "MODERATOR")
UserType.create!(:name => "ADMINISTRATOR")
end
def self.down
drop_table :user_types
end
end
class AddTypeIdToUsers < ActiveRecord::Migration
def self.up
add_column :users, :type_id, :integer
#Determine type via the admin flag
basic = UserType.find_by_name("BASIC")
admin = UserType.find_by_name("ADMINISTRATOR")
User.all.each {|u| u.update_attribute(:type_id, (u.admin?) ? admin.id : basic.id)}
#Remove the admin flag
remove_column :users, :admin
#Add foreign key
execute "alter table users add constraint fk_user_type_id
foreign key (type_id) references user_types (id)"
end
def self.down
#Re-add the admin flag
add_column :users, :admin, :boolean, :default => false
#Reset the admin flag (this is the problematic update code)
admin = UserType.find_by_name("ADMINISTRATOR")
execute "update users set admin=true where type_id=#{admin.id}"
#Remove foreign key constraint
execute "alter table users drop foreign key fk_user_type_id"
#Drop the type_id column
remove_column :users, :type_id
end
end
如您所见,有两个有问题的部分。首先是第一个模型中的行创建部分,如果我想连续运行所有迁移,这是必要的,然后是第二个迁移中的“更新”部分,它将“admin”列映射到“type_id”列。
有什么建议吗?