feat: limit imports

.gitignore

@@ -1,3 +1,4 @@
 /vendor
 /.idea
 /build
+.phpunit.result.cache

README.md

@@ -85,6 +85,12 @@ Import a `csv` with specific delimiter, enclosure characters and "gbk" encoding:
 $collection = (new FastExcel)->configureCsv(';', '#', 'gbk')->import('file.csv');
 ```
 
+Importing only the first 100 lines from a `csv` :
+
+```php
+$collection = (new FastExcel)->limitRows(100)->import('file.csv');
+```
+
 Import and insert to database:
 
 ```php

src/FastExcel.php

@@ -36,6 +36,11 @@ class FastExcel
      */
     private $start_row = 1;
 
+    /**
+     * @var ?int
+     */
+    private $end_row = null;
+
     /**
      * @var bool
      */
@@ -127,6 +132,16 @@ public function startRow(int $row)
         return $this;
     }
 
+    /**
+     * @return $this
+     */
+    public function limitRows(int $row = null)
+    {
+        $this->end_row = $row;
+
+        return $this;
+    }
+
     /**
      * @return $this
      */

src/Importable.php

@@ -12,6 +12,7 @@
  * Trait Importable.
  *
  * @property int  $start_row
+ * @property ?int  $end_row
  * @property bool $transpose
  * @property bool $with_header
  */
@@ -140,7 +141,7 @@ private function importSheet(SheetInterface $sheet, callable $callback = null)
 
         foreach ($sheet->getRowIterator() as $k => $rowAsObject) {
             $row = $rowAsObject->toArray();
-            if ($k >= $this->start_row) {
+            if ($k >= $this->start_row ) {
                 if ($this->with_header) {
                     if ($k == $this->start_row) {
                         $headers = $this->toStrings($row);
@@ -161,6 +162,10 @@ private function importSheet(SheetInterface $sheet, callable $callback = null)
                     $collection[] = empty($headers) ? $row : array_combine($headers, $row);
                 }
             }
+
+            if ($k > $this->end_row) {
+                break;
+            }
         }
 
         if ($this->transpose) {

tests/LimitTest.php

@@ -0,0 +1,36 @@
+<?php
+
+namespace Rap2hpoutre\FastExcel\Tests;
+
+use Rap2hpoutre\FastExcel\FastExcel;
+
+/**
+ * Class ChunkTest.
+ */
+class LimitTest extends TestCase
+{
+    public function collectionGenerator($n)
+    {
+        for ($i = 1; $i <= $n; $i++) {
+            yield collect(['a' => 'b', 'c' => 'd']);
+        }
+    }
+
+    public function arrayGenerator($n)
+    {
+        for ($i = 1; $i <= $n; $i++) {
+            yield ['a' => 'b', 'c' => 'd'];
+        }
+    }
+
+    public function testWithGenerator()
+    {
+        (new FastExcel($this->collectionGenerator(100)))->export(__DIR__.'/test-generator.xlsx');
+        $result = (new FastExcel())->limitRows(10)->import(__DIR__.'/test-generator.xlsx');
+        $this->assertEquals(
+            collect($this->arrayGenerator(10)),
+            $result
+        );
+        unlink(__DIR__.'/test-generator.xlsx');
+    }
+}