[Finder] Add optional pass limit to Finder::rows() to allow only gather maximum until provided limit

Author: samsonasik

src/Finder.php

@@ -16,6 +16,8 @@
 use function key;
 use function prev;
 
+use const PHP_INT_MAX;
+
 final class Finder
 {
     /**
@@ -123,10 +125,11 @@ public static function last(
      * @param callable(mixed $datum, int|string|null $key=): bool $filter
      * @return mixed[]
      */
-    public static function rows(iterable $data, callable $filter, bool $preserveKey = false): array
+    public static function rows(iterable $data, callable $filter, bool $preserveKey = false, int $limit = PHP_INT_MAX): array
     {
         $rows   = [];
         $newKey = 0;
+        $total  = 0;
 
         foreach ($data as $key => $datum) {
             $isFound = $filter($datum, $key);
@@ -146,6 +149,11 @@ public static function rows(iterable $data, callable $filter, bool $preserveKey
             }
 
             $rows[$rowKey] = $datum;
+
+            ++$total;
+            if ($total === $limit) {
+                break;
+            }
         }
 
         return $rows;

tests/FinderTest.php

@@ -323,4 +323,17 @@ public static function rowsDataProviderPreserveKey(): array
             ],
         ];
     }
+
+    public function testRowsWithLimit(): void
+    {
+        $data   = [1, 2];
+        $filter = static fn ($datum): bool => $datum >= 0;
+
+        $this->assertSame(
+            [
+                1,
+            ],
+            Finder::rows($data, $filter, limit: 1)
+        );
+    }
 }