PHP消息队列在分布式任务调度中的应用实践

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

use PhpAmqpLibConnectionAMQPStreamConnection;
use PhpAmqpLibMessageAMQPMessage;

// 连接RabbitMQ
$connection = new AMQPStreamConnection('localhost', 5672, 'guest', 'guest');
$channel = $connection->channel();

// 声明队列
$channel->queue_declare('task_queue', false, true, false, false);

// 从命令行获取任务数据
$data = implode(' ', array_slice($argv, 1));
if(empty($data)) {
    $data = "Hello World!";
}

// 创建消息
$msg = new AMQPMessage($data, array('delivery_mode' => AMQPMessage::DELIVERY_MODE_PERSISTENT));

// 发布消息到队列
$channel->basic_publish($msg, '', 'task_queue');

echo " [x] Sent '$data'
";

// 关闭连接
$channel->close();
$connection->close();
?>

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

use PhpAmqpLibConnectionAMQPStreamConnection;
use PhpAmqpLibMessageAMQPMessage;

// 连接RabbitMQ
$connection = new AMQPStreamConnection('localhost', 5672, 'guest', 'guest');
$channel = $connection->channel();

// 声明队列
$channel->queue_declare('task_queue', false, true, false, false);

echo " [*] Waiting for messages. To exit press CTRL+C
";

$callback = function($msg) {
  echo ' [x] Received ', $msg->body, "
";
  
  // 模拟任务的处理时间
  sleep(substr_count($msg->body, '.'));

  echo " [x] Done
";

  // 显示消息处理完成
  $msg->delivery_info['channel']->basic_ack($msg->delivery_info['delivery_tag']);
};

// 设置公平调度，防止某个消费者负载过重
$channel->basic_qos(null, 1, null);

// 消费消息队列中的任务
$channel->basic_consume('task_queue', '', false, false, false, false, $callback);

while(count($channel->callbacks)) {
    $channel->wait();
}

// 关闭连接
$channel->close();
$connection->close();
?>