lib/rsolr-ext/findable.rb in mwmitchell-rsolr-ext-0.7.0 vs lib/rsolr-ext/findable.rb in mwmitchell-rsolr-ext-0.7.1

@@ -52,18 +52,19 @@
       # yield each doc if a block is given
       response['response']['docs'].each_with_index do |doc,i|
         response['response']['docs'][i] = yield(doc)
       end
     end
-    if mode == :first
-      # return only one doc
-      response['response']['docs'].first
-    elsif opts[:include_response] == false
-      # return all docs
-      response['response']['docs']
-    else
-      # return the entire response
+    if opts[:include_response] == true
       response
+    else
+      if mode == :first
+        # return only one doc
+        response['response']['docs'].first
+      else
+        # return all docs
+        response['response']['docs']
+      end
     end
   end
   
   # find_by_id(10, :handler=>'catalog')
   def find_by_id(id, solr_params={}, opts={}, &blk)
\ No newline at end of file