完成自定义拦截器
- 实际开发工作中可能要求我们在生产者发送信息之前统一打一个标记或者修改一个类型就可以用到拦截器
1.创建生产者拦截器
public class UserDefineInterceptor implements ProducerInterceptor {
@Override
public ProducerRecord onSend(ProducerRecord record) {
System.out.println("发送之前record会在这里,我们可以对数据进行修改");
return new ProducerRecord(record.topic(),record.key(),record.value()+"------------fufu zhen shuai");
}
@Override
public void onAcknowledgement(RecordMetadata recordMetadata, Exception e) {
System.out.println("2222222");
System.out.println("metadata:"+recordMetadata+",exception:"+e);
}
@Override
public void close() {
System.out.println("close");
}
@Override
public void configure(Map<String, ?> map) {
System.out.println("configure");
}
}
2.创建生产者
public class KafkaProducerInterceptors {
public static void main(String[] args) {
Properties props = new Properties();
props.put(ProducerConfig.BOOTSTRAP_SERVERS_CONFIG,"ip:9092");
props.put(ProducerConfig.KEY_SERIALIZER_CLASS_CONFIG, StringSerializer.class.getName());
props.put(ProducerConfig.VALUE_SERIALIZER_CLASS_CONFIG,StringSerializer.class.getName());
props.put(ProducerConfig.INTERCEPTOR_CLASSES_CONFIG,UserDefineInterceptor.class.getName());
KafkaProducer<String, String> producer = new KafkaProducer<>(props);
for (int i = 0; i < 5; i++) {
ProducerRecord<String, String> record = new ProducerRecord<>("topic04", "key" + i, "value" + i);
producer.send(record);
}
producer.close();
}
}
3.创建消费者拦截器
public class UserCDefineInterceptor implements ConsumerInterceptor {
@Override
public ConsumerRecords onConsume(ConsumerRecords consumerRecords) {
System.out.println("接收信息入口 可以在这里对信息进行拦截修改");
Iterator<ConsumerRecord<String, String>> iterator = consumerRecords.iterator();
while (iterator.hasNext()){
ConsumerRecord<String, String> record = iterator.next();
String value = record.value();
System.out.println("hahahahah "+ value);
}
return consumerRecords;
}
@Override
public void onCommit(Map map) {
System.out.println("onCommit");
}
@Override
public void close() {
System.out.println("close");
}
@Override
public void configure(Map<String, ?> map) {
System.out.println("configure");
}
}
4.创建消费者
public class KafkaConsumerInterceptor {
public static void main(String[] args) {
Properties props = new Properties();
props.put(ConsumerConfig.BOOTSTRAP_SERVERS_CONFIG,"ip:9092");
props.put(ConsumerConfig.KEY_DESERIALIZER_CLASS_CONFIG, StringDeserializer.class.getName());
props.put(ConsumerConfig.VALUE_DESERIALIZER_CLASS_CONFIG,StringDeserializer.class.getName());
props.put(ConsumerConfig.INTERCEPTOR_CLASSES_CONFIG,UserCDefineInterceptor.class.getName());
props.put(ConsumerConfig.GROUP_ID_CONFIG,"g1");
KafkaConsumer<String,String> consumer = new KafkaConsumer(props);
consumer.subscribe(Arrays.asList("topic04"));
while(true){
ConsumerRecords<String, String> records = consumer.poll(Duration.ofSeconds(1));
if(!records.isEmpty()){
Iterator<ConsumerRecord<String, String>> iterator = records.iterator();
while(iterator.hasNext()){
ConsumerRecord<String, String> record = iterator.next();
String topic = record.topic();
int partition = record.partition();
long offset = record.offset();
String key = record.key();
String value = record.value();
long timestamp = record.timestamp();
System.out.println(topic+"\t"+partition+","+offset+"\t"+key+" "+value+" "+timestamp);
}
}
}
}
}