Make Stripe_Util_Set implement IteratorAggregate

topikito · Aug 6, 2014 · f1e12d7 · f1e12d7
1 parent 7f31c53
commit f1e12d7
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/Stripe/Util/Set.php b/lib/Stripe/Util/Set.php
@@ -1,6 +1,6 @@
 <?php
 
-class Stripe_Util_Set
+class Stripe_Util_Set implements IteratorAggregate
 {
     private $_elts;
 
@@ -27,9 +27,13 @@ public function discard($elt)
         unset($this->_elts[$elt]);
     }
 
-    // TODO: make Set support foreach
     public function toArray()
     {
         return array_keys($this->_elts);
     }
+
+    public function getIterator()
+    {
+        return new ArrayIterator($this->toArray());
+    }
 }