p.49 make shorter the function "total_charge"

kono · Apr 6, 2010 · 398773d · 398773d
1 parent 5134d57
commit 398773d
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/chapter1/Customer.rb b/chapter1/Customer.rb
@@ -32,11 +32,7 @@ def statement
   private
 
   def total_charge
-    result = 0
-    @rentals.each do |element|
-      result += element.charge
-    end
-    result
+    @rentals.inject(0){|sum, rental| sum + rental.charge}
   end
 
 end