156 lines
3.7 KiB
PHP
156 lines
3.7 KiB
PHP
<?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 queue.
|
|
*
|
|
* @var string
|
|
*/
|
|
protected $default;
|
|
|
|
/**
|
|
* The queue URL prefix.
|
|
*
|
|
* @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)
|
|
{
|
|
$response = $this->sqs->getQueueAttributes([
|
|
'QueueUrl' => $this->getQueue($queue),
|
|
'AttributeNames' => ['ApproximateNumberOfMessages'],
|
|
]);
|
|
|
|
$attributes = $response->get('Attributes');
|
|
|
|
return (int) $attributes['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 = [])
|
|
{
|
|
return $this->sqs->sendMessage([
|
|
'QueueUrl' => $this->getQueue($queue), 'MessageBody' => $payload,
|
|
])->get('MessageId');
|
|
}
|
|
|
|
/**
|
|
* 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)
|
|
{
|
|
return $this->sqs->sendMessage([
|
|
'QueueUrl' => $this->getQueue($queue),
|
|
'MessageBody' => $this->createPayload($job, $data),
|
|
'DelaySeconds' => $this->secondsUntil($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)
|
|
{
|
|
$response = $this->sqs->receiveMessage([
|
|
'QueueUrl' => $queue = $this->getQueue($queue),
|
|
'AttributeNames' => ['ApproximateReceiveCount'],
|
|
]);
|
|
|
|
if (! is_null($response['Messages']) && count($response['Messages']) > 0) {
|
|
return new SqsJob(
|
|
$this->container, $this->sqs, $response['Messages'][0],
|
|
$this->connectionName, $queue
|
|
);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get the queue or return the default.
|
|
*
|
|
* @param string|null $queue
|
|
* @return string
|
|
*/
|
|
public function getQueue($queue)
|
|
{
|
|
$queue = $queue ?: $this->default;
|
|
|
|
return filter_var($queue, FILTER_VALIDATE_URL) === false
|
|
? rtrim($this->prefix, '/').'/'.$queue : $queue;
|
|
}
|
|
|
|
/**
|
|
* Get the underlying SQS instance.
|
|
*
|
|
* @return \Aws\Sqs\SqsClient
|
|
*/
|
|
public function getSqs()
|
|
{
|
|
return $this->sqs;
|
|
}
|
|
}
|