simplify PendingBatch@ensureJobIsBatchable; memoize the classes which are batchable

Author: cosmastech

src/Illuminate/Bus/PendingBatch.php

@@ -47,6 +47,13 @@ class PendingBatch
      */
     public $options = [];
 
+    /**
+     * Batches that have been checked to contain the Batchable trait.
+     *
+     * @var array<class-string, bool>
+     */
+    protected static $batchableClasses = [];
+
     /**
      * Create a new pending batch instance.
      *
@@ -97,9 +104,13 @@ protected function ensureJobIsBatchable(object|array $job): void
                 return;
             }
 
-            if (! in_array(Batchable::class, class_uses_recursive($job))) {
+            if (! (static::$batchableClasses[$job::class] ?? false) && ! in_array(Batchable::class, class_uses_recursive($job))) {
+                static::$batchableClasses[$job::class] = false;
+
                 throw new RuntimeException(sprintf('Attempted to batch job [%s], but it does not use the Batchable trait.', $job::class));
             }
+
+            static::$batchableClasses[$job::class] = true;
         }
     }
 

tests/Bus/BusPendingBatchTest.php

@@ -231,4 +231,22 @@ public function test_it_throws_exception_if_batched_job_is_not_batchable(): void
 
         new PendingBatch(new Container, new Collection([$nonBatchableJob]));
     }
+
+    public function test_it_throws_an_exception_if_batched_job_contains_batch_with_nonbatchable_job(): void
+    {
+        $this->expectException(RuntimeException::class);
+
+        $container = new Container;
+        new PendingBatch(
+            $container,
+            new Collection(
+                [new PendingBatch($container, new Collection([new BatchableJob, new class {}]))]
+            )
+        );
+    }
+}
+
+class BatchableJob
+{
+    use Batchable;
 }