123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164 |
- <?php
-
- namespace Illuminate\Queue;
-
- use Pheanstalk\Pheanstalk;
- use Pheanstalk\Job as PheanstalkJob;
- use Illuminate\Queue\Jobs\BeanstalkdJob;
- use Illuminate\Contracts\Queue\Queue as QueueContract;
-
- class BeanstalkdQueue extends Queue implements QueueContract
- {
- /**
- * The Pheanstalk instance.
- *
- * @var \Pheanstalk\Pheanstalk
- */
- protected $pheanstalk;
-
- /**
- * The name of the default tube.
- *
- * @var string
- */
- protected $default;
-
- /**
- * The "time to run" for all pushed jobs.
- *
- * @var int
- */
- protected $timeToRun;
-
- /**
- * Create a new Beanstalkd queue instance.
- *
- * @param \Pheanstalk\Pheanstalk $pheanstalk
- * @param string $default
- * @param int $timeToRun
- * @return void
- */
- public function __construct(Pheanstalk $pheanstalk, $default, $timeToRun)
- {
- $this->default = $default;
- $this->timeToRun = $timeToRun;
- $this->pheanstalk = $pheanstalk;
- }
-
- /**
- * Get the size of the queue.
- *
- * @param string $queue
- * @return int
- */
- public function size($queue = null)
- {
- $queue = $this->getQueue($queue);
-
- return (int) $this->pheanstalk->statsTube($queue)->current_jobs_ready;
- }
-
- /**
- * Push a new job onto the queue.
- *
- * @param string $job
- * @param mixed $data
- * @param string $queue
- * @return mixed
- */
- public function push($job, $data = '', $queue = null)
- {
- return $this->pushRaw($this->createPayload($job, $data), $queue);
- }
-
- /**
- * Push a raw payload onto the queue.
- *
- * @param string $payload
- * @param string $queue
- * @param array $options
- * @return mixed
- */
- public function pushRaw($payload, $queue = null, array $options = [])
- {
- return $this->pheanstalk->useTube($this->getQueue($queue))->put(
- $payload, Pheanstalk::DEFAULT_PRIORITY, Pheanstalk::DEFAULT_DELAY, $this->timeToRun
- );
- }
-
- /**
- * Push a new job onto the queue after a delay.
- *
- * @param \DateTimeInterface|\DateInterval|int $delay
- * @param string $job
- * @param mixed $data
- * @param string $queue
- * @return mixed
- */
- public function later($delay, $job, $data = '', $queue = null)
- {
- $pheanstalk = $this->pheanstalk->useTube($this->getQueue($queue));
-
- return $pheanstalk->put(
- $this->createPayload($job, $data),
- Pheanstalk::DEFAULT_PRIORITY,
- $this->secondsUntil($delay),
- $this->timeToRun
- );
- }
-
- /**
- * Pop the next job off of the queue.
- *
- * @param string $queue
- * @return \Illuminate\Contracts\Queue\Job|null
- */
- public function pop($queue = null)
- {
- $queue = $this->getQueue($queue);
-
- $job = $this->pheanstalk->watchOnly($queue)->reserve(0);
-
- if ($job instanceof PheanstalkJob) {
- return new BeanstalkdJob(
- $this->container, $this->pheanstalk, $job, $this->connectionName, $queue
- );
- }
- }
-
- /**
- * Delete a message from the Beanstalk queue.
- *
- * @param string $queue
- * @param string $id
- * @return void
- */
- public function deleteMessage($queue, $id)
- {
- $queue = $this->getQueue($queue);
-
- $this->pheanstalk->useTube($queue)->delete(new PheanstalkJob($id, ''));
- }
-
- /**
- * Get the queue or return the default.
- *
- * @param string|null $queue
- * @return string
- */
- public function getQueue($queue)
- {
- return $queue ?: $this->default;
- }
-
- /**
- * Get the underlying Pheanstalk instance.
- *
- * @return \Pheanstalk\Pheanstalk
- */
- public function getPheanstalk()
- {
- return $this->pheanstalk;
- }
- }
|