<?php namespace Illuminate\Queue; use Aws\Sqs\SqsClient; use Illuminate\Queue\Jobs\SqsJob; use Illuminate\Contracts\Queue\Queue as QueueContract; class SqsQueue extends Queue implements QueueContract { /** * The Amazon SQS instance. * * @var \Aws\Sqs\SqsClient */ protected $sqs; /** * The name of the default tube. * * @var string */ protected $default; /** * The sqs prefix url. * * @var string */ protected $prefix; /** * Create a new Amazon SQS queue instance. * * @param \Aws\Sqs\SqsClient $sqs * @param string $default * @param string $prefix * @return void */ public function __construct(SqsClient $sqs, $default, $prefix = '') { $this->sqs = $sqs; $this->prefix = $prefix; $this->default = $default; } /** * Get the size of the queue. * * @param string $queue * @return int */ public function size($queue = null) { return (int) $this->sqs->getQueueAttributes([ 'QueueUrl' => $this->getQueue($queue), ])->get('ApproximateNumberOfMessages'); } /** * 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 = []) { $response = $this->sqs->sendMessage([ 'QueueUrl' => $this->getQueue($queue), 'MessageBody' => $payload, ]); return $response->get('MessageId'); } /** * Push a new job onto the queue after a delay. * * @param \DateTime|int $delay * @param string $job * @param mixed $data * @param string $queue * @return mixed */ public function later($delay, $job, $data = '', $queue = null) { $payload = $this->createPayload($job, $data); $delay = $this->getSeconds($delay); return $this->sqs->sendMessage([ 'QueueUrl' => $this->getQueue($queue), 'MessageBody' => $payload, 'DelaySeconds' => $delay, ])->get('MessageId'); } /** * 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); $response = $this->sqs->receiveMessage([ 'QueueUrl' => $queue, 'AttributeNames' => ['ApproximateReceiveCount'], ]); if (count($response['Messages']) > 0) { return new SqsJob($this->container, $this->sqs, $queue, $response['Messages'][0]); } } /** * Get the queue or return the default. * * @param string|null $queue * @return string */ public function getQueue($queue) { $queue = $queue ?: $this->default; if (filter_var($queue, FILTER_VALIDATE_URL) !== false) { return $queue; } return rtrim($this->prefix, '/').'/'.($queue); } /** * Get the underlying SQS instance. * * @return \Aws\Sqs\SqsClient */ public function getSqs() { return $this->sqs; } }