kafka-4--java-自定义拦截器

完成自定义拦截器

• 实际开发工作中可能要求我们在生产者发送信息之前统一打一个标记或者修改一个类型就可以用到拦截器

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");
    }

    /**
     * 发送信息通知 每次发送信息都会调用这个方法
     * @param recordMetadata
     * @param e
     */
    @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());
        //配置生产者自定义拦截器--UserDefineInterceptor
        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);
            //ProducerRecord<String, String> record = new ProducerRecord<>("topic01", "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());
        //配置消费者自定义拦截器 -- UserCDefineInterceptor
        props.put(ConsumerConfig.INTERCEPTOR_CLASSES_CONFIG,UserCDefineInterceptor.class.getName());
        props.put(ConsumerConfig.GROUP_ID_CONFIG,"g1");
        KafkaConsumer<String,String> consumer = new KafkaConsumer(props);
        //订阅相关的topics
        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);
                }
            }
        }
    }
}