cleaning up criteria

pbrumm · Sep 1, 2011 · 5f5fdd5 · 5f5fdd5
1 parent 688430d
commit 5f5fdd5
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/lib/sunspot/mongoid.rb b/lib/sunspot/mongoid.rb
@@ -32,13 +32,18 @@ def id
 
     class DataAccessor < Sunspot::Adapters::DataAccessor
       def load(id)
-        @clazz.find(BSON::ObjectID.from_string(id)) rescue nil
+        criteria(id).first
       end
 
       def load_all(ids)
-        @clazz.where(:_id.in => ids.map { |id| BSON::ObjectId.from_string(id) })
+        criteria(ids)
+      end
+
+      private
+
+      def criteria(id)
+        @clazz.criteria.id(id)
       end
-
     end
   end
 end