3 工作队列
工作队列(又称任务队列)的主要思想是避免立即执行资源密集型任务,而不得不等待它完成。相反我们安排任务在之后执行。我们把任务封装为消息并将其发送到队列。在后台运行的工作进程将弹出任务并最终执行作业。当有多个工作线程时,这些工作线程将一起处理这些任务。
3.1 轮询分发消息
在这个案例中我们会启动两个工作线程,一个消息发送线程,我们来看看他们两个工作线程是如何工作的。
3.1.1 抽取工具类
package com.zjw.rabbitmq.util;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.ConnectionFactory;
/**
* @author zjwblog <cn.zjwblog@gmail.com>
* @version 1.0
*/
public class RabbitMQUtil {
private static volatile Channel channel = null;
private static final Object lock = new Object();
public static Channel getChannel() throws Exception {
if (channel == null) {
synchronized (lock) {
if (channel == null) {
//创建一个连接工厂
ConnectionFactory factory = new ConnectionFactory();
factory.setHost("127.0.0.1");
factory.setUsername("admin");
factory.setPassword("admin");
Connection connection = factory.newConnection();
channel = connection.createChannel();
}
}
}
return channel;
}
}
3.1.2 启动两个消费者
package com.zjw.rabbitmq.workqueue;
import com.rabbitmq.client.Channel;
import com.zjw.rabbitmq.util.RabbitMQUtil;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
/**
* @author zjwblog <cn.zjwblog@gmail.com>
* @version 1.0
*/
public class Consumer01 {
private static final String QUEUE_NAME = "hello-world";
public static void main(String[] args) throws Exception {
Channel channel = RabbitMQUtil.getChannel();
String uuid = UUID.randomUUID().toString();
System.out.println("消费者[" + uuid + "] start: ");
channel.basicConsume(
QUEUE_NAME,
true,
(tag, msg) -> {
System.out.println("接收到消息: " + new String(msg.getBody()));
try {
TimeUnit.SECONDS.sleep(7);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println("消息 [ " + new String(msg.getBody()) + " ] 处理完成");
},
(consumerTag) -> System.out.println(consumerTag + "消费者取消消费接口回调逻辑")
);
}
}
package com.zjw.rabbitmq.workqueue;
import com.rabbitmq.client.Channel;
import com.zjw.rabbitmq.util.RabbitMQUtil;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
/**
* @author zjwblog <cn.zjwblog@gmail.com>
* @version 1.0
*/
public class Consumer02 {
private static final String QUEUE_NAME = "hello-world";
public static void main(String[] args) throws Exception {
Channel channel = RabbitMQUtil.getChannel();
String uuid = UUID.randomUUID().toString();
System.out.println("消费者[" + uuid + "] start: ");
channel.basicConsume(
QUEUE_NAME,
true,
(tag, msg) -> {
System.out.println("接收到消息: " + new String(msg.getBody()));
try {
TimeUnit.SECONDS.sleep(1);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println("消息 [ " + new String(msg.getBody()) + " ] 处理完成");
},
(consumerTag) -> System.out.println(consumerTag + "消费者取消消费接口回调逻辑")
);
}
}
3.1.3 启动生产者
package com.zjw.rabbitmq.workqueue;
import com.rabbitmq.client.Channel;
import com.zjw.rabbitmq.util.RabbitMQUtil;
import java.nio.charset.StandardCharsets;
import java.util.Scanner;
/**
* @author zjwblog <co.zjwblog@gmail.com>
* @version 1.0
*/
public class Producer {
private static final String QUEUE_NAME = "hello-world";
public static void main(String[] args) throws Exception {
try (Channel channel = RabbitMQUtil.getChannel();) {
channel.queueDeclare(QUEUE_NAME, false, false, false, null);
Scanner scanner = new Scanner(System.in);
while (scanner.hasNext()) {
String message = scanner.next();
channel.basicPublish("", QUEUE_NAME, null, message.getBytes(StandardCharsets.UTF_8));
System.out.println("发送消息 [ " + message + " ] 成功.");
}
}
}
}
3.2 消息应答
3.2.1 概念
消费者完成一个任务可能需要一段时间,如果其中一个消费者处理一个长的任务并仅只完成了部分突然它挂掉了,会发生什么情况。RabbitMQ 一旦向消费者传递了一条消息,便立即将该消息标记为删除。在这种情况下,突然有个消费者挂掉了,我们将丢失正在处理的消息。以及后续发送给该消费这的消息,因为它无法接收到。为了保证消息在发送过程中不丢失,rabbitmq 引入消息应答机制,消息应答就是:消费者在接收到消息并且处理该消息之后,告诉 rabbitmq 它已经处理了,rabbitmq 可以把该消息删除了。
3.2.2 自动应答
消息发送后立即被认为已经传送成功,这种模式需要在高吞吐量和数据传输安全性方面做权衡,因为这种模式如果消息在接收到之前,消费者那边出现连接或者 channel 关闭,那么消息就丢失了,当然另一方面这种模式消费者那边可以传递过载的消息,没有对传递的消息数量进行限制, 当然这样有可能使得消费者这边由于接收太多还来不及处理的消息,导致这些消息的积压,最终使得内存耗尽,最终这些消费者线程被操作系统杀死,所以这种模式仅适用在消费者可以高效并以某种速率能够处理这些消息的情况下使用。
3.2.3 消息应答的方法
-
Channel.basicAck
: (用于肯定确认) RabbitMQ已知道该消息并且成功的处理消息,可以将其丢弃了 -
Channel.basicNack
: 用于否定确认 -
Channel.basicReject
: 用于否定确认,与 Channel.basicNack 相比少一个参数不处理该消息了直接拒绝,可以将其丢弃了
3.2.4 Multiple 的解释
手动应答的好处是可以批量应答并且减少网络拥堵,multiple
的true
和false
代表不同意思。true
代表批量应答channel
上未应答的消息比如说channel
上有传送tag
的消息1,2,3,4
当前tag
是4
那么此时1-4
的这些还未应答的消息都会被确认。收到消息应答false
同上面相比只会应答tag=8
的消息1,2,3
这三个消息依然不会被确认收到消息应答
package com.zjw.rabbitmq.workqueue.bachack;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.DeliverCallback;
import com.zjw.rabbitmq.util.RabbitMQUtil;
import com.zjw.rabbitmq.util.SleepUtil;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
/**
* @author zjwblog <cn.zjwblog@gmail.com>
* @version 1.0
*/
public class Consumer {
private static final String QUEUE_NAME = "hello-world";
public static void main(String[] args) throws Exception {
Channel channel = RabbitMQUtil.getChannel();
channel.basicConsume(
QUEUE_NAME,
false, // 采用手动应答
(tag, msg) -> {
String message = new String(msg.getBody());
System.out.println("Consumer 接收到消息:" + message);
SleepUtil.sleep(1);
System.out.println("Consumer 处理消息[" +message+"]完成");
/*
* 1.消息标记 tag
* 2.是否批量应答未应答消息,开启批量应答可能会导致数据丢失
*
* 例如:Consumer已经消费了1,2,3,4,5条,当前是第5条数据,
* 之前的前4条都处理失败了,现在第5条处理成功,然后发送确认消息,
* 会确认掉前5条都消费完成,于是Broker会删除1 2 3 4 5 条数据,数据 1 2 3 4 丢失
*/
channel.basicAck(msg.getEnvelope().getDeliveryTag(), true);
},
tag -> System.out.println("[ " + tag + " ]消费者取消消费接口回调逻辑")
);
}
}
3.2.5 消息自动重新入队
如果消费者由于某些原因失去连接(其通道已关闭,连接已关闭或 TCP 连接丢失),导致消息未发送 ACK 确认,RabbitMQ将了解到消息未完全处理,并将对其重新排队。如果此时其他消费者可以处理,它将很快将其重新分发给另一个消费者。这样,即使某个消费者偶尔死亡,也可以确保不会丢失任何消息。
3.2.6 消息手动应答代码
默认消息采用的是自动应答,所以我们要想实现消息消费过程中不丢失,需要把自动应答改为手动应答,消费者在上面代码的基础上增加下面画红色部分代码。
消息生产者
package com.zjw.rabbitmq.workqueue.ack;
import com.rabbitmq.client.Channel;
import com.zjw.rabbitmq.util.RabbitMQUtil;
import java.nio.charset.StandardCharsets;
import java.util.Scanner;
/**
* @author zjwblog <co.zjwblog@gmail.com>
* @version 1.0
*/
public class Producer {
private static final String QUEUE_NAME = "hello-world";
public static void main(String[] args) throws Exception {
try (Channel channel = RabbitMQUtil.getChannel();) {
channel.queueDeclare(QUEUE_NAME, false, false, false, null);
Scanner scanner = new Scanner(System.in);
while (scanner.hasNext()) {
String message = scanner.next();
channel.basicPublish("", QUEUE_NAME, null, message.getBytes(StandardCharsets.UTF_8));
System.out.println("发送消息 [ " + message + " ] 成功.");
}
}
}
}
消费者 01
package com.zjw.rabbitmq.workqueue.ack;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.DeliverCallback;
import com.zjw.rabbitmq.util.RabbitMQUtil;
import com.zjw.rabbitmq.util.SleepUtil;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
/**
* @author zjwblog <cn.zjwblog@gmail.com>
* @version 1.0
*/
public class Consumer01 {
private static final String QUEUE_NAME = "hello-world";
public static void main(String[] args) throws Exception {
Channel channel = RabbitMQUtil.getChannel();
System.out.println("Consumer01 等待接收消息处理时间较长");
channel.basicConsume(
QUEUE_NAME,
false, // 采用手动应答
(tag, msg) -> {
String message = new String(msg.getBody());
System.out.println("Consumer01接收到消息:" + message);
SleepUtil.sleep(20);
System.out.println("Consumer01处理消息[" +message+"]完成");
/*
* 1.消息标记 tag
* 2.是否批量应答未应答消息
*/
channel.basicAck(msg.getEnvelope().getDeliveryTag(), false);
},
tag -> System.out.println("[ " + tag + " ]消费者取消消费接口回调逻辑")
);
}
}
消费者 02
package com.zjw.rabbitmq.workqueue.ack;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.DeliverCallback;
import com.zjw.rabbitmq.util.RabbitMQUtil;
import com.zjw.rabbitmq.util.SleepUtil;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
/**
* @author zjwblog <cn.zjwblog@gmail.com>
* @version 1.0
*/
public class Consumer02 {
private static final String QUEUE_NAME = "hello-world";
public static void main(String[] args) throws Exception {
Channel channel = RabbitMQUtil.getChannel();
System.out.println("Consumer02 等待接收消息处理时间较短");
channel.basicConsume(
QUEUE_NAME,
false, // 采用手动应答
(tag, msg) -> {
String message = new String(msg.getBody());
System.out.println("Consumer01接收到消息:" + message);
SleepUtil.sleep(2);
System.out.println("Consumer01处理消息[" +message+"]完成");
/*
* 1.消息标记 tag
* 2.是否批量应答未应答消息
*/
channel.basicAck(msg.getEnvelope().getDeliveryTag(), false);
},
tag -> System.out.println("[ " + tag + " ]消费者取消消费接口回调逻辑")
);
}
}
工具类
package com.zjw.rabbitmq.util;
import java.util.concurrent.TimeUnit;
/**
* @author zjwblog <cn.zjwblog@gmail.com>
* @version 1.0
*/
public class SleepUtil {
public static void sleep(int delay) {
try {
TimeUnit.SECONDS.sleep(delay);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
3.3 RabbitMQ 持久化
3.3.1 概念
刚刚我们已经看到了如何处理任务不丢失的情况,但是如何保障当RabbitMQ
服务停掉以后消息生产者发送过来的消息不丢失。默认情况下RabbitMQ
退出或由于某种原因崩溃时,它忽视队列和消息,除非告知它不要这样做。确保消息不会丢失需要做两件事:
-
需要将队列标记为持久化
-
需要将消息标记为持久化
3.3.2 队列如何实现持久化
之前我们创建的队列都是非持久化的,rabbitmq
如果重启的化,该队列就会被删除掉,如果要队列实现持久化 需要在声明队列的时候设置durable
参数。
但是需要注意的就是如果之前声明的队列不是持久化的,需要把原先队列先删除,或者重新创建一个持久化的队列,不然就会出现错误。
对于持久化队列,Broker
重启时队列也依然存在。
3.3.3 消息实现持久化
要想让消息实现持久化需要在消息生产者修改代码,MessageProperties.PERSISTENT_TEXT_PLAIN
添加这个属性。
将消息标记为持久化并不能完全保证不会丢失消息。尽管它告诉RabbitMQ
将消息保存到磁盘,但是这里依然存在当消息刚准备存储在磁盘的时候 但是还没有存储完,消息还在缓存的一个间隔点。此时并没有真正写入磁盘。持久性保证并不强,但是对于我们的简单任务队列而言,这已经绰绰有余了。
生产者
package com.zjw.rabbitmq.workqueue.durable;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.MessageProperties;
import com.zjw.rabbitmq.util.RabbitMQUtil;
import java.nio.charset.StandardCharsets;
import java.util.Scanner;
/**
* @author zjwblog <cn.zjwblog@gmail.com>
* @version 1.0
*/
public class Producer {
private static final String QUEUE_NAME = "durable-queue";
public static void main(String[] args) throws Exception {
try (Channel channel = RabbitMQUtil.getChannel()) {
// 第二个参数定义了queue为持久化队列
channel.queueDeclare(QUEUE_NAME, true, false, false, null);
Scanner scanner = new Scanner(System.in);
while (scanner.hasNext()) {
String message = scanner.next();
// 第三个参数定义了消息持久化
channel.basicPublish("", QUEUE_NAME, MessageProperties.PERSISTENT_TEXT_PLAIN, message.getBytes(StandardCharsets.UTF_8));
System.out.println("发送消息 [ " + message + " ] 成功.");
}
}
}
}
3.3.4 不公平分发
在最开始的时候我们学习到RabbitMQ
分发消息采用的轮询分发,但是在某种场景下这种策略并不是很好,比方说有两个消费者在处理任务,其中有个消费者1处理任务的速度非常快,而另外一个消费者2处理速度却很慢,这个时候我们还是采用轮询分发就会到这处理速度快的这个消费者很大一部分时间处于空闲状态,而处理慢的那个消费者一直在干活,这种分配方式在这种情况下其实就不太好,但是RabbitMQ
并不知道这种情况它依然很公平的进行分发。为了避免这种情况,我们可以设置参数channel.basicQos(1)
;
意思就是如果这个任务我还没有处理完或者我还没有应答你,你先别分配给我,我目前只能处理一个任务,然后rabbitmq
就会把该任务分配给没有那么忙的那个空闲消费者,当然如果所有的消费者都没有完成手上任务,队列还在不停的添加新任务,就可能会遇到队列被撑满的情况,这个时候就只能添加新的worker
或者改变其他存储任务的策略。
消费者:
// 消费者1
package com.zjw.rabbitmq.workqueue.qos;
import com.rabbitmq.client.Channel;
import com.zjw.rabbitmq.util.RabbitMQUtil;
import com.zjw.rabbitmq.util.SleepUtil;
/**
* @author zjwblog <cn.zjwblog@gmail.com>
* @version 1.0
*/
public class Consumer01 {
private static final String QUEUE_NAME = "durable-queue";
public static void main(String[] args) throws Exception {
Channel channel = RabbitMQUtil.getChannel();
// 设置预取值条数,当设置为1是,可以实现非公平分发消息
channel.basicQos(5);
channel.basicConsume(
QUEUE_NAME,
false,
(tag, msg) -> {
String message = new String(msg.getBody());
System.out.println("Consumer01 接收到消息:" + message);
SleepUtil.sleep(1);
System.out.println("Consumer01 处理消息[" +message+"]完成");
channel.basicAck(msg.getEnvelope().getDeliveryTag(), true);
},
tag -> System.out.println("[ " + tag + " ]消费者取消消费接口回调逻辑")
);
}
}
// 消费者2
package com.zjw.rabbitmq.workqueue.qos;
import com.rabbitmq.client.Channel;
import com.zjw.rabbitmq.util.RabbitMQUtil;
import com.zjw.rabbitmq.util.SleepUtil;
/**
* @author zjwblog <cn.zjwblog@gmail.com>
* @version 1.0
*/
public class Consumer02 {
private static final String QUEUE_NAME = "durable-queue";
public static void main(String[] args) throws Exception {
Channel channel = RabbitMQUtil.getChannel();
channel.basicQos(1);
channel.basicConsume(
QUEUE_NAME,
false, // 采用手动应答
(tag, msg) -> {
String message = new String(msg.getBody());
System.out.println("Consumer02 接收到消息:" + message);
SleepUtil.sleep(3);
System.out.println("Consumer02 处理消息[" +message+"]完成");
channel.basicAck(msg.getEnvelope().getDeliveryTag(), true);
},
tag -> System.out.println("[ " + tag + " ]消费者取消消费接口回调逻辑")
);
}
}
3.3.5 预取值
本身消息的发送就是异步发送的,所以在任何时候,channel
上肯定不止只有一个消息另外来自消费者的手动确认本质上也是异步的。因此这里就存在一个未确认的消息缓冲区,因此希望开发人员能限制此缓冲区的大小,以避免缓冲区里面无限制的未确认消息问题。这个时候就可以通过使用basic.qos
方法设置“预取计数”值来完成的。该值定义通道上允许的未确认消息的最大数量。一旦数量达到配置的数量,RabbitMQ
将停止在通道上传递更多消息,除非至少有一个未处理的消息被确认,例如,假设在通道上有未确认的消息5、6、7、8
,并且通道的预取计数设置为4
,此时RabbitMQ
将不会在该通道上再传递任何消息,除非至少有一个未应答的消息被ack
。比方说tag=6
这个消息刚刚被确认,RabbitMQ
将会感知这个情况到并再发送一条消息。消息应答和 QoS
预取值对用户吞吐量有重大影响。通常,增加预取将提高向消费者传递消息的速度。虽然自动应答传输消息速率是最佳的,但是,在这种情况下已传递但尚未处理的消息的数量也会增加,从而增加了消费者的内存消耗。应该小心使用具有无限预处理的自动确认模式或手动确认模式,消费者消费了大量的消息如果没有确认的话,会导致消费者连接节点的内存消耗变大,所以找到合适的预取值是一个反复试验的过程,不同的负载该值取值也不同100
到300
范围内的值通常可提供最佳的吞吐量,并且不会给消费者带来太大的风险。预取值为1
是最保守的。当然这将使吞吐量变得很低,特别是消费者连接延迟很严重的情况下,特别是在消费者连接等待时间较长的环境中。对于大多数应用来说,稍微高一点的值将是最佳的。
消费者
// 消费者
package com.zjw.rabbitmq.workqueue.qos;
import com.rabbitmq.client.Channel;
import com.zjw.rabbitmq.util.RabbitMQUtil;
import com.zjw.rabbitmq.util.SleepUtil;
/**
* @author zjwblog <cn.zjwblog@gmail.com>
* @version 1.0
*/
public class Consumer01 {
private static final String QUEUE_NAME = "durable-queue";
public static void main(String[] args) throws Exception {
Channel channel = RabbitMQUtil.getChannel();
// 设置预取值条数,当设置为1是,可以实现非公平分发消息
channel.basicQos(5);
channel.basicConsume(
QUEUE_NAME,
false,
(tag, msg) -> {
String message = new String(msg.getBody());
System.out.println("Consumer01 接收到消息:" + message);
SleepUtil.sleep(1);
System.out.println("Consumer01 处理消息[" +message+"]完成");
channel.basicAck(msg.getEnvelope().getDeliveryTag(), true);
},
tag -> System.out.println("[ " + tag + " ]消费者取消消费接口回调逻辑")
);
}
}
注:原始资料来源于尚硅谷,笔者学习过程中做了整理,加入了自己的实验代码