| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476 |
- <?php
- namespace Illuminate\Bus;
- use Closure;
- use Illuminate\Bus\Events\BatchDispatched;
- use Illuminate\Contracts\Container\Container;
- use Illuminate\Contracts\Events\Dispatcher as EventDispatcher;
- use Illuminate\Support\Arr;
- use Illuminate\Support\Collection;
- use Illuminate\Support\Traits\Conditionable;
- use Laravel\SerializableClosure\SerializableClosure;
- use RuntimeException;
- use Throwable;
- use function Illuminate\Support\enum_value;
- class PendingBatch
- {
- use Conditionable;
- /**
- * The IoC container instance.
- *
- * @var \Illuminate\Contracts\Container\Container
- */
- protected $container;
- /**
- * The batch name.
- *
- * @var string
- */
- public $name = '';
- /**
- * The jobs that belong to the batch.
- *
- * @var \Illuminate\Support\Collection
- */
- public $jobs;
- /**
- * The batch options.
- *
- * @var array
- */
- public $options = [];
- /**
- * Jobs that have been verified to contain the Batchable trait.
- *
- * @var array<class-string, bool>
- */
- protected static $batchableClasses = [];
- /**
- * Create a new pending batch instance.
- *
- * @param \Illuminate\Contracts\Container\Container $container
- * @param \Illuminate\Support\Collection $jobs
- */
- public function __construct(Container $container, Collection $jobs)
- {
- $this->container = $container;
- $this->jobs = $jobs->each(function (object|array $job) {
- $this->ensureJobIsBatchable($job);
- });
- }
- /**
- * Add jobs to the batch.
- *
- * @param iterable|object|array $jobs
- * @return $this
- */
- public function add($jobs)
- {
- $jobs = is_iterable($jobs) ? $jobs : Arr::wrap($jobs);
- foreach ($jobs as $job) {
- $this->ensureJobIsBatchable($job);
- $this->jobs->push($job);
- }
- return $this;
- }
- /**
- * Ensure the given job is batchable.
- *
- * @param object|array $job
- * @return void
- */
- protected function ensureJobIsBatchable(object|array $job): void
- {
- foreach (Arr::wrap($job) as $job) {
- if ($job instanceof PendingBatch || $job instanceof Closure) {
- return;
- }
- 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;
- }
- }
- /**
- * Add a callback to be executed when the batch is stored.
- *
- * @param callable $callback
- * @return $this
- */
- public function before($callback)
- {
- $this->registerCallback('before', $callback);
- return $this;
- }
- /**
- * Get the "before" callbacks that have been registered with the pending batch.
- *
- * @return array
- */
- public function beforeCallbacks()
- {
- return $this->options['before'] ?? [];
- }
- /**
- * Add a callback to be executed after a job in the batch have executed successfully.
- *
- * @param callable $callback
- * @return $this
- */
- public function progress($callback)
- {
- $this->registerCallback('progress', $callback);
- return $this;
- }
- /**
- * Get the "progress" callbacks that have been registered with the pending batch.
- *
- * @return array
- */
- public function progressCallbacks()
- {
- return $this->options['progress'] ?? [];
- }
- /**
- * Add a callback to be executed after all jobs in the batch have executed successfully.
- *
- * @param callable $callback
- * @return $this
- */
- public function then($callback)
- {
- $this->registerCallback('then', $callback);
- return $this;
- }
- /**
- * Get the "then" callbacks that have been registered with the pending batch.
- *
- * @return array
- */
- public function thenCallbacks()
- {
- return $this->options['then'] ?? [];
- }
- /**
- * Add a callback to be executed after the first failing job in the batch.
- *
- * @param callable $callback
- * @return $this
- */
- public function catch($callback)
- {
- $this->registerCallback('catch', $callback);
- return $this;
- }
- /**
- * Get the "catch" callbacks that have been registered with the pending batch.
- *
- * @return array
- */
- public function catchCallbacks()
- {
- return $this->options['catch'] ?? [];
- }
- /**
- * Add a callback to be executed after the batch has finished executing.
- *
- * @param callable $callback
- * @return $this
- */
- public function finally($callback)
- {
- $this->registerCallback('finally', $callback);
- return $this;
- }
- /**
- * Get the "finally" callbacks that have been registered with the pending batch.
- *
- * @return array
- */
- public function finallyCallbacks()
- {
- return $this->options['finally'] ?? [];
- }
- /**
- * Indicate that the batch should not be canceled when a job within the batch fails.
- *
- * Optionally, add callbacks to be executed upon each job failure.
- *
- * @template TParam of (Closure(\Illuminate\Bus\Batch, \Throwable|null): mixed)|(callable(\Illuminate\Bus\Batch, \Throwable|null): mixed)
- *
- * @param bool|TParam|array<array-key, TParam> $param
- * @return $this
- */
- public function allowFailures($param = true)
- {
- if (! is_bool($param)) {
- $param = Arr::wrap($param);
- foreach ($param as $callback) {
- if (is_callable($callback)) {
- $this->registerCallback('failure', $callback);
- }
- }
- }
- $this->options['allowFailures'] = ! ($param === false);
- return $this;
- }
- /**
- * Determine if the pending batch allows jobs to fail without cancelling the batch.
- *
- * @return bool
- */
- public function allowsFailures()
- {
- return Arr::get($this->options, 'allowFailures', false) === true;
- }
- /**
- * Get the "failure" callbacks that have been registered with the pending batch.
- *
- * @return array<array-key, Closure|callable>
- */
- public function failureCallbacks(): array
- {
- return $this->options['failure'] ?? [];
- }
- /**
- * Register a callback with proper serialization.
- */
- private function registerCallback(string $type, Closure|callable $callback): void
- {
- $this->options[$type][] = $callback instanceof Closure
- ? new SerializableClosure($callback)
- : $callback;
- }
- /**
- * Set the name for the batch.
- *
- * @param string $name
- * @return $this
- */
- public function name(string $name)
- {
- $this->name = $name;
- return $this;
- }
- /**
- * Specify the queue connection that the batched jobs should run on.
- *
- * @param string $connection
- * @return $this
- */
- public function onConnection(string $connection)
- {
- $this->options['connection'] = $connection;
- return $this;
- }
- /**
- * Get the connection used by the pending batch.
- *
- * @return string|null
- */
- public function connection()
- {
- return $this->options['connection'] ?? null;
- }
- /**
- * Specify the queue that the batched jobs should run on.
- *
- * @param \UnitEnum|string|null $queue
- * @return $this
- */
- public function onQueue($queue)
- {
- $this->options['queue'] = enum_value($queue);
- return $this;
- }
- /**
- * Get the queue used by the pending batch.
- *
- * @return string|null
- */
- public function queue()
- {
- return $this->options['queue'] ?? null;
- }
- /**
- * Add additional data into the batch's options array.
- *
- * @param string $key
- * @param mixed $value
- * @return $this
- */
- public function withOption(string $key, $value)
- {
- $this->options[$key] = $value;
- return $this;
- }
- /**
- * Dispatch the batch.
- *
- * @return \Illuminate\Bus\Batch
- *
- * @throws \Throwable
- */
- public function dispatch()
- {
- $repository = $this->container->make(BatchRepository::class);
- try {
- $batch = $this->store($repository);
- $batch = $batch->add($this->jobs);
- } catch (Throwable $e) {
- if (isset($batch)) {
- $repository->delete($batch->id);
- }
- throw $e;
- }
- $this->container->make(EventDispatcher::class)->dispatch(
- new BatchDispatched($batch)
- );
- return $batch;
- }
- /**
- * Dispatch the batch after the response is sent to the browser.
- *
- * @return \Illuminate\Bus\Batch
- */
- public function dispatchAfterResponse()
- {
- $repository = $this->container->make(BatchRepository::class);
- $batch = $this->store($repository);
- if ($batch) {
- $this->container->terminating(function () use ($batch) {
- $this->dispatchExistingBatch($batch);
- });
- }
- return $batch;
- }
- /**
- * Dispatch an existing batch.
- *
- * @param \Illuminate\Bus\Batch $batch
- * @return void
- *
- * @throws \Throwable
- */
- protected function dispatchExistingBatch($batch)
- {
- try {
- $batch = $batch->add($this->jobs);
- } catch (Throwable $e) {
- $batch->delete();
- throw $e;
- }
- $this->container->make(EventDispatcher::class)->dispatch(
- new BatchDispatched($batch)
- );
- }
- /**
- * Dispatch the batch if the given truth test passes.
- *
- * @param bool|\Closure $boolean
- * @return \Illuminate\Bus\Batch|null
- */
- public function dispatchIf($boolean)
- {
- return value($boolean) ? $this->dispatch() : null;
- }
- /**
- * Dispatch the batch unless the given truth test passes.
- *
- * @param bool|\Closure $boolean
- * @return \Illuminate\Bus\Batch|null
- */
- public function dispatchUnless($boolean)
- {
- return ! value($boolean) ? $this->dispatch() : null;
- }
- /**
- * Store the batch using the given repository.
- *
- * @param \Illuminate\Bus\BatchRepository $repository
- * @return \Illuminate\Bus\Batch
- */
- protected function store($repository)
- {
- $batch = $repository->store($this);
- (new Collection($this->beforeCallbacks()))->each(function ($handler) use ($batch) {
- try {
- return $handler($batch);
- } catch (Throwable $e) {
- if (function_exists('report')) {
- report($e);
- }
- }
- });
- return $batch;
- }
- }
|