Version 2: 🚀 Faster process with early validate filter before loop

Author: samsonasik

rector.php

@@ -3,7 +3,7 @@
 declare(strict_types=1);
 
 use Rector\Config\RectorConfig;
-use Rector\PHPUnit\Set\PHPUnitSetList;
+use Rector\TypeDeclaration\Rector\ClassMethod\BoolReturnTypeFromBooleanStrictReturnsRector;
 
 return RectorConfig::configure()
     ->withPhpSets(php81: true)
@@ -15,8 +15,11 @@
         privatization: true,
         typeDeclarations: true
     )
-    ->withSets([
-        PHPUnitSetList::PHPUNIT_100,
+    ->withComposerBased(phpunit: true)
+    ->withSkip([
+        BoolReturnTypeFromBooleanStrictReturnsRector::class => [
+            __DIR__ . '/tests/FilterTest.php',
+        ],
     ])
     ->withParallel()
     ->withRootFiles()

src/AtLeast.php

@@ -4,6 +4,7 @@
 
 namespace ArrayLookup;
 
+use ArrayLookup\Assert\Filter;
 use Traversable;
 use Webmozart\Assert\Assert;
 
@@ -47,14 +48,13 @@ private static function atLeastFoundTimes(
     ): bool {
         // usage must be higher than 0
         Assert::greaterThan($maxCount, 0);
+        // filter must be a callable with bool return type
+        Filter::boolean($filter);
 
         $totalFound = 0;
         foreach ($data as $key => $datum) {
             $isFound = $filter($datum, $key);
 
-            // returns of callable must be bool
-            Assert::boolean($isFound);
-
             if (! $isFound) {
                 continue;
             }

src/Collector.php

@@ -4,6 +4,7 @@
 
 namespace ArrayLookup;
 
+use ArrayLookup\Assert\Filter;
 use Traversable;
 use Webmozart\Assert\Assert;
 
@@ -71,6 +72,11 @@ public function getResults(): array
         $collectedData  = [];
         $isCallableWhen = is_callable($this->when);
 
+        if (is_callable($this->when)) {
+            // filter must be a callable with bool return type
+            Filter::boolean($this->when);
+        }
+
         foreach ($this->data as $key => $datum) {
             if ($isCallableWhen) {
                 /**
@@ -79,8 +85,6 @@ public function getResults(): array
                 $when    = $this->when;
                 $isFound = ($when)($datum, $key);
 
-                Assert::boolean($isFound);
-
                 if (! $isFound) {
                     continue;
                 }

src/Finder.php

@@ -5,6 +5,7 @@
 namespace ArrayLookup;
 
 use ArrayIterator;
+use ArrayLookup\Assert\Filter;
 use ArrayObject;
 use Traversable;
 use Webmozart\Assert\Assert;
@@ -24,12 +25,12 @@ final class Finder
      */
     public static function first(iterable $data, callable $filter, bool $returnKey = false): mixed
     {
+        // filter must be a callable with bool return type
+        Filter::boolean($filter);
+
         foreach ($data as $key => $datum) {
             $isFound = $filter($datum, $key);
 
-            // returns of callable must be bool
-            Assert::boolean($isFound);
-
             if (! $isFound) {
                 continue;
             }
@@ -71,6 +72,9 @@ public static function last(
         // ensure data is array for end(), key(), current(), prev() usage
         Assert::isArray($data);
 
+        // filter must be a callable with bool return type
+        Filter::boolean($filter);
+
         // Use end(), key(), current(), prev() usage instead of array_reverse()
         // to avoid immediatelly got "Out of memory" on many data
         // see https://3v4l.org/IHo2H vs https://3v4l.org/Wqejc
@@ -91,9 +95,6 @@ public static function last(
             $current = current($data);
             $isFound = $filter($current, $key);
 
-            // returns of callable must be bool
-            Assert::boolean($isFound);
-
             if (! $isFound) {
                 // go to previous row
                 prev($data);
@@ -133,12 +134,12 @@ public static function rows(
         $newKey     = 0;
         $totalFound = 0;
 
+        // filter must be a callable with bool return type
+        Filter::boolean($filter);
+
         foreach ($data as $key => $datum) {
             $isFound = $filter($datum, $key);
 
-            // returns of callable must be bool
-            Assert::boolean($isFound);
-
             if (! $isFound) {
                 continue;
             }

src/Only.php

@@ -4,6 +4,7 @@
 
 namespace ArrayLookup;
 
+use ArrayLookup\Assert\Filter;
 use Traversable;
 use Webmozart\Assert\Assert;
 
@@ -47,14 +48,13 @@ private static function onlyFoundTimes(
     ): bool {
         // usage must be higher than 0
         Assert::greaterThan($maxCount, 0);
+        // filter must be a callable with bool return type
+        Filter::boolean($filter);
 
         $totalFound = 0;
         foreach ($data as $key => $datum) {
             $isFound = $filter($datum, $key);
 
-            // returns of callable must be bool
-            Assert::boolean($isFound);
-
             if (! $isFound) {
                 continue;
             }