Adds IteratorReaders

src/Reader/CountableIteratorReader.php

@@ -0,0 +1,29 @@
+<?php
+
+namespace Ddeboer\DataImport\Reader;
+
+/**
+ * Use a class implementing both \Iterator and \Countable as a reader
+ *
+ * This class uses count() on iterators implementing \Countable interface
+ * and iterator_count in any further cases
+ *
+ * Be careful! iterator_count iterates through the whole iterator loading every data into the memory (for example from streams)
+ * It is not recommended for very big datasets.
+ *
+ * @author Márk Sági-Kazár <mark.sagikazar@gmail.com>
+ */
+class CountableIteratorReader extends IteratorReader implements CountableReader
+{
+    /**
+     * {@inheritdoc}
+     */
+    public function count()
+    {
+        if ($this->iterator instanceof \Countable) {
+            return count($this->iterator);
+        }
+
+        return iterator_count($this->iterator);
+    }
+}

src/Reader/IteratorReader.php

@@ -0,0 +1,21 @@
+<?php
+
+namespace Ddeboer\DataImport\Reader;
+
+use Ddeboer\DataImport\Reader;
+
+/**
+ * Use an iterator as a reader
+ *
+ * @author Márk Sági-Kazár <mark.sagikazar@gmail.com>
+ */
+class IteratorReader extends \IteratorIterator implements Reader
+{
+    /**
+     * {@inheritdoc}
+     */
+    public function getFields()
+    {
+        return array_keys($this->current());
+    }
+}