看著題目，有點一頭污水吧，事實上，沒有經歷過，很難去說ApplicationContextAware在什麼時候會用到，直接在一個bean對象里，你可以直接使用構造方法註入或者Autowired屬性註入的方式來使用其它的bean對象，這在springboot里是非常自然的，也是天然支持的；但如果你的這個bean不是由spring ioc自動註入的，而是通過攔截器動態配置的，這時你使用@Autowired時，是無法獲取到其它bean對象的；這時你需要使用ApplicationContextAware介面，再定義一個靜態的ApplicationContext實例，在你的攔截器執行方法里使用它就可以了。【應該和攔截器里的動態代理有關】

一個kafka的ConsumerInterceptor實例

在這個例子中，我們通過ConsumerInterceptor實現了一個TTL的延時隊列，當topic過期時，再通過KafkaTemplate將消息轉發到其它隊列里

• DelayPublisher.publish發送延時topic的方法

	/**
	 * 發送延時消息
	 * @param message 消息體
	 * @param delaySecondTime 多個秒後過期
	 * @param delayTopic 過期後發送到的話題
	 */
	public void publish(String message, long delaySecondTime, String delayTopic) {
		ProducerRecord producerRecord = new ProducerRecord<>(topic, 0, System.currentTimeMillis(), delayTopic, message,
				new RecordHeaders().add(new RecordHeader("ttl", toBytes(delaySecondTime))));
		kafkaTemplate.send(producerRecord);
	}

• ConsumerInterceptorTTL

/**
 * @author lind
 * @date 2023/8/18 8:33
 * @since 1.0.0
 */
@Component
public class ConsumerInterceptorTTL implements ConsumerInterceptor<String, String>, ApplicationContextAware {

	// 靜態化的上下文，用於獲取bean，因為ConsumerInterceptor是通過反射創建的，所以無法通過註入的方式獲取bean
	private static ApplicationContext applicationContext;

	@Override
	public ConsumerRecords<String, String> onConsume(ConsumerRecords<String, String> records) {
		long now = System.currentTimeMillis();
		Map<TopicPartition, List<ConsumerRecord<String, String>>> newRecords = new HashMap<>();
		for (TopicPartition tp : records.partitions()) {
			List<ConsumerRecord<String, String>> tpRecords = records.records(tp);
			List<ConsumerRecord<String, String>> newTpRecords = new ArrayList<>();
			for (ConsumerRecord<String, String> record : tpRecords) {
				Headers headers = record.headers();
				long ttl = -1;
				for (Header header : headers) {
					if (header.key().equals("ttl")) {
						ttl = toLong(header.value());
					}
				}
				// 消息超時判定
				if (ttl > 0 && now - record.timestamp() < ttl * 1000) {
					// 可以放在死信隊列中
					System.out.println("消息超時了,需要發到topic:" + record.key());
					KafkaTemplate kafkaTemplate = applicationContext.getBean(KafkaTemplate.class);
					kafkaTemplate.send(record.key(), record.value());
				}
				else { // 沒有設置TTL,不需要超時判定
					newTpRecords.add(record);
				}

			}
			if (!newRecords.isEmpty()) {
				newRecords.put(tp, newTpRecords);
			}
		}
		return new ConsumerRecords<>(newRecords);
	}

	@Override
	public void onCommit(Map<TopicPartition, OffsetAndMetadata> offsets) {
		offsets.forEach((tp, offset) -> System.out.println(tp + ":" + offset.offset()));
	}

	@Override
	public void close() {
	}

	@Override
	public void configure(Map<String, ?> configs) {

	}

	// 它的時機是在KafkaListenerAnnotationBeanPostProcessor的postProcessAfterInitialization方法中，applicationContext應該定時成static，否則在實例對象中，它的值可能是空
	@Override
	public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
		this.applicationContext = applicationContext;
	}

}

• 配置文件中註入攔截器

spring:
  kafka:
    consumer:
      properties:
        interceptor.classes: com.example.ConsumerInterceptorTTL 

作者：倉儲大叔，張占嶺，
榮譽：微軟MVP
QQ：853066980

支付寶掃一掃，為大叔打賞!