springboot~kafka中延(yan)時消息的(de)實現(xian)
應用場景
- 用戶下單5分鐘后,給他發短信
- 用戶下單30分鐘后,如果用戶不付款就自動取消訂單
kafka無死信隊列
kafka本身沒有這種延時隊(dui)(dui)列的(de)機制,像rabbitmq有自己(ji)的(de)死(si)信隊(dui)(dui)列,當一些(xie)消息在一定時間(jian)不消費時會發到死(si)信隊(dui)(dui)列,由死(si)信隊(dui)(dui)列來處理它們,上(shang)面的(de)兩個需求如果是rabbitmq可以通過死(si)信隊(dui)(dui)列實(shi)現的(de)。
kafka有生產者攔截器
通過對生產者攔截器實現一個TTL的檢查,然后再通過類似netty里的延時隊列組件來實現消息的延時發送,發到咱們的死信隊列里

- ProducerInterceptorTTL源碼
public class ProducerInterceptorTTL implements ProducerInterceptor<Integer, String>, ApplicationContextAware {
// 消息延時,單位秒
public static String TTL = "ttl";
// 死信隊列,延時后發送到的隊列,我們稱為死信隊列
public static String DEAD_TOPIC = "dead_topic";
// 靜態化的上下文,用于獲取bean,因為ConsumerInterceptor是通過反射創建的,所以無法通過注入的方式獲取bean
private static ApplicationContext applicationContext;
// 時間輪,用于延時發送消息
private static LindTimeWheel timeWheel = new LindTimeWheel(1000, 8);
@Override
public ProducerRecord onSend(ProducerRecord<Integer, String> record) {
final String topic = record.topic();
final Integer partition = record.partition();
final Integer key = record.key();
final String value = record.value();
final Long timestamp = record.timestamp();
final Headers headers = record.headers();
long ttl = -1;
String deadTopic = null;
for (Header header : headers) {
if (header.key().equals(TTL)) {
ttl = toLong(header.value());
}
if (header.key().equals(DEAD_TOPIC)) {
deadTopic = new String(header.value());
}
}
// 消息超時判定
if (deadTopic != null && ttl > 0) {
// 可以放在死信隊列中
String finalDeadTopic = deadTopic;
long finalTtl = ttl * 1000;
timeWheel.addTask(() -> {
System.out.println("消息超時了," + finalTtl + "需要發到topic:" + record.key());
KafkaTemplate kafkaTemplate = applicationContext.getBean(KafkaTemplate.class);
kafkaTemplate.send(finalDeadTopic, record.value());
}, finalTtl);
}
// 攔截器攔下來之后改變原來的消息內容
ProducerRecord<Integer, String> newRecord = new ProducerRecord<Integer, String>(topic, partition, timestamp,
key, value, headers);
// 傳遞新的消息
return newRecord;
}
@Override
public void onAcknowledgement(RecordMetadata recordMetadata, Exception e) {
}
@Override
public void close() {
}
@Override
public void configure(Map<String, ?> map) {
}
@Override
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
this.applicationContext = applicationContext;
}
}
- 注冊攔截器
spring:
kafka:
producer:
properties:
interceptor.classes: com.ruoyi.lawyer.delay.ProducerInterceptorTTL
- 延時消息在某個時間段之后會送出

