Added dependent destroy

lethunder · Apr 27, 2009 · 8739a75 · 8739a75
1 parent a30d272
commit 8739a75
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/lib/friendship.rb b/lib/friendship.rb
@@ -19,10 +19,10 @@ class Friendship < ActiveRecord::Base
   # associations
   belongs_to :user
   belongs_to :friend, :class_name => 'User', :foreign_key => 'friend_id'
-  belongs_to :message, :class_name => "FriendshipMessage", :foreign_key => "friendship_message_id"
+  belongs_to :message, :class_name => "FriendshipMessage", :foreign_key => "friendship_message_id", :dependent => :destroy
 
-  has_many :friendship_relations, :class_name => "FriendshipRelationType", :readonly => true
-  has_many :relations, :through => :friendship_relations
+  has_many :friendship_relations, :class_name => "FriendshipRelationType", :readonly => true, :dependent => :destroy
+  has_many :relations, :through => :friendship_relations, :class_name => "RelationType", :source => :relation
 
   # callback
   after_destroy do |f|

diff --git a/lib/relation_type.rb b/lib/relation_type.rb
@@ -1,3 +1,10 @@
 class RelationType < ActiveRecord::Base
   validates_presence_of :name
+
+  after_destroy :destroy_all_friendsip_relations
+
+  private
+    def destroy_all_friendsip_relations
+      FriendshipRelationType.delete_all :relation_id => self.id
+    end
 end