FIXED: sunspot rails to work with non-standard AR primary key, ie if …

…use the set_primary_key on a model
outoftime · Mar 10, 2010 · 8c19eed · 8c19eed
1 parent 8c813c2
commit 8c19eed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sunspot_rails/lib/sunspot/rails/adapters.rb b/sunspot_rails/lib/sunspot/rails/adapters.rb
@@ -47,7 +47,7 @@ def select=(value)
         # ActiveRecord::Base:: ActiveRecord model
         # 
         def load(id)
-          @clazz.find_by_id(id.to_i, options_for_find)
+          @clazz.send("find_by_#{@clazz.primary_key}", id.to_i, options_for_find)
         end
 
         # 
@@ -62,7 +62,7 @@ def load(id)
         # Array:: Collection of ActiveRecord models
         #
         def load_all(ids)
-          @clazz.find_all_by_id(ids.map { |id| id.to_i }, options_for_find)
+          @clazz.send("find_all_by_#{@clazz.primary_key}", ids.map { |id| id.to_i }, options_for_find)
         end
 
         private