randumb 0.1.2 → 0.1.3

data/lib/randumb.rb

@@ -8,7 +8,11 @@ module Randumb
     
     module Relation
       
-      def random(max_items = 1)
+      def random(max_items = nil)
+        # return only the first record if method was called without parameters
+        return_first_record = max_items.nil?
+        max_items ||= 1
+
         # take out limit from relation to use later
     
         relation = clone
@@ -32,7 +36,13 @@ module Randumb
           ids[rand_index] = id_results[rand_index]["id"] unless ids.has_key?(rand_index)
         end
 
-        klass.select(original_selects).includes(original_includes).find_all_by_id(ids.values)
+        records = klass.select(original_selects).includes(original_includes).find_all_by_id(ids.values)
+                
+        if return_first_record
+          records.first
+        else
+          records
+        end
       end
 
     end # Relation
@@ -40,7 +50,7 @@ module Randumb
     module Base
       
       # Class method
-      def random(max_items = 1)
+      def random(max_items = nil)
         relation.random(max_items)
       end
       
@@ -58,4 +68,4 @@ end
 
 class ActiveRecord::Base
   extend Randumb::ActiveRecord::Base
-end
+end

metadata

@@ -2,7 +2,7 @@
 name: randumb
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors: 
 - Zachary Kloepping
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-08-18 00:00:00 Z
+date: 2011-08-30 00:00:00 Z
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: rspec