| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144 |
- <?php
- namespace Illuminate\Bus;
- use Illuminate\Container\Container;
- use Illuminate\Contracts\Bus\Dispatcher;
- use Illuminate\Contracts\Queue\ShouldQueue;
- use Illuminate\Foundation\Bus\Dispatchable;
- use Illuminate\Queue\InteractsWithQueue;
- use Illuminate\Support\Collection;
- use Throwable;
- class ChainedBatch implements ShouldQueue
- {
- use Batchable, Dispatchable, InteractsWithQueue, Queueable;
- /**
- * The collection of batched jobs.
- *
- * @var \Illuminate\Support\Collection
- */
- public Collection $jobs;
- /**
- * The name of the batch.
- *
- * @var string
- */
- public string $name;
- /**
- * The batch options.
- *
- * @var array
- */
- public array $options;
- /**
- * Create a new chained batch instance.
- *
- * @param \Illuminate\Bus\PendingBatch $batch
- */
- public function __construct(PendingBatch $batch)
- {
- $this->jobs = static::prepareNestedBatches($batch->jobs);
- $this->name = $batch->name;
- $this->options = $batch->options;
- $this->queue = $batch->queue();
- $this->connection = $batch->connection();
- }
- /**
- * Prepare any nested batches within the given collection of jobs.
- *
- * @param \Illuminate\Support\Collection $jobs
- * @return \Illuminate\Support\Collection
- */
- public static function prepareNestedBatches(Collection $jobs): Collection
- {
- return $jobs->map(fn ($job) => match (true) {
- is_array($job) => static::prepareNestedBatches(new Collection($job))->all(),
- $job instanceof Collection => static::prepareNestedBatches($job),
- $job instanceof PendingBatch => new ChainedBatch($job),
- default => $job,
- });
- }
- /**
- * Handle the job.
- *
- * @return void
- */
- public function handle()
- {
- $this->attachRemainderOfChainToEndOfBatch(
- $this->toPendingBatch()
- )->dispatch();
- }
- /**
- * Convert the chained batch instance into a pending batch.
- *
- * @return \Illuminate\Bus\PendingBatch
- */
- public function toPendingBatch()
- {
- $batch = Container::getInstance()->make(Dispatcher::class)->batch($this->jobs);
- $batch->name = $this->name;
- $batch->options = $this->options;
- if ($this->queue) {
- $batch->onQueue($this->queue);
- }
- if ($this->connection) {
- $batch->onConnection($this->connection);
- }
- foreach ($this->chainCatchCallbacks ?? [] as $callback) {
- $batch->catch(function (Batch $batch, ?Throwable $exception) use ($callback) {
- if (! $batch->allowsFailures()) {
- $callback($exception);
- }
- });
- }
- return $batch;
- }
- /**
- * Move the remainder of the chain to a "finally" batch callback.
- *
- * @param \Illuminate\Bus\PendingBatch $batch
- * @return \Illuminate\Bus\PendingBatch
- */
- protected function attachRemainderOfChainToEndOfBatch(PendingBatch $batch)
- {
- if (! empty($this->chained)) {
- $next = unserialize(array_shift($this->chained));
- $next->chained = $this->chained;
- $next->onConnection($next->connection ?: $this->chainConnection);
- $next->onQueue($next->queue ?: $this->chainQueue);
- $next->chainConnection = $this->chainConnection;
- $next->chainQueue = $this->chainQueue;
- $next->chainCatchCallbacks = $this->chainCatchCallbacks;
- $batch->finally(function (Batch $batch) use ($next) {
- if (! $batch->cancelled()) {
- Container::getInstance()->make(Dispatcher::class)->dispatch($next);
- }
- });
- $this->chained = [];
- }
- return $batch;
- }
- }
|