acts_as_followable.rb's followed_by? method is working (again?)... and simplifie…

…d lookup there a lil bit.. no foreach iteration anymore Signed-off-by: Tom Cocca <tom.cocca@gmail.com>

acts_as_followable.rb's followed_by? method is working (again?)... and simplifie…
…d lookup there a lil bit.. no foreach iteration anymore Signed-off-by: Tom Cocca <tom.cocca@gmail.com>
Jörg Battermann · Tom Cocca
1 parent f53b8ae0
Showing 1 changed file with 10 additions and 6 deletions Show diff stats
lib/acts_as_followable.rb
@@ -29,14 +29,18 @@ module ActiveRecord #:nodoc:
         end
         # Returns true if the current instance is followed by the passed record.
-        def followed_by?(follwer)
-          rtn = false
-          self.follows.each do |f|
-            rtn = true if follwer.id == f.follower_id && parent_class_name(follwer) == f.follower_type 
-          end
-          rtn
+        def followed_by?(follower)
+          self.follows.find(:first, :conditions => ["follower_id = ? AND follower_type = ?", follower.id, parent_class_name(follower)]) ? true : false
         end
+        # Retrieves the parent class name if using STI.
+        def parent_class_name(obj)
+          if obj.class.superclass != ActiveRecord::Base
+            return obj.class.superclass.name
+          end
+
+          return obj.class.name
+        end
       end
     end