RabbitMQ是一个功能强大的消息代理,提供了多种消息传递模式来满足不同场景下的需求。本文将介绍RabbitMQ的四种常用消息传递模式:Work、Fanout、Direct、Topic,并给出相应的Java示例代码。
Work模式也被称为任务队列模式,它将任务分发给多个消费者,并由消费者竞争性地消费任务。每个任务只会被一个消费者处理。
import com.rabbitmq.client.*;
public class Worker {
private static final String TASK_QUEUE_NAME = "task_queue";
public static void main(String[] args) throws Exception {
ConnectionFactory factory = new ConnectionFactory();
factory.setHost("localhost");
final Connection connection = factory.newConnection();
final Channel channel = connection.createChannel();
channel.queueDeclare(TASK_QUEUE_NAME, true, false, false, null);
System.out.println(" [*] Waiting for messages. To exit press CTRL+C");
channel.basicQos(1);
Consumer consumer = new DefaultConsumer(channel) {
@Override
public void handleDelivery(String consumerTag, Envelope envelope, AMQP.BasicProperties properties, byte[] body) throws IOException {
String message = new String(body, "UTF-8");
System.out.println(" [x] Received '" + message + "'");
try {
doWork(message);
} finally {
System.out.println(" [x] Done");
channel.basicAck(envelope.getDeliveryTag(), false);
}
}
};
channel.basicConsume(TASK_QUEUE_NAME, false, consumer);
}
private static void doWork(String task) {
try {
Thread.sleep(1000);
} catch (InterruptedException _ignored) {
Thread.currentThread().interrupt();
}
}
}
应用场景 一个常见的应用场景是在Web应用中实现异步任务处理。例如,用户在网站上提交了一个长时间处理的任务(如生成报表、发送邮件等),为了提高用户体验,可以将任务提交到RabbitMQ的任务队列中,然后由后台的消费者进行异步处理。这样一来,用户在网站上提交任务后即可立即得到响应,而不必等待任务处理完成,提高了系统的响应速度和并发处理能力。
下面是一个使用Work模式的简单示例代码,包括生产者和消费者部分,用Java编写,并添加了详细的注释。
生产者(Producer)代码:
import com.rabbitmq.client.*;
public class Producer {
private static final String TASK_QUEUE_NAME = "task_queue";
public static void main(String[] args) throws Exception {
// 创建连接和频道
ConnectionFactory factory = new ConnectionFactory();
factory.setHost("localhost");
try (Connection connection = factory.newConnection();
Channel channel = connection.createChannel()) {
// 声明队列
channel.queueDeclare(TASK_QUEUE_NAME, true, false, false, null);
// 构造消息
String message = String.join(" ", args);
// 发送消息
channel.basicPublish("", TASK_QUEUE_NAME,
MessageProperties.PERSISTENT_TEXT_PLAIN,
message.getBytes("UTF-8"));
System.out.println(" [x] Sent '" + message + "'");
}
}
}
消费者(Consumer)代码:
import com.rabbitmq.client.*;
public class Consumer {
private static final String TASK_QUEUE_NAME = "task_queue";
public static void main(String[] args) throws Exception {
// 创建连接和频道
ConnectionFactory factory = new ConnectionFactory();
factory.setHost("localhost");
final Connection connection = factory.newConnection();
final Channel channel = connection.createChannel();
// 声明队列
channel.queueDeclare(TASK_QUEUE_NAME, true, false, false, null);
System.out.println(" [*] Waiting for messages. To exit press CTRL+C");
// 设置消息处理的回调函数
channel.basicConsume(TASK_QUEUE_NAME, false, new DefaultConsumer(channel) {
@Override
public void handleDelivery(String consumerTag, Envelope envelope, AMQP.BasicProperties properties, byte[] body) throws java.io.IOException {
String message = new String(body, "UTF-8");
// 模拟任务处理过程
try {
doWork(message);
} catch (InterruptedException e) {
e.printStackTrace();
} finally {
System.out.println(" [x] Done");
// 手动发送应答
channel.basicAck(envelope.getDeliveryTag(), false);
}
}
});
}
private static void doWork(String task) throws InterruptedException {
// 模拟任务处理过程,这里休眠1秒
Thread.sleep(1000);
}
}
在上述代码中,生产者(Producer)将消息发布到名为task_queue
的队列中,而消费者(Consumer)则从该队列中获取任务并进行处理。每个任务都会被一个消费者处理,通过模拟任务处理过程,演示了Work模式的基本使用方式。
Fanout模式是RabbitMQ中的一种消息传递模式,它将消息广播到所有绑定到Exchange的队列中,即使在消息发布之后才创建的队列,也能接收到消息。
原理解析 在Fanout模式中,生产者将消息发布到一个名为Exchange的交换机中,而不是直接发送到队列中。交换机会将收到的消息广播到与其绑定的所有队列中。因此,无论在消息发布之前还是之后创建的队列,只要它们与交换机进行了绑定,就能接收到交换机广播的消息。
在示例代码中,首先声明一个名为logs的Fanout类型的Exchange,并将消息发布到该Exchange中。消息发布时,并没有指定具体的队列,而是将消息发送到了Exchange中。Exchange会将消息广播到所有与其绑定的队列中,这就是Fanout模式的工作原理。
应用场景 一个常见的应用场景是日志处理系统。在一个分布式的日志系统中,通常会有多个日志消费者,它们分别负责处理不同级别(如info、error等)的日志。通过使用Fanout模式,可以将日志消息广播到所有相关的队列中,每个消费者只需要关注自己负责处理的日志级别,从而实现了日志的分发和处理。
通过Fanout模式,我们可以实现消息的广播传递,适用于多个消费者需要同时接收同一份消息的场景,例如日志处理系统、实时广播等。
import com.rabbitmq.client.*;
public class EmitLog {
private static final String EXCHANGE_NAME = "logs";
public static void main(String[] argv) throws Exception {
ConnectionFactory factory = new ConnectionFactory();
factory.setHost("localhost");
try (Connection connection = factory.newConnection();
Channel channel = connection.createChannel()) {
channel.exchangeDeclare(EXCHANGE_NAME, "fanout");
String message = getMessage(argv);
channel.basicPublish(EXCHANGE_NAME, "", null, message.getBytes("UTF-8"));
System.out.println(" [x] Sent '" + message + "'");
}
}
private static String getMessage(String[] strings) {
if (strings.length < 1) {
return "info: Hello World!";
}
return String.join(" ", strings);
}
}
Direct模式将消息路由到与消息的RoutingKey完全匹配的队列中。 应用场景 一个常见的应用场景是日志级别过滤。在一个分布式的日志系统中,可能有多个消费者负责处理不同级别的日志(如info、error、warning等)。通过使用Direct模式,生产者可以根据日志的级别指定不同的RoutingKey,并将日志消息发布到Exchange中。Exchange会将日志消息路由到与其RoutingKey完全匹配的队列中,从而实现了日志的级别过滤和分发。
import com.rabbitmq.client.*;
public class EmitLogDirect {
private static final String EXCHANGE_NAME = "direct_logs";
public static void main(String[] argv) throws Exception {
ConnectionFactory factory = new ConnectionFactory();
factory.setHost("localhost");
try (Connection connection = factory.newConnection();
Channel channel = connection.createChannel()) {
channel.exchangeDeclare(EXCHANGE_NAME, "direct");
String severity = getSeverity(argv);
String message = getMessage(argv);
channel.basicPublish(EXCHANGE_NAME, severity, null, message.getBytes("UTF-8"));
System.out.println(" [x] Sent '" + severity + "':'" + message + "'");
}
}
private static String getSeverity(String[] strings) {
if (strings.length < 1) {
return "info";
}
return strings[0];
}
private static String getMessage(String[] strings) {
if (strings.length < 2) {
return "Hello World!";
}
return String.join(" ", strings);
}
}
Topic模式是RabbitMQ中的一种消息传递模式,它将消息发送到与匹配通配符的RoutingKey相匹配的队列中.
原理解析 在Topic模式中,生产者将消息发布到一个名为Exchange的交换机中,并且在发送消息时需要指定一个RoutingKey。RoutingKey可以使用通配符(和#)来匹配多个队列,其中表示匹配一个单词,#表示匹配零个或多个单词。交换机会将收到的消息根据RoutingKey和通配符匹配规则将消息路由到与之匹配的队列中。
在示例代码中,我们首先声明一个名为topic_logs的Topic类型的Exchange,并指定消息的RoutingKey。然后将消息发布到Exchange中,Exchange会根据消息的RoutingKey和通配符匹配规则将消息路由到与之匹配的队列中。
应用场景 一个常见的应用场景是日志过滤器。在一个分布式的日志系统中,可能有多个消费者负责处理不同模块或不同级别的日志。通过使用Topic模式,生产者可以根据日志的模块或级别指定特定的RoutingKey,并将日志消息发布到Exchange中。消费者可以使用通配符来订阅感兴趣的日志模块或级别,Exchange会将日志消息路由到与之匹配的队列中,从而实现了日志的模块化过滤和分发。
import com.rabbitmq.client.*;
public class EmitLogTopic {
private static final String EXCHANGE_NAME = "topic_logs";
public static void main(String[] argv) throws Exception {
ConnectionFactory factory = new ConnectionFactory();
factory.setHost("localhost");
try (Connection connection = factory.newConnection();
Channel channel = connection.createChannel()) {
channel.exchangeDeclare(EXCHANGE_NAME, "topic");
String routingKey = getRouting(argv);
String message = getMessage(argv);
channel.basicPublish(EXCHANGE_NAME, routingKey, null, message.getBytes("UTF-8"));
System.out.println(" [x] Sent '" + routingKey + "':'" + message + "'");
}
}
private static String getRouting(String[] strings) {
if (strings.length < 1) {
return "anonymous.info";
}
return strings[0];
}
private static String
getMessage(String[] strings) {
if (strings.length < 2) {
return "Hello World!";
}
return String.join(" ", strings);
}
}