| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417 |
- <?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 Throwable;
- 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 = [];
- /**
- * Create a new pending batch instance.
- *
- * @param \Illuminate\Contracts\Container\Container $container
- * @param \Illuminate\Support\Collection $jobs
- * @return void
- */
- public function __construct(Container $container, Collection $jobs)
- {
- $this->container = $container;
- $this->jobs = $jobs;
- }
- /**
- * 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->jobs->push($job);
- }
- return $this;
- }
- /**
- * Add a callback to be executed when the batch is stored.
- *
- * @param callable $callback
- * @return $this
- */
- public function before($callback)
- {
- $this->options['before'][] = $callback instanceof Closure
- ? new SerializableClosure($callback)
- : $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->options['progress'][] = $callback instanceof Closure
- ? new SerializableClosure($callback)
- : $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->options['then'][] = $callback instanceof Closure
- ? new SerializableClosure($callback)
- : $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->options['catch'][] = $callback instanceof Closure
- ? new SerializableClosure($callback)
- : $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->options['finally'][] = $callback instanceof Closure
- ? new SerializableClosure($callback)
- : $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 cancelled when a job within the batch fails.
- *
- * @param bool $allowFailures
- * @return $this
- */
- public function allowFailures($allowFailures = true)
- {
- $this->options['allowFailures'] = $allowFailures;
- 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;
- }
- /**
- * 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 string $queue
- * @return $this
- */
- public function onQueue(string $queue)
- {
- $this->options['queue'] = $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) {
- if (isset($batch)) {
- $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);
- collect($this->beforeCallbacks())->each(function ($handler) use ($batch) {
- try {
- return $handler($batch);
- } catch (Throwable $e) {
- if (function_exists('report')) {
- report($e);
- }
- }
- });
- return $batch;
- }
- }
|