From 02ac5dd9f1bdf0b2c99a9a86d3f1a029ad09a1c0 Mon Sep 17 00:00:00 2001
From: Jun Hiroe <Jun.Hiroe@gmail.com>
Date: Tue, 18 Mar 2014 22:48:57 +0900
Subject: Add Enumerable#count

---
 mrbgems/mruby-enum-ext/mrblib/enum.rb | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

(limited to 'mrbgems/mruby-enum-ext/mrblib')

diff --git a/mrbgems/mruby-enum-ext/mrblib/enum.rb b/mrbgems/mruby-enum-ext/mrblib/enum.rb
index 90f321596..1ca62b673 100644
--- a/mrbgems/mruby-enum-ext/mrblib/enum.rb
+++ b/mrbgems/mruby-enum-ext/mrblib/enum.rb
@@ -208,4 +208,22 @@ module Enumerable
       a
     end
   end
+
+  def count(v=nil, &block)
+    count = 0
+    if block
+      self.each do |e|
+        count += 1 if block.call(e)
+      end
+    else
+      if v == nil
+        self.each { count += 1 }
+      else
+        self.each do |e|
+          count += 1 if e == v 
+        end
+      end
+    end
+    count
+  end
 end
-- 
cgit v1.2.3