lib/beerdb/server.rb in beerdb-0.6.5 vs lib/beerdb/server.rb in beerdb-0.6.6

@@ -70,12 +70,11 @@
 
   get '/beer/:key' do |key|
 
     if ['r', 'rnd', 'rand', 'random'].include?( key )
       # special key for random beer
-      beer_id = rand( Beer.count )+1   # nb: assumes record ids sequence from 1-n
-      beer = Beer.find( beer_id )
+      beer = Beer.rnd
     else
       beer = Beer.find_by_key!( key )
     end
 
     brewery = {}
@@ -106,11 +105,10 @@
 
   get '/brewery/:key' do |key|
 
     if ['r', 'rnd', 'rand', 'random'].include?( key )
       # special key for random brewery
-      brewery_id = rand( Brewery.count )+1   # nb: assumes record ids sequence from 1-n
-      brewery = Brewery.find( brewery_id )
+      brewery = Brewery.rnd
     else
       brewery = Brewery.find_by_key!( key )
     end