php-使用RabbitMQ实现延时队列

44次阅读

共计 2908 个字符,预计需要花费 8 分钟才能阅读完成。

前言

延时队列的作用不再累述
本文使用 rabbitmq 的 queue 可以设置 ttl 时间,将到期的 message 设为死信,message 会被 push 到 delay_queue,消费 delay_queue 即可实现延时队列功能

code

对 RabbitMQ 进行简单的封装

<?php


namespace RabbitMQ;

use PhpAmqpLib\Connection\AMQPStreamConnection;
use PhpAmqpLib\Message\AMQPMessage;

class RabbitMQ
{

    private $host = '127.0.0.1';
    private $port = 5672;
    private $user = 'guest';
    private $password = 'guest';
    protected $connection;
    protected $channel;


    /**
     * RabbitMQ constructor.
     */
    public function __construct()
    {$this->connection = new AMQPStreamConnection($this->host, $this->port, $this->user, $this->password);
        $this->channel    = $this->connection->channel();}

    /**
     * 生成信息
     * @param $message
     */
    public function sendMessage($message, $routeKey, $exchange = '', $properties = [])
    {
        $data = new AMQPMessage($message, $properties);
        $this->channel->basic_publish($data, $exchange, $routeKey);
    }

    /**
     * 消费消息
     * @param $queueName
     * @param $callback
     * @throws \ErrorException
     */
    public function consumeMessage($queueName,$callback)
    {$this->channel->basic_consume($queueName, '', false, false, false, false, $callback);
        while ($this->channel->is_consuming()) {$this->channel->wait();
        }
    }

    /**
     * @throws \Exception
     */
    public function __destruct()
    {$this->channel->close();
        $this->connection->close();}
}

创建延时队列

<?php


namespace RabbitMQ;

use PhpAmqpLib\Exchange\AMQPExchangeType;
use PhpAmqpLib\Message\AMQPMessage;
use PhpAmqpLib\Wire\AMQPTable;

/**
 * 使用 RabbitMQ 实现延时队列功能
 * Class DelayQueue
 * @package RabbitMQ
 */
class DelayQueue extends RabbitMQ
{

    /**
     * 创建延时队列
     * @param $ttl
     * @param $delayExName
     * @param $delayQueueName
     * @param $queueName
     */
    public function createQueue($ttl, $delayExName, $delayQueueName, $queueName)
    {
        $args = new AMQPTable([
            'x-dead-letter-exchange'    => $delayExName,
            'x-message-ttl'             => $ttl, // 消息存活时间
            'x-dead-letter-routing-key' => $queueName
        ]);
        $this->channel->queue_declare($queueName, false, true, false, false, false, $args);
        // 绑定死信 queue
        $this->channel->exchange_declare($delayExName, AMQPExchangeType::DIRECT, false, true, false);
        $this->channel->queue_declare($delayQueueName, false, true, false, false);
        $this->channel->queue_bind($delayQueueName, $delayExName, $queueName, false);
    }
}

生产者,代码很简单,看看运行之后的效果,订单的 message 越来越多

<?php
require_once '../vendor/autoload.php';

// 生产者

$delay = new \RabbitMQ\DelayQueue();

$ttl            = 1000 * 100;// 订单 100s 后超时
$delayExName    = 'delay-order-exchange';// 超时 exchange
$delayQueueName = 'delay-order-queue';// 超时 queue
$queueName      = 'ttl-order-queue';// 订单 queue

$delay->createQueue($ttl, $delayExName, $delayQueueName, $queueName);

//100 个订单信息,每个订单超时时间都是 10s
for ($i = 0; $i < 100; $i++) {
    $data = [
        'order_id' => $i + 1,
        'remark'   => 'this is a order test'
    ];
    $delay->sendMessage(json_encode($data), $queueName);
    sleep(1);
}

消费者,看看消费之后的,过一会会观察到,已经有到期 message 被 push 到了 delay_order_queue

消费者也消费到了 message

<?php
require_once '../vendor/autoload.php';

// 消费者

$delay = new \RabbitMQ\DelayQueue();

$delayQueueName = 'delay-order-queue';

$callback = function ($msg) {
    echo $msg->body . PHP_EOL;
    $msg->delivery_info['channel']->basic_ack($msg->delivery_info['delivery_tag']);

    // 处理订单超时逻辑,给用户推送提醒等等。。。sleep(10);
};

/**
 * 消费已经超时的订单信息,进行处理
 */
$delay->consumeMessage($delayQueueName, $callback);

代码见:https://github.com/jiaoyang3/…

正文完
 0