kafka producer interceptor拦截器(五)

心已赠人 2023-08-17 16:16 118阅读 0赞

producer在发送数据时，会经过拦截器和序列化，最后到达相应的分区。在经过拦截器时，我们可以对发送的数据做进步的处理。

要正确的使用拦截器需要以下步骤：

1.实现拦截器ProducerInterceptor的方法

2.在producer的prop中配置

prop.put("interceptor.classes", "com.xxx.interceptor.xxxInterceptor")

如果是拦截器链的话，在后面追加即可

prop.put("interceptor.classes", ""com.xxx.interceptor.xxxInterceptor1,com.xxx.interceptor.xxxInterceptor2");

**生产者的拦截器需要实现ProducerInterceptor接口中的方法来实现**

@Override
    　　public void configure(Map<String, ?> arg0) {}　　#获取broker的配置信息
           
        @Override
        public void close() {}　　　　　　　#在producer关闭时调用此方法　　　　　　　　　   
      
        @Override
        public void onAcknowledgement(RecordMetadata recordMetadata, Exception e) {} #数据在写到broker时，无论是否成功的回调
       
        @Override
        public ProducerRecord<String, String> onSend(ProducerRecord<String, String> producerRecord) {}　　#拦截的信息

**生产者:**

public class ProducerDemo {
        
        private static final Logger LOG = LoggerFactory.getLogger(ProducerDemo.class);
        
        public static void main(String[] args) throws InterruptedException, ExecutionException {
          
            //1.加载配置信息
            Properties prop = loadProperties();
            
            //2.创建生产者
            KafkaProducer<String,String> producer = new KafkaProducer<>(prop);
            
            //3.发送内容
            String sendContent = "hello_kafka";
            
            IntStream.range(0, 10).forEach(i ->{
                try {
                    ProducerRecord<String,String> record = new ProducerRecord<>("test1",sendContent+"_"+i);
                    Future<RecordMetadata> future = producer.send(record);
                    RecordMetadata recordMetadata = future.get();
                    LOG.info("发送的数据是 :{},offset:是{},partition是：{}",sendContent,recordMetadata.offset(),recordMetadata.partition());
                } catch (Exception e) {
                    e.printStackTrace();
                }
                
            });
                    
            producer.close();    //回调拦截器中的close方法
            
        }
            
        //配置文件的设置
        public static Properties loadProperties() {
            Properties prop = new Properties();
            prop.put("bootstrap.servers", "192.168.100.144:9092,192.168.100.145:9092,192.168.100.146:9092");
            prop.put("key.serializer", "org.apache.kafka.common.serialization.StringSerializer");
            prop.put("value.serializer", "org.apache.kafka.common.serialization.StringSerializer");
            prop.put("interceptor.classes", "com.zpb.interceptor.ProducerInterceptorDemo,com.zpb.interceptor.ProducerInterceptorDemo2");
            prop.put("acks", "all");    //发送到所有的ISR队列中
            return prop;
        }
    }

**拦截器一：**

public class ProducerInterceptorDemo implements ProducerInterceptor<String, String>{
        
        private static final Logger LOG = LoggerFactory.getLogger(ProducerInterceptorDemo.class);
    
        private volatile long succNum = 0;
        
        private volatile long failNum = 0;
        
        @Override
        public void configure(Map<String, ?> arg0) {
            LOG.info("configure ==>"+arg0);
        }
    
        @Override
        public void close() {
            double succRatio = succNum/succNum+failNum;
            LOG.info("成功率是："+succRatio*100);
        }
    
        @Override
        public void onAcknowledgement(RecordMetadata recordMetadata, Exception e) {
            
            if(null == e){
                succNum++;
            }else{
                failNum++;
            }
            
        }
    
        @Override
        public ProducerRecord<String, String> onSend(ProducerRecord<String, String> producerRecord) {
            
            String prefixValue = producerRecord.value()+"prefix_1";
            
            return new ProducerRecord<String, String>(producerRecord.topic(),prefixValue);
        }
    }

**拦截器二：**

public class ProducerInterceptorDemo2 implements ProducerInterceptor<String, String>{
    
        @Override
        public void configure(Map<String, ?> configs) {
            
        }
    
        @Override
        public ProducerRecord<String, String> onSend(ProducerRecord<String, String> record) {
            String prefixValue = record.value()+"prefix_2";
            return new ProducerRecord<String, String>(record.topic(), prefixValue);
        }
    
        @Override
        public void onAcknowledgement(RecordMetadata metadata, Exception exception) {
            
        }
    
        @Override
        public void close() {
            
        }
    }

转载于:https://www.cnblogs.com/MrRightZhao/p/11345447.html